Merge pull request #179 from esnible/custom-host-part2

Feature: Allow AGENT_ENDPOINT env var to customize URL on Agent card

Author: esnible

a2a/a2a_contact_extractor/__main__.py

@@ -4,6 +4,7 @@
 """
 
 import logging
+import os
 
 import click
 import httpx
@@ -95,7 +96,8 @@ def get_agent_card(host: str, port: int):
     return AgentCard(
         name="Marvin Contact Extractor",
         description="Extracts structured contact information from text using Marvin's extraction capabilities",
-        url=f"http://{host}:{port}/",
+        # Allow env var AGENT_ENDPOINT to override the URL in the agent card
+        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
         version="1.0.0",
         defaultInputModes=ExtractorAgent.SUPPORTED_CONTENT_TYPES,
         defaultOutputModes=ExtractorAgent.SUPPORTED_CONTENT_TYPES,

a2a/a2a_currency_converter/app/__main__.py

@@ -52,7 +52,7 @@ def main(host, port):
             name="Currency Agent",
             description="Helps with exchange rates for currencies",
             # Allow env var AGENT_ENDPOINT to override the URL in the agent card
-            url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}/"),
+            url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
             version="1.0.0",
             default_input_modes=CurrencyAgent.SUPPORTED_CONTENT_TYPES,
             default_output_modes=CurrencyAgent.SUPPORTED_CONTENT_TYPES,

a2a/cheerup_agent/src/cheerup_agent/agent.py

@@ -1,4 +1,5 @@
 import logging
+import os
 from textwrap import dedent
 
 import uvicorn
@@ -48,7 +49,8 @@ def get_agent_card(host: str, port: int):
             - Be a positive, supportive friend to chat with
             """,
         ),
-        url=f"http://{host}:{port}/",
+        # Allow env var AGENT_ENDPOINT to override the URL in the agent card
+        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
         version="1.0.0",
         default_input_modes=["text"],
         default_output_modes=["text"],

a2a/file_organizer/src/file_organizer/agent.py

@@ -48,7 +48,8 @@ def get_agent_card(host: str, port: int):
             - **MCP Tool Calling** – uses a MCP tool to organize files.
             """,
         ),
-        url=f"http://{host}:{port}/",
+        # Allow env var AGENT_ENDPOINT to override the URL in the agent card
+        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
         version="1.0.0",
         default_input_modes=["text"],
         default_output_modes=["text"],

a2a/generic_agent/src/generic_agent/agent.py

@@ -1,4 +1,5 @@
 import logging
+import os
 from textwrap import dedent
 
 import uvicorn
@@ -49,7 +50,8 @@ def get_agent_card(host: str, port: int) -> AgentCard:
             This agent provides assistance for various tasks using different MCP tools.{mcp_section}
             """,
         ),
-        url=f"http://{host}:{port}/",
+        # Allow env var AGENT_ENDPOINT to override the URL in the agent card
+        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
         version=config.AGENT_VERSION,
         default_input_modes=["text"],
         default_output_modes=["text"],

a2a/git_issue_agent/a2a_agent.py

@@ -3,6 +3,7 @@
 """
 
 import logging
+import os
 import sys
 import traceback
 
@@ -53,7 +54,8 @@ def get_agent_card(host: str, port: int):
     return AgentCard(
         name="Github issue agent",
         description="Answer queries about Github issues",
-        url=f"http://{host}:{port}/",
+        # Allow env var AGENT_ENDPOINT to override the URL in the agent card
+        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
         version="1.0.0",
         default_input_modes=["text"],
         default_output_modes=["text"],

a2a/image_service/src/image_service/agent.py

@@ -51,7 +51,7 @@ def get_agent_card(host: str, port: int):
             """
         ),
         # Allow env var AGENT_ENDPOINT to override the URL in the agent card
-        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}/"),
+        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
         version="1.0.0",
         default_input_modes=["text"],
         default_output_modes=["text"],

a2a/recipe_agent/src/recipe_agent/agent.py

@@ -1,4 +1,5 @@
 import logging
+import os
 from textwrap import dedent
 
 import uvicorn
@@ -46,7 +47,8 @@ def get_agent_card(host: str, port: int):
             - Get 2-3 recipe suggestions with instructions
             """,
         ),
-        url=f"http://{host}:{port}/",
+        # Allow env var AGENT_ENDPOINT to override the URL in the agent card
+        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
         version="1.0.0",
         default_input_modes=["text"],
         default_output_modes=["text"],

a2a/reservation_service/src/reservation_service/agent.py

@@ -60,7 +60,8 @@ def get_agent_card(host: str, port: int):
             - **Real-time Availability** – Checks current restaurant availability
             """,
         ),
-        url=f"http://{host}:{port}/",
+        # Allow env var AGENT_ENDPOINT to override the URL in the agent card
+        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
         version="1.0.0",
         default_input_modes=["text"],
         default_output_modes=["text"],

a2a/slack_researcher/a2a_agent.py

@@ -3,6 +3,7 @@
 """
 
 import logging
+import os
 import sys
 import traceback
 from typing import Callable
@@ -54,7 +55,8 @@ def get_agent_card(host: str, port: int):
     return AgentCard(
         name="Web Research Agent",
         description="Answer queries by searching through a given slack server",
-        url=f"http://{host}:{port}/",
+        # Allow env var AGENT_ENDPOINT to override the URL in the agent card
+        url=os.getenv("AGENT_ENDPOINT", f"http://{host}:{port}").rstrip("/") + "/",
         version="1.0.0",
         default_input_modes=["text"],
         default_output_modes=["text"],