Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
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 |
-
|
37 |
-
|
38 |
-
return f"Simulando búsqueda web para: {query}"
|
39 |
except Exception as e:
|
40 |
-
return f"Error
|
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."""
|