From 882eb1617461f647f8d2d5dbc4172414473b8934 Mon Sep 17 00:00:00 2001 From: "xiaozhiqing.xzq" Date: Fri, 11 Jul 2025 17:27:50 +0800 Subject: [PATCH] format --- .../datasource/vdb/tablestore/tablestore_vector.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/api/core/rag/datasource/vdb/tablestore/tablestore_vector.py b/api/core/rag/datasource/vdb/tablestore/tablestore_vector.py index 4c09f05927..552068c99e 100644 --- a/api/core/rag/datasource/vdb/tablestore/tablestore_vector.py +++ b/api/core/rag/datasource/vdb/tablestore/tablestore_vector.py @@ -59,17 +59,19 @@ class TableStoreVector(BaseVector): docs = [] request = BatchGetRowRequest() columns_to_get = [Field.METADATA_KEY.value, Field.CONTENT_KEY.value] - rows_to_get = [ - [("id", _id)] for _id in ids - ] + rows_to_get = [[("id", _id)] for _id in ids] request.add(TableInBatchGetRowItem(self._table_name, rows_to_get, columns_to_get, None, 1)) result = self._tablestore_client.batch_get_row(request) table_result = result.get_result_by_table(self._table_name) for item in table_result: if item.is_ok and item.row: - kv = {k:v for k,v,t in item.row.attribute_columns} - docs.append(Document(page_content=kv[Field.CONTENT_KEY.value], metadata=json.loads(kv[Field.METADATA_KEY.value]))) + kv = {k: v for k, v, t in item.row.attribute_columns} + docs.append( + Document( + page_content=kv[Field.CONTENT_KEY.value], metadata=json.loads(kv[Field.METADATA_KEY.value]) + ) + ) return docs def get_type(self) -> str: