format

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 = [
+        rows_to_get = [[("id", _id)] for _id in ids]
-            [("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}
+                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])))
+                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: