jjvelezo commited on
Commit
c10c11f
·
verified ·
1 Parent(s): 38a441d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -15,10 +15,10 @@ def setup_llm(api_key: str, model_name: str = "gemini-2.0-flash"):
15
 
16
  # Herramientas adicionales
17
  class EnhancedAgent:
18
- def __init__(self, api_key: str, model_name: str = "gemini-2.0-flash"):
19
- self.api_key = api_key
20
  self.model_name = model_name
21
- self.llm = setup_llm(api_key, model_name)
22
  self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
23
 
24
  # Eliminamos las herramientas de smolagent_tools
 
15
 
16
  # Herramientas adicionales
17
  class EnhancedAgent:
18
+ def __init__(self, model_name: str = "gemini-2.0-flash"):
19
+ self.api_key = os.getenv("GOOGLE_API_KEY", "") # Asume que está definida en Hugging Face Secrets
20
  self.model_name = model_name
21
+ self.llm = setup_llm(self.api_key, self.model_name)
22
  self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
23
 
24
  # Eliminamos las herramientas de smolagent_tools