From e8605882784925c761414b1ffc8f29a3fe91a76e Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Mon, 1 Apr 2024 11:55:08 -0700 Subject: [PATCH] type checking fixes --- vision_agent/agent/vision_agent.py | 2 +- vision_agent/llm/llm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index 80d56a3e..1aed604f 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -431,7 +431,7 @@ def __init__( OpenAILLM(temperature=0.1) if answer_model is None else answer_model ) self.reflect_model = ( - OpenAILMM(temperature=0.0) if reflect_model is None else reflect_model + OpenAILMM(temperature=0.1) if reflect_model is None else reflect_model ) self.max_retries = max_retries diff --git a/vision_agent/llm/llm.py b/vision_agent/llm/llm.py index 47bd4185..e97bcdeb 100644 --- a/vision_agent/llm/llm.py +++ b/vision_agent/llm/llm.py @@ -48,7 +48,7 @@ def generate(self, prompt: str) -> str: messages=[ {"role": "user", "content": prompt}, ], - **self.kwargs, # type: ignore + **self.kwargs, ) return cast(str, response.choices[0].message.content)