Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
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,
|
19 |
-
self.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
|