We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ee24831 commit 8597060Copy full SHA for 8597060
1 file changed
src/agent.py
@@ -33,10 +33,9 @@ def ___init__(self, function: LowLevelILFunction | MediumLevelILFunction |
33
# Get the list of available engines.
34
engines: list[Engine] = openai.Engine.list().data
35
# Ensure the user's selected engine is available.
36
- # pylint: disable=unnecessary-comprehension
37
- if engine not in [i.id for i in engines]:
+ if engine not in [e.id for e in engines]:
38
InvalidEngineException(f'Invalid engine: {engine}. Valid engines '
39
- f'are: {[id for id in engines.id]}')
+ f'are: {[e.id for e in engines]}')
40
41
# Set instance attributes.
42
self.function = function
0 commit comments