Scott Cogan
commited on
Commit
·
dd0227b
1
Parent(s):
d63af7e
requirements update for llm compat
Browse files
app.py
CHANGED
|
@@ -312,18 +312,20 @@ class BasicAgent:
|
|
| 312 |
|
| 313 |
# Create tool configuration for Gemini
|
| 314 |
genai_tool = {
|
| 315 |
-
"
|
| 316 |
-
|
| 317 |
-
|
| 318 |
-
"
|
| 319 |
-
|
| 320 |
-
"
|
| 321 |
-
"
|
| 322 |
-
|
| 323 |
-
|
| 324 |
-
|
| 325 |
-
|
| 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)")
|