如何为一个Runnable附加回调：一步步的指导
引言
在现代编程中，回调（Callback）是一种重要的技术，可用于控制异步代码执行、事件处理和链式操作等。本文将介绍如何将回调附加到一个Runnable上，通过回调机制来监控和处理运行状态及结果。适用于开发需要动态执行任务的应用场景。
主要内容
1. 回调的基本概念
回调是指一段代码在特定时间点或条件满足时被执行。它通常作为参数传递给另一个代码块，并在某个时刻调用，可以用于处理任务完成后的操作、错误处理等。
2. 自定义回调处理器
创建自定义回调处理器（Handler）可以帮助我们更灵活地处理不同的事件和状态。
3. 链式Runnable的概念
链式Runnable指的是通过将多个Runnable对象按照顺序依次执行的一种编程模式。它可以显著减少代码的复杂度。
4. 在Runnable中附加回调
为了使回调在多个链式Runnable中重复使用，可以通过with_config()方法将回调绑定到整个链上。这可以确保回调在所有子组件中传播。
5. 示例代码
以下是一个完整的示例代码，演示了如何将回调附加到Runnable链中：
from typing import Any, Dict, List
from langchain_anthropic import ChatAnthropic
from langchain_core.callbacks import BaseCallbackHandler
from langchain_core.messages import BaseMessage
from langchain_core.outputs import LLMResult
from langchain_core.prompts import ChatPromptTemplate
# 自定义回调处理器
class LoggingHandler(BaseCallbackHandler):
def on_chat_model_start(
self, serialized: Dict[str, Any], messages: List[List[BaseMessage]], **kwargs
) -> None:
print("Chat model started")
def on_llm_end(self, response: LLMResult, **kwargs) -> None:
print(f"Chat model ended, response: {response}")
def on_chain_start(
self, serialized: Dict[str, Any], inputs: Dict[str, Any], **kwargs
) -> None:
print(f"Chain {serialized.get('name')} started")
def on_chain_end(self, outputs: Dict[str, Any], **kwargs) -> None:
print(f"Chain ended, outputs: {outputs}")
# 回调列表
callbacks = [LoggingHandler()]
# 使用指定模型
llm = ChatAnthropic(model="claude-3-sonnet-20240229")
# 定义提示模板
prompt = ChatPromptTemplate.from_template("What is 1 + {number}?")
# 创建链式操作
chain = prompt | llm
# 将回调绑定到链上
chain_with_callbacks = chain.with_config(callbacks=callbacks)
# 执行链式操作
chain_with_callbacks.invoke({"number": "2"})
6. 常见问题和解决方案
问题一：回调没有被触发
解决方案：确保回调处理器的方法名和参数与事件正确匹配。例如，检查方法名on_chat_model_start和on_llm_end是否正确拼写。
问题二：回调执行顺序混乱
解决方案：检查回调的绑定顺序和链式操作的执行顺序，避免回调绑定在嵌套层次不正确的位置。
总结和进一步学习资源
通过本文，你学到了如何将回调附加到一个Runnable链中，并通过自定义回调处理器灵活处理事件和状态。你可以进一步阅读官方文档和其他相关资源以深入了解：
- ChatAnthropic API # 使用API代理服务提高访问稳定性
- BaseCallbackHandler文档 # 使用API代理服务提高访问稳定性
- BaseMessage文档 # 使用API代理服务提高访问稳定性
- LLMResult文档 # 使用API代理服务提高访问稳定性
- ChatPromptTemplate文档 # 使用API代理服务提高访问稳定性
参考资料
如果这篇文章对你有帮助，欢迎点赞并关注我的博客。您的支持是我持续创作的动力！
---END---
