From 6c3c9913dbed78cc2490db99ab458c62bf87affb Mon Sep 17 00:00:00 2001
From: Abhimanyu Saharan <asaharan@onemindservices.com>
Date: Mon, 16 Feb 2026 01:46:06 +0530
Subject: [PATCH] feat: update agent heartbeat endpoint to require no request
 payload

---
 backend/app/api/agent.py                      | 10 ++++------
 backend/tests/test_openapi_agent_role_tags.py |  7 +++++++
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/backend/app/api/agent.py b/backend/app/api/agent.py
index da400e4a..a290420b 100644
--- a/backend/app/api/agent.py
+++ b/backend/app/api/agent.py
@@ -27,7 +27,6 @@ from app.models.tasks import Task
 from app.schemas.agents import (
     AgentCreate,
     AgentHeartbeat,
-    AgentHeartbeatCreate,
     AgentNudge,
     AgentRead,
 )
@@ -1323,8 +1322,8 @@ async def nudge_agent(
     tags=AGENT_ALL_ROLE_TAGS,
     summary="Upsert agent heartbeat",
     description=(
-        "Record liveness for the authenticated agent's current status.\n\n"
-        "Use this when the agent heartbeat loop reports status changes."
+        "Record liveness for the authenticated agent.\n\n"
+        "Use this when the agent heartbeat loop checks in."
     ),
     openapi_extra={
         "x-llm-intent": "agent_heartbeat",
@@ -1339,7 +1338,7 @@ async def nudge_agent(
         "x-required-actor": "any_agent",
         "x-prerequisites": [
             "Authenticated agent token",
-            "Valid AgentHeartbeatCreate payload",
+            "No request payload required",
         ],
         "x-side-effects": [
             "Updates agent heartbeat and status metadata",
@@ -1372,7 +1371,6 @@ async def nudge_agent(
     },
 )
 async def agent_heartbeat(
-    payload: AgentHeartbeatCreate,
     session: AsyncSession = SESSION_DEP,
     agent_ctx: AgentAuthContext = AGENT_CTX_DEP,
 ) -> AgentRead:
@@ -1383,7 +1381,7 @@ async def agent_heartbeat(
     # Heartbeats must apply to the authenticated agent; agent names are not unique.
     return await agents_api.heartbeat_agent(
         agent_id=str(agent_ctx.agent.id),
-        payload=AgentHeartbeat(status=payload.status),
+        payload=AgentHeartbeat(),
         session=session,
         actor=_actor(agent_ctx),
     )
diff --git a/backend/tests/test_openapi_agent_role_tags.py b/backend/tests/test_openapi_agent_role_tags.py
index ff00e166..2784f274 100644
--- a/backend/tests/test_openapi_agent_role_tags.py
+++ b/backend/tests/test_openapi_agent_role_tags.py
@@ -101,6 +101,13 @@ def test_openapi_agent_role_endpoint_descriptions_exist() -> None:
     )
 
 
+def test_openapi_agent_heartbeat_requires_no_request_body() -> None:
+    """Authenticated heartbeats should infer identity from token without payload."""
+    schema = app.openapi()
+    op = schema["paths"]["/api/v1/agent/heartbeat"]["post"]
+    assert "requestBody" not in op
+
+
 def test_openapi_agent_tool_endpoints_include_llm_hints() -> None:
     """Tool-facing agent endpoints should expose structured usage hints and operation IDs."""
     schema = app.openapi()