neo4j暴露公网ip接口——给大模型联通知识图谱

特别鸣谢

我的领导，我的脑子，我的学习能力，感动了

1. 搭建知识图谱数据库（见上一章博客）

这里不加赘述了，请参考上一篇博客搭建

2. FastApi包装接口

这里注意：NEO4J_URI不得写http:,只能写neo4j或者bolt

还需要注意APP_HOST不得是127.0.0.1，必须是0.0.0.0，不然只有局域网才能访问！

这是.env文件：

bash 复制代码

# app
APP_HOST=0.0.0.0
APP_PORT=8000

# neo4j
NEO4J_URI=neo4j://xxxxxx:7687
NEO4J_USER=neo4j
NEO4J_PASSWORD=你自己的密码
NEO4J_VERSION=5
NEO4J_DATABASE=数据库的名字
NEO4J_PORT=7474

# http://127.0.0.1:8000/api/node/search?name=布洛芬
# http://120.26.32.188:8000/api/node/search?name=布洛芬

这是router的查询相关内容：

python 复制代码

#!/usr/bin/python3

import logging
from fastapi import APIRouter, status
from fastapi.responses import JSONResponse
from utils.neo4j_provider import asyncDriver
from settings import settings

router = APIRouter()


# 定义一个根路由
@router.get("/add")
def add_node():
    # TODO 往 neo4j 里创建新的节点
    data = {
        'code': 0,
        'message': '',
        'data': 'add success'
    }
    return JSONResponse(content=data, status_code=status.HTTP_200_OK)


@router.route("/search")
async def get_search(q: str = None):
    if q is None:
        return []
    cql = """
        MATCH (p:商品名称) 
        WHERE p.name CONTAINS $name 
        RETURN p
    """
    records, _, _ = await asyncDriver.execute_query(
        cql,
        name=q.query_params['name'],
        database_=settings.NEO4J_DATABASE,
        routing_="r",
    )
    
    # 转换节点为可序列化的字典
    serialized_records = []
    for record in records:
        node = record["p"]
        serialized_records.append({
            "id": node.element_id,
            "labels": list(node.labels),
            "properties": dict(node)
        })
    
    return JSONResponse(content=serialized_records, status_code=status.HTTP_200_OK)

这是main.py函数

python 复制代码

import logging
from settings import settings

from app_init import create_application

import colorama

# 初始化 colorama 模块 在控制台、命令行输出彩色文字的模块,可以跨平台使用
colorama.init(autoreset=True)

app = create_application()
# app = FastAPI()

if __name__ == "__main__":
    import uvicorn

    logging.root.setLevel(logging.DEBUG)
    logging.info("Starting on  %s:%d ", settings.APP_HOST, settings.APP_PORT)

    # main:app main下面的 app，相当于注入
    # main: main.py 文件(也可理解为Python模块).
    # app: main.py 中 app = FastAPI()
    # 语句创建的app对象.
    # --reload: 在代码改变后重启服务器，只能在开发的时候使用
    uvicorn.run("main:app", host=settings.APP_HOST, port=settings.APP_PORT)

neo4j_provider.py

python 复制代码

#!/usr/bin/python3
import os

from neo4j import GraphDatabase, AsyncGraphDatabase, basic_auth, Driver, AsyncDriver
from settings import settings


# Neo4j 数据库操作类
class Neo4jProvider:
    """创建 Neo4j 数据库连接"""
    def __init__(self) -> None:
        # 获取环境变量值，如果没有就返回默认值
        self.url = settings.NEO4J_URI
        self.username = settings.NEO4J_USER
        self.password = settings.NEO4J_PASSWORD
        self.neo4j_version = settings.NEO4J_VERSION
        self.database = settings.NEO4J_DATABASE
        self.port = int(settings.NEO4J_PORT)
    # 同步驱动
    def driver(self) -> Driver:
        print("self.url", self.url)
        return GraphDatabase.driver(self.url, auth=basic_auth(self.username, self.password))
    # 异步驱动
    def async_driver(self) -> AsyncDriver:
        return AsyncGraphDatabase.driver(self.url, auth=basic_auth(self.username, self.password))
# 同步驱动。暴露给外面调用
driver = Neo4jProvider().driver()
# 异步驱动。暴露给外面调用
asyncDriver = Neo4jProvider().async_driver()

3. 公网ip查询结果展示

4. 与大模型连接效果展示

（总结不易，欢迎各位金主爸爸打赏，点赞收藏！）