Spaces:
Build error
Build error
battle timer on
Browse files
agents.py
CHANGED
@@ -236,6 +236,7 @@ class GeminiAgent(LLMAgentBase):
|
|
236 |
def __init__(self, api_key: str = None, model: str = "gemini-1.5-flash", avatar: str = "steven", *args, **kwargs):
|
237 |
# Set avatar before calling parent constructor
|
238 |
kwargs['avatar'] = avatar
|
|
|
239 |
super().__init__(*args, **kwargs)
|
240 |
self.model_name = model
|
241 |
used_api_key = api_key or os.environ.get("GOOGLE_API_KEY")
|
@@ -308,6 +309,7 @@ class OpenAIAgent(LLMAgentBase):
|
|
308 |
def __init__(self, api_key: str = None, model: str = "gpt-4o", avatar: str = "giovanni", *args, **kwargs):
|
309 |
# Set avatar before calling parent constructor
|
310 |
kwargs['avatar'] = avatar
|
|
|
311 |
super().__init__(*args, **kwargs)
|
312 |
self.model = model
|
313 |
used_api_key = api_key or os.environ.get("OPENAI_API_KEY")
|
@@ -371,6 +373,7 @@ class MistralAgent(LLMAgentBase):
|
|
371 |
def __init__(self, api_key: str = None, model: str = "mistral-large-latest", avatar: str = "alder", *args, **kwargs):
|
372 |
# Set avatar before calling parent constructor
|
373 |
kwargs['avatar'] = avatar
|
|
|
374 |
super().__init__(*args, **kwargs)
|
375 |
self.model = model
|
376 |
used_api_key = api_key or os.environ.get("MISTRAL_API_KEY")
|
|
|
236 |
def __init__(self, api_key: str = None, model: str = "gemini-1.5-flash", avatar: str = "steven", *args, **kwargs):
|
237 |
# Set avatar before calling parent constructor
|
238 |
kwargs['avatar'] = avatar
|
239 |
+
kwargs['start_timer_on_battle_start'] = True
|
240 |
super().__init__(*args, **kwargs)
|
241 |
self.model_name = model
|
242 |
used_api_key = api_key or os.environ.get("GOOGLE_API_KEY")
|
|
|
309 |
def __init__(self, api_key: str = None, model: str = "gpt-4o", avatar: str = "giovanni", *args, **kwargs):
|
310 |
# Set avatar before calling parent constructor
|
311 |
kwargs['avatar'] = avatar
|
312 |
+
kwargs['start_timer_on_battle_start'] = True
|
313 |
super().__init__(*args, **kwargs)
|
314 |
self.model = model
|
315 |
used_api_key = api_key or os.environ.get("OPENAI_API_KEY")
|
|
|
373 |
def __init__(self, api_key: str = None, model: str = "mistral-large-latest", avatar: str = "alder", *args, **kwargs):
|
374 |
# Set avatar before calling parent constructor
|
375 |
kwargs['avatar'] = avatar
|
376 |
+
kwargs['start_timer_on_battle_start'] = True
|
377 |
super().__init__(*args, **kwargs)
|
378 |
self.model = model
|
379 |
used_api_key = api_key or os.environ.get("MISTRAL_API_KEY")
|