Pārlūkot izejas kodu

第一次提交全部代码

yulongyan_citu 10 mēneši atpakaļ
vecāks
revīzija
ff600e1e94
2 mainītis faili ar 4 papildinājumiem un 1 dzēšanām
  1. 1 1
      database/create_db.py
  2. 3 0
      graph/graph_retrieval.py

+ 1 - 1
database/create_db.py

@@ -2,7 +2,7 @@ import sys
 import os
 from embedding.embedding import get_embedding
 from langchain_community.document_loaders import PyMuPDFLoader,Docx2txtLoader,TextLoader,DirectoryLoader
-from langchain_chroma import Chroma
+from langchain_chroma.vectorstores import Chroma
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 sys.path.append(os.path.dirname(os.path.dirname(__file__)))
 # 首先实现基本配置

+ 3 - 0
graph/graph_retrieval.py

@@ -17,6 +17,9 @@ def connect():
     os.environ["NEO4J_USERNAME"] = "neo4j"
     os.environ["NEO4J_PASSWORD"] = "!@#qwe123^&*"
 
+    # os.environ["NEO4J_URI"] = "bolt://192.168.3.91:27687"
+    # os.environ["NEO4J_USERNAME"] = "neo4j"
+    # os.environ["NEO4J_PASSWORD"] = "citu2099@@CCA."
     graph = Neo4jGraph()
     return graph