特別鳴謝
我的領導,我的腦子,我的學習能力,感動了
1. 搭建知識圖譜數據庫(見上一章博客)
這里不加贅述了,請參考上一篇博客搭建
2. FastApi包裝接口
這里注意:NEO4J_URI不得寫http:,只能寫neo4j或者bolt
還需要注意APP_HOST不得是127.0.0.1,必須是0.0.0.0,不然只有局域網才能訪問!
這是.env文件:
# 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的查詢相關內容:
#!/usr/bin/python3import logging
from fastapi import APIRouter, status
from fastapi.responses import JSONResponse
from utils.neo4j_provider import asyncDriver
from settings import settingsrouter = 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函數
import logging
from settings import settingsfrom app_init import create_applicationimport colorama# 初始化 colorama 模塊 在控制臺、命令行輸出彩色文字的模塊,可以跨平臺使用
colorama.init(autoreset=True)app = create_application()
# app = FastAPI()if __name__ == "__main__":import uvicornlogging.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
#!/usr/bin/python3
import osfrom neo4j import GraphDatabase, AsyncGraphDatabase, basic_auth, Driver, AsyncDriver
from settings import settings# Neo4j 數據庫操作類
class Neo4jProvider:"""創建 Neo4j 數據庫連接"""def __init__(self) -> None:# 獲取環境變量值,如果沒有就返回默認值self.url = settings.NEO4J_URIself.username = settings.NEO4J_USERself.password = settings.NEO4J_PASSWORDself.neo4j_version = settings.NEO4J_VERSIONself.database = settings.NEO4J_DATABASEself.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. 與大模型連接效果展示
(總結不易,歡迎各位金主爸爸打賞,點贊收藏!)