type checking fixes

landing-ai · Apr 1, 2024 · e860588 · e860588
1 parent de63f5b
commit e860588
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)