basic linting

Author: ioanarm

tests/loki_test.py

@@ -24,6 +24,7 @@
 
 pytestmark = pytest.mark.anyio
 
+
 @pytest.fixture
 def mcp_url():
     return os.environ.get("MCP_GRAFANA_URL", DEFAULT_MCP_URL)
@@ -81,7 +82,9 @@ async def test_loki(model: str, mcp_client: ClientSession):
     # Verify Loki datasource exists
     loki_datasources = [ds for ds in datasources_data if ds.get("type") == "loki"]
     assert len(loki_datasources) > 0, "No Loki datasource found"
-    print(f"\nFound Loki datasource: {loki_datasources[0]['name']} (uid: {loki_datasources[0]['uid']})")
+    print(
+        f"\nFound Loki datasource: {loki_datasources[0]['name']} (uid: {loki_datasources[0]['uid']})"
+    )
 
     # Call the LLM including the tool call result.
     response = await acompletion(
@@ -149,7 +152,9 @@ async def test_loki_container_labels(model: str, mcp_client: ClientSession):
     # Verify Loki datasource exists
     loki_datasources = [ds for ds in datasources_data if ds.get("type") == "loki"]
     assert len(loki_datasources) > 0, "No Loki datasource found"
-    print(f"\nFound Loki datasource: {loki_datasources[0]['name']} (uid: {loki_datasources[0]['uid']})")
+    print(
+        f"\nFound Loki datasource: {loki_datasources[0]['name']} (uid: {loki_datasources[0]['uid']})"
+    )
 
     # Call the LLM including the tool call result.
     response = await acompletion(
@@ -165,10 +170,7 @@ async def test_loki_container_labels(model: str, mcp_client: ClientSession):
             response,
             mcp_client,
             "list_loki_label_values",
-            {
-                "datasourceUid": "loki",
-                "labelName": "container"
-            },
+            {"datasourceUid": "loki", "labelName": "container"},
         )
     )
 
@@ -219,8 +221,12 @@ async def assert_and_handle_tool_call(
 
         if expected_args:
             for key, value in expected_args.items():
-                assert key in arguments, f"Missing required argument '{key}' in tool call"
-                assert arguments[key] == value, f"Argument '{key}' has wrong value. Expected: {value}, Got: {arguments[key]}"
+                assert key in arguments, (
+                    f"Missing required argument '{key}' in tool call"
+                )
+                assert arguments[key] == value, (
+                    f"Argument '{key}' has wrong value. Expected: {value}, Got: {arguments[key]}"
+                )
         result = await mcp_client.call_tool(tool_call.function.name, arguments)
         # Assume each tool returns a single text content for now
         assert len(result.content) == 1