feat: Update core agent logic, code execution utilities, and LLM configuration.

2026-01-07 16:41:38 +08:00
parent 3a2f90aef5
commit 621e546b43
7 changed files with 73 additions and 14 deletions
--- a/data_analysis_agent.py
+++ b/data_analysis_agent.py
@@ -89,7 +89,11 @@ class DataAnalysisAgent:
                return self._handle_generate_code(response, yaml_data)

        except Exception as e:
-            print(f"⚠️ 解析响应失败: {str(e)}，按generate_code处理")
+            print(f"⚠️ 解析响应失败: {str(e)}，尝试提取代码并按generate_code处理")
+            # 即使YAML解析失败，也尝试提取代码
+            extracted_code = extract_code_from_response(response)
+            if extracted_code:
+                 return self._handle_generate_code(response, {"code": extracted_code})
            return self._handle_generate_code(response, {})

    def _handle_analysis_complete(