Remove unused exception.

A model is always returned in the current configuration.

Relate #14.

Author: WhatTheFuzz

src/agent.py

@@ -12,8 +12,6 @@
 from binaryninja.settings import Settings
 from binaryninja import log
 
-from .exceptions import InvalidEngineException
-
 
 class Agent:
 
@@ -31,7 +29,6 @@ class Agent:
 
     def __init__(self,
                 function: Union[LowLevelILFunction, MediumLevelILFunction, HighLevelILFunction],
-                engine: str,
                 path_to_api_key: Optional[Path]=None) -> None:
 
         # Read the API key from the environment variable.
@@ -142,4 +139,3 @@ def send_query(self, query: str) -> str:
             max_tokens=2_048
         )
         return response.choices[0].text
-

src/entry.py

@@ -9,7 +9,6 @@
 def check_function(bv: BinaryView, func: Function) -> bool:
     agent: Agent = Agent(
         function=func,
-        engine='text-davinci-003',
         path_to_api_key=API_KEY_PATH
     )
     query: str = agent.generate_query(func)

src/exceptions.py

@@ -1,8 +1,3 @@
-from openai.error import OpenAIError
-
-class InvalidEngineException(OpenAIError):
-    pass
-
 class RegisterSettingsGroupException(Exception):
     pass