Update messagers/message_outputer.py

messagers/message_outputer.py

@@ -1,5 +1,5 @@
 import json
-import time
+import time 
 
 
 class OpenaiStreamOutputer:
@@ -7,22 +7,16 @@ class OpenaiStreamOutputer:
     Create chat completion - OpenAI API Documentation
     * https://platform.openai.com/docs/api-reference/chat/create
     """
-    
 
     def __init__(self):
-        current_time = int(time.time())
+        current_time = int(time.time()) 
         self.default_data = {
+            "created": current_time,
             "id": "chatcmpl-hugginface",
             "object": "chat.completion.chunk",
-            "created": current_time,
             # "content_type": "Completions",
             "model": "hugginface",
             "choices": [],
-            "usage": {
-                "prompt_tokens": 0,
-                "completion_tokens": 0,
-                "total_tokens": 0
-            }
         }
 
     def data_to_string(self, data={}, content_type=""):
@@ -70,10 +64,4 @@ class OpenaiStreamOutputer:
                     "finish_reason": None,
                 }
             ]
-
-        data["usage"]["prompt_tokens"] += tokens_count
-        data["usage"]["completion_tokens"] += len(content.split())
-        data["usage"]["total_tokens"] = data["usage"]["prompt_tokens"] + data["usage"]["completion_tokens"]
-        
-        return self.data_to_string(data, content_type)
-
+        return self.data_to_string(data, content_type)