From 0399ca60dd2452d6acace21ca23f5ff57a510a65 Mon Sep 17 00:00:00 2001
From: ymq1 <safecorner@163.com>
Date: Fri, 22 Aug 2025 11:59:45 +0800
Subject: [PATCH] bugfix

---
 llmengine/base_chat_llm.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/llmengine/base_chat_llm.py b/llmengine/base_chat_llm.py
index 54954ac..7c3419a 100644
--- a/llmengine/base_chat_llm.py
+++ b/llmengine/base_chat_llm.py
@@ -145,6 +145,7 @@ class BaseChatLLM:
 						kwargs=kwargs)
 		thread.start()
 		txt = ''
+		i_tokens = o_tokens = 0
 		i = 0
 		for d in self.output_generator(streamer):
 			if i == 0:
@@ -153,8 +154,8 @@ class BaseChatLLM:
 			if d['choices'][0]['finish_reason'] != 'stop':
 				txt += d['choices'][0]['delta']['content']
 			else:
-				i_tokens = d['input_tokens']
-				o_tokens = d['output_tokens']
+				o_tokens = d.get('output_tokens', 0)
+		i_tokens  = input_len
 
 		t2 = time()
 		return {
@@ -164,7 +165,6 @@ class BaseChatLLM:
 			"model":self.model_id,
 			"response_time": t2 - t1,
 			"finish_time": t3 - t1,
-			"output_tokens": output_tokens,
 			"choices":[
 				{
 					"index":0,