Update apis/chat_api.py

apis/chat_api.py

@@ -116,6 +116,17 @@ class ChatAPIApp:
         model_name: str = Field(..., example="bert-base-uncased")
         api_key: str = Field(..., example="your_hf_api_key_here")
 
+    class EmbeddingResponseItem(BaseModel):
+        object: str
+        index: int
+        embedding: List[List[List[float]]]  # Adjust this based on the actual structure
+    
+    class EmbeddingResponse(BaseModel):
+        object: str
+        data: List[EmbeddingResponseItem]
+        model: str
+        usage: dict
+
 
     class ChatCompletionsPostItem(BaseModel):
         model: str = Field(
@@ -237,7 +248,7 @@ class ChatAPIApp:
                 prefix + "/embedding",  # Use the specific prefix for this route
                 summary="Generate embeddings for the given texts",
                 include_in_schema=include_in_schema,
-                response_model=List  # Adapt based on your actual response model
+                response_model=EmbeddingResponse  # Adapt based on your actual response model
             )(self.embedding)