Spaces:
Running
Running
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")
|