golang 结合 cobra 使用 chatgpt qdrant 实现 ai知识库 cli

潇湘莫大

于 2023-09-22 00:49:42 发布

阅读量500

点赞数

文章标签： golang chatgpt 人工智能

本文链接：https://blog.youkuaiyun.com/qq_21956483/article/details/133153414

版权

golang 结合 cobra 使用 chatgpt qdrant 实现 ai知识库 cli

流程

将数据集通过 openai embedding 得到向量+组装payload,存入 qdrant
用户进行问题搜索,通过 openai embedding 得到向量,从 qdrant 中搜索相似度大于0.8的数据
从 qdrant 中取出数据得到参考答案
将问题标题+参考答案,组装成promot 向gpt进行提问,得到偏向于已有知识库设定的扩展知识回答

kabi 知识库的导入和搜索

仓库地址:https://github.com/webws/embedding-knowledge-base

kabi 是使用 golang 基于 openai chatgpt embedding + qdrant 实现知识库的导入和问答

❯ kabi -h
a local knowledge base, based on chatgpt and qdrant

usage:
  kbai [flags]
  kbai [command]

available commands:
  completion  generate the autocompletion script for the specified shell
  help        help about any command
  import      import data to vector database
  search      ask the knowledge base example: kbai a