akhaliq HF Staff commited on
Commit
f9931c4
·
1 Parent(s): 87df3aa
Files changed (1) hide show
  1. app.py +5 -10
app.py CHANGED
@@ -253,17 +253,15 @@ def create_multimodal_message(text, image=None):
253
  # Uses 'advanced' search_depth and auto_parameters=True for speed and relevance
254
 
255
  def perform_web_search(query: str, max_results: int = 5, include_domains=None, exclude_domains=None) -> str:
256
- """Perform web search using Tavily and return formatted results (fast, prompt-focused)"""
257
  if not tavily_client:
258
  return "Web search is not available. Please set the TAVILY_API_KEY environment variable."
259
 
260
  try:
261
- # Use advanced search for better results, auto_parameters for prompt intent
262
  search_params = {
263
- "auto_parameters": True,
264
  "search_depth": "advanced",
265
- "max_results": min(max(1, max_results), 20),
266
- "include_answer": True
267
  }
268
  if include_domains is not None:
269
  search_params["include_domains"] = include_domains
@@ -272,9 +270,6 @@ def perform_web_search(query: str, max_results: int = 5, include_domains=None, e
272
 
273
  response = tavily_client.search(query, **search_params)
274
 
275
- answer = response.get('answer')
276
- formatted_answer = f"**AI Answer:**\n{answer}\n\n" if answer else ""
277
-
278
  search_results = []
279
  for result in response.get('results', []):
280
  title = result.get('title', 'No title')
@@ -283,9 +278,9 @@ def perform_web_search(query: str, max_results: int = 5, include_domains=None, e
283
  search_results.append(f"Title: {title}\nURL: {url}\nContent: {content}\n")
284
 
285
  if search_results:
286
- return formatted_answer + "Web Search Results:\n\n" + "\n---\n".join(search_results)
287
  else:
288
- return formatted_answer + "No search results found."
289
 
290
  except Exception as e:
291
  return f"Search error: {str(e)}"
 
253
  # Uses 'advanced' search_depth and auto_parameters=True for speed and relevance
254
 
255
  def perform_web_search(query: str, max_results: int = 5, include_domains=None, exclude_domains=None) -> str:
256
+ """Perform web search using Tavily with default parameters"""
257
  if not tavily_client:
258
  return "Web search is not available. Please set the TAVILY_API_KEY environment variable."
259
 
260
  try:
261
+ # Use Tavily defaults with advanced search depth for better results
262
  search_params = {
 
263
  "search_depth": "advanced",
264
+ "max_results": min(max(1, max_results), 20)
 
265
  }
266
  if include_domains is not None:
267
  search_params["include_domains"] = include_domains
 
270
 
271
  response = tavily_client.search(query, **search_params)
272
 
 
 
 
273
  search_results = []
274
  for result in response.get('results', []):
275
  title = result.get('title', 'No title')
 
278
  search_results.append(f"Title: {title}\nURL: {url}\nContent: {content}\n")
279
 
280
  if search_results:
281
+ return "Web Search Results:\n\n" + "\n---\n".join(search_results)
282
  else:
283
+ return "No search results found."
284
 
285
  except Exception as e:
286
  return f"Search error: {str(e)}"