Skip to main content

A framework designed to produce long-texts with GPT or other large language models.

Project description

🦜🦜🦜 textlong

PyPI version

textlong 的目标是基于大语言模型提供结构化的长文本生成能力。

注意:以下文字全部使用textlong自动生成:

《textlong 使用指南》

1 环境准备

1.1 使用 dotenv 管理环境变量

将 APIKEY 和项目配置保存到.env文件,再加载到进程的环境变量中,这是很好的实践策略。

这需要使用 dotenv 包,它可以帮助我们管理项目中的环境变量。

创建和配置.env文件,你需要在你项目的根目录下创建一个名为.env的文件(注意,文件名以点开始)。在这个文件中,你可以定义你的环境变量,例如:

ZHIPUAI_API_KEY="你的智谱AI API密钥"
TEXTLONG_FOLDER="你的项目目录"

为此,你可能需要先安装 python-dotenv 包:

pip install python-dotenv

然后在 Python 代码中,使用以下代码片段来加载.env文件中的环境变量:

from dotenv import load_dotenv, find_dotenv
load_dotenv(find_dotenv(), override=True)

1.2 申请大语言模型

当你决定申请大语言模型服务时,可以选择美国公司如 OpenAI 或 Claude 提供的服务,也可以选择中国的智谱 AI 等国内服务。这些服务通常都会提供详细的申请流程指引。申请过程中,你需要填写相关信息,并按照指引完成相应的步骤。成功申请后,你会获得一个 API_KEY,这是你调用这些大语言模型服务的唯一凭证。

获得 API_KEY 后,为了确保其安全性和便捷性,你应该将其配置到项目根目录下的.env文件中,例如:

你的.env 文件

ZHIPUAI_API_KEY="YOUR_API_KEY_NAME"

这种方式不仅能保护你的敏感信息,还能让你的代码更加整洁、易于管理。

2 textlong 的安装与加载

2.1 安装 textlong 包

在 Python 中安装 textlong 包非常简单,以下命令会尝试安装最新版本的 textlong:

pip install textlong

为了确保安装的是最新版本,可以在命令中添加--upgrade选项,如下:

pip install --upgrade textlong

3 使用 Project 创作长文

3.1 在 jupyter 环境中使用 textlong

安装 JupyterLab

在使用 textlong 之前,首先需要确保你的环境中安装了 JupyterLab。JupyterLab 是一个交互式的开发环境,非常适合进行数据分析和机器学习项目。你可以通过以下命令安装 JupyterLab:

pip install jupyterlab

安装完成后,可以通过运行以下命令来启动 JupyterLab:

jupyter-lab

建立笔记

在新的笔记中,首先需要导入 textlong 包。在进行导入之前,请确保已经按照前面的步骤正确安装了 textlong 包:

首先,加载你的环境变量,这样你就可以安全地访问 API 密钥:

from dotenv import load_dotenv, find_dotenv
load_dotenv(find_dotenv(), override=True)

接下来,导入 textlong 包和所需的模型:

from textlong import Project
from langchain_zhipu import ChatZhipuAI

现在,你可以初始化一个 Project 对象,如下所示:

p = Project(ChatZhipuAI(model="glm-4"), project_folder="demo")
p

这里project_folder是你为 textlong 项目指定的唯一标识符。确保你使用的是正确的 API 模型名称和项目 ID。

恭喜你,现在可以使用 textlong 包来创作长文了!

3.2 创作提纲

从一个简单的例子开始,假设我们要创作的是一部 10000 字的修仙小说,标题为《我修了个假仙》,主角是夏小兰,男一号是周成,每个章节都将以意外和打脸的线索推动情节。

接下来,使用 textlong 的from_idea方法来创建提纲。该方法接收三个参数:提纲文件的路径、文章创作的任务描述,以及 prompt_id 指定为OUTLINE

task = "请帮我创作10000字的修仙小说,标题为《我修了个假仙》,主角是夏小兰,男一号是周成,每一个章节都适用意外、打脸的线索推动。"
p.from_idea("提纲.md", task, prompt_id="OUTLINE")

当你运行上述代码后,textlong 会根据你的任务描述自动生成一份提纲。提纲将包括小说的章节划分和每个章节的关键情节。你可以根据这个提纲来进行文章的扩写。

