fix formatting

Author: dillonalaird

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,

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)