devops-bot-sdk 1.4.150__tar.gz → 1.4.151__tar.gz

{devops_bot_sdk-1.4.150 → devops_bot_sdk-1.4.151}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: devops-bot-sdk
-Version: 1.4.150
+Version: 1.4.151
 Summary: DevOps Bot Desktop SDK — thin client for the AgentOS Electron desktop app
 Author: noumanaziz2128
 License-Expression: LicenseRef-Proprietary

{devops_bot_sdk-1.4.150 → devops_bot_sdk-1.4.151}/devops_bot_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: devops-bot-sdk
-Version: 1.4.150
+Version: 1.4.151
 Summary: DevOps Bot Desktop SDK — thin client for the AgentOS Electron desktop app
 Author: noumanaziz2128
 License-Expression: LicenseRef-Proprietary

{devops_bot_sdk-1.4.150 → devops_bot_sdk-1.4.151}/sdk/__init__.py

@@ -1,6 +1,6 @@
 """AgentOS Desktop SDK — thin HTTPS/SSE client for the Electron app.
 
-Version: 1.4.150
+Version: 1.4.151
 
 Public surface:
   BackendClient.from_config()        — create client from ~/.agentos/config.toml
@@ -30,7 +30,7 @@ Rules:
   - All data egress through submit_webhook only
 """
 
-__version__ = "1.4.150"   # SINGLE SOURCE OF TRUTH — bump on every change; pyproject,
+__version__ = "1.4.151"   # SINGLE SOURCE OF TRUTH — bump on every change; pyproject,
 __author__  = "AgentOS"   # sdk.client.SDK_VERSION and the UI server version all read this.
 
 from sdk.client import BackendClient

{devops_bot_sdk-1.4.150 → devops_bot_sdk-1.4.151}/sdk/agents/inventory.py

