Update apis/chat_api.py

apis/chat_api.py

@@ -20,6 +20,19 @@ from mocks.stream_chat_mocker import stream_chat_mock
 
 from fastapi.middleware.cors import CORSMiddleware
 
+
+class EmbeddingResponseItem(BaseModel):
+    object: str = "embedding"
+    index: int
+    embedding: List[float]  # Adjust if your structure is different
+
+class EmbeddingResponse(BaseModel):
+    object: str = "list"
+    data: List[EmbeddingResponseItem]
+    model: str
+    usage: Dict[str, Any]
+
+
 class ChatAPIApp:
     def __init__(self):
         self.app = FastAPI(
@@ -116,18 +129,6 @@ 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(
             default="mixtral-8x7b",