34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
|
import json
|
|||
|
|
import re
|
|||
|
|
import concurrent
|
|||
|
|
from fastapi.concurrency import run_in_threadpool
|
|||
|
|
from langchain.tools import YouTubeSearchTool
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
from server.chat import utils
|
|||
|
|
from server.chat.utils import doc_to_list, solve_knowledge_map,solve_mental_data,shared_variable
|
|||
|
|
from server.knowledge_base.kb_doc_api import search_docs
|
|||
|
|
from server.knowledge_base.model.kb_document_model import DocumentWithVSId
|
|||
|
|
from configs import kb_config
|
|||
|
|
|
|||
|
|
|
|||
|
|
def do_nothing(query: str):
|
|||
|
|
"""
|
|||
|
|
什么都不做不调用工具设置
|
|||
|
|
"""
|
|||
|
|
# 以下三行逻辑控制反问，如果不需要反问注释掉即可，但可能会带来的问题包括该agent将始终会使用工具
|
|||
|
|
|
|||
|
|
return f"\n不需要调用工具了"
|
|||
|
|
|
|||
|
|
def get_next_tip(query: str):
|
|||
|
|
"""
|
|||
|
|
什么都不做不调用工具设置
|
|||
|
|
"""
|
|||
|
|
# 以下三行逻辑控制反问，如果不需要反问注释掉即可，但可能会带来的问题包括该agent将始终会使用工具
|
|||
|
|
res = utils.get_shared_variable(query)
|
|||
|
|
res["END"] = "ok"
|
|||
|
|
utils.set_shared_variable(query,res)
|
|||
|
|
|
|||
|
|
return f"\n提示：你已经使用过环节跳转了，可以开始输出正文了"
|
|||
|
|
|
|||
|
|
class doNothingInput(BaseModel):
|
|||
|
|
query: str = Field(...,description="查询对象")
|