@@ -26,6 +26,7 @@ def _map_backend_agent(row: dict, source: str) -> dict:
     sys_prompt = prompt.get("system_prompt", "") if isinstance(prompt, dict) else ""
     pvars = prompt.get("variables", []) if isinstance(prompt, dict) else []
     return {
+        "id": row.get("id"),                 # backend agentId — used for activity webhook
         "name": row.get("name"),
         "description": sys_prompt[:200],
         "variables": [v.get("name") for v in pvars if isinstance(v, dict) and v.get("name")],

{devops_bot_sdk-1.4.150 → devops_bot_sdk-1.4.151}/sdk/client.py

@@ -37,6 +37,11 @@ from sdk.sse import _check_status, stream_with_reconnect
 
 logger = logging.getLogger(__name__)
 
+# Process-lifetime cache of agent NAME → backend agentId (""=known non-backend).
+# Used by resolve_agent_id so the agent-activity webhook fires only for backend
+# agents without a lookup on every run.
+_AGENT_ID_CACHE: dict[str, str] = {}
+
 # ── Ticket status-change observers ────────────────────────────────────────
 # Best-effort, synchronous hooks fired when the client changes a ticket's backend
 # status (In Progress / Backlog / To Do). The CLI loop registers one to track
@@ -1036,6 +1041,58 @@ class BackendClient:
             logger.warning("client.log_activity error err=%s", exc)
             return False
 
+    async def log_agent_activity(self, activity: str, agent_id: str) -> bool:
+        """POST {ML_GATEWAY_URL}/ml-api/agent-activity — record an agent-activity
+        event for a **backend** agent (body ``{activity, agentId}``). Best-effort;
+        never raises. No-op when ``agent_id`` is empty (local-only agents are not
+        tracked — the backend is the source of truth for agents).
+        """
+        from sdk.config import ML_GATEWAY_URL
+
+        aid = (agent_id or "").strip()
+        if not aid:
+            return False
+        url = f"{ML_GATEWAY_URL.rstrip('/')}/ml-api/agent-activity"
+        headers = {**self._headers, "accept": "*/*", "developer-name": "null", "product": ""}
+        try:
+            async with httpx.AsyncClient(timeout=10.0) as client:
+                resp = await client.post(
+                    url, json={"activity": str(activity), "agentId": aid}, headers=headers,
+                )
+            if resp.status_code >= 400:
+                logger.warning("client.log_agent_activity failed status=%s body=%s",
+                               resp.status_code, resp.text[:200])
+                return False
+            return True
+        except Exception as exc:  # noqa: BLE001
+            logger.warning("client.log_agent_activity error err=%s", exc)
+            return False
+
+    async def resolve_agent_id(self, name: str) -> str | None:
+        """Best-effort backend ``agentId`` for an agent NAME (custom + department).
+
+        Returns None when the name isn't a backend agent (e.g. a local-only manifest)
+        — the caller then skips activity emit, so only backend agents are tracked.
+        Cached per process (the id↔name mapping is stable within a session).
+        """
+        nm = (name or "").strip()
+        if not nm:
+            return None
+        if nm in _AGENT_ID_CACHE:
+            return _AGENT_ID_CACHE[nm] or None
+        rows: list[dict] = []
+        for fetch in (self.list_custom_agents, self.list_department_agents):
+            try:
+                rows += await fetch()
+            except Exception:  # noqa: BLE001 — best-effort across both sources
+                pass
+        for r in rows:
+            if r.get("name") == nm and r.get("id"):
+                _AGENT_ID_CACHE[nm] = str(r["id"])
+                return _AGENT_ID_CACHE[nm]
+        _AGENT_ID_CACHE[nm] = ""     # negative-cache: not a backend agent
+        return None
+
     async def notify_whatsapp(
         self,
         task_id: str,

{devops_bot_sdk-1.4.150 → devops_bot_sdk-1.4.151}/sdk/config.py

@@ -25,7 +25,10 @@ _CONFIG_FILE = _CONFIG_DIR / "config.toml"
 _SENTINEL    = "co_"
 
 AI_DEFAULT_URL = "https://devopsbot-ai.apiswagger.co.uk"
-BE_DEFAULT_URL = "https://devopsbot-be.apiswagger.co.uk"
+# ML backend / gateway — all /ml-api, /tasks-ml, /approvals-ml, agent-activity, etc.
+# now go through the ML gateway (was devopsbot-be.apiswagger.co.uk).
+BE_DEFAULT_URL = "https://gateway-ml.apiswagger.co.uk"
+ML_GATEWAY_URL = "https://gateway-ml.apiswagger.co.uk"
 
 
 # ── Machine-derived Fernet key ────────────────────────────────────────

{devops_bot_sdk-1.4.150 → devops_bot_sdk-1.4.151}/sdk/ui/api.py

@@ -304,6 +304,28 @@ async def logs_stream() -> StreamingResponse:
     )
 
 
+# ── Agent-activity webhook ──────────────────────────────────────────────────
+# Every custom/default agent run/chat records to the LOCAL usage ledger AND emits
+# a `{activity, agentId}` event to the ML gateway (POST /ml-api/agent-activity),
+# which stores it for RBAC-scoped reporting via the chat endpoint. Only agents that
+# exist in the backend (have an agentId) are emitted — local-only agents are skipped.
+# Best-effort: never raises into the run.
+
+async def _emit_agent_activity(name: str, kind: str, result: dict, duration_s: float = 0.0) -> None:
+    try:
+        from sdk.client import BackendClient
+        client = BackendClient.from_config()
+        agent_id = await client.resolve_agent_id(name)
+        if not agent_id:                       # local-only agent → not tracked
+            return
+        verb = "Chatted with" if kind == "chat" else "Ran"
+        status = "ok" if result.get("ok") else "error"
+        detail = f" ({round(duration_s, 1)}s)" if duration_s else ""
+        await client.log_agent_activity(f"{verb} agent '{name}' — {status}{detail}", agent_id)
+    except Exception:  # noqa: BLE001 — activity emit must never break a run
+        pass
+
+
 # ── Agents: catalog, creation, chat, default agents ────────────────────────
 
 @router.get("/connectors", dependencies=[_auth])
@@ -689,12 +711,14 @@ async def agent_chat(body: AgentChatRequest) -> dict:
         started = _time.monotonic()
         result = await _run()
 
+        elapsed = _time.monotonic() - started
         usage_mod.record(
-            name, kind="chat", duration_s=_time.monotonic() - started,
+            name, kind="chat", duration_s=elapsed,
             input_tokens=result.get("input_tokens", 0),
             output_tokens=result.get("output_tokens", 0),
             cost_usd=result.get("cost_usd", 0.0), ok=bool(result.get("ok")),
         )
+        await _emit_agent_activity(name, "chat", result, elapsed)
         return result
 
     # Run detached from this request: navigating away can't kill it. Poll
@@ -732,12 +756,14 @@ async def agent_run(body: AgentRunRequest) -> dict:
         started = _time.monotonic()
         result = await _run()
 
+        elapsed = _time.monotonic() - started
         usage_mod.record(
-            name, kind="run", duration_s=_time.monotonic() - started,
+            name, kind="run", duration_s=elapsed,
             input_tokens=result.get("input_tokens", 0),
             output_tokens=result.get("output_tokens", 0),
             cost_usd=result.get("cost_usd", 0.0), ok=bool(result.get("ok")),
         )
+        await _emit_agent_activity(name, "run", result, elapsed)
         return result
 
     return {"job_id": jobs.start(work())}

{devops_bot_sdk-1.4.150 → devops_bot_sdk-1.4.151}/sdk/updater.py

@@ -132,10 +132,10 @@ def _ml_api_base() -> str:
     try:
         from sdk.config import get_base_url
         base = get_base_url()
-        # devopsbot-ai.apiswagger.co.uk  →  devopsbot-be.apiswagger.co.uk
-        return base.replace("devopsbot-ai", "devopsbot-be")
+        # devopsbot-ai.apiswagger.co.uk  →  gateway-ml.apiswagger.co.uk
+        return base.replace("devopsbot-ai.apiswagger.co.uk", "gateway-ml.apiswagger.co.uk")
     except Exception:  # noqa: BLE001
-        return "https://devopsbot-be.apiswagger.co.uk"
+        return "https://gateway-ml.apiswagger.co.uk"
 
 
 def _env_github_token() -> str | None: