Jofthomas commited on
Commit
5c1a880
·
1 Parent(s): 88beb6d

battle timer on

Browse files
Files changed (1) hide show
  1. agents.py +3 -0
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")