Spaces:
Running
Running
Update agents.py
Browse files
agents.py
CHANGED
@@ -11,7 +11,7 @@ from google import genai
|
|
11 |
from google.genai import types
|
12 |
|
13 |
# --- Mistral AI ---
|
14 |
-
from mistralai
|
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 =
|
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())
|