ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-20 21:23:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Enhance

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-06-01 18:25:20 +02:00
parent 094990e818
commit 5af8f77379
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lollms/functions/story_writing.py
View File

@ -150,7 +150,8 @@ def write_story_section(prompt_ideas: str, llm: Any, story_file_path: str, story
f"{start_header_id_template}story_section_writer{end_header_id_template}"
]
)
new_section = llm.fast_gen(prompt).strip()
new_section = f"## {current_section}\n\n"
new_section += llm.fast_gen(prompt).strip()
# Append the new section to the story file
story_path.write_text(story_content + "\n" + new_section)