Merge pull request lightspeed-core#1308 from tisnik/lcore-1438-minor-refactoring

LCORE-1438: minor refactoring

Author: tisnik

src/a2a_storage/__init__.py

@@ -16,8 +16,8 @@
 
 __all__ = [
     "A2AContextStore",
+    "A2AStorageFactory",
     "InMemoryA2AContextStore",
-    "SQLiteA2AContextStore",
     "PostgresA2AContextStore",
-    "A2AStorageFactory",
+    "SQLiteA2AContextStore",
 ]

src/observability/__init__.py

@@ -11,4 +11,4 @@
 from observability.formats import InferenceEventData, build_inference_event
 from observability.splunk import send_splunk_event
 
-__all__ = ["send_splunk_event", "InferenceEventData", "build_inference_event"]
+__all__ = ["InferenceEventData", "build_inference_event", "send_splunk_event"]

tests/e2e/utils/utils.py

@@ -268,7 +268,7 @@ def restart_container(container_name: str) -> None:
             check=True,
         )
     except (subprocess.CalledProcessError, subprocess.TimeoutExpired) as e:
-        print(f"Failed to restart container {container_name}: {str(e.stderr)}")
+        print(f"Failed to restart container {container_name}: {e.stderr!s}")
         raise
 
     # Wait for container to be healthy

tests/unit/app/endpoints/test_streaming_query.py

@@ -2369,8 +2369,7 @@ def build_mcp_tool_call_side_effect(
             # Remove item from dict to simulate real behavior
             # arguments parameter is required by function signature but unused here
             _ = arguments
-            if output_index in mcp_call_items:
-                del mcp_call_items[output_index]
+            mcp_call_items.pop(output_index, None)
             return mock_tool_call
 
         mocker.patch(