Update apis/chat_api.py

apis/chat_api.py

@@ -125,9 +125,9 @@ class ChatAPIApp:
         return None
 
     class QueryRequest(BaseModel):
-        texts: List[str]
+        input: str
         model_name: str = Field(..., example="bert-base-uncased")
-        api_key: str = Field(..., example="your_hf_api_key_here")
+        encoding_format: str
 
     class ChatCompletionsPostItem(BaseModel):
         model: str = Field(
@@ -187,7 +187,7 @@ class ChatAPIApp:
             data_response = streamer.chat_return_dict(stream_response)
             return data_response
 
-    async def chat_embedding(self, texts, model_name, api_key):
+    async def chat_embedding(self, texts, model_name, api_key, api_key: str = Depends(extract_api_key)):
         api_url = f"https://api-inference.huggingface.co/pipeline/feature-extraction/{model_name}"
         headers = {"Authorization": f"Bearer {api_key}"}
         response = requests.post(api_url, headers=headers, json={"inputs": texts})
@@ -206,7 +206,7 @@ class ChatAPIApp:
             try:
                 for attempt in range(3):  # Retry logic
                     try:
-                        embeddings = await self.chat_embedding(request.texts, request.model_name, request.api_key)
+                        embeddings = await self.chat_embedding(request.input, request.model, request.api_key)
                         data = [
                             {"object": "embedding", "index": i, "embedding": embedding}
                             for i, embedding in enumerate(embeddings)