gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update src/llm_models/utils_model.py

Browse Source 
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Todysheep
2025-06-23 16:30:25 +08:00
committed by GitHub
parent c7a804e286
commit 7961a1f04c
1 changed files with 7 additions and 2 deletions
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()