如何在图数据库上添加语义层

您可以使用数据库查询从 Neo4j 等图数据库中检索信息。一种选择是使用大型语言模型（LLM）生成 Cypher 语句。虽然这种方法提供了极佳的灵活性，但解决方案可能不稳定，无法始终生成精确的 Cypher 语句。与其生成 Cypher 语句，我们不如将 Cypher 模板作为工具，在语义层中实现，供 LLM 代理进行交互。

设置

首先，获取所需的包并设置环境变量

%pip install --upgrade --quiet  langchain langchain-neo4j langchain-openai

在本指南中，我们默认使用 OpenAI 模型，但您可以根据需要替换为其他模型提供商的模型。

import getpass
import os

os.environ["OPENAI_API_KEY"] = getpass.getpass()

# Uncomment the below to use LangSmith. Not required.
# os.environ["LANGSMITH_API_KEY"] = getpass.getpass()
# os.environ["LANGSMITH_TRACING"] = "true"

 ········

接下来，我们需要定义 Neo4j 凭据。请按照这些安装步骤来设置 Neo4j 数据库。

os.environ["NEO4J_URI"] = "bolt://:7687"
os.environ["NEO4J_USERNAME"] = "neo4j"
os.environ["NEO4J_PASSWORD"] = "password"

以下示例将创建一个与 Neo4j 数据库的连接，并使用有关电影及其演员的示例数据填充该数据库。

from langchain_neo4j import Neo4jGraph

graph = Neo4jGraph(refresh_schema=False)

# Import movie information

movies_query = """
LOAD CSV WITH HEADERS FROM 
'https://raw.githubusercontent.com/tomasonjo/blog-datasets/main/movies/movies_small.csv'
AS row
MERGE (m:Movie {id:row.movieId})
SET m.released = date(row.released),
    m.title = row.title,
    m.imdbRating = toFloat(row.imdbRating)
FOREACH (director in split(row.director, '|') | 
    MERGE (p:Person {name:trim(director)})
    MERGE (p)-[:DIRECTED]->(m))
FOREACH (actor in split(row.actors, '|') | 
    MERGE (p:Person {name:trim(actor)})
    MERGE (p)-[:ACTED_IN]->(m))
FOREACH (genre in split(row.genres, '|') | 
    MERGE (g:Genre {name:trim(genre)})
    MERGE (m)-[:IN_GENRE]->(g))
"""

graph.query(movies_query)

API 参考：Neo4jGraph

[]

带有 Cypher 模板的自定义工具

语义层包含向 LLM 暴露的各种工具，LLM 可以使用这些工具与知识图谱进行交互。这些工具的复杂性各不相同。您可以将语义层中的每个工具视为一个函数。

我们将实现的功能是检索有关电影或其演员的信息。

description_query = """
MATCH (m:Movie|Person)
WHERE m.title CONTAINS $candidate OR m.name CONTAINS $candidate
MATCH (m)-[r:ACTED_IN|IN_GENRE]-(t)
WITH m, type(r) as type, collect(coalesce(t.name, t.title)) as names
WITH m, type+": "+reduce(s="", n IN names | s + n + ", ") as types
WITH m, collect(types) as contexts
WITH m, "type:" + labels(m)[0] + "\ntitle: "+ coalesce(m.title, m.name) 
       + "\nyear: "+coalesce(m.released,"") +"\n" +
       reduce(s="", c in contexts | s + substring(c, 0, size(c)-2) +"\n") as context
RETURN context LIMIT 1
"""


def get_information(entity: str) -> str:
    try:
        data = graph.query(description_query, params={"candidate": entity})
        return data[0]["context"]
    except IndexError:
        return "No information was found"

您可以观察到，我们已经定义了用于检索信息的 Cypher 语句。因此，我们可以避免生成 Cypher 语句，并仅使用 LLM 代理来填充输入参数。为了向 LLM 代理提供何时使用该工具及其输入参数的额外信息，我们将该函数封装为一个工具。

