langchain1.X学习笔记-29-中间件Middleware之自定义中间件(二)包装风格钩子

文章目录

• 1 模型初始化
• 2 Wrap-style hooks函数用法
• • 2.1 基本用法(wrap_model_call)
  • • 2.1.1 基于装饰器实现(@wrap_model_call)
    • 2.1.2 基于类实现(继承AgentMiddleware)
    • 2.1.3 场景1：重试逻辑
    • 2.1.4 场景2：响应缓存
    • 2.1.5 场景3：修改系统提示
  • 2.2 基本用法(wrap_tool_call)
  • • 2.2.1 基于装饰器实现(@wrap_tool_call)
    • 2.2.2 基于类实现(继承AgentMiddleware)
    • 2.2.3 场景：用于监控工具
  • 2.3 两种方法的统一
  • 2.4 参数说明

Wrap-style(包装式)：它们像洋葱一样完整包裹了模型或工具的实际调用过程。它们完全掌握被包裹函数的控制权，可以决定是调用0次(直接短路/缓存命中)、调用1次(正常透传)，还是调用多次(实现重试逻辑)。

1 模型初始化

from langchain_openaiimportChatOpenAI from dotenvimportload_dotenvimportos load_dotenv(override=True)#通过ChatOpenAI 连接模型API_KEY=os.getenv("API_KEY")BASE_URL=</