Minor changes for making tests pass.

aisuite/providers/azure_provider.py

@@ -2,8 +2,6 @@
 import json
 from aisuite.provider import Provider
 from aisuite.framework import ChatCompletionResponse
-from typing import override
-import sys
 
 
 class AzureProvider(Provider):
@@ -13,7 +11,6 @@ def __init__(self, **config):
         if not self.api_key:
             raise ValueError("api_key is required in the config")
 
-    @override
     def chat_completions_create(self, model, messages, **kwargs):
         # TODO: Need to decide if we need to use base_url or just ignore it.
         # TODO: Remove the hardcoded region name to use environment variable.

aisuite/providers/gcp_provider.py

@@ -1,11 +1,9 @@
 from aisuite.provider import Provider
-from typing import override
 
 
 class GcpProvider(Provider):
     def __init__(self) -> None:
         pass
 
-    @override
     def chat_completions_create(self, model, messages):
         raise ValueError("GCP Provider not yet implemented.")

aisuite/providers/groq_provider.py

@@ -1,11 +1,9 @@
 from aisuite.provider import Provider
-from typing import override
 
 
 class GroqProvider(Provider):
     def __init__(self) -> None:
         pass
 
-    @override
     def chat_completions_create(self, model, messages):
         raise ValueError("Groq provider not yet implemented.")