from typing import Optional, Type

from langchain_core.tools import BaseTool
from pydantic import BaseModel, Field


class InformationInput(BaseModel):
    entity: str = Field(description="movie or a person mentioned in the question")


class InformationTool(BaseTool):
    name: str = "Information"
    description: str = (
        "useful for when you need to answer questions about various actors or movies"
    )
    args_schema: Type[BaseModel] = InformationInput

    def _run(
        self,
        entity: str,
    ) -> str:
        """Use the tool."""
        return get_information(entity)

    async def _arun(
        self,
        entity: str,
    ) -> str:
        """Use the tool asynchronously."""
        return get_information(entity)

API 参考：BaseTool

LangGraph 代理

我们将使用 LangGraph 实现一个简单的 ReAct 代理。

该代理包含一个 LLM 和工具步骤。当我们与代理交互时，我们将首先调用 LLM 来决定是否应使用工具。然后我们将运行一个循环

如果代理表示要执行操作（即调用工具），我们将运行工具并将结果传递回代理。如果代理没有要求运行工具，我们将完成（响应用户）。

代码实现非常简单。首先，我们将工具绑定到 LLM，并定义助手步骤。

from langchain_core.messages import HumanMessage, SystemMessage
from langchain_openai import ChatOpenAI
from langgraph.graph import MessagesState

llm = ChatOpenAI(model="gpt-4o")

tools = [InformationTool()]
llm_with_tools = llm.bind_tools(tools)

# System message
sys_msg = SystemMessage(
    content="You are a helpful assistant tasked with finding and explaining relevant information about movies."
)


# Node
def assistant(state: MessagesState):
    return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])]}

API 参考：HumanMessage | SystemMessage | ChatOpenAI

接下来我们定义 LangGraph 流程。

from IPython.display import Image, display
from langgraph.graph import END, START, StateGraph
from langgraph.prebuilt import ToolNode, tools_condition

# Graph
builder = StateGraph(MessagesState)

# Define nodes: these do the work
builder.add_node("assistant", assistant)
builder.add_node("tools", ToolNode(tools))

# Define edges: these determine how the control flow moves
builder.add_edge(START, "assistant")
builder.add_conditional_edges(
    "assistant",
    # If the latest message (result) from assistant is a tool call -> tools_condition routes to tools
    # If the latest message (result) from assistant is a not a tool call -> tools_condition routes to END
    tools_condition,
)
builder.add_edge("tools", "assistant")
react_graph = builder.compile()

# Show
display(Image(react_graph.get_graph(xray=True).draw_mermaid_png()))

API 参考：StateGraph | ToolNode | tools_condition

现在我们用一个示例问题来测试一下这个工作流程。

input_messages = [HumanMessage(content="Who played in the Casino?")]
messages = react_graph.invoke({"messages": input_messages})
for m in messages["messages"]:
    m.pretty_print()

================================[1m Human Message [0m=================================

Who played in the Casino?
==================================[1m Ai Message [0m==================================
Tool Calls:
  Information (call_j4usgFStGtBM16fuguRaeoGc)
 Call ID: call_j4usgFStGtBM16fuguRaeoGc
  Args:
    entity: Casino
=================================[1m Tool Message [0m=================================
Name: Information

type:Movie
title: Casino
year: 1995-11-22
ACTED_IN: Robert De Niro, Joe Pesci, Sharon Stone, James Woods
IN_GENRE: Drama, Crime

==================================[1m Ai Message [0m==================================

The movie "Casino," released in 1995, features the following actors:

- Robert De Niro
- Joe Pesci
- Sharon Stone
- James Woods

The film is in the Drama and Crime genres.

设置​

带有 Cypher 模板的自定义工具​

LangGraph 代理​

设置

带有 Cypher 模板的自定义工具

LangGraph 代理