From f974bf39bb08b29aab9b81ae3d88922413641335 Mon Sep 17 00:00:00 2001 From: ArnoChen Date: Fri, 7 Feb 2025 23:13:28 +0800 Subject: [PATCH] format format --- lightrag/api/lightrag_server.py | 4 +--- lightrag/kg/postgres_impl.py | 4 +++- lightrag/kg/tidb_impl.py | 6 +++--- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index 836411d1..23507dd1 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -536,9 +536,7 @@ def parse_args() -> argparse.Namespace: parser.add_argument( "--namespace-prefix", type=str, - default=get_env_value( - "NAMESPACE_PREFIX", "" - ), + default=get_env_value("NAMESPACE_PREFIX", ""), help="Prefix of the namespace", ) diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py index 2f6d1bc4..a031a0c3 100644 --- a/lightrag/kg/postgres_impl.py +++ b/lightrag/kg/postgres_impl.py @@ -298,7 +298,9 @@ class PGKVStorage(BaseKVStorage): async def index_done_callback(self): for n in ("full_docs", "text_chunks"): if self.namespace.endswith(n): - logger.info("full doc and chunk data had been saved into postgresql db!") + logger.info( + "full doc and chunk data had been saved into postgresql db!" + ) break diff --git a/lightrag/kg/tidb_impl.py b/lightrag/kg/tidb_impl.py index 548ca8e0..4a21d067 100644 --- a/lightrag/kg/tidb_impl.py +++ b/lightrag/kg/tidb_impl.py @@ -190,7 +190,7 @@ class TiDBKVStorage(BaseKVStorage): "tokens": item["tokens"], "chunk_order_index": item["chunk_order_index"], "full_doc_id": item["full_doc_id"], - "content_vector": f"{item["__vector__"].tolist()}", + "content_vector": f"{item['__vector__'].tolist()}", "workspace": self.db.workspace, } ) @@ -297,7 +297,7 @@ class TiDBVectorDBStorage(BaseVectorStorage): "id": item["id"], "name": item["entity_name"], "content": item["content"], - "content_vector": f"{item["content_vector"].tolist()}", + "content_vector": f"{item['content_vector'].tolist()}", "workspace": self.db.workspace, } # update entity_id if node inserted by graph_storage_instance before @@ -319,7 +319,7 @@ class TiDBVectorDBStorage(BaseVectorStorage): "source_name": item["src_id"], "target_name": item["tgt_id"], "content": item["content"], - "content_vector": f"{item["content_vector"].tolist()}", + "content_vector": f"{item['content_vector'].tolist()}", "workspace": self.db.workspace, } # update relation_id if node inserted by graph_storage_instance before