LangGraph 模拟大模型调用工具失败后重试直到成功

锋行天下2026-09-15 22:31

模拟用户输入问题后，大模型分析要调用工具，但是工具节点有一定几率调用失败，失败后重新调用，直到成功

python 复制代码 
from langchain.tools import tool
from typing import Literal
from langchain_core.messages import SystemMessage, HumanMessage, ToolMessage, content
from dotenv import load_dotenv
from langgraph.graph import MessagesState, StateGraph,START,END
from rich import print as rprint
from IPython.display import display
from random import randint
load_dotenv()
from langchain.chat_models import init_chat_model
llm=init_chat_model(
  model="deepseek-chat"
)
@tool(parse_docstring=True)
def get_weather(city: str):
  """
  获取指定城市的天气情况

  Args:
      city 城市名称

  """
  return f"{city}天气晴朗，适合玩出游玩"
tool_map={
  "get_weather":get_weather,
}
llm_with_tools=llm.bind_tools([get_weather])
class GlobalState(MessagesState):
  user_input: str
  result: str

def input_node(state:GlobalState):
  value=state["user_input"]
  return {
    "messages": [HumanMessage(value),]
  }
def llm_node(state:MessagesState):
  ai_msg=llm_with_tools.invoke(state["messages"])
  return {
    "messages": [ai_msg]
  }
def router(state:MessagesState)-> Literal["tool_node","output_node"]:
  messages=state["messages"]
  last=messages[-1]
  if last.tool_calls:
    return "tool_node"
  else:
    return "output_node"

def tool_node(state:MessagesState):
  messages=state["messages"].copy()
  ai_msg=messages[-1]
  num=5
  tool_calls=ai_msg.tool_calls
  for tool_item in tool_calls:
    if randint(0,9) < num:
      con="网络错误，工具调用失败，请重试"
    else:
      tool=tool_map[tool_item['name']]
      res=tool.invoke(tool_item)
      con=res.content
    messages.append(ToolMessage(content=con,tool_call_id=tool_item["id"]))
  return {
    "messages": messages
  }
def output_node(state:GlobalState):
  messages=state["messages"]
  return {
    "result":messages[-1].content,
  }
builder=StateGraph(GlobalState)
builder.add_node(input_node)
builder.add_node(llm_node)
builder.add_node(tool_node)
builder.add_node(output_node)

builder.add_edge(START,"input_node")
builder.add_edge("input_node","llm_node")
builder.add_conditional_edges("llm_node",router)
builder.add_edge("tool_node","llm_node")
builder.add_edge("output_node",END)

graph=builder.compile()
display(graph)

res=graph.invoke({
  "user_input": "查询今天上海的天气",
  "messages": [SystemMessage(content="如果工具调用失败，必须重新调用直到成功为止")]
})
rprint(res)

以下输出运行信息