生成的结果如下:

# 《我修了个假仙》

## 第一章:星辰之女的觉醒

<OUTLINE>
夏小兰一直被认为拥有神秘力量,却在一次意外中被揭露出她的能力并不如外界传言那般强大,实际上她修了个“假仙”。这次觉醒让她在村中受尽白眼,决心踏上寻找真正修仙之道的旅程。

扩写要求:

- 预估字数:1000
- 创意要点:意外揭露,能力不足,决心追求真相
- 创作思路:通过主角的失落和觉醒,引出后续成长和冒险
- 涉及实体名称:夏小兰,星辰之女

</OUTLINE>

...

3.3 文章扩写

接下来,是根据提纲扩写:

p.from_outline(
    output_file="我修了个假仙人.md",
    input_file="提纲.md"
)

运行上述代码,textlong 将根据“提纲.md”中的结构,生成一篇名为“我修了个假仙人.md”的文章初稿。

也许你希望对生成的文章进行微调,以更好地符合你的创作风格或文章需求。 为此,你可以像下面这样调整from_outline方法的参数:

p.from_outline(
    output_file="我修了个假仙人.md",
    input_file="提纲.md",
    task="多使用人物细节、对话描写、打斗描写,减少抽象叙事"
)

在这个调整后的代码中,我们添加了一个额外的参数task,用于指导 AI 在生成文章时要特别关注的方面。这样,生成的文章将包含更多的人物细节、对话和打斗描写,同时减少抽象的叙述。

生成结果如下:

# 《我修了个假仙》

## 第一章:星辰之女的觉醒

夏小兰,一直被村民视为拥有神秘力量的星辰之女。然而,在那次意外中,她的真实能力被揭穿,原来她修了个“假仙”。村中的嘲笑与白眼,让她痛不欲生。夏小兰站在村口,紧握双拳,眼中闪过坚定:“我一定要找到真正的修仙之道,证明给所有人看!”

那天,阳光透过云层洒在她的脸上,仿佛预示着她即将踏上的艰难旅程。夏小兰迈开坚定的步伐,离开这个让她充满失落的地方,去寻找属于自己的真相。

...

3.4 引用素材管理

可能你希望在创作过程中,在每个步骤都共享一些知识,例如在创作长篇小说时的人物设定。

你可以先使用用from_idea方法生成关于小说人物的设定:

task = "我要写一个修仙小说,主角是夏小兰,男一号是周成,请帮我设想一下这两个人的出身,要非常魔幻。"
p.from_idea("人物设定.md", task)

接下来,通过from_outline方法中的kg_files参数来引用它们。如下示例:

p.from_outline(
    output_file="我修了个假仙人.md",
    input_file="提纲.md",
    task="多使用人物细节、对话描写、打斗描写,减少抽象叙事",
    kg_files=["人物设定.md"]
)

注意,from_idea方法中同样可以使用kg_files参数。

4 一键直出

你在上述项目中使用过的方法都已经被日志记录,因此可以通过save_scripts将你手工执行过的动作保存到 project_script.yml脚本文件,再执行run_scripts实现一键直出。

保存自动化脚本 首先使用save_scripts保存可执行的脚本清单,这会生成或更新项目文件夹中的 project_script.yml文件:

# 加载
p.save_script()

查看project_script.yml

project_script.yml的结构很简单,你也可以收工编辑或对生成或的脚本裁剪:

- cmd: from_idea
  kwargs:
    output_file: 人物设定.md
    task: 我要写一个IT职场小说,主角是夏小兰,男一号是周成,请帮我设想一下这两个人的出身,要非常悲惨。
  modified_at: "2024-06-18 10:31:07"
- cmd: from_idea
  kwargs:
    output_file: 写作提纲.md
    prompt_id: OUTLINE
    task: 请帮我创作500字的职场小说,标题为《我修了个假仙》,主角是夏小兰,男一号是周成,每一个章节都适用意外、打脸的线索推动。
  modified_at: "2024-06-18 10:31:26"

确认这就是你需要的执行脚本,运行脚本重新生成结果:

# 执行
p.run_script()

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

textlong-0.3.7.tar.gz (42.2 kB view hashes)

Uploaded Source

Built Distribution

textlong-0.3.7-py3-none-any.whl (53.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page