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="查询对象")