Update apis/chat_api.py

apis/chat_api.py

@@ -116,7 +116,7 @@ class ChatAPIApp:
         model_name: str = Field(..., example="bert-base-uncased")
         api_key: str = Field(..., example="your_hf_api_key_here")
 
-    async def send_request_to_hugging_face(texts, model_name, api_key):
+    async def chat_embedding(texts, model_name, 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})
@@ -186,11 +186,13 @@ class ChatAPIApp:
             data_response = streamer.chat_return_dict(stream_response)
             return data_response
 
+        
+
     async def embedding(request: QueryRequest):
         try:
             for attempt in range(3):  # Retry logic
                 try:
-                    embeddings = await send_request_to_hugging_face(request.texts, request.model_name, request.api_key)
+                    embeddings = await chat_embedding(request.texts, request.model_name, request.api_key)
                     data = [
                         {"object": "embedding", "index": i, "embedding": embedding}
                         for i, embedding in enumerate(embeddings)