Skip to content

Commit 521339b

Browse files
committed
Fix ruff issues
1 parent 4e19479 commit 521339b

3 files changed

Lines changed: 12 additions & 10 deletions

File tree

examples/semantic_search_example.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,9 @@ def example_openai_agent_loop():
257257
tool = tools.get_tool(tool_call.function.name)
258258
if tool:
259259
result = tool.execute(tool_call.function.arguments)
260-
print(f" Response keys: {list(result.keys()) if isinstance(result, dict) else type(result)}")
260+
print(
261+
f" Response keys: {list(result.keys()) if isinstance(result, dict) else type(result)}"
262+
)
261263
else:
262264
print(f"OpenAI responded with text: {response.choices[0].message.content}")
263265

stackone_ai/utility_tools.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -349,12 +349,14 @@ def execute_search(arguments: str | JsonDict | None = None) -> JsonDict:
349349
norm_name = _normalize_action_name(r.action_name)
350350
if norm_name not in seen:
351351
seen.add(norm_name)
352-
tools_data.append({
353-
"name": norm_name,
354-
"description": r.description,
355-
"score": r.similarity_score,
356-
"connector": r.connector_key,
357-
})
352+
tools_data.append(
353+
{
354+
"name": norm_name,
355+
"description": r.description,
356+
"score": r.similarity_score,
357+
"connector": r.connector_key,
358+
}
359+
)
358360

359361
return {"tools": tools_data[:limit]}
360362

tests/test_semantic_search.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -935,9 +935,7 @@ class TestSemanticSearchDeduplication:
935935

936936
@patch.object(SemanticSearchClient, "search")
937937
@patch("stackone_ai.toolset._fetch_mcp_tools")
938-
def test_search_tools_deduplicates_versions(
939-
self, mock_fetch: MagicMock, mock_search: MagicMock
940-
) -> None:
938+
def test_search_tools_deduplicates_versions(self, mock_fetch: MagicMock, mock_search: MagicMock) -> None:
941939
"""Test that search_tools deduplicates multiple API versions of the same action."""
942940
from stackone_ai import StackOneToolSet
943941
from stackone_ai.toolset import _McpToolDefinition

0 commit comments

Comments
 (0)