add history to chat apis
This commit is contained in:
parent
2d49746a8d
commit
c17572b6a6
|
|
@ -11,9 +11,9 @@ from fastapi.middleware.cors import CORSMiddleware
|
||||||
from starlette.responses import RedirectResponse
|
from starlette.responses import RedirectResponse
|
||||||
from server.chat import (chat, knowledge_base_chat, openai_chat,
|
from server.chat import (chat, knowledge_base_chat, openai_chat,
|
||||||
search_engine_chat)
|
search_engine_chat)
|
||||||
# from server.knowledge_base import (list_kbs, create_kb, delete_kb,
|
from server.knowledge_base import (list_kbs, create_kb, delete_kb,
|
||||||
# list_docs, upload_doc, delete_doc,
|
list_docs, upload_doc, delete_doc,
|
||||||
# update_doc, recreate_vector_store)
|
update_doc, recreate_vector_store)
|
||||||
from server.utils import BaseResponse, ListResponse
|
from server.utils import BaseResponse, ListResponse
|
||||||
|
|
||||||
nltk.data.path = [NLTK_DATA_PATH] + nltk.data.path
|
nltk.data.path = [NLTK_DATA_PATH] + nltk.data.path
|
||||||
|
|
@ -58,52 +58,52 @@ def create_app():
|
||||||
tags=["Chat"],
|
tags=["Chat"],
|
||||||
summary="与搜索引擎对话")(search_engine_chat)
|
summary="与搜索引擎对话")(search_engine_chat)
|
||||||
|
|
||||||
# # Tag: Knowledge Base Management
|
# Tag: Knowledge Base Management
|
||||||
# app.get("/knowledge_base/list_knowledge_bases",
|
app.get("/knowledge_base/list_knowledge_bases",
|
||||||
# tags=["Knowledge Base Management"],
|
tags=["Knowledge Base Management"],
|
||||||
# response_model=ListResponse,
|
response_model=ListResponse,
|
||||||
# summary="获取知识库列表")(list_kbs)
|
summary="获取知识库列表")(list_kbs)
|
||||||
#
|
|
||||||
# app.post("/knowledge_base/create_knowledge_base",
|
app.post("/knowledge_base/create_knowledge_base",
|
||||||
|
tags=["Knowledge Base Management"],
|
||||||
|
response_model=BaseResponse,
|
||||||
|
summary="创建知识库"
|
||||||
|
)(create_kb)
|
||||||
|
|
||||||
|
app.delete("/knowledge_base/delete_knowledge_base",
|
||||||
|
tags=["Knowledge Base Management"],
|
||||||
|
response_model=BaseResponse,
|
||||||
|
summary="删除知识库"
|
||||||
|
)(delete_kb)
|
||||||
|
|
||||||
|
app.get("/knowledge_base/list_docs",
|
||||||
|
tags=["Knowledge Base Management"],
|
||||||
|
response_model=ListResponse,
|
||||||
|
summary="获取知识库内的文件列表"
|
||||||
|
)(list_docs)
|
||||||
|
|
||||||
|
app.post("/knowledge_base/upload_doc",
|
||||||
|
tags=["Knowledge Base Management"],
|
||||||
|
response_model=BaseResponse,
|
||||||
|
summary="上传文件到知识库"
|
||||||
|
)(upload_doc)
|
||||||
|
|
||||||
|
app.delete("/knowledge_base/delete_doc",
|
||||||
|
tags=["Knowledge Base Management"],
|
||||||
|
response_model=BaseResponse,
|
||||||
|
summary="删除知识库内的文件"
|
||||||
|
)(delete_doc)
|
||||||
|
|
||||||
|
# app.post("/knowledge_base/update_doc",
|
||||||
# tags=["Knowledge Base Management"],
|
# tags=["Knowledge Base Management"],
|
||||||
# response_model=BaseResponse,
|
# response_model=BaseResponse,
|
||||||
# summary="创建知识库"
|
# summary="上传文件到知识库,并删除另一个文件"
|
||||||
# )(create_kb)
|
# )(update_doc)
|
||||||
#
|
|
||||||
# app.delete("/knowledge_base/delete_knowledge_base",
|
app.post("/knowledge_base/recreate_vector_store",
|
||||||
# tags=["Knowledge Base Management"],
|
tags=["Knowledge Base Management"],
|
||||||
# response_model=BaseResponse,
|
summary="根据content中文档重建向量库,流式输出处理进度。"
|
||||||
# summary="删除知识库"
|
)(recreate_vector_store)
|
||||||
# )(delete_kb)
|
|
||||||
#
|
|
||||||
# app.get("/knowledge_base/list_docs",
|
|
||||||
# tags=["Knowledge Base Management"],
|
|
||||||
# response_model=ListResponse,
|
|
||||||
# summary="获取知识库内的文件列表"
|
|
||||||
# )(list_docs)
|
|
||||||
#
|
|
||||||
# app.post("/knowledge_base/upload_doc",
|
|
||||||
# tags=["Knowledge Base Management"],
|
|
||||||
# response_model=BaseResponse,
|
|
||||||
# summary="上传文件到知识库"
|
|
||||||
# )(upload_doc)
|
|
||||||
#
|
|
||||||
# app.delete("/knowledge_base/delete_doc",
|
|
||||||
# tags=["Knowledge Base Management"],
|
|
||||||
# response_model=BaseResponse,
|
|
||||||
# summary="删除知识库内的文件"
|
|
||||||
# )(delete_doc)
|
|
||||||
#
|
|
||||||
# # app.post("/knowledge_base/update_doc",
|
|
||||||
# # tags=["Knowledge Base Management"],
|
|
||||||
# # response_model=BaseResponse,
|
|
||||||
# # summary="上传文件到知识库,并删除另一个文件"
|
|
||||||
# # )(update_doc)
|
|
||||||
#
|
|
||||||
# app.post("/knowledge_base/recreate_vector_store",
|
|
||||||
# tags=["Knowledge Base Management"],
|
|
||||||
# summary="根据content中文档重建向量库,流式输出处理进度。"
|
|
||||||
# )(recreate_vector_store)
|
|
||||||
return app
|
return app
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue