Jofthomas commited on
Commit
60ec1bc
·
verified ·
1 Parent(s): 7604820

Update agents.py

Browse files
Files changed (1) hide show
  1. agents.py +0 -4
agents.py CHANGED
@@ -12,7 +12,6 @@ from google.genai import types
12
 
13
  # --- Mistral AI ---
14
  from mistralai.async_client import MistralAsyncClient
15
- from mistralai.exceptions import MistralAPIError
16
 
17
  # --- Poke-Env ---
18
  from poke_env.player import Player
@@ -396,9 +395,6 @@ class MistralAgent(LLMAgentBase):
396
  # Model decided not to call a tool
397
  return {"error": f"Mistral did not return a tool call. Response: {message.content}"}
398
 
399
- except MistralAPIError as e:
400
- print(f"Error during Mistral API call: {e}")
401
- return {"error": f"Mistral API Error: {str(e)}"}
402
  except Exception as e:
403
  print(f"Error during Mistral API call: {e}")
404
  return {"error": f"Unexpected error: {str(e)}"}
 
12
 
13
  # --- Mistral AI ---
14
  from mistralai.async_client import MistralAsyncClient
 
15
 
16
  # --- Poke-Env ---
17
  from poke_env.player import Player
 
395
  # Model decided not to call a tool
396
  return {"error": f"Mistral did not return a tool call. Response: {message.content}"}
397
 
 
 
 
398
  except Exception as e:
399
  print(f"Error during Mistral API call: {e}")
400
  return {"error": f"Unexpected error: {str(e)}"}