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())
|