Update agent.py
Browse files
agent.py
CHANGED
@@ -254,7 +254,7 @@ class IntelligentSourceRouter:
|
|
254 |
- web_search: For all other information (current events, general facts, weather, how-to guides, etc.)
|
255 |
Respond with ONLY "arxiv" or "web_search".
|
256 |
"""
|
257 |
-
response =
|
258 |
selected_source = response.text.strip().lower()
|
259 |
|
260 |
results = [f"**Query**: {query}", f"**Selected Source**: {selected_source}", "="*50]
|
@@ -293,7 +293,7 @@ class IntelligentSourceRouter:
|
|
293 |
- web_search: For all other information (current events, general facts, weather, how-to guides, etc.)
|
294 |
Respond with ONLY "arxiv" or "web_search".
|
295 |
"""
|
296 |
-
response =
|
297 |
selected_source = response.text.strip().lower()
|
298 |
|
299 |
results = [f"**Query**: {query}", f"**Selected Source**: {selected_source}", "="*50]
|
|
|
254 |
- web_search: For all other information (current events, general facts, weather, how-to guides, etc.)
|
255 |
Respond with ONLY "arxiv" or "web_search".
|
256 |
"""
|
257 |
+
response = proj_llm.complete(intent_prompt)
|
258 |
selected_source = response.text.strip().lower()
|
259 |
|
260 |
results = [f"**Query**: {query}", f"**Selected Source**: {selected_source}", "="*50]
|
|
|
293 |
- web_search: For all other information (current events, general facts, weather, how-to guides, etc.)
|
294 |
Respond with ONLY "arxiv" or "web_search".
|
295 |
"""
|
296 |
+
response = proj_llm.complete(intent_prompt)
|
297 |
selected_source = response.text.strip().lower()
|
298 |
|
299 |
results = [f"**Query**: {query}", f"**Selected Source**: {selected_source}", "="*50]
|