新增属性图的配置

This commit is contained in:
wanyaokun
2024-09-14 09:49:51 +08:00
parent df7ebf3d4f
commit a8f99adaca
2 changed files with 12 additions and 2 deletions
+11 -1
View File
@@ -65,6 +65,13 @@ VECTOR_STORE_TYPE=chroma
# The name of the collection in your vector database
VECTOR_STORE_COLLECTION=default
GRAPH_STORE_TYPE =
#---------- neo4j - PropertyGraph ----------------
NEO4J_USERNAME = neo4j
NEO4J_PASSWORD = neo4j
NEO4J_URL = bolt://10.1.6.40:7687
# The API endpoint for your vector database
# VECTOR_STORE_HOST=
@@ -76,6 +83,7 @@ VECTOR_STORE_COLLECTION=default
# Otherwise, use VECTOR_STORE__HOST and VECTOR_STORE__PORT config above
VECTOR_STORE_PATH=./storage_vector
BM_RETRIEVER_PATH =./storage_bm
GRAPH_STORAGE_PATH =./storage_graph
PHOENIX_API_KEY=123456
@@ -110,4 +118,6 @@ PROJECT_TITLE = "您好,我是博微工程理解小助手,您可以问我有
CHAT_UPLOAD_FILECACHE = "./output/uploaded"
JIEBA_DATA=./nltk_data
NLTK_DATA=./nltk_data
NLTK_DATA=./nltk_data
PYTHONUTF8=1
+1 -1
View File
@@ -40,7 +40,7 @@ class PropertyGraph:
if GRAPH_STORE_TYPE == 'neo4j':
index = PropertyGraphIndex.from_existing(property_graph_store= get_Neo4j_Graph_Store(self._prjFlag))
else:
GRAPH_STORAGE_DIR = os.getenv("GRAPH_STORAGE_DIR", "storage_graph")
GRAPH_STORAGE_DIR = os.getenv("GRAPH_STORAGE_PATH", "storage_graph")
prjCachePath = GRAPH_STORAGE_DIR + f"/{self._prjFlag}"
if not os.path.exists(prjCachePath):
return None