Update agents.py

agents.py

@@ -233,7 +233,9 @@ class LLMAgentBase(Player):
 # --- Google Gemini Agent ---
 class GeminiAgent(LLMAgentBase):
     """Uses Google Gemini API for decisions."""
-    def __init__(self, api_key: str = None, model: str = "gemini-1.5-flash", *args, **kwargs):
+    def __init__(self, api_key: str = None, model: str = "gemini-1.5-flash", avatar: str = "steven", *args, **kwargs):
+        # Set avatar before calling parent constructor
+        kwargs['avatar'] = avatar
         super().__init__(*args, **kwargs)
         self.model_name = model
         used_api_key = api_key or os.environ.get("GOOGLE_API_KEY")
@@ -303,7 +305,9 @@ class GeminiAgent(LLMAgentBase):
 # --- OpenAI Agent ---
 class OpenAIAgent(LLMAgentBase):
     """Uses OpenAI API for decisions."""
-    def __init__(self, api_key: str = None, model: str = "gpt-4o", *args, **kwargs):
+    def __init__(self, api_key: str = None, model: str = "gpt-4o", avatar: str = "giovanni", *args, **kwargs):
+        # Set avatar before calling parent constructor
+        kwargs['avatar'] = avatar
         super().__init__(*args, **kwargs)
         self.model = model
         used_api_key = api_key or os.environ.get("OPENAI_API_KEY")
@@ -364,7 +368,9 @@ class OpenAIAgent(LLMAgentBase):
 # --- Mistral Agent ---
 class MistralAgent(LLMAgentBase):
     """Uses Mistral AI API for decisions."""
-    def __init__(self, api_key: str = None, model: str = "mistral-large-latest", *args, **kwargs):
+    def __init__(self, api_key: str = None, model: str = "mistral-large-latest", avatar: str = "alder", *args, **kwargs):
+        # Set avatar before calling parent constructor
+        kwargs['avatar'] = avatar
         super().__init__(*args, **kwargs)
         self.model = model
         used_api_key = api_key or os.environ.get("MISTRAL_API_KEY")