携带错误信息的重试

完整代码如下：

# !/user/bin/env python
# coding: utf-8
"""
@Time       : 2025/4/29 14:40
@Auther     : clip@gmail.com
@File       : 3.携带错误信息的重试.py
"""
from typing import Any

import dotenv
from langchain_core.messages import AIMessage, ToolMessage, HumanMessage, ToolCall
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.runnables import RunnableConfig
from langchain_core.tools import tool
from langchain_openai import ChatOpenAI


dotenv.load_dotenv()


class CustomToolException(Exception):
    '''自定义的工具错误异常'''

    def __init__(self, tool_call: ToolCall, exception: Exception) -> None:
        super().__init__()
        self.tool_call = tool_call
        self.exception = exception


@tool
def complex_tool(int_arg: int, float_arg: float, dict_arg: dict) -> float:
    '''使用复杂工具进行复杂计算操作'''
    return int_arg + float_arg

def tool_custom_exception(msg: AIMessage, config: RunnableConfig) ->  Any:
    try:
        return complex_tool.invoke(msg.tool_calls[0]['args'], config)
    except Exception as e:
        raise CustomToolException(msg.tool_calls[0], e)

def exception_to_messages(inputs: dict) -> dict:
    # 1.从inputs中分离出异常信息
    print(inputs)
    exception = inputs.pop('exception')
    print(exception)
    print(exception.tool_call)
    # 2.根据异常信息组装占位消息列表
    messages = [
        AIMessage(content='', tool_calls=[exception.tool_call]),
        ToolMessage(tool_call_id=exception.tool_call['id'], content=str(exception.exception)),
        HumanMessage(content='最后一次工具调用引发了异常，请分析异常信息，使用更正的参数再次调用该工具')
    ]
    inputs['last_output'] = messages
    return inputs


# 1.创建prompt,并预留占位符，用于存储错误输出信息
prompt = ChatPromptTemplate.from_messages([
    ('human', '{query}'),
    ('placeholder', '{last_output}')
])

# 2.创建大语言模型并绑定该工具
llm = ChatOpenAI(model='gpt-4o', temperature=0)
llm_with_tool = llm.bind_tools(tools=[complex_tool])

# 3.创建链并执行工具
chain = prompt | llm_with_tool | tool_custom_exception
self_correcting_chain = chain.with_fallbacks(
    [exception_to_messages | chain], exception_key='exception'
)

# 3.调用链
print(self_correcting_chain.invoke({'query': '使用复杂工具，对应的参数为5和12.3'}))