简单的LangGraph示例

丹宇码农

已于 2024-05-01 09:26:45 修改

阅读量2.2k

点赞数 7

CC 4.0 BY-SA版权

分类专栏： AI 文章标签： langchain LangGraph 多智能体 Demo

于 2024-05-01 09:21:10 首次发布

本文链接：https://blog.youkuaiyun.com/happyweb/article/details/138366645

在学习智能体，然后又接触到LangGraph，参照文档尝试了一个简单的LangGraph demo。

一、环境准备：

pip install langchain
pip install langchain_openai
pip install langgraph

二、代码：

from typing import TypedDict, Annotated, Sequence
import operator
from langchain_core.messages import BaseMessage
from langchain.tools.render import format_tool_to_openai_function
from langchain_openai import ChatOpenAI
from langgraph.prebuilt import ToolExecutor
from langchain_community.tools.tavily_search import TavilySearchResults
from langgraph.prebuilt import ToolInvocation
import json
from langchain_core.messages import FunctionMessage
from langgraph.graph import StateGraph, END
from langchain_core.messages import HumanMessage

# Import things that are needed generically
from langchain.pydantic_v1 import BaseModel, Field
from langchain.tools import BaseTool, StructuredTool, tool

# 加载 .env 到环境变量,这样就能读取到 .env文件中的 OPENAI_API_KEY和OPENAI_BASE_URL这个设置
from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv())

# 自定义工具
@tool
def search(query: str) -> str:
    """Look up things online."""
    print(f"search: {query}")
    return "sunny"
    
    
@tool
def multiply(a: int, b: int) -> int:
    """Multiply two numbers."""
    return a * b    

tools = [search,multiply]

tool_executor = ToolExecutor(tools)


# We will set streaming=True so that we can stream t