Update src/refinement/refinement.py

src/refinement/refinement.py

@@ -15,7 +15,7 @@ from playwright.sync_api import sync_playwright
 
 
 class VLMCommenter:
-    def __init__(self, api_key, prompt, provider="openai", model_name=None):
+    def __init__(self, api_key, prompt, provider="openai", model_name=None, config=None):
         """
         :param api_key: API Key
         :param prompt: 提示词文本
@@ -94,7 +94,7 @@ class VLMCommenter:
 
 
 class LLMReviser:
-    def __init__(self, api_key, prompt, provider="openai", model_name=None):
+    def __init__(self, api_key, prompt, provider="openai", model_name=None, config=None):
         """
         :param api_key: API Key
         :param prompt: 提示词文本
@@ -317,8 +317,8 @@ def refine_one_slide(input_path, output_path, prompts, outline, max_iterations,
 
     platform = "gemini" if "gemini" in model.lower() else "openai"
 
-    vlm = VLMCommenter(api_key, commenter_prompt, provider=platform, model_name=model)
-    reviser = LLMReviser(api_key, reviser_prompt, provider=platform, model_name=model)
+    vlm = VLMCommenter(api_key, commenter_prompt, provider=platform, model_name=model, config=config)
+    reviser = LLMReviser(api_key, reviser_prompt, provider=platform, model_name=model, config=config)
 
     current_input = input_path
     critic_his = ""