Merge pull request #18 from BenedictKing/main

🐛 fix: api could be a list
yym68686 · Oct 13, 2024 · cb70818 · cb70818
2 parents 1808468 + 859400e
commit cb70818
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/utils.py b/utils.py
@@ -39,6 +39,8 @@ def update_initial_model(api_url, api):
     try:
         endpoint = BaseAPI(api_url=api_url)
         endpoint_models_url = endpoint.v1_models
+        if isinstance(api, list):
+            api = api[0]
         response = httpx.get(
             endpoint_models_url,
             headers={"Authorization": f"Bearer {api}"},
@@ -55,6 +57,8 @@ def update_initial_model(api_url, api):
         return models_id
     except Exception as e:
         print("error:", e)
+        import traceback
+        traceback.print_exc()
         return []
 
 def update_config(config_data):