fixing lints

Author: splunk-dtaborski

splunklib/ai/engines/langchain.py

@@ -280,12 +280,12 @@ async def awrap_tool_call(
 
                     if resp.name.startswith(AGENT_PREFIX):
                         resp.artifact = SubagentFailureResult(
-                            str(resp.content)
-                        )  # pyright: ignore[reportUnknownArgumentType]
+                            str(resp.content)  # pyright: ignore[reportUnknownArgumentType]
+                        )
                     else:
                         resp.artifact = ToolFailureResult(
-                            str(resp.content)
-                        )  # pyright: ignore[reportUnknownArgumentType]
+                            str(resp.content)  # pyright: ignore[reportUnknownArgumentType]
+                        )
 
                 return resp
 
@@ -971,9 +971,9 @@ async def _sdk_handler(request: ToolRequest) -> ToolResponse:
         lc_request = _convert_tool_request_to_lc(request, original_request)
         result = await handler(lc_request)
         sdk_result = _convert_tool_message_from_lc(result)
-        assert isinstance(
-            sdk_result, ToolMessage
-        ), "Expected tool response from tool middleware handler"
+        assert isinstance(sdk_result, ToolMessage), (
+            "Expected tool response from tool middleware handler"
+        )
         return ToolResponse(sdk_result.result)
 
     return _sdk_handler
@@ -991,9 +991,9 @@ async def _sdk_handler(
         lc_request = _convert_subagent_request_to_lc(request, original_request)
         result = await handler(lc_request)
         sdk_result = _convert_tool_message_from_lc(result)
-        assert isinstance(
-            sdk_result, SubagentMessage
-        ), "Expected subagent response from subagent middleware handler"
+        assert isinstance(sdk_result, SubagentMessage), (
+            "Expected subagent response from subagent middleware handler"
+        )
         return SubagentResponse(sdk_result.result)
 
     return _sdk_handler
@@ -1186,18 +1186,16 @@ def _convert_tool_message_from_lc(
             )
         case LC_ToolMessage():
             # If this is reached, we likely passed an invalid tool name to LangChain.
-            assert (
-                message.name is not None
-            ), "LangChain responded with a nameless tool call"
+            assert message.name is not None, (
+                "LangChain responded with a nameless tool call"
+            )
 
             if message.name.startswith(TOOL_STRATEGY_TOOL_PREFIX):
                 return StructuredOutputMessage(
                     name=message.name.removeprefix(TOOL_STRATEGY_TOOL_PREFIX),
                     call_id=message.tool_call_id,
                     status=message.status,
-                    content=str(
-                        message.content
-                    ),  # pyright: ignore[reportUnknownArgumentType]
+                    content=str(message.content),  # pyright: ignore[reportUnknownArgumentType]
                 )
 
             assert isinstance(message.artifact, ToolResult) or isinstance(
@@ -1357,9 +1355,7 @@ async def _tool_call(
         except ToolException as e:
             raise LC_ToolException(*e.args) from e
         except LC_ToolException:
-            assert (
-                False
-            ), (  # noqa: PT015
+            assert False, (  # noqa: PT015
                 "ToolException from LangChain should not be raised in tool.func"
             )
 
@@ -1650,9 +1646,7 @@ def _get_approximate_token_counter(model: BaseChatModel) -> LC_TokenCounter:
     # NOTE: This is adapted from the backend provider library
     # 3.3 was estimated in an offline experiment, comparing with Claude's token-counting
     # API: https://platform.claude.com/docs/en/build-with-claude/token-counting
-    if (
-        model._llm_type == ANTHROPIC_CHAT_MODEL_TYPE
-    ):  # pyright: ignore[reportPrivateUsage]
+    if model._llm_type == ANTHROPIC_CHAT_MODEL_TYPE:  # pyright: ignore[reportPrivateUsage]
         return partial(count_tokens_approximately, chars_per_token=3.3)
     return count_tokens_approximately
 

tests/unit/ai/test_default_limits.py

@@ -75,13 +75,9 @@ def test_default_values_match_constants(self) -> None:
         token = next(m for m in mw if isinstance(m, TokenLimitMiddleware))
         step = next(m for m in mw if isinstance(m, StepLimitMiddleware))
         timeout = next(m for m in mw if isinstance(m, TimeoutLimitMiddleware))
-        assert (
-            token._limit == DEFAULT_TOKEN_LIMIT
-        )  # pyright: ignore[reportPrivateUsage]
+        assert token._limit == DEFAULT_TOKEN_LIMIT  # pyright: ignore[reportPrivateUsage]
         assert step._limit == DEFAULT_STEP_LIMIT  # pyright: ignore[reportPrivateUsage]
-        assert (
-            timeout._seconds == DEFAULT_TIMEOUT_SECONDS
-        )  # pyright: ignore[reportPrivateUsage]
+        assert timeout._seconds == DEFAULT_TIMEOUT_SECONDS  # pyright: ignore[reportPrivateUsage]
 
     def test_user_token_limit_suppresses_default(self) -> None:
         agent = _make_agent(middleware=[TokenLimitMiddleware(50_000)])
@@ -153,9 +149,7 @@ async def test_deadline_is_none_before_first_invoke(self) -> None:
 
     async def test_timeout_fires_when_deadline_exceeded(self) -> None:
         mw = TimeoutLimitMiddleware(60.0)
-        mw._deadline = (
-            monotonic() - 1.0
-        )  # pyright: ignore[reportPrivateUsage]  # already in the past
+        mw._deadline = monotonic() - 1.0  # pyright: ignore[reportPrivateUsage]  # already in the past
 
         state = AgentState(
             messages=[],