Skip to content

Commit

Permalink
adding gpt4t as default model for old vision agent
Browse files Browse the repository at this point in the history
  • Loading branch information
shankar-vision-eng committed May 13, 2024
1 parent 72f17fe commit f3b0a56
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions vision_agent/agent/vision_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -464,15 +464,17 @@ def __init__(
report_progress_callback: a callback to report the progress of the agent. This is useful for streaming logs in a web application where multiple VisionAgent instances are running in parallel. This callback ensures that the progress are not mixed up.
"""
self.task_model = (
OpenAILLM(json_mode=True, temperature=0.1)
OpenAILLM(model_name="gpt-4-turbo", json_mode=True, temperature=0.0)
if task_model is None
else task_model
)
self.answer_model = (
OpenAILLM(temperature=0.1) if answer_model is None else answer_model
OpenAILLM(model_name="gpt-4-turbo", temperature=0.0)
if answer_model is None
else answer_model
)
self.reflect_model = (
OpenAILMM(json_mode=True, temperature=0.1)
OpenAILMM(model_name="gpt-4-turbo", json_mode=True, temperature=0.0)
if reflect_model is None
else reflect_model
)
Expand Down

0 comments on commit f3b0a56

Please sign in to comment.