[fix] (llm_analyzer) 健壮的处理，安全访问LLM服务返回的嵌套字段

2026-09-22 20:01:04 +00:00 · 2025-09-11 22:20:26 +08:00
parent 8d0b21045d
commit f22438eb94
1 changed files with 15 additions and 2 deletions
@@ -56,8 +56,21 @@ class LLMAnalyzer:
                            except Exception as json_err:
                                error_text = await resp.text()
                                logger.error(f"自定义LLM服务商响应JSON解析失败: {json_err}, 内容: {error_text}")
-                            # 兼容 OpenAI 格式
-                            content = response_json["choices"][0]["message"]["content"]
+                                return None
+                            # 兼容 OpenAI 格式，安全访问嵌套字段
+                            content = None
+                            try:
+                                choices = response_json.get("choices")
+                                if choices and isinstance(choices, list) and len(choices) > 0:
+                                    message = choices[0].get("message")
+                                    if message and isinstance(message, dict):
+                                        content = message.get("content")
+                                if content is None:
+                                    logger.error(f"自定义LLM响应格式异常: {response_json}")
+                                    return None
+                            except Exception as key_err:
+                                logger.error(f"自定义LLM响应结构解析失败: {key_err}, 响应内容: {response_json}")
+                                return None
                            # 构造一个兼容原有逻辑的对象
                            class CustomResponse:
                                completion_text = content