From 38f73a4379ff042cfbc93856b89da166fb36bfce Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 19 Mar 2024 16:17:09 -0700 Subject: [PATCH] fix formatting --- vision_agent/agent/easytool.py | 3 ++- vision_agent/llm/llm.py | 8 +++++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/vision_agent/agent/easytool.py b/vision_agent/agent/easytool.py index 4690a829..b80f1a7b 100644 --- a/vision_agent/agent/easytool.py +++ b/vision_agent/agent/easytool.py @@ -160,7 +160,7 @@ def retrieval( tool_name = tool_instructions["name"] parameters = choose_parameter(model, question, tool_usage, previous_log) - if parameters is None: # TODO + if parameters is None: # TODO pass tool_results = [{"tool_name": tool_name, "parameters": parameters}] @@ -212,6 +212,7 @@ class EasyTool(Agent): >>> print(resp) >>> "It will travel approximately 31.03 kilometers in 29 minutes." """ + def __init__( self, task_model: Optional[Union[LLM, LMM]] = None, diff --git a/vision_agent/llm/llm.py b/vision_agent/llm/llm.py index 91e3ebe8..dcf61762 100644 --- a/vision_agent/llm/llm.py +++ b/vision_agent/llm/llm.py @@ -31,7 +31,9 @@ def __call__(self, input: Union[str, List[Dict[str, str]]]) -> str: class OpenAILLM(LLM): r"""An LLM class for any OpenAI LLM model.""" - def __init__(self, model_name: str = "gpt-4-turbo-preview", json_mode: bool = False): + def __init__( + self, model_name: str = "gpt-4-turbo-preview", json_mode: bool = False + ): self.model_name = model_name self.client = OpenAI() self.json_mode = json_mode @@ -43,7 +45,7 @@ def generate(self, prompt: str) -> str: messages=[ {"role": "user", "content": prompt}, ], - **kwargs, # type: ignore + **kwargs, # type: ignore ) return cast(str, response.choices[0].message.content) @@ -53,7 +55,7 @@ def chat(self, chat: List[Dict[str, str]]) -> str: response = self.client.chat.completions.create( model=self.model_name, messages=chat, # type: ignore - **kwargs, # type: ignore + **kwargs, # type: ignore ) return cast(str, response.choices[0].message.content)