Fix use vision model for openai in try

sunjae1294 · Nov 6, 2024 · 2e18985 · 2e18985
1 parent 54c935e
commit 2e18985
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/examples/try.py b/examples/try.py
@@ -13,12 +13,11 @@
 
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
+import argparse
 import asyncio
 
 from src import Agent
 
-import argparse
-
 logging.basicConfig(level=logging.INFO)
 
 
@@ -28,7 +27,7 @@ def get_llm(provider: str):
 			model_name='claude-3-5-sonnet-20240620', timeout=25, stop=None, temperature=0.3
 		)
 	elif provider == 'openai':
-		return ChatOpenAI(model='gpt-4')
+		return ChatOpenAI(model='gpt-4o', temperature=0.3)
 	else:
 		raise ValueError(f'Unsupported provider: {provider}')