gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Create indexes on the ID columns for all PostgreSQL tables

Browse source
This commit is contained in:
yangdx 2025-04-03 17:29:52 +08:00
parent 1eb40a69f5
commit 91b6064645
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
lightrag/kg/postgres_impl.py
View file

@ -114,6 +114,23 @@ class PostgreSQLDB:
f"PostgreSQL, Failed to create table {k} in database, Please verify the connection with PostgreSQL database, Got: {e}"
)
raise e
# Create index for id column in each table
try:
index_name = f"idx_{k.lower()}_id"
check_index_sql = f"""
SELECT 1 FROM pg_indexes
WHERE indexname = '{index_name}'
AND tablename = '{k.lower()}'
"""
index_exists = await self.query(check_index_sql)
if not index_exists:
create_index_sql = f"CREATE INDEX {index_name} ON {k}(id)"
logger.info(f"PostgreSQL, Creating index {index_name} on table {k}")
await self.execute(create_index_sql)
except Exception as e:
logger.error(f"PostgreSQL, Failed to create index on table {k}, Got: {e}")
async def query(
self,