jjvelezo commited on
Commit
a267c57
·
verified ·
1 Parent(s): 649464d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -5
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
- search_tool = WebSearchTool()
38
- results = search_tool.search(query, domain)
39
- return results
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