aip-agents-binary 0.5.20__py3-none-manylinux_2_31_x86_64.whl

aip_agents/examples/hello_world_a2a_google_adk_client.py

@@ -0,0 +1,49 @@
+"""Example of a General Assistant GoogleADKAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import json
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    agent_name = "GoogleAssistantAgent"
+    # Create the agent with simplified instructions tailored to the agent name
+    assistant_agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="You are a helpful assistant that can help with various tasks by delegating to specialized agents.",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8002"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    query = "What is the weather in Jakarta?"
+    logger.info(f"Processing Query: {query}")
+
+    response = assistant_agent.send_to_agent(agent_cards[0], query)
+    logger.info(f"Raw response: \n\n{json.dumps(response, indent=4)}Q")
+    logger.info(f"Agent Response: {response['content']}")
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_google_adk_client.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""

aip_agents/examples/hello_world_a2a_google_adk_client_agent.py

@@ -0,0 +1,48 @@
+"""Example of a General Assistant GoogleADKAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    agent_name = "GoogleAssistantAgent"
+    assistant_agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="You are a helpful assistant that can help with various tasks by delegating to specialized agents.",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8002"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    # Register agents
+    assistant_agent.register_a2a_agents(agent_cards)
+
+    query = "What is the weather in Jakarta?"
+    logger.info(f"Processing Query: {query}")
+
+    response = assistant_agent.run(query)
+    logger.info(f"Assistant Agent Response: {response['output']}")
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_google_adk_client_agent.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""

aip_agents/examples/hello_world_a2a_google_adk_client_streaming.py

