bibibi12345 commited on
Commit
1531ddf
·
1 Parent(s): f5ecd80

added openai model back

Browse files
Files changed (1) hide show
  1. app/routes/models_api.py +4 -4
app/routes/models_api.py CHANGED
@@ -90,14 +90,14 @@ async def list_models(fastapi_request: Request, api_key: str = Depends(get_api_k
90
  for model_id in raw_vertex_models: # Iterate through the original list of base models
91
  if EXPERIMENTAL_MARKER in model_id:
92
  suffixed_id = f"{model_id}{OPENAI_DIRECT_SUFFIX}"
93
- print("===========")
94
- print(suffixed_id)
95
  # Check if already added (e.g. if remote config somehow already listed it)
96
  if not any(m['id'] == suffixed_id for m in dynamic_models_data):
97
  dynamic_models_data.append({
98
  "id": suffixed_id, "object": "model", "created": current_time, "owned_by": "google",
99
  "permission": [], "root": model_id, "parent": None
100
  })
101
- final_models_data_map = {m["id"]: m for m in dynamic_models_data}
 
 
102
 
103
- return {"object": "list", "data": list(final_models_data_map.values())}
 
90
  for model_id in raw_vertex_models: # Iterate through the original list of base models
91
  if EXPERIMENTAL_MARKER in model_id:
92
  suffixed_id = f"{model_id}{OPENAI_DIRECT_SUFFIX}"
 
 
93
  # Check if already added (e.g. if remote config somehow already listed it)
94
  if not any(m['id'] == suffixed_id for m in dynamic_models_data):
95
  dynamic_models_data.append({
96
  "id": suffixed_id, "object": "model", "created": current_time, "owned_by": "google",
97
  "permission": [], "root": model_id, "parent": None
98
  })
99
+ # final_models_data_map = {m["id"]: m for m in dynamic_models_data}
100
+ # model_list = list(final_models_data_map.values())
101
+ # model_list.sort()
102
 
103
+ return {"object": "list", "data": final_models_data_map}