init

Author: smoreinis

src/agentex/lib/environment_variables.py

@@ -34,6 +34,10 @@ class EnvVarKeys(str, Enum):
     WORKFLOW_TASK_QUEUE = "WORKFLOW_TASK_QUEUE"
     # Auth Configuration
     AUTH_PRINCIPAL_B64 = "AUTH_PRINCIPAL_B64"
+    # Build Information
+    AGENT_COMMIT = "AGENT_COMMIT"
+    # The URL of the agent's source code
+    AGENT_CODE_URL = "AGENT_CODE_URL"
 
 
 class Environment(str, Enum):
@@ -63,6 +67,9 @@ class EnvironmentVariables(BaseModel):
     WORKFLOW_TASK_QUEUE: str | None = None
     WORKFLOW_NAME: str | None = None
     AUTH_PRINCIPAL_B64: str | None = None
+    # Build Information
+    AGENT_COMMIT: str | None = None
+    AGENT_CODE_URL: str | None = None
 
     @classmethod
     def refresh(cls) -> EnvironmentVariables:

src/agentex/lib/sdk/fastacp/fastacp.py

@@ -1,3 +1,8 @@
+import inspect
+import json
+import os
+from pathlib import Path
+
 from typing import Literal
 from agentex.lib.sdk.fastacp.base.base_acp_server import BaseACPServer
 from agentex.lib.sdk.fastacp.impl.agentic_base_acp import AgenticBaseACP
@@ -8,6 +13,7 @@
     BaseACPConfig,
     SyncACPConfig,
 )
+from agentex.lib.utils.logging import make_logger
 
 # Add new mappings between ACP types and configs here
 # Add new mappings between ACP types and implementations here
@@ -16,6 +22,7 @@
     "base": AgenticBaseACP,
 }
 
+logger = make_logger(__name__)
 
 class FastACP:
     """Factory for creating FastACP instances
@@ -51,6 +58,22 @@ def create_agentic_acp(config: AgenticACPConfig, **kwargs) -> BaseACPServer:
         else:
             return implementation_class.create(**kwargs)
 
+    @staticmethod
+    def maybe_get_build_info() -> None:
+        """If a build-info.json file is present, load it and set the AGENT_COMMIT and AGENT_CODE_URL environment variables"""
+        acp_root = Path(inspect.stack()[1].filename).resolve().parents[0]
+        build_info_path = acp_root / "build-info.json"
+        if build_info_path.exists():
+            try:
+                with open(build_info_path, "r") as f:
+                    build_info = json.load(f)
+                    if build_info.get("agent_commit"):
+                        os.environ["AGENT_COMMIT"] = build_info.get("agent_commit")
+                    if build_info.get("agent_repo") and build_info.get("agent_path"):
+                        os.environ["AGENT_CODE_URL"] = build_info.get("agent_repo") + "/" + build_info.get("agent_path")
+            except Exception as e:
+                logger.error(f"Error loading build info: {e}")
+
     @staticmethod
     def create(
         acp_type: Literal["sync", "agentic"], config: BaseACPConfig | None = None, **kwargs
@@ -63,6 +86,8 @@ def create(
             **kwargs: Additional configuration parameters
         """ 
 
+        FastACP.maybe_get_build_info()
+        
         if acp_type == "sync":
             sync_config = config if isinstance(config, SyncACPConfig) else None
             return FastACP.create_sync_acp(sync_config, **kwargs)

src/agentex/lib/utils/registration.py

@@ -44,6 +44,11 @@ async def register_agent(env_vars: EnvironmentVariables):
     if env_vars.AGENT_ID:
         registration_data["agent_id"] = env_vars.AGENT_ID
 
+    if env_vars.AGENT_COMMIT:
+        registration_data["commit_hash"] = env_vars.AGENT_COMMIT
+    if env_vars.AGENT_CODE_URL:
+        registration_data["code_url"] = env_vars.AGENT_CODE_URL
+
     # Make the registration request
     registration_url = f"{env_vars.AGENTEX_BASE_URL.rstrip('/')}/agents/register"
     # Retry logic with configurable attempts and delay