Spaces:

Jofthomas
/

twitch_streaming

Running

Jofthomas commited on Apr 16

Commit

693f765

verified ·

1 Parent(s): 6a4f566

Update agents.py

Files changed (1) hide show

agents.py CHANGED Viewed

@@ -367,7 +367,7 @@ class MistralAgent(LLMAgentBase):
         user_prompt = f"Current Battle State:\n{battle_state}\n\nChoose the best action by calling the appropriate function ('choose_move' or 'choose_switch')."
         try:
-            response = await self.mistral_client.chat(
                 model=self.model,
                 messages=[
                     {"role": "system", "content": system_prompt},

         user_prompt = f"Current Battle State:\n{battle_state}\n\nChoose the best action by calling the appropriate function ('choose_move' or 'choose_switch')."
         try:
+            response = await self.mistral_client.chat.complete(
                 model=self.model,
                 messages=[
                     {"role": "system", "content": system_prompt},