Skip to content

ElementQi/zhipuRAG

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

一个利用FastAPI搭建的基于智普API的RAG模块

需要在main.py中设置好api_key (zhipuAPI)

在当前文件夹下执行:

uvicorn main:app --reload

然后访问

localhost:8000

在选定好index并点击确定后,在对话框中进行对话即可

一个测试的对话例子:

You: 为我总结一下小明干了什么
Model: 还没有载入index, 请先选择index
You: 为我总结一下小明干了什么
Model: 对不起,似乎有些混淆。在您提供的文档中,并没有提到一个名叫小明的人物。您可能是在引用另一个对话或故事。如果您能提供更多关于小明的背景信息或上下文,我会很乐意为您总结他的行动。如果您是指上述文件中的故事,那么我可以为您总结Ava或Leo的行为。请指明您需要关于哪一个人物的总结。
You: Ava或Leo的行为
Model: 当然可以。 Ava的故事总结: Ava是一个充满好奇心的年轻村民,她探索了一个被村民们很少提及的神秘路径。这条路径据说通往一个被遗忘的梦想王国。在探索过程中,Ava发现了一个充满色彩和魔法的清晰,以及一棵结满各种颜色果实的树。这些果实代表着被人们遗弃的梦想。通过品尝果实,Ava找回了她曾经遗忘的绘画梦想。她带着新的灵感和明确的目标回到村庄,成为了村民们深爱的艺术家,并提醒大家追求梦想永远不会太晚。 Leo的故事总结: Leo是一个偶然来到老钟表匠Elias店里的年轻人。Elias以其精湛的钟表工艺而闻名,他有一个能操纵时间的秘密时钟。当Leo为了能和病重的祖母多度过一些时间而感到绝望时,Elias感应到Leo纯洁的心,并将这个神奇的时钟托付给他。Leo珍惜每一刻与祖母相处的时间,使她的最后日子充满了快乐。时钟在完成了它的使命后神秘消失。Elias满意地笑了,因为他知道他的礼物带来了幸福,同时也提醒人们时间是最宝贵的礼物。
You: 简短一些
Model: Ava探索了神秘路径,找回了遗忘的绘画梦想。 Leo得到了能操纵时间的时钟,用来陪伴病重的祖母。

你可以自行在data文件夹中添加被检索的文件

你可以自行选择自己需要的embbeding模型等

注意,从HuggingFace上读取模型可能会用到科学上网

About

A RAG model Based on Zhipu API using FastAPI

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published