diff --git a/dsLightRag/Config/Config.py b/dsLightRag/Config/Config.py index c926e41d..f7a1e1ed 100644 --- a/dsLightRag/Config/Config.py +++ b/dsLightRag/Config/Config.py @@ -29,4 +29,12 @@ MYSQL_HOST = "127.0.0.1" MYSQL_PORT = 22066 MYSQL_USER = "root" MYSQL_PASSWORD = "DsideaL147258369" -MYSQL_DB_NAME = "base_db" \ No newline at end of file +MYSQL_DB_NAME = "base_db" + +# POSTGRESQL配置信息 +AGE_GRAPH_NAME = "dickens" +POSTGRES_HOST = "10.10.14.208" +POSTGRES_PORT = 5432 +POSTGRES_USER = "postgres" +POSTGRES_PASSWORD = "DsideaL147258369" +POSTGRES_DATABASE = "rag" \ No newline at end of file diff --git a/dsLightRag/Config/__pycache__/Config.cpython-310.pyc b/dsLightRag/Config/__pycache__/Config.cpython-310.pyc index 4f046fee..c8accaf3 100644 Binary files a/dsLightRag/Config/__pycache__/Config.cpython-310.pyc and b/dsLightRag/Config/__pycache__/Config.cpython-310.pyc differ diff --git a/dsLightRag/Util/LightRagUtil.py b/dsLightRag/Util/LightRagUtil.py index ab487359..f0c88c47 100644 --- a/dsLightRag/Util/LightRagUtil.py +++ b/dsLightRag/Util/LightRagUtil.py @@ -142,16 +142,13 @@ def create_embedding_func(): ), ) - - # AGE -os.environ["AGE_GRAPH_NAME"] = "dickens" -os.environ["POSTGRES_HOST"] = "10.10.14.208" -os.environ["POSTGRES_PORT"] = "5432" -os.environ["POSTGRES_USER"] = "postgres" -os.environ["POSTGRES_PASSWORD"] = "postgres" -os.environ["POSTGRES_DATABASE"] = "rag" - +os.environ["AGE_GRAPH_NAME"] = AGE_GRAPH_NAME +os.environ["POSTGRES_HOST"] = POSTGRES_HOST +os.environ["POSTGRES_PORT"] = str(POSTGRES_PORT) +os.environ["POSTGRES_USER"] = POSTGRES_USER +os.environ["POSTGRES_PASSWORD"] = POSTGRES_PASSWORD +os.environ["POSTGRES_DATABASE"] = POSTGRES_DATABASE async def initialize_pg_rag(WORKING_DIR): rag = LightRAG( diff --git a/dsLightRag/Util/__pycache__/LightRagUtil.cpython-310.pyc b/dsLightRag/Util/__pycache__/LightRagUtil.cpython-310.pyc index a1118699..6c06b136 100644 Binary files a/dsLightRag/Util/__pycache__/LightRagUtil.cpython-310.pyc and b/dsLightRag/Util/__pycache__/LightRagUtil.cpython-310.pyc differ