agent optimisation

agent/gaia_agent.py

@@ -12,7 +12,7 @@ from langchain_community.chat_models import ChatOpenAI
 
 def create_langchain_agent() -> AgentExecutor:
     llm = ChatOpenAI(
-        model_name="gpt-4",  # Or "gpt-3.5-turbo"
+        model_name="gpt-4o",  # Or "gpt-3.5-turbo"
         temperature=0.3,
         openai_api_key=os.getenv("OPENAI_API_KEY"),
     )
@@ -24,7 +24,15 @@ def create_langchain_agent() -> AgentExecutor:
         Tool(name="Chess Image Solver", func=solve_chess_image, description="Use this to solve a chess puzzle from an image."),
         Tool(name="File Parser", func=parse_file_and_summarize, description="Use this to process Excel or CSV files for analytics."),
     ]
+    
+    agent = initialize_agent(
+        tools=tools,
+        llm=llm,
+        agent=AgentType.OPENAI_FUNCTIONS,
+        verbose=True
+    )
 
+    '''
     agent = initialize_agent(
         tools=tools,
         llm=llm,
@@ -46,4 +54,5 @@ def create_langchain_agent() -> AgentExecutor:
             ),
         }
     )
+    '''
     return agent