Fix CI lint errors

Author: Shashikant86

examples/demo_semantic_search.py

@@ -184,6 +184,7 @@ def run_demo(use_local: bool, lambda_url: str, api_key: str | None) -> None:
             print("Error: STACKONE_API_KEY required for production mode")
             print("Use --local flag for local Lambda mode")
             exit(1)
+        assert api_key is not None  # narrowing for type checker
         sem_client = SemanticSearchClient(api_key=api_key)
         semantic_search = sem_client.search
         client = None
@@ -216,7 +217,7 @@ def run_demo(use_local: bool, lambda_url: str, api_key: str | None) -> None:
         query = demo["query"]
         why = demo["why"]
 
-        print(f"\n  [{i}/{len(DEMO_QUERIES)}] Query: \"{query}\"")
+        print(f'\n  [{i}/{len(DEMO_QUERIES)}] Query: "{query}"')
         print(f"  Why interesting: {why}")
         print()
 
@@ -243,7 +244,7 @@ def run_demo(use_local: bool, lambda_url: str, api_key: str | None) -> None:
             s_name = sem_names[j] if j < len(sem_names) else ""
             s_score = sem_scores[j] if j < len(sem_scores) else ""
             l_display = f"  {l_name[:w]:<{w}}"
-            s_display = f"  {s_name[:w - 8]:<{w - 8}} ({s_score})" if s_name else ""
+            s_display = f"  {s_name[: w - 8]:<{w - 8}} ({s_score})" if s_name else ""
             print(f"{l_display} |{s_display}")
 
         input("\n  Press Enter for next query...")

stackone_ai/semantic_search.py

@@ -108,9 +108,7 @@ def search(
             data = response.json()
             return SemanticSearchResponse(**data)
         except httpx.HTTPStatusError as e:
-            raise SemanticSearchError(
-                f"API error: {e.response.status_code} - {e.response.text}"
-            ) from e
+            raise SemanticSearchError(f"API error: {e.response.status_code} - {e.response.text}") from e
         except httpx.RequestError as e:
             raise SemanticSearchError(f"Request failed: {e}") from e
         except Exception as e:

tests/benchmark_search.py

@@ -171,10 +171,12 @@ def fetch_all_actions(self) -> list[SemanticSearchResult]:
         seen: dict[str, SemanticSearchResult] = {}
         for query in broad_queries:
             try:
-                data = self._invoke({
-                    "type": "search",
-                    "payload": {"query": query, "top_k": 500},
-                })
+                data = self._invoke(
+                    {
+                        "type": "search",
+                        "payload": {"query": query, "top_k": 500},
+                    }
+                )
                 for r in data.get("results", []):
                     name = r.get("action_name", "")
                     if name and name not in seen:

tests/test_semantic_search.py

@@ -593,12 +593,14 @@ def make_tool(name: str) -> StackOneTool:
                 _api_key="test-key",
             )
 
-        tools = Tools([
-            make_tool("bamboohr_create_employee"),
-            make_tool("bamboohr_list_employees"),
-            make_tool("hibob_create_employee"),
-            make_tool("slack_send_message"),
-        ])
+        tools = Tools(
+            [
+                make_tool("bamboohr_create_employee"),
+                make_tool("bamboohr_list_employees"),
+                make_tool("hibob_create_employee"),
+                make_tool("slack_send_message"),
+            ]
+        )
 
         connectors = tools.get_connectors()
 
@@ -623,12 +625,14 @@ def make_tool(name: str) -> StackOneTool:
                 _api_key="test-key",
             )
 
-        tools = Tools([
-            make_tool("bamboohr_create_employee"),
-            make_tool("bamboohr_list_employees"),
-            make_tool("hibob_create_employee"),
-            make_tool("slack_send_message"),
-        ])
+        tools = Tools(
+            [
+                make_tool("bamboohr_create_employee"),
+                make_tool("bamboohr_list_employees"),
+                make_tool("hibob_create_employee"),
+                make_tool("slack_send_message"),
+            ]
+        )
 
         # Filter by single connector
         bamboo_tools = tools.filter_by_connector(["bamboohr"])
@@ -647,9 +651,7 @@ def test_filter_by_connector_case_insensitive(self) -> None:
         tool = StackOneTool(
             description="Creates employee",
             parameters=ToolParameters(type="object", properties={}),
-            _execute_config=ExecuteConfig(
-                name="bamboohr_create_employee", method="POST", url="", headers={}
-            ),
+            _execute_config=ExecuteConfig(name="bamboohr_create_employee", method="POST", url="", headers={}),
             _api_key="test-key",
         )
         tools = Tools([tool])
@@ -666,9 +668,7 @@ def test_filter_by_connector_returns_new_tools(self) -> None:
         tool = StackOneTool(
             description="Creates employee",
             parameters=ToolParameters(type="object", properties={}),
-            _execute_config=ExecuteConfig(
-                name="bamboohr_create_employee", method="POST", url="", headers={}
-            ),
+            _execute_config=ExecuteConfig(name="bamboohr_create_employee", method="POST", url="", headers={}),
             _api_key="test-key",
         )
         tools = Tools([tool])