Update apis/chat_api.py

apis/chat_api.py

@@ -337,6 +337,11 @@ class ChatAPIApp:
 
 
     class EmbeddingRequest(BaseModel):
+        model: str
+        input: str
+        options: Optional[dict] = None
+
+    class OllamaEmbeddingRequest(BaseModel):
         model: str
         prompt: str
         options: Optional[dict] = None
@@ -345,7 +350,7 @@ class ChatAPIApp:
         try:
             model = request.model
             model_kwargs = request.options
-            embeddings = self.embeddings[model].encode(request.prompt, convert_to_tensor=True)#, **model_kwargs)
+            embeddings = self.embeddings[model].encode(request.input, convert_to_tensor=True)#, **model_kwargs)
             return {
                 "object":"list", 
                 "data":[
@@ -357,7 +362,7 @@ class ChatAPIApp:
         except ValueError as e:
             raise HTTPException(status_code=400, detail=str(e))
 
-    def get_embeddings_ollama(self, request: EmbeddingRequest, api_key: str = Depends(extract_api_key)):
+    def get_embeddings_ollama(self, request: OllamaEmbeddingRequest, api_key: str = Depends(extract_api_key)):
         try:
             model = request.model
             model_kwargs = request.options