Update api_usage.py

api_usage.py

@@ -200,11 +200,8 @@ def check_key_gemini_availability(key):
         #print(e)
         return 'Error while making request.', None
 
-def check_key_azure_availability(key):
-    try:
-        endpoint = key.split(';')[0]
-        api_key = key.split(';')[1]
-        
+def check_key_azure_availability(endpoint, api_key):
+    try:        
         if endpoint.startswith('http'):
             url = f'{endpoint}/openai/models?api-version=2023-03-15-preview'
         else:
@@ -287,21 +284,21 @@ def get_azure_status(endpoint, api_key, deployments_list):
     has_turbo = False
     list_model = {}
     for model, deploy in azure_deploy.items():
-        if model == 'gpt-4-32k':
+        if model.startswith('gpt-4-32k'):
             list_model[model] = deploy
             has_32k = True
-        elif model == 'gpt-4':
+        elif model.startswith('gpt-4'):
             list_model[model] = deploy
             has_gpt4 = True
-        elif model == 'gpt-35-turbo':
+        elif model.startswith('gpt-35-turbo'):
             list_model[model] = deploy
             has_turbo = True
                 
     if not list_model: #has_32k == False and has_gpt4 == False and has_turbo == False:
-        return "No GPT model to check.", has_turbo, has_gpt4, has_gpt4turbo, has_32k 
+        return "No GPT model to check.", has_32k, has_gpt4turbo, has_gpt4, has_turbo
     else:
         if has_gpt4:
-            has_gpt4turbo = check_gpt4turbo(key, list_model['gpt-4'])
+            has_gpt4turbo = check_gpt4turbo(endpoint, api_key, list_model['gpt-4'])
         
         pozz_res = {}