From c6fa756eb9545c8c7558287ef6c6dedb684bb98f Mon Sep 17 00:00:00 2001 From: Giridhar M Date: Sat, 24 May 2025 18:34:22 +0530 Subject: [PATCH] agent shift to codeAgent --- mxtoai/agents/email_agent.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mxtoai/agents/email_agent.py b/mxtoai/agents/email_agent.py index f0c115a..e45fbfc 100644 --- a/mxtoai/agents/email_agent.py +++ b/mxtoai/agents/email_agent.py @@ -7,7 +7,7 @@ from dotenv import load_dotenv # Update imports to use proper classes from smolagents -from smolagents import Tool, ToolCallingAgent +from smolagents import Tool, CodeAgent # Add imports for the new default tools from smolagents.default_tools import ( @@ -126,12 +126,12 @@ def __init__( def _init_agent(self): """ - Initialize the ToolCallingAgent with Azure OpenAI. + Initialize the CodeAgent with Azure OpenAI. """ # Initialize the routed model with the default model group self.routed_model = RoutedLiteLLMModel() - self.agent = ToolCallingAgent( + self.agent = CodeAgent( model=self.routed_model, tools=self.available_tools, max_steps=12,