Update blueprint.py

novel_generator/blueprint.py

@@ -48,6 +48,7 @@ def Chapter_blueprint_generate(
     llm_model: str,
     filepath: str,
     number_of_chapters: int,
+    user_guidance: str = "",  # 新增参数
     temperature: float = 0.7,
     max_tokens: int = 4096,
     timeout: int = 600
@@ -106,7 +107,8 @@ def Chapter_blueprint_generate(
                 chapter_list=limited_blueprint,
                 number_of_chapters=number_of_chapters,
                 n=current_start,
-                m=current_end
+                m=current_end,
+                user_guidance=user_guidance  # 新增参数
             )
             logging.info(f"Generating chapters [{current_start}..{current_end}] in a chunk...")
             chunk_result = invoke_with_cleaning(llm_adapter, chunk_prompt)
@@ -126,7 +128,8 @@ def Chapter_blueprint_generate(
     if chunk_size >= number_of_chapters:
         prompt = chapter_blueprint_prompt.format(
             novel_architecture=architecture_text,
-            number_of_chapters=number_of_chapters
+            number_of_chapters=number_of_chapters,
+            user_guidance=user_guidance  # 新增参数
         )
         blueprint_text = invoke_with_cleaning(llm_adapter, prompt)
         if not blueprint_text.strip():
@@ -149,7 +152,8 @@ def Chapter_blueprint_generate(
             chapter_list=limited_blueprint,
             number_of_chapters=number_of_chapters,
             n=current_start,
-            m=current_end
+            m=current_end,
+            user_guidance=user_guidance  # 新增参数
         )
         logging.info(f"Generating chapters [{current_start}..{current_end}] in a chunk...")
         chunk_result = invoke_with_cleaning(llm_adapter, chunk_prompt)