From 7961a1f04c08553befcb95fe4e97d97c6c0fb50d Mon Sep 17 00:00:00 2001
From: Todysheep <97968466+Todysheep@users.noreply.github.com>
Date: Mon, 23 Jun 2025 16:30:25 +0800
Subject: [PATCH] Update src/llm_models/utils_model.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
---
 src/llm_models/utils_model.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/llm_models/utils_model.py b/src/llm_models/utils_model.py
index 12f396758..52f5f2139 100644
--- a/src/llm_models/utils_model.py
+++ b/src/llm_models/utils_model.py
@@ -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()