mriusero commited on
Commit
40a900a
·
1 Parent(s): 7f568ce

fix: add class

Browse files
Files changed (1) hide show
  1. src/agent/mistral_agent.py +1 -1
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-large"
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,