Skip to content

Commit 5eaa3c5

Browse files
Fix the Ruff CI issue
1 parent d62943d commit 5eaa3c5

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

stackone_ai/utility_tools.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -350,9 +350,7 @@ def execute_search(arguments: str | JsonDict | None = None) -> JsonDict:
350350
max_workers = min(len(connectors_to_search), 10)
351351
with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as pool:
352352
futures = {
353-
pool.submit(
354-
semantic_client.search, query=query, connector=c, top_k=limit
355-
): c
353+
pool.submit(semantic_client.search, query=query, connector=c, top_k=limit): c
356354
for c in connectors_to_search
357355
}
358356
for future in concurrent.futures.as_completed(futures):

tests/test_semantic_search.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -503,9 +503,7 @@ def test_utility_tools_semantic_search(self) -> None:
503503
utility = tools.utility_tools(semantic_client=mock_client)
504504
assert len(utility) == 2
505505
# Should pass available connectors from the tools collection
506-
mock_create.assert_called_once_with(
507-
mock_client, available_connectors={"test"}
508-
)
506+
mock_create.assert_called_once_with(mock_client, available_connectors={"test"})
509507

510508

511509
class TestSemanticToolSearch:

0 commit comments

Comments
 (0)