使用RunnableParallel进行并行执行

码龄2年

关注

原创最新推荐文章于 2026-08-23 21:17:30 发布 252 阅读 GEO检测

使用LangChain中的RunnableParallel进行并行化 RunnableParallel 允许我们并行执行多个 Runnable 对象，并将这些 Runnable 的输出作为映射返回。这对于并行运行独立进程尤其有用，因为映射中的每个 Runnable 都会并行执行。阅读详情

在AI开发中，提高任务的执行效率和响应速度是非常关键的。这篇文章将为你介绍如何使用RunnableParallel来并行执行任务，从而提升AI应用的性能。以下内容将涵盖技术背景、核心原理、代码实现示例、应用场景分析以及实践建议。

技术背景介绍

在LangChain中，RunnableParallel用于同时执行多个任务。它接收的输入是一个字典，其中的值是可执行的对象（如函数）。这些任务会并行运行，并返回一个包含每个任务结果的字典。这种方式可以有效地提高处理效率，尤其在需要同时处理多个独立任务的场景中。

核心原理解析

实现并行化的核心在于任务的独立性。RunnableParallel通过并行执行各子任务，并将结果整合到一个字典中。值得注意的是，在LangChain中，类型转换是自动完成的，这意味着即使不显式包装任务，它们也会被识别和处理为RunnableParallel。

代码实现演示

以下是一个使用RunnableParallel的示例代码，该代码会同时生成关于bear主题的笑话和两行诗：

from langchain_core.prompts import ChatPromptTemplate
from langchain_core.runnables import RunnableParallel
from langchain_openai import ChatOpenAI

# 初始化AI模型
model = ChatOpenAI()

# 笑话生成链
joke_chain = ChatPromptTemplate.from_template("tell me a joke about {topic}") | model

# 诗歌生成链
poem_chain = ChatPromptTemplate.from_template("write a 2-line poem about {topic}") | model

# 使用RunnableParallel并行执行
map_chain = RunnableParallel(joke=joke_chain, poem=poem_chain)

# 执行并输出结果
results = map_chain.invoke({"topic": "bear"})
print(results)