@@ -0,0 +1,60 @@
+"""Example of a General Assistant GoogleADKAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A with streaming.
+4. Handling streaming responses and providing relevant advice.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import asyncio
+
+from a2a.types import TaskState
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+async def main():
+    """Main function demonstrating the General Assistant agent with streaming A2A capabilities."""
+    agent_name = "GoogleADKWeatherAgent"
+
+    # Create the agent with simplified instructions and our tool
+    assistant_agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="You are a helpful assistant that can help with various tasks by delegating to specialized agents.",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8002"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    query = "What is the weather in Jakarta?"
+    logger.info(f"Processing Query: {query}")
+
+    agent_result = ""
+    async for chunk in assistant_agent.astream_to_agent(agent_card=agent_cards[0], message=query):
+        task_state = chunk.get("task_state")
+        content = chunk.get("content", "")
+        if task_state == TaskState.working and content:
+            logger.info(f"Event Working: {content}")
+        elif task_state == TaskState.completed and content:
+            agent_result += content
+
+    logger.info(f"Agent Response: {agent_result}")
+    return agent_result
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_google_adk_client_streaming.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+
+async def main():
+    """Main function demonstrating the General Assistant agent with streaming A2A capabilities."""

aip_agents/examples/hello_world_a2a_google_adk_server.py

@@ -0,0 +1,79 @@
+"""Example A2A server for a Google ADK Weather Service.
+
+This server instantiates a Google ADK agent with weather lookup capabilities and serves it
+via the A2A protocol using GoogleADKAgent's to_a2a method.
+
+To run this server:
+    python examples/hello_world_a2a_google_adk_server.py
+
+It will listen on http://localhost:8002 by default.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import click
+import uvicorn
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.tools.adk_weather_tool import weather_tool
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+SERVER_AGENT_NAME = "GoogleADKWeatherAgent"
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8002, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the Google ADK Weather A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """
+    logger.info(f"Starting {SERVER_AGENT_NAME} on http://{host}:{port}")
+
+    agent_card = AgentCard(
+        name=SERVER_AGENT_NAME,
+        description="A weather agent based on Google ADK that provides weather information for cities.",
+        url=f"http://{host}:{port}",
+        version="1.0.0",
+        defaultInputModes=["text"],
+        defaultOutputModes=["text"],
+        capabilities=AgentCapabilities(streaming=True),
+        skills=[
+            AgentSkill(
+                id="weather",
+                name="Weather Lookup",
+                description="Provides current weather information for cities.",
+                examples=["What's the weather in Tokyo?", "Get weather for London"],
+                tags=["weather"],
+            )
+        ],
+        tags=["weather"],
+    )
+
+    # Create a Google ADK agent
+    agent = GoogleADKAgent(
+        model="gemini-2.0-flash",
+        name=SERVER_AGENT_NAME,
+        description="A weather agent that provides weather information for cities.",
+        instruction="""You are a weather agent that provides weather information for cities.
+        Always use the weather_tool for looking up weather data. Format your responses clearly and professionally.""",
+        tools=[weather_tool],
+    )
+
+    # Convert the agent to an A2A server directly
+    app = agent.to_a2a(agent_card=agent_card)
+
+    uvicorn.run(app, host=host, port=port)
+    logger.info("A2A application configured. Starting Uvicorn server...")
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_google_adk_server.pyi

@@ -0,0 +1,15 @@
+from _typeshed import Incomplete
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.tools.adk_weather_tool import weather_tool as weather_tool
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+SERVER_AGENT_NAME: str
+
+def main(host: str, port: int):
+    """Runs the Google ADK Weather A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """

aip_agents/examples/hello_world_a2a_langchain_client.py

@@ -0,0 +1,39 @@
+"""Example of a General Assistant LangChainAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    assistant_agent = LangChainAgent(
+        name="AssistantAgentLangChain",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    response = assistant_agent.send_to_agent(agent_cards[0], message="What is the weather in Jakarta?")
+    print(response.get("content", str(response)))
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_langchain_client.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""

aip_agents/examples/hello_world_a2a_langchain_client_agent.py

@@ -0,0 +1,39 @@
+"""Example of a General Assistant LangChainAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    assistant_agent = LangChainAgent(
+        name="AssistantAgentLangChain",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+    assistant_agent.register_a2a_agents(agent_cards)
+
+    response = assistant_agent.run("What is the weather in Jakarta?")
+    print(response["output"])
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_langchain_client_agent.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""

aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.py

@@ -0,0 +1,37 @@
+"""Example of a General Assistant LangChainAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    assistant_agent = LangChainAgent(
+        name="AssistantAgentLangChain",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model="openai/gpt-4o",
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    response = assistant_agent.send_to_agent(agent_cards[0], message="What is the weather in Jakarta?")
+    print(response.get("content", str(response)))
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""

aip_agents/examples/hello_world_a2a_langchain_client_streaming.py

@@ -0,0 +1,41 @@
+"""Example of a General Assistant LangChainAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A with streaming.
+4. Handling streaming responses and providing relevant advice.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+async def main():
+    """Main function demonstrating the General Assistant agent with streaming A2A capabilities."""
+    agent = LangChainAgent(
+        name="AssistantAgentLangChain",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model=ChatOpenAI(model="gpt-4.1", streaming=True, temperature=0),
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    agent_cards = agent.discover_agents(client_a2a_config)
+
+    async for chunk in agent.astream_to_agent(agent_card=agent_cards[0], message="What is the weather in Jakarta?"):
+        print(chunk.get("content", ""))
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langchain_client_streaming.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+async def main() -> None:
+    """Main function demonstrating the General Assistant agent with streaming A2A capabilities."""

aip_agents/examples/hello_world_a2a_langchain_reference_client_streaming.py

@@ -0,0 +1,60 @@
+"""Example A2A streaming client for the Google Search Tool and Time Tool Server.
+
+This client demonstrates streaming interaction with the reference server
+that provides time and mock Google search functionality via A2A protocol.
+
+To use this client:
+1. First start the reference server: python hello_world_a2a_langchain_reference_server.py
+2. Then run this client: python hello_world_a2a_langchain_reference_client_streaming.py
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+import pprint
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+async def main():
+    """Main function demonstrating streaming A2A client for the reference server."""
+    agent = LangChainAgent(
+        name="ReferenceClientAgent",
+        instruction="""You are a helpful assistant that can delegate tasks to specialized agents.
+        You can ask the mock retrieval server for mock data or mock Google search results.""",
+        model=ChatOpenAI(model="gpt-4.1", streaming=True, temperature=0),
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8455"],
+    )
+    agent_cards = agent.discover_agents(client_a2a_config)
+
+    if not agent_cards:
+        print("No agents discovered. Make sure the reference server is running on http://localhost:8455")
+        return
+
+    print(f"Discovered agent: {agent_cards[0].name}")
+    print(f"Description: {agent_cards[0].description}")
+    print(f"Skills: {[skill.name for skill in agent_cards[0].skills]}")
+
+    # Test queries to demonstrate the server's capabilities
+    test_queries = "Search for information about NeoAI and retrieve some mock data about artificial intelligence"
+
+    print(f"\n{'=' * 60}")
+    print(f"Query: {test_queries}")
+    print(f"{'=' * 60}")
+
+    async for chunk in agent.astream_to_agent(agent_card=agent_cards[0], message=test_queries):
+        print("-" * 50)
+        pprint.pprint(chunk)
+        print("-" * 50)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langchain_reference_client_streaming.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+async def main() -> None:
+    """Main function demonstrating streaming A2A client for the reference server."""

aip_agents/examples/hello_world_a2a_langchain_reference_server.py

@@ -0,0 +1,105 @@
+"""Example A2A server for a LangChainAgent Weather Service.
+
+This server instantiates a LangChainAgent with weather lookup capabilities and serves it
+via the A2A protocol using the to_a2a convenience method.
+
+To run this server:
+    python examples/hello_world_a2a_langchain_server.py
+
+It will listen on http://localhost:8001 by default.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import click
+import uvicorn
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.examples.tools import google_serper_tool, mock_retrieval_tool, time_tool
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+SERVER_AGENT_NAME = "Mock Retrieval Tool and Google Search Tool Server"
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8455, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the LangChain Mock Retrieval and Google Search A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """
+    logger.info(f"Starting {SERVER_AGENT_NAME} on http://{host}:{port}")
+
+    agent_card = AgentCard(
+        name=SERVER_AGENT_NAME,
+        description="A server that provides mock retrieval data and mock google search information",
+        url=f"http://{host}:{port}",
+        version="1.0.0",
+        defaultInputModes=["text"],
+        defaultOutputModes=["text"],
+        capabilities=AgentCapabilities(streaming=True),
+        skills=[
+            AgentSkill(
+                id="mock_retrieval",
+                name="Mock Data Retrieval",
+                description="Retrieves mock data and returns references for testing.",
+                examples=["Retrieve mock data", "Get test information"],
+                tags=["retrieval", "mock"],
+            ),
+            AgentSkill(
+                id="google_serper",
+                name="Mock Google Search",
+                description="Provides mock google search information.",
+                examples=["What is NeoAI?", "Get NeoAI"],
+                tags=["google serper"],
+            ),
+        ],
+        tags=["retrieval", "mock", "google serper"],
+    )
+
+    time_delegate_agent = LangChainAgent(
+        name="TimeLookupAgent",
+        instruction="""You are a specialized time lookup assistant.
+        Always answer user queries by calling the `time_tool` to obtain the most up-to-date information.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0, streaming=True),
+        tools=[time_tool()],
+    )
+
+    google_delegate_agent = LangChainAgent(
+        name="GoogleSerperAgent",
+        instruction="""You are a specialized research assistant that must use the `google_serper_tool`
+        for every query. Provide concise summaries using the tool output.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0, streaming=True),
+        tools=[google_serper_tool()],
+    )
+
+    langchain_agent = LangChainAgent(
+        name=SERVER_AGENT_NAME,
+        instruction="""You are a coordinator agent that delegates research tasks to sub-agents but uses
+        mock_retrieval_tool directly. For research questions, delegate to `GoogleSerperAgent`.
+        For retrieval queries, use the mock_retrieval_tool directly. Synthesize their responses
+        and present a clear combined answer.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0, streaming=True),
+        tools=[mock_retrieval_tool()],  # Direct tool usage
+        agents=[time_delegate_agent, google_delegate_agent],  # Delegation for others
+    )
+
+    app = langchain_agent.to_a2a(
+        agent_card=agent_card,
+    )
+
+    uvicorn.run(app, host=host, port=port)
+    logger.info("A2A application configured. Starting Uvicorn server...")
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_langchain_reference_server.pyi

@@ -0,0 +1,15 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.examples.tools import google_serper_tool as google_serper_tool, mock_retrieval_tool as mock_retrieval_tool, time_tool as time_tool
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+SERVER_AGENT_NAME: str
+
+def main(host: str, port: int):
+    """Runs the LangChain Mock Retrieval and Google Search A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """