jjvelezo commited on
Commit
b015e1c
·
verified ·
1 Parent(s): fc53e98

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -5
agent.py CHANGED
@@ -4,6 +4,8 @@ from langchain.agents import AgentExecutor, Tool
4
  from langchain.chat_models import ChatGoogleGenerativeAI
5
  import genai
6
  from gemini import GeminiAgent
 
 
7
 
8
  # Función para configuración de llm
9
  def setup_llm(api_key: str, model_name: str = "gemini-2.0-flash"):
@@ -31,13 +33,11 @@ class EnhancedAgent:
31
  )
32
 
33
  def _web_search(self, query: str, domain: Optional[str] = None) -> str:
34
- """Realiza una búsqueda en la web usando DuckDuckGo con limitación de tasa y reintentos."""
35
  try:
36
- # Puedes usar otro motor de búsqueda si lo deseas aquí
37
- # Para ilustrar, solo devolveremos una respuesta estática
38
- return f"Simulando búsqueda web para: {query}"
39
  except Exception as e:
40
- return f"Error al realizar la búsqueda: {str(e)}"
41
 
42
  def run(self, query: str) -> str:
43
  """Procesa las consultas del usuario con reintentos y manejo de errores."""
 
4
  from langchain.chat_models import ChatGoogleGenerativeAI
5
  import genai
6
  from gemini import GeminiAgent
7
+ from typing import Optional
8
+ from duckduckgo_search import DDGS
9
 
10
  # Función para configuración de llm
11
  def setup_llm(api_key: str, model_name: str = "gemini-2.0-flash"):
 
33
  )
34
 
35
  def _web_search(self, query: str, domain: Optional[str] = None) -> str:
 
36
  try:
37
+ results = DDGS().text(query, max_results=3)
38
+ return "\n".join([r["body"] for r in results])
 
39
  except Exception as e:
40
+ return f"Error en la búsqueda: {str(e)}"
41
 
42
  def run(self, query: str) -> str:
43
  """Procesa las consultas del usuario con reintentos y manejo de errores."""