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

fix: add logger to configure_vchordrq() and format code

Browse source
This commit is contained in:
wmsnp 2025-11-18 15:31:08 +08:00
parent d07023c962
commit f4bf5d279c
No known key found for this signature in database
GPG key ID: 707D185257DA9195
1 changed files with 14 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
lightrag/kg/postgres_impl.py
View file

@ -417,8 +417,8 @@ class PostgreSQLDB:
try: try:
await connection.execute(f"SET vchordrq.probes TO '{self.vchordrq_probes}'") await connection.execute(f"SET vchordrq.probes TO '{self.vchordrq_probes}'")
await connection.execute(f"SET vchordrq.epsilon TO {self.vchordrq_epsilon}") await connection.execute(f"SET vchordrq.epsilon TO {self.vchordrq_epsilon}")
except Exception: except Exception as e:
pass logger.error(f"Failed to set vchordrq.probes or vchordrq.epsilon: {e}")
async def _migrate_llm_cache_schema(self): async def _migrate_llm_cache_schema(self):
"""Migrate LLM cache schema: add new columns and remove deprecated mode field""" """Migrate LLM cache schema: add new columns and remove deprecated mode field"""
@ -1388,12 +1388,14 @@ class PostgreSQLDB:
CREATE INDEX {{vector_index_name}} CREATE INDEX {{vector_index_name}}
ON {{k}} USING vchordrq (content_vector vector_cosine_ops) ON {{k}} USING vchordrq (content_vector vector_cosine_ops)
{f'WITH (options = $${self.vchordrq_build_options}$$)' if self.vchordrq_build_options else ''} {f'WITH (options = $${self.vchordrq_build_options}$$)' if self.vchordrq_build_options else ''}
""" """,
} }
embedding_dim = int(os.environ.get("EMBEDDING_DIM", 1024)) embedding_dim = int(os.environ.get("EMBEDDING_DIM", 1024))
for k in vdb_tables: for k in vdb_tables:
vector_index_name = f"idx_{k.lower()}_{self.vector_index_type.lower()}_cosine" vector_index_name = (
f"idx_{k.lower()}_{self.vector_index_type.lower()}_cosine"
)
check_vector_index_sql = f""" check_vector_index_sql = f"""
SELECT 1 FROM pg_indexes SELECT 1 FROM pg_indexes
WHERE indexname = '{vector_index_name}' AND tablename = '{k.lower()}' WHERE indexname = '{vector_index_name}' AND tablename = '{k.lower()}'
@ -1405,8 +1407,14 @@ class PostgreSQLDB:
alter_sql = f"ALTER TABLE {k} ALTER COLUMN content_vector TYPE VECTOR({embedding_dim})" alter_sql = f"ALTER TABLE {k} ALTER COLUMN content_vector TYPE VECTOR({embedding_dim})"
await self.execute(alter_sql) await self.execute(alter_sql)
logger.debug(f"Ensured vector dimension for {k}") logger.debug(f"Ensured vector dimension for {k}")
logger.info(f"Creating {self.vector_index_type} index {vector_index_name} on table {k}") logger.info(
await self.execute(create_sql[self.vector_index_type].format(vector_index_name=vector_index_name, k=k)) f"Creating {self.vector_index_type} index {vector_index_name} on table {k}"
)
await self.execute(
create_sql[self.vector_index_type].format(
vector_index_name=vector_index_name, k=k
)
)
logger.info( logger.info(
f"Successfully created vector index {vector_index_name} on table {k}" f"Successfully created vector index {vector_index_name} on table {k}"
) )