Scott Cogan commited on
Commit
dd0227b
·
1 Parent(s): d63af7e

requirements update for llm compat

Browse files
Files changed (1) hide show
  1. app.py +14 -12
app.py CHANGED
@@ -312,18 +312,20 @@ class BasicAgent:
312
 
313
  # Create tool configuration for Gemini
314
  genai_tool = {
315
- "name": "google_search",
316
- "description": "Search for information on the web",
317
- "parameters": {
318
- "type": "object",
319
- "properties": {
320
- "query": {
321
- "type": "string",
322
- "description": "The search query"
323
- }
324
- },
325
- "required": ["query"]
326
- }
 
 
327
  }
328
 
329
  logger.info("Attempting to use primary LLM (Gemini)")
 
312
 
313
  # Create tool configuration for Gemini
314
  genai_tool = {
315
+ "function_declarations": [{
316
+ "name": "google_search",
317
+ "description": "Search for information on the web",
318
+ "parameters": {
319
+ "type": "object",
320
+ "properties": {
321
+ "query": {
322
+ "type": "string",
323
+ "description": "The search query"
324
+ }
325
+ },
326
+ "required": ["query"]
327
+ }
328
+ }]
329
  }
330
 
331
  logger.info("Attempting to use primary LLM (Gemini)")