Coool2 commited on
Commit
6aee14a
·
verified ·
1 Parent(s): 5e30500

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -249,7 +249,7 @@ class IntelligentSourceRouter:
249
  self.arxiv_tool = ArxivToolSpec().to_tool_list()[0]
250
  self.duckduckgo_tool = DuckDuckGoSearchToolSpec().to_tool_list()[1]
251
 
252
- def detect_intent_and_extract_content(self, query: str, max_results = 3) -> str:
253
  # Use your LLM to decide between arxiv and web_search
254
  intent_prompt = f"""
255
  Analyze this query and determine if it's scientific research or general information:
@@ -278,11 +278,11 @@ class IntelligentSourceRouter:
278
  )
279
  result = "\n".join(formatted)
280
  results.append(f"**Web Search Results:**\n{result}")
 
281
  except Exception as e:
282
  results.append(f"**Search failed**: {str(e)}")
283
  return "\n\n".join(results)
284
 
285
-
286
  # Initialize router
287
  intelligent_router = IntelligentSourceRouter()
288
 
 
249
  self.arxiv_tool = ArxivToolSpec().to_tool_list()[0]
250
  self.duckduckgo_tool = DuckDuckGoSearchToolSpec().to_tool_list()[1]
251
 
252
+ def detect_intent_and_extract_content(self, query: str, max_results = 1) -> str:
253
  # Use your LLM to decide between arxiv and web_search
254
  intent_prompt = f"""
255
  Analyze this query and determine if it's scientific research or general information:
 
278
  )
279
  result = "\n".join(formatted)
280
  results.append(f"**Web Search Results:**\n{result}")
281
+ print (results)
282
  except Exception as e:
283
  results.append(f"**Search failed**: {str(e)}")
284
  return "\n\n".join(results)
285
 
 
286
  # Initialize router
287
  intelligent_router = IntelligentSourceRouter()
288