wuzhiqiang/ai-admin-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 文档检索接口

Browse Source
This commit is contained in:
martsforever
2025-09-07 20:57:44 +08:00
parent 949b4eda8b
commit 3c7683a827
2 changed files with 18 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/controller/add_knowledge_route.py
+5 -7
View File
@@ -7,16 +7,14 @@ from starlette.requests import Request
from app.utils.db_utils import AsyncSessionDep
from app.utils.knowledge_utils import knowledge_service
from app.utils.milvus_utils import milvus_service
from app.utils.milvus_utils import milvus_service, KnowledgeQueryParam
def add_knowledge_route(app):
# @app.get("/knowledge/search")
# async def knowledge_search(text: str = Param()):
# print("text", text)
# query_result = await milvus_service.async_search(text or "hello")
# return query_result
#
@app.post("/knowledge/search")
async def knowledge_search(param: KnowledgeQueryParam):
return await milvus_service.async_search(param)
# @app.post("/knowledge/embed_text")
# async def knowledge_embed_text(text_list: List[str]):
# id_list = await next_id(len(text_list))