gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-04-03 17:31:01 +08:00
parent 91b6064645
commit a4b22f5343
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lightrag/kg/postgres_impl.py
View File

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