From 338db8d7cf59be844e386038aba78309a7b68032 Mon Sep 17 00:00:00 2001
From: CNlaojing <154053522+CNlaojing@users.noreply.github.com>
Date: Wed, 5 Mar 2025 02:16:04 +0800
Subject: [PATCH] Update common.py

---
 novel_generator/common.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/novel_generator/common.py b/novel_generator/common.py
index f31fed7..1a620da 100644
--- a/novel_generator/common.py
+++ b/novel_generator/common.py
@@ -44,16 +44,24 @@ def debug_log(prompt: str, response_content: str):
 
 def invoke_with_cleaning(llm_adapter, prompt: str) -> str:
     """
-    调用 LLM，增加重试和清洗逻辑
-    如果多次失败，则返回空字符串以继续流程，而不是中断。
+    增强版调用方法，支持自定义prompt结构
     """
     def _invoke(prompt):
         return llm_adapter.invoke(prompt)
 
     response = call_with_retry(func=_invoke, max_retries=3, fallback_return="", prompt=prompt)
+    
     if not response:
         logging.warning("No response from model after retry. Return empty.")
         return ""
+    
+    # 增强清洗逻辑
     cleaned_text = remove_think_tags(response)
+    
+    # 移除可能的多余标记
+    cleaned_text = re.sub(r'^```markdown\s*', '', cleaned_text, flags=re.IGNORECASE)
+    cleaned_text = re.sub(r'\s*```$', '', cleaned_text)
+    
     debug_log(prompt, cleaned_text)
     return cleaned_text.strip()
+