test: fix sys.modules stub leakage and apply repo formatting

test_claude_agents_* now remove their placeholder packages after loading, which previously blocked real imports of the temporal plugin tree in later-collected tests. Formats touched files with ruff and narrows span.output types in new tests for pyright.

Author: levilentz

src/agentex/lib/adk/__init__.py

@@ -41,13 +41,11 @@
     "events",
     "agent_task_tracker",
     "TurnSpan",
-
     # Checkpointing / LangGraph
     "create_checkpointer",
     "create_langgraph_tracing_handler",
     "stream_langgraph_events",
     "convert_langgraph_to_agentex_events",
-
     # Providers
     "providers",
     # Utils

src/agentex/lib/adk/_modules/_langgraph_tracing.py

@@ -13,8 +13,8 @@
 
 from agentex.types.span import Span
 from agentex.lib.utils.logging import make_logger
-from agentex.lib.adk._modules.tracing import TracingModule
 from agentex.lib.core.tracing.usage import usage_from_counts
+from agentex.lib.adk._modules.tracing import TracingModule
 
 logger = make_logger(__name__)
 
@@ -251,11 +251,7 @@ def _extract_usage(response: LLMResult, msg: Any) -> dict[str, Any] | None:
     llm_output = getattr(response, "llm_output", None) or {}
     token_usage = llm_output.get("token_usage") or llm_output.get("usage")
     if isinstance(token_usage, dict) and token_usage:
-        return {
-            key: value
-            for key, value in token_usage.items()
-            if isinstance(value, (int, float))
-        }
+        return {key: value for key, value in token_usage.items() if isinstance(value, (int, float))}
     return None
 
 

src/agentex/lib/adk/_modules/tracing.py

@@ -234,9 +234,7 @@ async def turn_span(
 
         Example::
 
-            async with adk.tracing.turn_span(
-                trace_id=task.id, name="turn", input={...}, task_id=task.id
-            ) as turn:
+            async with adk.tracing.turn_span(trace_id=task.id, name="turn", input={...}, task_id=task.id) as turn:
                 result = await run_llm_calls()
                 turn.output = {"response": result.text}
                 turn.record_usage(usage=result.usage, cost_usd=result.cost_usd)

src/agentex/lib/core/services/adk/providers/litellm.py

@@ -114,11 +114,7 @@ async def chat_completion_auto_send(
                 ),
             ) as streaming_context:
                 completion = await self.llm_gateway.acompletion(**llm_config.model_dump())
-                if (
-                    completion.choices
-                    and len(completion.choices) > 0
-                    and completion.choices[0].message
-                ):
+                if completion.choices and len(completion.choices) > 0 and completion.choices[0].message:
                     final_content = TextContent(
                         author="agent",
                         content=completion.choices[0].message.content or "",
@@ -178,9 +174,7 @@ async def chat_completion_stream(
         ) as span:
             # Direct streaming outside temporal - yield each chunk as it comes
             chunks: list[Completion] = []
-            async for chunk in self.llm_gateway.acompletion_stream(
-                **completion_kwargs
-            ):
+            async for chunk in self.llm_gateway.acompletion_stream(**completion_kwargs):
                 chunks.append(chunk)
                 yield chunk
             if span:
@@ -231,18 +225,12 @@ async def chat_completion_stream_auto_send(
             ) as streaming_context:
                 # Get the streaming response
                 chunks = []
-                async for response in self.llm_gateway.acompletion_stream(
-                    **completion_kwargs
-                ):
+                async for response in self.llm_gateway.acompletion_stream(**completion_kwargs):
                     heartbeat_if_in_workflow("chat completion streaming")
                     # Store every chunk for final message assembly, including
                     # the usage-only final chunk, which has no choices
                     chunks.append(response)
-                    if (
-                        response.choices
-                        and len(response.choices) > 0
-                        and response.choices[0].delta
-                    ):
+                    if response.choices and len(response.choices) > 0 and response.choices[0].delta:
                         delta = response.choices[0].delta.content
                         if delta:
                             # Stream the chunk via the context manager
@@ -257,11 +245,7 @@ async def chat_completion_stream_auto_send(
 
                 # Update the final message content
                 complete_message = concat_completion_chunks(chunks)
-                if (
-                    complete_message
-                    and complete_message.choices
-                    and complete_message.choices[0].message
-                ):
+                if complete_message and complete_message.choices and complete_message.choices[0].message:
                     final_content = TextContent(
                         author="agent",
                         content=complete_message.choices[0].message.content or "",