Jofthomas commited on
Commit
6a4f566
·
verified ·
1 Parent(s): 6e6b08f

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +2 -2
agents.py CHANGED
@@ -11,7 +11,7 @@ from google import genai
11
  from google.genai import types
12
 
13
  # --- Mistral AI ---
14
- from mistralai.async_client import MistralAsyncClient
15
 
16
  # --- Poke-Env ---
17
  from poke_env.player import Player
@@ -351,7 +351,7 @@ class MistralAgent(LLMAgentBase):
351
  used_api_key = api_key or os.environ.get("MISTRAL_API_KEY")
352
  if not used_api_key:
353
  raise ValueError("Mistral API key not provided or found in MISTRAL_API_KEY env var.")
354
- self.mistral_client = MistralAsyncClient(api_key=used_api_key)
355
 
356
  # Convert standard schema to Mistral's tool format
357
  self.mistral_tools = list(self.standard_tools.values())
 
11
  from google.genai import types
12
 
13
  # --- Mistral AI ---
14
+ from mistralai import Mistral
15
 
16
  # --- Poke-Env ---
17
  from poke_env.player import Player
 
351
  used_api_key = api_key or os.environ.get("MISTRAL_API_KEY")
352
  if not used_api_key:
353
  raise ValueError("Mistral API key not provided or found in MISTRAL_API_KEY env var.")
354
+ self.mistral_client = Mistral(api_key=used_api_key)
355
 
356
  # Convert standard schema to Mistral's tool format
357
  self.mistral_tools = list(self.standard_tools.values())