Coool2 commited on
Commit
0c9e128
·
verified ·
1 Parent(s): a479d3c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
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 = text_llm.complete(intent_prompt)
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 = text_llm.complete(intent_prompt)
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]