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

format

Browse Source 
format
This commit is contained in:
ArnoChen
2025-02-07 23:13:28 +08:00
parent 88d691deb9
commit f974bf39bb
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/api/lightrag_server.py
View File

@@ -536,9 +536,7 @@ def parse_args() -> argparse.Namespace:
parser.add_argument( parser.add_argument(
"--namespace-prefix", "--namespace-prefix",
type=str, type=str,
default=get_env_value( default=get_env_value("NAMESPACE_PREFIX", ""),
"NAMESPACE_PREFIX", ""
),
help="Prefix of the namespace", help="Prefix of the namespace",
) )

4
lightrag/kg/postgres_impl.py
View File

@@ -298,7 +298,9 @@ class PGKVStorage(BaseKVStorage):
async def index_done_callback(self): async def index_done_callback(self):
for n in ("full_docs", "text_chunks"): for n in ("full_docs", "text_chunks"):
if self.namespace.endswith(n): 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 break

6
lightrag/kg/tidb_impl.py
View File

@@ -190,7 +190,7 @@ class TiDBKVStorage(BaseKVStorage):
"tokens": item["tokens"], "tokens": item["tokens"],
"chunk_order_index": item["chunk_order_index"], "chunk_order_index": item["chunk_order_index"],
"full_doc_id": item["full_doc_id"], "full_doc_id": item["full_doc_id"],
"content_vector": f"{item["__vector__"].tolist()}", "content_vector": f"{item['__vector__'].tolist()}",
"workspace": self.db.workspace, "workspace": self.db.workspace,
} }
) )
@@ -297,7 +297,7 @@ class TiDBVectorDBStorage(BaseVectorStorage):
"id": item["id"], "id": item["id"],
"name": item["entity_name"], "name": item["entity_name"],
"content": item["content"], "content": item["content"],
"content_vector": f"{item["content_vector"].tolist()}", "content_vector": f"{item['content_vector'].tolist()}",
"workspace": self.db.workspace, "workspace": self.db.workspace,
} }
# update entity_id if node inserted by graph_storage_instance before # update entity_id if node inserted by graph_storage_instance before
@@ -319,7 +319,7 @@ class TiDBVectorDBStorage(BaseVectorStorage):
"source_name": item["src_id"], "source_name": item["src_id"],
"target_name": item["tgt_id"], "target_name": item["tgt_id"],
"content": item["content"], "content": item["content"],
"content_vector": f"{item["content_vector"].tolist()}", "content_vector": f"{item['content_vector'].tolist()}",
"workspace": self.db.workspace, "workspace": self.db.workspace,
} }
# update relation_id if node inserted by graph_storage_instance before # update relation_id if node inserted by graph_storage_instance before