Update apis/chat_api.py

apis/chat_api.py

@@ -44,6 +44,11 @@ class ChatAPIApp:
     def get_available_models(self):
         return {"object": "list", "data": AVAILABLE_MODELS_DICTS}
 
+    def get_available_models_ollama(self):
+        ollama_models_dict = AVAILABLE_MODELS_DICTS.copy()
+        ollama_models_dict['name'] = ollama_models_dict.pop('id')
+        return {"object": "list", "models":ollama_models_dict}
+
     def extract_api_key(
         credentials: HTTPAuthorizationCredentials = Depends(HTTPBearer()),
     ):
@@ -323,7 +328,7 @@ class ChatAPIApp:
             "/api/tags",
             summary="Get Available Models Ollama",
             include_in_schema=True,
-        )(self.get_available_models)
+        )(self.get_available_models_ollama)
         
         self.app.get(
             "/readme",