akhaliq HF Staff commited on
Commit
87df3aa
·
1 Parent(s): 78fc423

use search depth

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -250,7 +250,7 @@ def create_multimodal_message(text, image=None):
250
  return {"role": "user", "content": content}
251
 
252
  # Updated for faster Tavily search and closer prompt usage
253
- # Uses 'basic' 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)"""
@@ -258,10 +258,10 @@ def perform_web_search(query: str, max_results: int = 5, include_domains=None, e
258
  return "Web search is not available. Please set the TAVILY_API_KEY environment variable."
259
 
260
  try:
261
- # Use basic search for speed, auto_parameters for prompt intent
262
  search_params = {
263
  "auto_parameters": True,
264
- "search_depth": "basic",
265
  "max_results": min(max(1, max_results), 20),
266
  "include_answer": True
267
  }
 
250
  return {"role": "user", "content": content}
251
 
252
  # Updated for faster Tavily search and closer prompt usage
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)"""
 
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
  }