Skip to content

Commit 568a95f

Browse files
committed
fixing linting issues
1 parent c8fb585 commit 568a95f

2 files changed

Lines changed: 10 additions & 3 deletions

File tree

integrations/pgvector/src/haystack_integrations/document_stores/pgvector/document_store.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
22
#
33
# SPDX-License-Identifier: Apache-2.0
4+
45
from typing import Any, Literal, Optional, Union, overload
56

67
from haystack import default_from_dict, default_to_dict, logging
@@ -1136,7 +1137,9 @@ def update_by_filter(self, filters: dict[str, Any], meta: dict[str, Any]) -> int
11361137
msg = "meta must be a non-empty dictionary"
11371138
raise ValueError(msg)
11381139

1139-
update_sql = SQL("UPDATE {schema_name}.{table_name} SET meta = COALESCE(meta, '{{}}'::jsonb) || %s::jsonb").format(
1140+
update_sql = SQL(
1141+
"UPDATE {schema_name}.{table_name} SET meta = COALESCE(meta, '{{}}'::jsonb) || %s::jsonb"
1142+
).format(
11401143
schema_name=Identifier(self.schema_name),
11411144
table_name=Identifier(self.table_name),
11421145
)
@@ -1184,7 +1187,9 @@ async def update_by_filter_async(self, filters: dict[str, Any], meta: dict[str,
11841187
msg = "meta must be a non-empty dictionary"
11851188
raise ValueError(msg)
11861189

1187-
update_sql = SQL("UPDATE {schema_name}.{table_name} SET meta = COALESCE(meta, '{{}}'::jsonb) || %s::jsonb").format(
1190+
update_sql = SQL(
1191+
"UPDATE {schema_name}.{table_name} SET meta = COALESCE(meta, '{{}}'::jsonb) || %s::jsonb"
1192+
).format(
11881193
schema_name=Identifier(self.schema_name),
11891194
table_name=Identifier(self.table_name),
11901195
)

integrations/pgvector/tests/test_document_store_async.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,9 @@ async def test_update_by_filter_async_multiple_fields(self, document_store: Pgve
245245
assert doc.meta["year"] == 2023 # Original field should still be present
246246

247247
# Verify category B was not updated
248-
b_docs = await document_store.filter_documents_async(filters={"field": "meta.category", "operator": "==", "value": "B"})
248+
b_docs = await document_store.filter_documents_async(
249+
filters={"field": "meta.category", "operator": "==", "value": "B"}
250+
)
249251
assert len(b_docs) == 1
250252
assert "status" not in b_docs[0].meta
251253
assert "priority" not in b_docs[0].meta

0 commit comments

Comments
 (0)