【知识图谱】4、LLM大模型结合neo4j图数据库实现AI问答的功能

原创

已于 2024-09-01 19:13:42 修改 · 3.4k 阅读

27 ·

CC 4.0 BY-SA版权

文章标签：

#数据库 #python #fastapi #人工智能 #知识图谱

于 2024-09-01 19:12:44 首次发布

昨天写了一篇文章，使用fastapi直接操作neo4j图数据库插入数据的例子，本文实现LLM大模型结合neo4j图数据库实现AI问答功能。

废话不多说，先上代码

import gradio as gr
from fastapi import FastAPI, HTTPException, Request
from pydantic import BaseModel
from langchain_openai import ChatOpenAI
from langchain.prompts import (
    ChatPromptTemplate,
    MessagesPlaceholder,
    SystemMessagePromptTemplate,
    HumanMessagePromptTemplate,
)
from langchain.chains import LLMChain
from langchain.memory import ConversationBufferMemory
from langchain_community.graphs import Neo4jGraph
from langchain.chains import GraphCypherQAChain
import asyncio
from typing import List
import json

# Initialize FastAPI
app = FastAPI()

# Initialize Neo4j with timeout
try:
    graph = Neo4jGraph(
        url="bolt://localhost:7687",
        username="neo4j",
        password="password",
        database="neo4j",