工具调用处理

概述

提供检测以及解析工具调用参数的实用工具。

检测工具调用

检测消息是否包含工具调用，核心函数是 has_tool_calling。

代码示例

import datetime
from langchain_core.tools import tool
from langchain_dev_utils.tool_calling import has_tool_calling

@tool
def get_current_time() -> str:
    """获取当前时间戳"""
    return str(datetime.datetime.now().timestamp())

response = model.bind_tools([get_current_time]).invoke("现在几点了？")
print(has_tool_calling(response))

解析工具调用参数

提供一个实用函数来解析工具调用参数，从消息中提取参数信息，核心函数是 parse_tool_calling。

代码示例

import datetime
from langchain_core.tools import tool
from langchain_dev_utils.tool_calling import has_tool_calling, parse_tool_calling

@tool
def get_current_time() -> str:
    """获取当前时间戳"""
    return str(datetime.datetime.now().timestamp())

response = model.bind_tools([get_current_time]).invoke("现在几点了？")

if has_tool_calling(response):
    name, args = parse_tool_calling(
        response, first_tool_call_only=True
    )
    print(name, args)