Spaces:
Running
Running
Update agents.py
Browse files
agents.py
CHANGED
@@ -367,7 +367,7 @@ class MistralAgent(LLMAgentBase):
|
|
367 |
user_prompt = f"Current Battle State:\n{battle_state}\n\nChoose the best action by calling the appropriate function ('choose_move' or 'choose_switch')."
|
368 |
|
369 |
try:
|
370 |
-
response = await self.mistral_client.chat(
|
371 |
model=self.model,
|
372 |
messages=[
|
373 |
{"role": "system", "content": system_prompt},
|
|
|
367 |
user_prompt = f"Current Battle State:\n{battle_state}\n\nChoose the best action by calling the appropriate function ('choose_move' or 'choose_switch')."
|
368 |
|
369 |
try:
|
370 |
+
response = await self.mistral_client.chat.complete(
|
371 |
model=self.model,
|
372 |
messages=[
|
373 |
{"role": "system", "content": system_prompt},
|