From 75ff3306f95b1793133f87c07d8d5e56db71b4c0 Mon Sep 17 00:00:00 2001 From: Stas Moreinis Date: Wed, 5 Nov 2025 15:01:58 -0800 Subject: [PATCH 1/2] Update base_acp_server.py --- src/agentex/lib/sdk/fastacp/base/base_acp_server.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/agentex/lib/sdk/fastacp/base/base_acp_server.py b/src/agentex/lib/sdk/fastacp/base/base_acp_server.py index eb945161..cea3c0c5 100644 --- a/src/agentex/lib/sdk/fastacp/base/base_acp_server.py +++ b/src/agentex/lib/sdk/fastacp/base/base_acp_server.py @@ -78,6 +78,9 @@ def __init__(self): self.add_middleware(RequestIDMiddleware) self._handlers: dict[RPCMethod, Callable] = {} + # Agent info to return in healthz + self.agent_id: str | None = None + @classmethod def create(cls): """Create and initialize BaseACPServer instance""" @@ -94,6 +97,7 @@ def get_lifespan_function(self): @asynccontextmanager async def lifespan_context(app: FastAPI): # noqa: ARG001 env_vars = EnvironmentVariables.refresh() + self.agent_id = env_vars.AGENT_ID if env_vars.AGENTEX_BASE_URL: await register_agent(env_vars) else: @@ -105,7 +109,10 @@ async def lifespan_context(app: FastAPI): # noqa: ARG001 async def _healthz(self): """Health check endpoint""" - return {"status": "healthy"} + result = {"status": "healthy"} + if self.agent_id: + result["agent_id"] = self.agent_id + return result def _wrap_handler(self, fn: Callable[..., Awaitable[Any]]): """Wraps handler functions to provide JSON-RPC 2.0 response format""" From d1dd0f098a10123ae5f3c61ce63d612be8e617cb Mon Sep 17 00:00:00 2001 From: Stas Moreinis Date: Wed, 5 Nov 2025 16:05:47 -0800 Subject: [PATCH 2/2] Update base_acp_server.py --- src/agentex/lib/sdk/fastacp/base/base_acp_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/agentex/lib/sdk/fastacp/base/base_acp_server.py b/src/agentex/lib/sdk/fastacp/base/base_acp_server.py index cea3c0c5..b625eaa1 100644 --- a/src/agentex/lib/sdk/fastacp/base/base_acp_server.py +++ b/src/agentex/lib/sdk/fastacp/base/base_acp_server.py @@ -97,9 +97,9 @@ def get_lifespan_function(self): @asynccontextmanager async def lifespan_context(app: FastAPI): # noqa: ARG001 env_vars = EnvironmentVariables.refresh() - self.agent_id = env_vars.AGENT_ID if env_vars.AGENTEX_BASE_URL: await register_agent(env_vars) + self.agent_id = env_vars.AGENT_ID else: logger.warning("AGENTEX_BASE_URL not set, skipping agent registration")