@qa-gentic/stlc-agents 1.0.23 → 1.0.25

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@qa-gentic/stlc-agents",
-  "version": "1.0.23",
+  "version": "1.0.25",
   "description": "QA STLC Agents — five MCP servers + skills for AI-powered test case, Gherkin, Playwright generation, and Helix-QA file writing against Azure DevOps and Jira Cloud. Full pipeline for both: fetch → test cases → Gherkin → Playwright → Helix-QA. Works with Claude Code, GitHub Copilot, Cursor, Windsurf.",
   "keywords": [
     "playwright",

package/src/stlc_agents/agent_gherkin_generator/server.py

@@ -17,6 +17,7 @@ Skills: see skills/generate-gherkin.md
 import asyncio
 import json
 import sys
+import time
 
 from dotenv import load_dotenv
 from mcp.server import Server
@@ -31,6 +32,7 @@ from stlc_agents.agent_gherkin_generator.tools.ado_gherkin import (
     attach_work_item_file as _attach_wi_file,
     validate_gherkin_content as _validate_gherkin,
 )
+from stlc_agents.shared.cost_tracker import track
 
 load_dotenv()
 
@@ -377,6 +379,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_feature_hierarchy":
             result = await asyncio.to_thread(
@@ -415,7 +418,7 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                         "Fix the errors below before attaching to ADO."
                     ),
                 }
-                return [types.TextContent(type="text", text=json.dumps(result, indent=2, ensure_ascii=False))]
+                return track(result, tool_name=name, server="qa-gherkin-generator", t0=t0)
 
             result = await asyncio.to_thread(
                 _attach_feature,
@@ -445,7 +448,7 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                         "Fix the errors below before attaching to ADO."
                     ),
                 }
-                return [types.TextContent(type="text", text=json.dumps(result, indent=2, ensure_ascii=False))]
+                return track(result, tool_name=name, server="qa-gherkin-generator", t0=t0)
 
             result = await asyncio.to_thread(
                 _attach_wi_file,
@@ -506,13 +509,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-gherkin-generator", 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-gherkin-generator", t0=t0)
 
 
 async def _run():

package/src/stlc_agents/agent_helix_writer/server.py

@@ -23,6 +23,7 @@ import asyncio
 import json
 import re
 import sys
+import time
 
 from dotenv import load_dotenv
 from mcp.server import Server
@@ -36,6 +37,7 @@ from stlc_agents.agent_helix_writer.tools.helix_write import (
     list_helix_tree       as _list_tree,
     update_helix_file     as _update_file,
 )
+from stlc_agents.shared.cost_tracker import track
 
 load_dotenv()
 
@@ -448,6 +450,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 == "inspect_helix_project":
             result = await asyncio.to_thread(_inspect, arguments["helix_root"])
@@ -467,7 +470,7 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                         "below and retry. No files were written to disk."
                     ),
                 }
-                return [types.TextContent(type="text", text=json.dumps(result, indent=2, ensure_ascii=False))]
+                return track(result, tool_name=name, server="qa-helix-writer", t0=t0)
 
             result = await asyncio.to_thread(
                 _write_files,
@@ -507,13 +510,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-helix-writer", 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-helix-writer", t0=t0)
 
 
 async def _run():

package/src/stlc_agents/agent_jira_manager/server.py

@@ -22,6 +22,7 @@ Skills: see skills/qa-jira-manager.md
 import asyncio
 import json
 import sys
+import time
 
 from dotenv import load_dotenv
 from mcp.server import Server
@@ -36,6 +37,7 @@ from stlc_agents.agent_jira_manager.tools.jira_workitem import (
     get_linked_test_cases as _get_linked_test_cases,
     attach_gherkin_to_issue as _attach_gherkin,
 )
+from stlc_agents.shared.cost_tracker import track
 
 load_dotenv()
 
@@ -411,6 +413,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:
         cloud_id = (arguments.get("cloud_id") or "").strip() or get_cloud_id()
 
@@ -442,7 +445,7 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                         "below and retry. No test cases were created in Jira."
                     ),
                 }
-                return [types.TextContent(type="text", text=json.dumps(result, indent=2, ensure_ascii=False))]
+                return track(result, tool_name=name, server="qa-jira-manager", t0=t0)
 
             # ── Peek at the issue to get type and project_key ─────────────
             issue_data: dict = {}
@@ -484,7 +487,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-jira-manager", t0=t0)
 
             if not project_key:
                 result = {
@@ -494,7 +497,7 @@ async def call_tool(name: str, arguments: dict) -> list[types.TextContent]:
                         "Pass project_key explicitly, e.g. 'PROJ'."
                     ),
                 }
-                return [types.TextContent(type="text", text=json.dumps(result, indent=2, ensure_ascii=False))]
+                return track(result, tool_name=name, server="qa-jira-manager", t0=t0)
 
             # ── Create test cases ─────────────────────────────────────────
             created = []
@@ -638,13 +641,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-jira-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-jira-manager", t0=t0)
 
 
 # ---------------------------------------------------------------------------

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
@@ -30,6 +31,7 @@ from stlc_agents.agent_test_case_manager.tools.ado_workitem import (
     link_test_cases_to_work_item as _link_test_cases,
     get_linked_test_cases as _get_linked_test_cases,
 )
+from stlc_agents.shared.cost_tracker import track
 
 load_dotenv()
 
@@ -351,6 +353,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 +383,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 +402,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 +451,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
 
@@ -565,13 +568,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)
 
 
 # ---------------------------------------------------------------------------