Update agents.py

agents.py

@@ -263,7 +263,7 @@ class GeminiAgent(LLMAgentBase):
         try:
             # Configure tools using the Gemini API format
             tools = genai.types.Tool(function_declarations=self.function_declarations)
-            config = genai.types.GenerateContentConfig(tools=[tools])
+            config = genai.types.GenerateContentConfig(tools=[tools],automatic_function_calling=types.AutomaticFunctionCallingConfig(disable=True))
             
             # Send request to the model
             response = self.genai_client.models.generate_content(
@@ -271,27 +271,16 @@ class GeminiAgent(LLMAgentBase):
                 contents=prompt,
                 config=config
             )
-            print("GEMINI RESPONSE : ", response)
-            
-            # Check for function calls in the response
-            if (hasattr(response, 'candidates') and 
-                response.candidates and 
-                hasattr(response.candidates[0], 'content') and
-                hasattr(response.candidates[0].content, 'parts') and
-                response.candidates[0].content.parts and
-                hasattr(response.candidates[0].content.parts[0], 'function_call')):
-                
-                function_call = response.candidates[0].content.parts[0].function_call
-                function_name = function_call.name
-                # Get arguments
-                arguments = {}
-                if hasattr(function_call, 'args'):
-                    arguments = function_call.args
-                
-                if function_name in self.standard_tools:
+            try:
+                function_calls = response.function_calls
+                if function_calls:
+                    function_name = function_calls[0].name
+                    arguments = function_calls[0].args
                     return {"decision": {"name": function_name, "arguments": arguments}}
                 else:
-                    return {"error": f"Model called unknown function '{function_name}'."}
+                    return {"error": "Gemini did not return a function call."}
+            except Exception as e:
+                    return {"error": f"Model called unknown @function '{function_name}'."}
             
             # No function call found
             return {"error": "Gemini did not return a function call."}