Update apis/chat_api.py

apis/chat_api.py

@@ -208,6 +208,13 @@ class ChatAPIApp:
                 )
 
             data_response = streamer.chat_return_dict(stream_response)
+            print(data_response)
+            data_response = {
+              "model": data_response.get('model'),
+              "created_at": data_response.get('created'),
+              "response": data_response["choices"][0]["message"]["content"],
+              "done": True,
+            }
             return data_response
         except HfApiException as e:
             raise HTTPException(status_code=e.status_code, detail=e.detail)
@@ -342,7 +349,7 @@ class ChatAPIApp:
             return {
                 "object":"list", 
                 "data":[
-                    "object": "embedding", "index": 0, "embedding": embeddings.tolist()
+                    {"object": "embedding", "index": 0, "embedding": embeddings.tolist()}
                 ], 
                 "model": model, 
                 "usage":{},