mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-25 17:22:55 +00:00
fix: remove unnecessary print statement from model invocation
This commit is contained in:
parent
05b59a4d35
commit
18b40f3664
|
|
@ -16,7 +16,7 @@ from langchain_openai import ChatOpenAI
|
|||
from langchain_openai.chat_models.base import _create_usage_metadata
|
||||
|
||||
from common.config.tokenizer_manage_config import TokenizerManage
|
||||
|
||||
from common.utils.logger import maxkb_logger
|
||||
|
||||
def custom_get_token_ids(text: str):
|
||||
tokenizer = TokenizerManage.get_tokenizer()
|
||||
|
|
@ -103,13 +103,13 @@ class BaseChatOpenAI(ChatOpenAI):
|
|||
future = executor.submit(super().get_num_tokens_from_messages, messages, tools)
|
||||
try:
|
||||
response = future.result()
|
||||
print("请求成功(未超时)")
|
||||
maxkb_logger.info("请求成功(未超时)")
|
||||
return response
|
||||
except Exception as e:
|
||||
if isinstance(e, ReadTimeout):
|
||||
raise # 继续抛出
|
||||
else:
|
||||
print("except:", e)
|
||||
maxkb_logger.error("except:", e)
|
||||
tokenizer = TokenizerManage.get_tokenizer()
|
||||
return sum([len(tokenizer.encode(get_buffer_string([m]))) for m in messages])
|
||||
|
||||
|
|
|
|||
|
|
@ -159,7 +159,6 @@ class XFSparkSpeechToText(MaxKBBaseModel, BaseSpeechToText):
|
|||
"audio": str(base64.b64encode(buf), 'utf-8'),
|
||||
"encoding": "lame"}
|
||||
}
|
||||
print(d)
|
||||
d = json.dumps(d)
|
||||
await ws.send(d)
|
||||
status = STATUS_CONTINUE_FRAME
|
||||
|
|
|
|||
Loading…
Reference in New Issue