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

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +1 -1
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},