Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
@@ -4,7 +4,6 @@ from langchain.agents import AgentExecutor, Tool
|
|
4 |
from langchain.chat_models import ChatGoogleGenerativeAI
|
5 |
import genai
|
6 |
from gemini import GeminiAgent
|
7 |
-
from smolagent_tools import WikipediaSearchTool, WebSearchTool, SmolagentToolWrapper
|
8 |
|
9 |
# Función para configuración de llm
|
10 |
def setup_llm(api_key: str, model_name: str = "gemini-2.0-flash"):
|
@@ -20,8 +19,8 @@ class EnhancedAgent:
|
|
20 |
self.llm = setup_llm(api_key, model_name)
|
21 |
self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
|
22 |
|
|
|
23 |
self.tools = [
|
24 |
-
SmolagentToolWrapper(WikipediaSearchTool()),
|
25 |
Tool(name="web_search", func=self._web_search, description="Realizar búsqueda web avanzada")
|
26 |
]
|
27 |
|
@@ -34,9 +33,9 @@ class EnhancedAgent:
|
|
34 |
def _web_search(self, query: str, domain: Optional[str] = None) -> str:
|
35 |
"""Realiza una búsqueda en la web usando DuckDuckGo con limitación de tasa y reintentos."""
|
36 |
try:
|
37 |
-
|
38 |
-
|
39 |
-
return
|
40 |
except Exception as e:
|
41 |
return f"Error al realizar la búsqueda: {str(e)}"
|
42 |
|
|
|
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"):
|
|
|
19 |
self.llm = setup_llm(api_key, model_name)
|
20 |
self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
|
21 |
|
22 |
+
# Eliminamos las herramientas de smolagent_tools
|
23 |
self.tools = [
|
|
|
24 |
Tool(name="web_search", func=self._web_search, description="Realizar búsqueda web avanzada")
|
25 |
]
|
26 |
|
|
|
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 |
|