mirror of
https://gitee.com/270580156/weiyu.git
synced 2026-05-14 19:27:53 +00:00
48 lines
2.0 KiB
Python
48 lines
2.0 KiB
Python
|
'''
|
|||
|
|
Author: jackning 270580156@qq.com
|
|||
|
|
Date: 2024-08-29 09:55:35
|
|||
|
|
LastEditors: jackning 270580156@qq.com
|
|||
|
|
LastEditTime: 2024-08-31 19:19:57
|
|||
|
|
Description: bytedesk.com https://github.com/Bytedesk/bytedesk
|
|||
|
|
Please be aware of the BSL license restrictions before installing Bytedesk IM –
|
|||
|
|
selling, reselling, or hosting Bytedesk IM as a service is a breach of the terms and automatically terminates your rights under the license.
|
|||
|
|
仅支持企业内部员工自用，严禁私自用于销售、二次销售或者部署SaaS方式销售
|
|||
|
|
Business Source License 1.1: https://github.com/Bytedesk/bytedesk/blob/main/LICENSE
|
|||
|
|
contact: 270580156@qq.com
|
|||
|
|
技术/商务联系：270580156@qq.com
|
|||
|
|
Copyright (c) 2024 by bytedesk.com, All Rights Reserved.
|
|||
|
|
'''
|
|||
|
|
#
|
|||
|
|
import logging
|
|||
|
|
from fastapi import APIRouter, Request
|
|||
|
|
from app.redisVector import myredisVector
|
|||
|
|
|
|||
|
|
router = APIRouter(
|
|||
|
|
prefix='/chat',
|
|||
|
|
tags=['chat v1 apis']
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# http://127.0.0.1:9007/api/v1/chat/query?kbuid=1461090177253570&query=报名条件
|
|||
|
|
# http://127.0.0.1:9007/api/v1/chat/query?kbuid=1461487033909519&query=DataStructure
|
|||
|
|
@router.get("/query")
|
|||
|
|
def query(kbuid: str, query: str):
|
|||
|
|
# 测试搜索结果
|
|||
|
|
search_results = myredisVector.search_docs(kbUid=kbuid, query=query)
|
|||
|
|
# search_results = myredisVector.search_as_retriever(kbUid=kbuid, query=query)
|
|||
|
|
return {
|
|||
|
|
"results": search_results
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# # http://127.0.0.1:9007/api/v1/chat/stream?kbuid=1461090177253570&query=报名条件
|
|||
|
|
# # http://127.0.0.1:9007/api/v1/chat/stream?kbuid=1461487033909519&query=DataStructure
|
|||
|
|
# @router.get("/stream")
|
|||
|
|
# async def query(kbuid: str, query: str):
|
|||
|
|
# logging.info(f'stream: {kbuid}, {query}')
|
|||
|
|
# # TODO: query from db/cache, if match then return, if not then goto llm
|
|||
|
|
# search_results = myredisVector.search_docs(kbUid=kbuid, query=query)
|
|||
|
|
# logging.info(f'搜索结果: count={ len(search_results) }')
|
|||
|
|
# await myredisVector.query_llm(messageUid='', threadTopic='', kbUid=kbuid, question=query, search_results=search_results)
|
|||
|
|
# return {
|
|||
|
|
# 'message': 'ok'
|
|||
|
|
# }
|