Spaces:
Sleeping
Sleeping
mriusero
commited on
Commit
·
40a900a
1
Parent(s):
7f568ce
fix: add class
Browse files
src/agent/mistral_agent.py
CHANGED
@@ -18,7 +18,7 @@ class MistralAgent:
|
|
18 |
self.api_key = os.getenv("MISTRAL_API_KEY")
|
19 |
self.agent_id = os.getenv("AGENT_ID")
|
20 |
self.client = Mistral(api_key=self.api_key)
|
21 |
-
self.model = "mistral-
|
22 |
self.prompt = None
|
23 |
self.names_to_functions = {
|
24 |
"calculate_sum": calculate_sum,
|
|
|
18 |
self.api_key = os.getenv("MISTRAL_API_KEY")
|
19 |
self.agent_id = os.getenv("AGENT_ID")
|
20 |
self.client = Mistral(api_key=self.api_key)
|
21 |
+
self.model = "mistral-medium-latest"
|
22 |
self.prompt = None
|
23 |
self.names_to_functions = {
|
24 |
"calculate_sum": calculate_sum,
|