@qa-gentic/stlc-agents 1.0.23 → 1.0.26

package/src/stlc_agents/agent_test_case_manager/server.py

@@ -17,6 +17,7 @@ import asyncio
 import json
 import os
 import sys
+import time
 
 from dotenv import load_dotenv
 from mcp.server import Server
@@ -29,7 +30,9 @@ from stlc_agents.agent_test_case_manager.tools.ado_workitem import (
     create_test_case as _create_test_case,
     link_test_cases_to_work_item as _link_test_cases,
     get_linked_test_cases as _get_linked_test_cases,
+    add_tag_to_work_item as _add_tag,
 )
+from stlc_agents.shared.cost_tracker import track
 
 load_dotenv()
 
@@ -351,6 +354,7 @@ async def list_tools() -> list[types.Tool]:
 
 @app.call_tool()
 async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
+    t0 = time.monotonic()
     try:
         if name == "fetch_work_item":
             result = await asyncio.to_thread(
@@ -380,7 +384,7 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                         "below and retry. No test cases were created in ADO."
                     ),
                 }
-                return [types.TextContent(type="text", text=json.dumps(result, indent=2, ensure_ascii=False))]
+                return track(result, tool_name=name, server="qa-test-case-manager", t0=t0)
 
             # ── Epic guard — always check work item type before creating TCs ─
             try:
@@ -399,7 +403,7 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                             "Feature or PBI/Bug ID instead."
                         ),
                     }
-                    return [types.TextContent(type="text", text=json.dumps(result, indent=2, ensure_ascii=False))]
+                    return track(result, tool_name=name, server="qa-test-case-manager", t0=t0)
             except Exception:
                 pass  # if the peek fails, continue and let the real call surface the error
 
@@ -448,7 +452,7 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                             "Reply 'yes' or 'confirm' to proceed, or 'no' / 'cancel' to abort."
                         ),
                     }
-                    return [types.TextContent(type="text", text=json.dumps(result, indent=2, ensure_ascii=False))]
+                    return track(result, tool_name=name, server="qa-test-case-manager", t0=t0)
             except Exception:
                 pass  # wi_data may not be set if the earlier peek failed; continue normally
 
@@ -480,6 +484,16 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                 except Exception as e:
                     link_result = {"error": str(e)}
 
+            # Add tag to the parent work item after linking
+            tag_result = {}
+            if created:
+                try:
+                    tag_result = await asyncio.to_thread(
+                        _add_tag, org, project, wi_id, "STLCAgentTestCases"
+                    )
+                except Exception as e:
+                    tag_result = {"error": str(e)}
+
             result = {
                 "summary": {
                     "requested": len(test_cases),
@@ -490,6 +504,7 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                 "created_test_cases": created,
                 "failed": failed,
                 "link_result": link_result,
+                "tag_result": tag_result,
                 "_validation": {
                     "valid": len(failed) == 0 and bool(link_result.get("success", True)),
                     "input_validation": input_validation,
@@ -565,13 +580,11 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
         else:
             result = {"error": f"Unknown tool: {name}"}
 
-        return [types.TextContent(type="text", text=json.dumps(result, indent=2, ensure_ascii=False))]
+        return track(result, tool_name=name, server="qa-test-case-manager", t0=t0)
 
     except Exception as exc:
-        return [types.TextContent(
-            type="text",
-            text=json.dumps({"error": str(exc), "tool": name}, indent=2),
-        )]
+        err_result = {"error": str(exc), "tool": name}
+        return track(err_result, tool_name=name, server="qa-test-case-manager", t0=t0)
 
 
 # ---------------------------------------------------------------------------

package/src/stlc_agents/agent_test_case_manager/tools/ado_workitem.py

@@ -8,6 +8,8 @@ Public API:
   create_test_case(org_url, project, title, steps, ...)  -> dict
   link_test_cases_to_work_item(org_url, project, wi_id, tc_ids) -> dict
   get_linked_test_cases(org_url, project, work_item_id)  -> dict
+  add_comment_to_work_item(org_url, project, work_item_id, text) -> dict
+  add_tag_to_work_item(org_url, project, work_item_id, tag)      -> dict
 """
 from __future__ import annotations
 
@@ -175,8 +177,13 @@ def link_test_cases_to_work_item(
     project: str,
     work_item_id: int,
     test_case_ids: List[int],
+    link_comment: str = "STLC-Agent generated test case",
 ) -> dict:
-    """Create TestedBy-Forward links from a work item to test cases."""
+    """Create TestedBy-Forward links from a work item to test cases.
+
+    link_comment is stored as attributes.comment on each relation and appears
+    in the Links tab Comments column in Azure DevOps.
+    """
     org_url = org_url.rstrip("/")
     headers = get_auth_headers("application/json-patch+json")
 
@@ -187,6 +194,7 @@ def link_test_cases_to_work_item(
             "value": {
                 "rel": "Microsoft.VSTS.Common.TestedBy-Forward",
                 "url": f"{org_url}/{project}/_apis/wit/workItems/{tc_id}",
+                "attributes": {"comment": link_comment},
             },
         }
         for tc_id in test_case_ids
@@ -249,6 +257,62 @@ def get_linked_test_cases(org_url: str, project: str, work_item_id: int) -> dict
     return {"work_item_id": work_item_id, "linked_test_cases": linked, "count": len(linked)}
 
 
+# ---------------------------------------------------------------------------
+# add_comment_to_work_item
+# ---------------------------------------------------------------------------
+
+def add_comment_to_work_item(org_url: str, project: str, work_item_id: int, text: str) -> dict:
+    """Add a comment to a work item via the ADO comments API."""
+    org_url = org_url.rstrip("/")
+    headers = get_auth_headers()
+
+    resp = requests.post(
+        f"{org_url}/{project}/_apis/wit/workitems/{work_item_id}/comments",
+        headers=headers,
+        params={"api-version": "7.1-preview.3"},
+        json={"text": text},
+        timeout=30,
+    )
+    resp.raise_for_status()
+    return {"success": True, "comment_id": resp.json().get("id")}
+
+
+# ---------------------------------------------------------------------------
+# add_tag_to_work_item
+# ---------------------------------------------------------------------------
+
+def add_tag_to_work_item(org_url: str, project: str, work_item_id: int, tag: str) -> dict:
+    """Append a tag to a work item's System.Tags field (no-op if already present)."""
+    org_url = org_url.rstrip("/")
+
+    fetch_resp = requests.get(
+        f"{org_url}/{project}/_apis/wit/workitems/{work_item_id}",
+        headers=get_auth_headers(),
+        params={"api-version": _API},
+        timeout=30,
+    )
+    fetch_resp.raise_for_status()
+    existing_str = fetch_resp.json().get("fields", {}).get("System.Tags", "") or ""
+    existing = [t.strip() for t in existing_str.split(";") if t.strip()]
+
+    if tag in existing:
+        return {"success": True, "tag": tag, "already_present": True}
+
+    existing.append(tag)
+    new_tags_str = "; ".join(existing)
+
+    patch = [{"op": "add", "path": "/fields/System.Tags", "value": new_tags_str}]
+    patch_resp = requests.patch(
+        f"{org_url}/{project}/_apis/wit/workitems/{work_item_id}",
+        headers=get_auth_headers("application/json-patch+json"),
+        params={"api-version": _API},
+        json=patch,
+        timeout=30,
+    )
+    patch_resp.raise_for_status()
+    return {"success": True, "tag": tag, "tags": new_tags_str}
+
+
 # ---------------------------------------------------------------------------
 # Helpers
 # ---------------------------------------------------------------------------