BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update src/llm_models/utils_model.py 

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
pull/1057/head
Todysheep 2025-06-23 16:30:25 +08:00 committed by GitHub
parent c7a804e286
commit 7961a1f04c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/llm_models/utils_model.py
View File

@ -130,8 +130,13 @@ class LLMRequest:
self.pri_out = model.get("pri_out", 0)
self.max_tokens = model.get("max_tokens", global_config.model.model_max_output_length)
# print(f"max_tokens: {self.max_tokens}")
self.custom_params = model.get("custom_params", "{}")
self.custom_params = json.loads(self.custom_params)
custom_params_str = model.get("custom_params", "{}")
try:
self.custom_params = json.loads(custom_params_str)
except json.JSONDecodeError as e:
logger.error(f"Invalid JSON in custom_params for model '{self.model_name}': {custom_params_str}")
self.custom_params = {}
# 获取数据库实例
self._init_database()