requirements update for llm compat

app.py

@@ -310,22 +310,24 @@ class BasicAgent:
                     else:
                         messages_with_system = [self.sys_msg] + messages
                     
+                    # Define tools in Gemini's format
+                    tools = [{
+                        "name": "google_search",
+                        "description": "Search for information on the web",
+                        "parameters": {
+                            "properties": {
+                                "query": {
+                                    "description": "The search query",
+                                    "type": "string"
+                                }
+                            },
+                            "required": ["query"]
+                        }
+                    }]
+                    
                     response = self.primary_llm.invoke(
                         messages_with_system,
-                        tools=[{"type": "function", "function": {
-                            "name": "google_search",
-                            "description": "Search for information on the web",
-                            "parameters": {
-                                "type": "object",
-                                "properties": {
-                                    "query": {
-                                        "type": "string",
-                                        "description": "The search query"
-                                    }
-                                },
-                                "required": ["query"]
-                            }
-                        }}]
+                        tools=tools
                     )
                 except Exception as e:
                     error_str = str(e)
@@ -337,20 +339,23 @@ class BasicAgent:
                                     # For OpenAI, we can use the system message directly
                                     response = self.fallback_llm.invoke(
                                         [self.sys_msg] + messages,
-                                        tools=[{"type": "function", "function": {
-                                            "name": "google_search",
-                                            "description": "Search for information on the web",
-                                            "parameters": {
-                                                "type": "object",
-                                                "properties": {
-                                                    "query": {
-                                                        "type": "string",
-                                                        "description": "The search query"
-                                                    }
-                                                },
-                                                "required": ["query"]
+                                        tools=[{
+                                            "type": "function",
+                                            "function": {
+                                                "name": "google_search",
+                                                "description": "Search for information on the web",
+                                                "parameters": {
+                                                    "type": "object",
+                                                    "properties": {
+                                                        "query": {
+                                                            "type": "string",
+                                                            "description": "The search query"
+                                                        }
+                                                    },
+                                                    "required": ["query"]
+                                                }
                                             }
-                                        }}]
+                                        }]
                                     )
                                     logger.info("Successfully used fallback LLM")
                                 except Exception as fallback_error: