AI大模型应用实战：如何用langchain打造自己的AI工作流

最新推荐文章于 2025-04-29 11:05:00 发布

冻感糕人~

最新推荐文章于 2025-04-29 11:05:00 发布

阅读量919

点赞数 34

文章标签：人工智能 langchain AI大模型 LLM AI 大语言模型大模型应用

本文链接：https://blog.youkuaiyun.com/python12345_/article/details/146884977

版权

什么是 LangChain？

LangChain 是一个功能强大的 AI 框架，专门用于构建基于大语言模型（LLM）的智能应用。它不仅提供了基础的 LLM 调用接口，还通过 Prompt 模板、记忆（Memory）、智能代理（Agents）、知识检索（Retrieval）等模块，让 AI 具备更强的推理、搜索、执行任务的能力

官网：https://www.langchain.com
git：https://github.com/langchain-ai/langchain

什么是AI工作流

工作流，是由各种类型的节点和边组成一个类似流程图的东西。AI工作流，就是融合了AI能力的工作流，例如有些节点就是一个AI大模型。像coze的工作流

工作流可以做什么

有些工作他是有固定流程的，例如hr筛选招聘简历，可能你的输入是一个压缩包，第一步你需要加压压缩包，第二步你需要一个个简历过一遍，然后记录是否符合要求，最后再整理符合要求的简历，给到部门负责人进行第2轮的筛选。这些不管谁做，什么时候做，都有固定的流程步骤，这类的事情，我们就可以做成一个工作流，让他自动的做这些事情。工作流其实并不是什么新事物，是很早就有的概念，也有相关应用，但在AI时代，升级成AI工作流后，工作流就变得异常强大，可以胜任更多以前做不了的事情。

今天就教你不用coze，怎么用langchain写一个工作流

举例

下面这个例子是演示如何根据输入，判断这个输入是属于什么级别的事情，分支节点有：

普通事情
重要事情
重要紧急事情
紧急事情

1.首先定义数据结构

我们需要定义一个贯彻整个流程输入输出的数据结构

// 定义状态注解
const StateAnnotation = Annotation.Root({
  input: Annotation<string>,
  decision: Annotation<string>,
  output: Annotation<string>,
});

2.定义和实现各类节点

// 开始节点：接收输入并进行初步处理
asyncfunction startNode(state: typeof StateAnnotation.State) {
console.log("Executing startNode");
// 假设这里对输入进行了一些初步处理
return { ...state }; // 直接返回状态，不做修改
}

// 路由节点：根据输入文本决定后续路径
asyncfunction router(state: typeof StateAnnotation.State) {
console.log("Executing router");
const { input } = state;
let decision = "handleDefault"; // 默认路径

// 模拟异步操作，延时 1 秒
await sleep(1000);

if (input.includes("urgent") && input.includes("important")) {
    decision = "handleBoth";
  } elseif (input.includes("urgent")) {
    decision = "handleUrgent";
  } elseif (input.includes("important")) {
    decision = "handleImportant";
  }

return { ...state, decision };
}

// 处理紧急任务
asyncfunction handleUrgent(state: typeof StateAnnotation.State) {
console.log("Executing handleUrgent");
return { output: "Urgent task handled" };
}

// 处理重要任务
asyncfunction handleImportant(state: typeof StateAnnotation.State) {
console.log("Executing handleImportant");
return { output: "Important task handled" };
}

// 处理紧急且重要的任务
asyncfunction handleBoth(state: typeof StateAnnotation.State) {
console.log("Executing handleBoth");
return { output: "Both urgent and important tasks handled" };
}

// 处理默认任务
asyncfunction handleDefault(state: typeof StateAnnotation.State) {
console.log("Executing handleDefault",state);
return { output: "Default task handled" };
}

3.构建工作流

其实就是定义节点名字，和连线

// 构建工作流
const workflow = new StateGraph(StateAnnotation)
  .addNode("startNode", startNode)
  .addNode("router", router)
  .addNode("handleUrgent", handleUrgent)
  .addNode("handleImportant", handleImportant)
  .addNode("handleBoth", handleBoth)
  .addNode("handleDefault", handleDefault)
  .addEdge("__start__", "startNode") // 开始节点不是路由节点
  .addEdge("startNode", "router") // 从开始节点到路由节点
  .addConditionalEdges(
    "router",
    (state) => state.decision, // 条件函数：根据 decision 字段决定后续路径
    ["handleUrgent", "handleImportant", "handleBoth", "handleDefault"]
  )
  .compile();

重点说明

addNode 定义流程里面的节点
addEdge 流程里面的连线
addConditionalEdges 条件节点需要采用条件连线，即分支

4.测试

const inputs = [
  "This is an urgent task",
  "This is an important task",
  "This is both urgent and important",
  "This is a regular task"
];

5.输出结果

6.工作流异常处理

1.超时处理工作流的节点可能出现很长时间都没有答复，这个时候我们需要给这个流程配置个超时时间，不能让流程一直卡着。

 const result = await workflow.invoke({ input: "This is an urgent task" }, { timeout: 10000 }); // 设置全局超时时间（整个流程，非单个节点）

2.异常处理工作流执行过程可能出现节点执行异常，包括上述的超时也算异常，异常的处理方式跟普通的代码一样。

(async () => {
 try {
   const result = await workflow.invoke({ input: "This is an urgent task" }, { timeout: 3000 }); // 设置全局超时时间
   console.log(`Output: ${result.output}`);
 } catch (error) {
   console.error(`Workflow failed: ${error.message}`,error);
 }
})();

7.完整代码

import { StateGraph, Annotation } from"@langchain/langgraph";

// 封装 sleep 函数
function sleep(ms: number): Promise<void> {
returnnewPromise((resolve) => setTimeout(resolve, ms));
}

// 定义状态注解
const StateAnnotation = Annotation.Root({
input: Annotation<string>,
decision: Annotation<string>,
output: Annotation<string>,
});

// 开始节点：接收输入并进行初步处理
asyncfunction startNode(state: typeof StateAnnotation.State) {
console.log("Executing startNode");
// 假设这里对输入进行了一些初步处理
return { ...state }; // 直接返回状态，不做修改
}

// 路由节点：根据输入文本决定后续路径
asyncfunction router(state: typeof StateAnnotation.State) {
console.log("Executing router");
const { input } = state;
let decision = "handleDefault"; // 默认路径

// 模拟异步操作，延时 1 秒
await sleep(1000);

if (input.includes("urgent") && input.includes("important")) {
    decision = "handleBoth";
  } elseif (input.includes("urgent")) {
    decision = "handleUrgent";
  } elseif (input.includes("important")) {
    decision = "handleImportant";
  }

return { ...state, decision };
}

// 处理紧急任务
asyncfunction handleUrgent(state: typeof StateAnnotation.State) {
console.log("Executing handleUrgent");
return { output: "Urgent task handled" };
}

// 处理重要任务
asyncfunction handleImportant(state: typeof StateAnnotation.State) {
console.log("Executing handleImportant");
return { output: "Important task handled" };
}

// 处理紧急且重要的任务
asyncfunction handleBoth(state: typeof StateAnnotation.State) {
console.log("Executing handleBoth");
return { output: "Both urgent and important tasks handled" };
}

// 处理默认任务
asyncfunction handleDefault(state: typeof StateAnnotation.State) {
console.log("Executing handleDefault",state);
return { output: "Default task handled" };
}

// 构建工作流
const workflow = new StateGraph(StateAnnotation)
  .addNode("startNode", startNode)
  .addNode("router", router)
  .addNode("handleUrgent", handleUrgent)
  .addNode("handleImportant", handleImportant)
  .addNode("handleBoth", handleBoth)
  .addNode("handleDefault", handleDefault)
  .addEdge("__start__", "startNode") // 开始节点不是路由节点
  .addEdge("startNode", "router") // 从开始节点到路由节点
  .addConditionalEdges(
    "router",
    (state) => state.decision, // 条件函数：根据 decision 字段决定后续路径
    ["handleUrgent", "handleImportant", "handleBoth", "handleDefault"]
  )
  .compile();

// 测试工作流
const inputs = [
"This is an urgent task",
"This is an important task",
"This is both urgent and important",
"This is a regular task"
];

for (const input of inputs) {
console.log(`\nProcessing input: ${input}`);
const result = await workflow.invoke({ input });
console.log(`Output: ${result.output}`);
}

结语

看到这里，大家不知道会不会觉得，工作流是不是把普通简单的事情复杂化了？例如同样的事情，如果不采用工作流的做法，自己写个代码更简单，更快就能完成。确实是这样。但相信你写的代码肯定是把工作流里面的流程写在某个函数里面，这样的代码相比工作流，灵活性就要差很多，例如我们想做一个跟coze一样可以随意变化的工作流，就需要将流程的搭建过程抽象出来，支持动态关联。