From 373b85a3f83542762a66c8b8c7bbb10f24703e77 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=A9=AC=E4=B8=80=E4=B8=81?= <1769123563@qq.com>
Date: Fri, 21 Nov 2025 05:33:47 +0800
Subject: [PATCH] Repair the Logic of Automatically Repairing Charts

---
 ReportEngine/utils/chart_repair_api.py | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/ReportEngine/utils/chart_repair_api.py b/ReportEngine/utils/chart_repair_api.py
index 861914d..d892e27 100644
--- a/ReportEngine/utils/chart_repair_api.py
+++ b/ReportEngine/utils/chart_repair_api.py
@@ -145,13 +145,12 @@ def create_llm_repair_functions() -> List:
         def repair_with_report_engine(widget_block: Dict[str, Any], errors: List[str]) -> Optional[Dict[str, Any]]:
             """使用ReportEngine的LLM修复图表"""
             try:
-                from llm_client import LLMClient
+                from ReportEngine.llms import LLMClient
 
                 client = LLMClient(
                     api_key=settings.REPORT_ENGINE_API_KEY,
                     base_url=settings.REPORT_ENGINE_BASE_URL,
                     model_name=settings.REPORT_ENGINE_MODEL_NAME or "gpt-4",
-                    provider="openai"
                 )
 
                 prompt = build_chart_repair_prompt(widget_block, errors)
@@ -180,13 +179,12 @@ def create_llm_repair_functions() -> List:
         def repair_with_forum_engine(widget_block: Dict[str, Any], errors: List[str]) -> Optional[Dict[str, Any]]:
             """使用ForumEngine的LLM修复图表"""
             try:
-                from llm_client import LLMClient
+                from ReportEngine.llms import LLMClient
 
                 client = LLMClient(
                     api_key=settings.FORUM_HOST_API_KEY,
                     base_url=settings.FORUM_HOST_BASE_URL,
                     model_name=settings.FORUM_HOST_MODEL_NAME or "gpt-4",
-                    provider="openai"
                 )
 
                 prompt = build_chart_repair_prompt(widget_block, errors)
@@ -214,13 +212,12 @@ def create_llm_repair_functions() -> List:
         def repair_with_insight_engine(widget_block: Dict[str, Any], errors: List[str]) -> Optional[Dict[str, Any]]:
             """使用InsightEngine的LLM修复图表"""
             try:
-                from llm_client import LLMClient
+                from ReportEngine.llms import LLMClient
 
                 client = LLMClient(
                     api_key=settings.INSIGHT_ENGINE_API_KEY,
                     base_url=settings.INSIGHT_ENGINE_BASE_URL,
                     model_name=settings.INSIGHT_ENGINE_MODEL_NAME or "gpt-4",
-                    provider="openai"
                 )
 
                 prompt = build_chart_repair_prompt(widget_block, errors)
@@ -248,13 +245,12 @@ def create_llm_repair_functions() -> List:
         def repair_with_media_engine(widget_block: Dict[str, Any], errors: List[str]) -> Optional[Dict[str, Any]]:
             """使用MediaEngine的LLM修复图表"""
             try:
-                from llm_client import LLMClient
+                from ReportEngine.llms import LLMClient
 
                 client = LLMClient(
                     api_key=settings.MEDIA_ENGINE_API_KEY,
                     base_url=settings.MEDIA_ENGINE_BASE_URL,
                     model_name=settings.MEDIA_ENGINE_MODEL_NAME or "gpt-4",
-                    provider="openai"
                 )
 
                 prompt = build_chart_repair_prompt(widget_block, errors)