Ver código fonte

enhancement: OpenAI API

api.together.xyz, api.replicate.com
Timothy J. Baek 1 ano atrás
pai
commit
9832e6edba
1 arquivos alterados com 5 adições e 2 exclusões
  1. 5 2
      backend/apps/openai/main.py

+ 5 - 2
backend/apps/openai/main.py

@@ -171,6 +171,7 @@ async def fetch_url(url, key):
 
 
 
 
 def merge_models_lists(model_lists):
 def merge_models_lists(model_lists):
+    log.info(f"merge_models_lists {model_lists}")
     merged_list = []
     merged_list = []
 
 
     for idx, models in enumerate(model_lists):
     for idx, models in enumerate(model_lists):
@@ -199,14 +200,16 @@ async def get_all_models():
         ]
         ]
 
 
         responses = await asyncio.gather(*tasks)
         responses = await asyncio.gather(*tasks)
+        log.info(f"get_all_models:responses() {responses}")
+
         models = {
         models = {
             "data": merge_models_lists(
             "data": merge_models_lists(
                 list(
                 list(
                     map(
                     map(
                         lambda response: (
                         lambda response: (
                             response["data"]
                             response["data"]
-                            if response and "data" in response
-                            else None
+                            if (response and "data" in response)
+                            else (response if isinstance(response, list) else None)
                         ),
                         ),
                         responses,
                         responses,
                     )
                     )