小说生成Agent系统

项目描述

这是一个基于Python和PostgreSQL的智能小说生成代理系统，具备以下核心功能：

• 自动生成小说章节
• 记忆管理和上下文追踪
• 数据持久化存储
• 基于OpenAI的内容生成

系统架构

• 数据库层：PostgreSQL
• 生成模型：OpenAI GPT
• 记忆管理：自定义内存追踪系统

安装依赖

pip install -r requirements.txt

配置说明

1. 配置PostgreSQL数据库连接
2. 设置OpenAI API Key
3. 根据需要调整生成参数

使用示例

from novel_agent import NovelAgentController

# 初始化Agent
novel_agent = NovelAgentController("your_openai_api_key")

# 生成新章节
new_chapter = novel_agent.create_new_chapter(
    genre="科幻", 
    characters="主角张三"
)

功能模块

• 数据库管理
• 内容生成
• 记忆追踪
• 故事情节分析

注意事项

• 需要有效的OpenAI API Key
• 确保PostgreSQL数据库正常运行
• 建议设置API调用频率限制