python 复制代码 
{
    'messages': [
        SystemMessage(
            content='如果工具调用失败，必须重新调用直到成功为止',
            additional_kwargs={},
            response_metadata={},
            id='4b3be4ef-f86a-4195-8e43-9c0964907641'
        ),
        HumanMessage(
            content='查询今天上海的天气',
            additional_kwargs={},
            response_metadata={},
            id='e23b10cd-d4d8-4dd4-bc46-1e6896e7db95'
        ),
        AIMessage(
            content='',
            additional_kwargs={'refusal': None},
            response_metadata={
                'token_usage': {
                    'completion_tokens': 38,
                    'prompt_tokens': 279,
                    'total_tokens': 317,
                    'completion_tokens_details': None,
                    'prompt_tokens_details': {
                        'audio_tokens': None,
                        'cache_write_tokens': None,
                        'cached_tokens': 128,
                        'image_tokens': None,
                        'text_tokens': None
                    },
                    'prompt_cache_hit_tokens': 128,
                    'prompt_cache_miss_tokens': 151
                },
                'model_provider': 'deepseek',
                'model_name': 'deepseek-flash',
                'system_fingerprint': 'aeb56401ca74e127821c4f9126dcb669',
                'id': '45365c27-4880-40a2-90a4-4e5990d4562e',
                'finish_reason': 'tool_calls',
                'logprobs': None
            },
            id='lc_run--01a0a54c-8f8e-7041-951b-e4044cd8bfad-0',
            tool_calls=[
                {
                    'name': 'get_weather',
                    'args': {'city': '上海'},
                    'id': 'call_00_sSlG4W0XzN44MjY3alrk3781',
                    'type': 'tool_call'
                }
            ],
            invalid_tool_calls=[],
            usage_metadata={
                'input_tokens': 279,
                'output_tokens': 38,
                'total_tokens': 317,
                'input_token_details': {'cache_read': 128},
                'output_token_details': {}
            }
        ),
        ToolMessage(
            content='网络错误，工具调用失败，请重试',
            id='b01f5ccd-a212-4d51-8584-eade0bfc52eb',
            tool_call_id='call_00_sSlG4W0XzN44MjY3alrk3781'
        ),
        AIMessage(
            content='',
            additional_kwargs={'refusal': None},
            response_metadata={
                'token_usage': {
                    'completion_tokens': 38,
                    'prompt_tokens': 339,
                    'total_tokens': 377,
                    'completion_tokens_details': None,
                    'prompt_tokens_details': {
                        'audio_tokens': None,
                        'cache_write_tokens': None,
                        'cached_tokens': 128,
                        'image_tokens': None,
                        'text_tokens': None
                    },
                    'prompt_cache_hit_tokens': 128,
                    'prompt_cache_miss_tokens': 211
                },
                'model_provider': 'deepseek',
                'model_name': 'deepseek-flash',
                'system_fingerprint': 'aeb56401ca74e127821c4f9126dcb669',
                'id': '705a84d7-a91e-4fef-aeb9-94f0769211e5',
                'finish_reason': 'tool_calls',
                'logprobs': None
            },
            id='lc_run--01a0a54c-91a8-7d30-bf20-a473f4d70501-0',
            tool_calls=[
                {
                    'name': 'get_weather',
                    'args': {'city': '上海'},
                    'id': 'call_00_bUbtuHCra9EjoNEuKAZt7966',
                    'type': 'tool_call'
                }
            ],
            invalid_tool_calls=[],
            usage_metadata={
                'input_tokens': 339,
                'output_tokens': 38,
                'total_tokens': 377,
                'input_token_details': {'cache_read': 128},
                'output_token_details': {}
            }
        ),
        ToolMessage(
            content='网络错误，工具调用失败，请重试',
            id='4c65954a-970c-4adb-867c-1347980350a1',
            tool_call_id='call_00_bUbtuHCra9EjoNEuKAZt7966'
        ),
        AIMessage(
            content='',
            additional_kwargs={'refusal': None},
            response_metadata={
                'token_usage': {
                    'completion_tokens': 38,
                    'prompt_tokens': 399,
                    'total_tokens': 437,
                    'completion_tokens_details': None,
                    'prompt_tokens_details': {
                        'audio_tokens': None,
                        'cache_write_tokens': None,
                        'cached_tokens': 256,
                        'image_tokens': None,
                        'text_tokens': None
                    },
                    'prompt_cache_hit_tokens': 256,
                    'prompt_cache_miss_tokens': 143
                },
                'model_provider': 'deepseek',
                'model_name': 'deepseek-flash',
                'system_fingerprint': 'aeb56401ca74e127821c4f9126dcb669',
                'id': 'eccb0c64-fd61-49e5-82d7-263379046b0b',
                'finish_reason': 'tool_calls',
                'logprobs': None
            },
            id='lc_run--01a0a54c-93cc-7532-bd46-6878fdb6c76b-0',
            tool_calls=[
                {
                    'name': 'get_weather',
                    'args': {'city': '上海'},
                    'id': 'call_00_FyDzz3agPKX6iHrfyogA5964',
                    'type': 'tool_call'
                }
            ],
            invalid_tool_calls=[],
            usage_metadata={
                'input_tokens': 399,
                'output_tokens': 38,
                'total_tokens': 437,
                'input_token_details': {'cache_read': 256},
                'output_token_details': {}
            }
        ),
        ToolMessage(
            content='网络错误，工具调用失败，请重试',
            id='463605cd-ba7e-4179-9eaa-77087554f61e',
            tool_call_id='call_00_FyDzz3agPKX6iHrfyogA5964'
        ),
        AIMessage(
            content='',
            additional_kwargs={'refusal': None},
            response_metadata={
                'token_usage': {
                    'completion_tokens': 38,
                    'prompt_tokens': 459,
                    'total_tokens': 497,
                    'completion_tokens_details': None,
                    'prompt_tokens_details': {
                        'audio_tokens': None,
                        'cache_write_tokens': None,
                        'cached_tokens': 256,
                        'image_tokens': None,
                        'text_tokens': None
                    },
                    'prompt_cache_hit_tokens': 256,
                    'prompt_cache_miss_tokens': 203
                },
                'model_provider': 'deepseek',
                'model_name': 'deepseek-flash',
                'system_fingerprint': 'aeb56401ca74e127821c4f9126dcb669',
                'id': '916b62bd-85bd-4bb4-963b-97d0ee2b3bf8',
                'finish_reason': 'tool_calls',
                'logprobs': None
            },
            id='lc_run--01a0a54c-964a-7540-9e66-46340e12711b-0',
            tool_calls=[
                {
                    'name': 'get_weather',
                    'args': {'city': '上海'},
                    'id': 'call_00_XYyBO6xKEDSBteh2Jfsn3542',
                    'type': 'tool_call'
                }
            ],
            invalid_tool_calls=[],
            usage_metadata={
                'input_tokens': 459,
                'output_tokens': 38,
                'total_tokens': 497,
                'input_token_details': {'cache_read': 256},
                'output_token_details': {}
            }
        ),
        ToolMessage(
            content='上海天气晴朗，适合玩出游玩',
            id='1b7a5c0e-90d7-4130-ac8a-cd0a96690bdf',
            tool_call_id='call_00_XYyBO6xKEDSBteh2Jfsn3542'
        ),
        AIMessage(
            content='今天上海的天气是**晴朗**，适合外出游玩。☀️\n\n（小提示：工具前几次调用失败了，重试后才获取到结
果。）',
            additional_kwargs={'refusal': None},
            response_metadata={
                'token_usage': {
                    'completion_tokens': 33,
                    'prompt_tokens': 517,
                    'total_tokens': 550,
                    'completion_tokens_details': None,
                    'prompt_tokens_details': {
                        'audio_tokens': None,
                        'cache_write_tokens': None,
                        'cached_tokens': 256,
                        'image_tokens': None,
                        'text_tokens': None
                    },
                    'prompt_cache_hit_tokens': 256,
                    'prompt_cache_miss_tokens': 261
                },
                'model_provider': 'deepseek',
                'model_name': 'deepseek-flash',
                'system_fingerprint': 'aeb56401ca74e127821c4f9126dcb669',
                'id': 'e98704ab-a962-4a03-bedf-aa2ee8a165c3',
                'finish_reason': 'stop',
                'logprobs': None
            },
            id='lc_run--01a0a54c-993e-7330-99a8-52ff579a623c-0',
            tool_calls=[],
            invalid_tool_calls=[],
            usage_metadata={
                'input_tokens': 517,
                'output_tokens': 33,
                'total_tokens': 550,
                'input_token_details': {'cache_read': 256},
                'output_token_details': {}
            }
        )
    ],
    'user_input': '查询今天上海的天气',
    'result': 
'今天上海的天气是**晴朗**，适合外出游玩。☀️\n\n（小提示：工具前几次调用失败了，重试后才获取到结果。）'
}
相关推荐
张彦峰ZYF4 小时前
越过边界之前：Reward Seeker 实验如何改写智能体安全的时间逻辑
人工智能·ai agent·anthropic·智能体安全·reward seeker·“控制评价过程”泛化跨领域策略·envbom
拓恒技术4 小时前
无人机巡检AI算法的工程化落地：从需求到交付的5个核心环节
人工智能·算法·无人机
罗西的思考7 小时前
AI 时代的 “房谋杜断”：如何培养能力、品味与判断力
人工智能·算法·机器学习
字节跳动视频云技术团队7 小时前
720P 直升 8K，解密 2026 抖音美好奇妙夜巨幕超分黑科技
人工智能·音视频开发
打工仔折腾 AI7 小时前
在 Ubuntu ARM64 部署 Q1 V3.0 MuJoCo 模型：加载、单步推理与执行器响应解析
linux·运维·人工智能·python·深度学习·ubuntu·性能优化
逐米时代8 小时前
从试点验证到协同深化：制造业AI转型三年路线图设计方法
人工智能
GlobalInfo8 小时前
全球及中国风电叶片检测机器人行业研究报告（2026版）：市场现状、行业前景与占有率分析
运维·网络·人工智能·机器人
stormzhangV8 小时前
AI 越强，失业越严重
人工智能·aigc
夏文强9 小时前
Rerank 与混合检索：怎么让 RAG 从「能用」变成「好用」
人工智能·大模型·向量数据库·rag·检索增强生成
救救孩子把9 小时前
AI面试八股文-23_23-Agent可观测性与生产监控面试题
人工智能·面试·职场和发展·agent·langsmith·langfuase
热门推荐
01国内可直接用、免费额度/永久免费的大模型API清单（含 SiliconFlow、火山、阿里、智谱、百度、Kimi、DeepSeek、DMXAPI 等）02如何新建文件夹？ 电脑新建文件夹的4种方法03GitHub 镜像站点04历年考研数学一、数学二、数学三真题试卷及答案PDF05几个好用的ip纯净度检测网站06苹果iPA游戏软件资源下载网站07昇腾910B系列芯片完全指南：四款型号对比、服务器选型与大模型部署实战08蜘蛛磁力 搜索引擎大全，如何使用蜘蛛磁力查找磁力链接09ai安全工具:CyberStrikeAI安装部署与使用10Windows 10/11 如何退出微软账户并改用本地账户登录（含找不到选项的解决方法）