Update apis/chat_api.py

apis/chat_api.py

@@ -187,7 +187,7 @@ class ChatAPIApp:
             data_response = streamer.chat_return_dict(stream_response)
             return data_response
 
-    async def chat_embedding(self, input, model_name, api_key):
+    async def chat_embedding(self, input, model_name, 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": input})
@@ -206,7 +206,7 @@ class ChatAPIApp:
             try:
                 for attempt in range(3):  # Retry logic
                     try:
-                        embeddings = await self.chat_embedding(request.input, request.model, request.api_key)
+                        embeddings = await self.chat_embedding(request.input, request.model, api_key)
                         data = [
                             {"object": "embedding", "index": i, "embedding": embedding}
                             for i, embedding in enumerate(embeddings)
@@ -214,7 +214,7 @@ class ChatAPIApp:
                         return {
                             "object": "list",
                             "data": data,
-                            "model": request.model_name,
+                            "model": request.model,
                             "usage": {"prompt_tokens": len(request.input), "total_tokens": len(request.input)}
                         }
                     except RuntimeError as e: