Nancy1906 commited on
Commit
4036a74
·
verified ·
1 Parent(s): 44a3f99
Files changed (1) hide show
  1. my_tools.py +2 -2
my_tools.py CHANGED
@@ -180,7 +180,7 @@ class GeminiLLM(LLM):
180
  # -------------------------------------------------------------------
181
  HEADERS = {'User-Agent': 'Mozilla/5.0'}
182
 
183
- def buscar_web(query, num_results=5):
184
  retriever = TavilySearchAPIRetriever(api_key=os.getenv("TAVILY_API_KEY"), k=num_results)
185
  results = retriever.invoke(query)
186
  # Formatear resultados para pasarlos al LLM
@@ -265,7 +265,7 @@ def read_excel_data(file_path: str, sheet_name=0) -> str:
265
  except Exception as e:
266
  return f"Error read_excel_data: {e}"
267
 
268
- def classify_botanical(items_list_str: str) -> str:
269
  """
270
  Splits an input list of foods (English names) into botanical Vegetables,
271
  Fruits, and Others, and returns the three groups as comma-separated lists.
 
180
  # -------------------------------------------------------------------
181
  HEADERS = {'User-Agent': 'Mozilla/5.0'}
182
 
183
+ def buscar_web(query, num_results=8):
184
  retriever = TavilySearchAPIRetriever(api_key=os.getenv("TAVILY_API_KEY"), k=num_results)
185
  results = retriever.invoke(query)
186
  # Formatear resultados para pasarlos al LLM
 
265
  except Exception as e:
266
  return f"Error read_excel_data: {e}"
267
 
268
+ def classify_botanical_foods(items_list_str: str) -> str:
269
  """
270
  Splits an input list of foods (English names) into botanical Vegetables,
271
  Fruits, and Others, and returns the three groups as comma-separated lists.