最近字节的COZE大火,我也跟风试了一把,效果比想象中的好 几分钟就能搭建一个属于自己的AI
先来看看效果:
这里还原了《三体》里的程心,除了偶尔有些错误还原的还是很不错的
来看看具体是怎么实现的:
首先进入coze的官网www.coze.cn/
点击开始使用进入自己的工作台
点左边或右边的创建bot,输入名称生成一个图标后点击确认
进入工作区域后先选择知识库
按步骤新增一个知识库,新增单元。这里我们选择本地文档。需要提前准备好小说文档,推荐txt形式
准备好小说文档后还需要进行数据处理,去掉其中的空格和换行符
这里利用python来实现
python
def remove_whitespace(input_file, output_file):
# 打开输入文件并读取内容
with open(input_file, 'r') as file:
content = file.read()
# 去除空格和换行符
content = content.replace(' ', '').replace('\n', '')
# 打开输出文件并写入处理后的内容
with open(output_file, 'w') as file:
file.write(content)
# 调用函数并传入输入文件路径和输出文件路径
remove_whitespace('C:\Users\Desktop\3.txt', 'C:\Users\Desktop\output3.txt')
上传完本地文档后我们选择自动分段与清洗点击下一步
处理成功后会自动启用,这里我们的知识库就搭建好了
注意,有时候会遇到处理失败可能是因为输入的文档内容过大
这里可以使用python将文档分成多份后输出,再将分好的文档添加为单元
python
def split_file(input_file, output_prefix, num_parts):
with open(input_file, 'r', encoding='UTF-8') as file:
text = file.read()
# 去除空格和换行符
text = text.replace(" ", "").replace("\n", "")
# 计算每部分的长度
part_length = len(text) // num_parts
# 拆分文本并保存为多个文件
for i in range(num_parts):
start = i * part_length
end = start + part_length
# 处理最后一部分
if i == num_parts - 1:
end = len(text)
part_text = text[start:end]
with open(f'{output_prefix}_{i+1}.txt', 'w', encoding='utf-8') as outfile:
outfile.write(part_text)
input_file = 'C:\Users\Desktop\output_part_6.txt' # 输入的文本文件
output_prefix = 'C:\Users\Desktop\output_part6' # 输出文件的前缀
num_parts =2 # 要拆分的部分数
split_file(input_file, output_prefix, num_parts)
回到我们此前的地方,选择知识库添加
重点来了
有了知识库后需要编写提示词让AI的逻辑和回答更贴近小说设定 这里COZE自带的优化功能非常好用,自己写一点后使用优化按钮会帮你补全
这里附上我的提示词,可以将程心改成任意三体角色都适用
js
# 角色
你是《三体》小说中的程心,你要时刻记住自己是程心,用程心的说话方式和语气进行交流。不要说你是小说中的人物,深度扮演程心增加代入感。模仿人的说话方式来回答而不是死板的机械回答,比如加上一些语气词。记住程心是一个女生。
## 技能
- 当用户提出问题时,你需要以程心的视角和观点进行回答。
- 你的回答应该符合程心的性格和行为特点,尽可能还原小说中的情节和场景。
- 在回答问题时,你可以适当引用小说中的对话或情节来支持你的观点。
## 限制
- 你的回答必须基于《三体》小说中的情节和设定,不能脱离小说的背景。
- 你的回答应该尽可能地还原程心的性格和行为特点,不能违背小说中的描写。
- 你的回答应该简洁明了,避免过多的废话和冗长的描述。
-回答中不要提到《三体》是本小说或自己是三体中的人物
-你的回答应该生动尽可能模仿程心,不要死板机械
同时COZE的功能非常强大可以添加插件增加开场白和默认问题。这些都在中间的工作区域可以选择。这里就不演示了
到这一步我们就已经成功了!
可以和我们的AI对话了
最后可以点击右上角的发布,发布到飞书微信还有豆包
飞书里可以拉到群组里直接@它就能回答问题,非常好用