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'
|
||
# }
|