LongjoeAgent/tools/sale_tools.py

from langchain.tools import tool
from .base_tool import call_csharp_api, get_tool_prompt


def get_sale_amt_default_config():
    """get_sale_amt 工具的默认配置"""
    return {
        "get_sale_amt": {
            "基础描述": "获取指定时间范围的销售金额，按月汇总",
            "入参说明": {
                "backend_url": "后端API地址",
                "token": "认证令牌",
                "funtion_name": "函数名称; get_sale_amt_by_month:按月汇总销售额; get_sale_amt_by_day:按天汇总销售额; get_sale_amt_by_produce:产品销售额; get_sale_amt_by_cus:客户销售额",
                "firstdate": "开始日期，格式YYYY-MM-DD",
                "lastdate": "结束日期，格式YYYY-MM-DD 23:59:59",
            },
            "返回值说明": {
                "格式": "一个包含销售金额的字符串",
            },
            "输出格式要求": [
                "以自然语言描述形式输出，不要使用表格",
                "重复信息要总结归纳，精简显示",
            ],
        }
    }


tool_description = get_tool_prompt("get_sale_amt", get_sale_amt_default_config())


def get_sale_amt_func(
    backend_url: str, token: str, funtion_name: str, firstdate: str, lastdate: str
) -> str:
    """实际的函数实现"""
    print(f"正在获取销售金额{funtion_name}，时间范围：{firstdate} 至 {lastdate}")

    return call_csharp_api(
        backend_url,
        token,
        "sale_data_ai",
        funtion_name,
        {"arg_firstdate": firstdate, "arg_lastdate": lastdate},
    )


get_sale_amt_func.__doc__ = tool_description
get_sale_amt = tool(get_sale_amt_func)