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")
         | 
