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 05:49:19 +08:00
committed by GitHub
parent d0a5d9bcee
commit 51fe16b18f
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
novel_generator/architecture.py
+11 -3
View File
@@ -55,6 +55,7 @@ def Novel_architecture_generate(
number_of_chapters: int, number_of_chapters: int,
word_number: int, word_number: int,
filepath: str, filepath: str,
user_guidance: str = "", # 新增参数
temperature: float = 0.7, temperature: float = 0.7,
max_tokens: int = 2048, max_tokens: int = 2048,
timeout: int = 600 timeout: int = 600
@@ -105,7 +106,10 @@ def Novel_architecture_generate(
# Step2: 角色动力学 # Step2: 角色动力学
if "character_dynamics_result" not in partial_data: if "character_dynamics_result" not in partial_data:
logging.info("Step2: Generating character_dynamics_prompt ...") logging.info("Step2: Generating character_dynamics_prompt ...")
prompt_character = character_dynamics_prompt.format(core_seed=partial_data["core_seed_result"].strip()) prompt_character = character_dynamics_prompt.format(
core_seed=partial_data["core_seed_result"].strip(),
user_guidance=user_guidance
)
character_dynamics_result = invoke_with_cleaning(llm_adapter, prompt_character) character_dynamics_result = invoke_with_cleaning(llm_adapter, prompt_character)
if not character_dynamics_result.strip(): if not character_dynamics_result.strip():
logging.warning("character_dynamics_prompt generation failed.") logging.warning("character_dynamics_prompt generation failed.")
@@ -135,7 +139,10 @@ def Novel_architecture_generate(
# Step3: 世界观 # Step3: 世界观
if "world_building_result" not in partial_data: if "world_building_result" not in partial_data:
logging.info("Step3: Generating world_building_prompt ...") logging.info("Step3: Generating world_building_prompt ...")
prompt_world = world_building_prompt.format(core_seed=partial_data["core_seed_result"].strip()) prompt_world = world_building_prompt.format(
core_seed=partial_data["core_seed_result"].strip(),
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():
logging.warning("world_building_prompt generation failed.") logging.warning("world_building_prompt generation failed.")
@@ -151,7 +158,8 @@ def Novel_architecture_generate(
prompt_plot = plot_architecture_prompt.format( prompt_plot = plot_architecture_prompt.format(
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
) )
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():