lizhilun/AI_NovelGenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update architecture.py

Browse Source
This commit is contained in:
CNlaojing
2025-03-13 07:15:29 +08:00
committed by GitHub
parent 33944c0266
commit 54acd6b6b4
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
novel_generator/architecture.py
+4 -3
View File
@@ -92,7 +92,8 @@ def Novel_architecture_generate(
topic=topic, topic=topic,
genre=genre, genre=genre,
number_of_chapters=number_of_chapters, number_of_chapters=number_of_chapters,
word_number=word_number word_number=word_number,
user_guidance=user_guidance # 修复：添加内容指导
) )
core_seed_result = invoke_with_cleaning(llm_adapter, prompt_core) core_seed_result = invoke_with_cleaning(llm_adapter, prompt_core)
if not core_seed_result.strip(): if not core_seed_result.strip():
@@ -141,7 +142,7 @@ def Novel_architecture_generate(
logging.info("Step3: Generating world_building_prompt ...") logging.info("Step3: Generating world_building_prompt ...")
prompt_world = world_building_prompt.format( prompt_world = world_building_prompt.format(
core_seed=partial_data["core_seed_result"].strip(), core_seed=partial_data["core_seed_result"].strip(),
user_guidance=user_guidance user_guidance=user_guidance # 修复：添加用户指导
) )
world_building_result = invoke_with_cleaning(llm_adapter, prompt_world) world_building_result = invoke_with_cleaning(llm_adapter, prompt_world)
if not world_building_result.strip(): if not world_building_result.strip():
@@ -159,7 +160,7 @@ def Novel_architecture_generate(
core_seed=partial_data["core_seed_result"].strip(), core_seed=partial_data["core_seed_result"].strip(),
character_dynamics=partial_data["character_dynamics_result"].strip(), character_dynamics=partial_data["character_dynamics_result"].strip(),
world_building=partial_data["world_building_result"].strip(), world_building=partial_data["world_building_result"].strip(),
user_guidance=user_guidance user_guidance=user_guidance # 修复：添加用户指导
) )
plot_arch_result = invoke_with_cleaning(llm_adapter, prompt_plot) plot_arch_result = invoke_with_cleaning(llm_adapter, prompt_plot)
if not plot_arch_result.strip(): if not plot_arch_result.strip():