aigency 0.0.1rc136602178__tar.gz → 0.0.1rc163779164__tar.gz

{aigency-0.0.1rc136602178 → aigency-0.0.1rc163779164}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aigency
-Version: 0.0.1rc136602178
+Version: 0.0.1rc163779164
 Summary: Add your description here
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown

aigency-0.0.1rc163779164/aigency/agents/client.py

@@ -0,0 +1,59 @@
+import httpx
+ 
+from a2a.client.client import ClientConfig
+from a2a.client.client_factory import ClientFactory
+from a2a.types import AgentCard, Message, SendMessageResponse
+#TODO: Enable when auth is implemented
+#from a2a.client.auth.interceptor import AuthInterceptor
+
+
+class AgentClient:
+    """A class to hold the connections to the remote agents.
+
+    This class manages connections to remote agents using the A2A protocol.
+    It provides methods for retrieving agent information and sending messages
+    to remote agents.
+
+    Attributes:
+        _httpx_client (httpx.AsyncClient): The HTTP client used for asynchronous requests.
+        agent_card (AgentCard): The agent card containing metadata about the remote agent.
+    """
+
+    def __init__(self, agent_card: AgentCard):
+        """Initialize a connection to a remote agent.
+
+        Args:
+            agent_card (AgentCard): The agent card containing metadata about the remote agent.
+
+        Raises:
+            None
+
+        Returns:
+            None
+        """
+        self._httpx_client = httpx.AsyncClient(timeout=60)
+        self.card = agent_card
+
+        config = ClientConfig(httpx_client=self._httpx_client)
+        factory = ClientFactory(config=config)
+        self.agent_client = factory.create(agent_card)
+
+    def get_agent(self) -> AgentCard:
+        """Get the agent card for this remote agent connection.
+
+        Returns:
+            AgentCard: The agent card containing metadata about the remote agent.
+        """
+        return self.card
+
+    async def send_message(self, message_request: Message) -> SendMessageResponse:
+        """Send a message to the remote agent.
+
+        Args:
+            message_request (Message): The message request to send to the remote agent.
+
+        Returns:
+            SendMessageResponse: The response from the remote agent.
+        """
+        async for response in self.agent_client.send_message(message_request):
+            yield response

aigency-0.0.1rc163779164/aigency/agents/communicator.py

@@ -0,0 +1,107 @@
+import logging
+import uuid
+from typing import Any
+from a2a.types import Message, Task
+from google.adk.tools.tool_context import ToolContext
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+class Communicator:
+    """Base class for Agent-to-Agent communication."""
+
+    def __init__(self, remote_agent_connections=None):
+        self.remote_agent_connections = remote_agent_connections
+
+    async def send_message(self, agent_name: str, task: str, tool_context: ToolContext):
+        """Delegate a task to a specified remote agent.
+
+        This method sends a message to a remote agent, requesting it to perform a task.
+        It handles the creation of the message payload and manages the communication
+        with the remote agent.
+
+        Args:
+            agent_name: Name of the remote agent to send the task to.
+            task: Detailed description of the task for the remote agent to perform.
+            tool_context: Context object containing state and other information.
+
+        Returns:
+            Task object if successful, None otherwise.
+
+        Raises:
+            ValueError: If the specified agent is not found in the available connections.
+        """
+        logger.info(
+            f"`send_message` triggered with agent_name: {agent_name}, task: {task}"
+        )
+        if agent_name not in self.remote_agent_connections:
+            logger.error(
+                f"LLM tried to call '{agent_name}' but it was not found. "
+                f"Available agents: {list(self.remote_agent_connections.keys())}"
+            )
+            raise ValueError(f"Agent '{agent_name}' not found.")
+
+        state = tool_context.state
+        client = self.remote_agent_connections[agent_name]
+
+        if "remote_agent_contexts" not in state:
+            state["remote_agent_contexts"] = {}
+
+        if agent_name not in state["remote_agent_contexts"]:
+            logger.debug(f"Creating new context for agent: {agent_name}")
+            state["remote_agent_contexts"][agent_name] = {
+                "context_id": str(uuid.uuid4())
+            }
+        context_id = state["remote_agent_contexts"][agent_name]["context_id"]
+        task_id = state.get("task_id", None)
+        message_id = state.get("input_message_metadata", {}).get(
+            "message_id", str(uuid.uuid4())
+        )
+
+        payload = self.create_send_message_payload(task, task_id, context_id)
+        payload["message"]["message_id"] = message_id
+        logger.debug("`send_message` triggered with payload: %s", payload)
+
+        send_response = None
+        async for resp in client.send_message(
+            message_request=Message(**payload.get("message"))
+        ):
+            send_response = resp
+
+        if isinstance(send_response, tuple):
+            send_response, _ = send_response
+
+        if not isinstance(send_response, Task):
+            return None
+        return send_response
+
+    @staticmethod
+    def create_send_message_payload(
+        text: str,
+        task_id: str | None = None,
+        context_id: str | None = None,
+    ) -> dict[str, Any]:
+        """Create a message payload for sending to a remote agent.
+
+        Args:
+            text: The text content of the message.
+            task_id: Optional task ID to associate with the message.
+            context_id: Optional context ID to associate with the message.
+
+        Returns:
+            dict: A dictionary containing the formatted message payload ready
+                to be sent to a remote agent.
+        """
+        payload: dict[str, Any] = {
+            "message": {
+                "role": "user",
+                "parts": [{"type": "text", "text": text}],
+                "message_id": uuid.uuid4().hex,
+            },
+        }
+        if task_id:
+            payload["message"]["task_id"] = task_id
+        if context_id:
+            payload["message"]["context_id"] = context_id
+        return payload

aigency-0.0.1rc163779164/aigency/agents/generator.py

@@ -0,0 +1,134 @@
+import logging
+import httpx
+from google.adk.agents import Agent
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+
+from aigency.agents.executor import AgentA2AExecutor
+from aigency.agents.client import AgentClient
+from aigency.schemas.aigency_config import AigencyConfig
+from aigency.tools.generator import ToolGenerator
+from aigency.utils.utils import generate_url, safe_async_run
+from aigency.agents.communicator import Communicator
+
+from google.adk.artifacts import InMemoryArtifactService
+from google.adk.memory.in_memory_memory_service import InMemoryMemoryService
+from google.adk.runners import Runner
+from google.adk.sessions import InMemorySessionService
+
+from a2a.client import A2ACardResolver
+
+
+class AgentA2AGenerator:
+
+    @staticmethod
+    def create_agent(agent_config: AigencyConfig) -> Agent:
+
+        tools = [
+            ToolGenerator.create_tool(tool_cfg) for tool_cfg in agent_config.agent.tools
+        ]
+
+        remote_agents = agent_config.agent.remote_agents
+        if remote_agents:
+            remote_agent_connections = AgentA2AGenerator.build_remote_agent_connections(
+                agent_config
+            )
+            logger.info(f"Remote agent connections: {remote_agent_connections}")
+            communicator = Communicator(
+                remote_agent_connections=remote_agent_connections
+            )
+            tools.append(communicator.send_message)
+
+        return Agent(
+            name=agent_config.metadata.name,
+            model=agent_config.agent.model.name,
+            instruction=agent_config.agent.instruction,
+            tools=tools,
+        )
+
+    @staticmethod
+    def build_agent_card(agent_config: AigencyConfig) -> AgentCard:
+
+        # TODO: Parse properly
+        capabilities = AgentCapabilities(
+            streaming=agent_config.service.capabilities.streaming
+        )
+
+        skills = [
+            AgentSkill(
+                id=skill.id,
+                name=skill.name,
+                description=skill.description,
+                tags=skill.tags,
+                examples=skill.examples,
+            )
+            for skill in agent_config.agent.skills
+        ]
+
+        return AgentCard(
+            name=agent_config.metadata.name,
+            description=agent_config.metadata.description,
+            url=agent_config.service.url,
+            version=agent_config.metadata.version,
+            default_input_modes=agent_config.service.interface.default_input_modes,
+            default_output_modes=agent_config.service.interface.default_output_modes,
+            capabilities=capabilities,
+            skills=skills,
+        )
+
+    @staticmethod
+    def build_executor(agent: Agent, agent_card: AgentCard) -> AgentA2AExecutor:
+
+        runner = Runner(
+            app_name=agent.name,
+            agent=agent,
+            artifact_service=InMemoryArtifactService(),
+            session_service=InMemorySessionService(),
+            memory_service=InMemoryMemoryService(),
+        )
+
+        return AgentA2AExecutor(runner=runner, card=agent_card)
+
+    @staticmethod
+    def build_remote_agent_connections(agent_config: AigencyConfig):
+        """Initialize connections to all remote agents asynchronously.
+
+        Tests each connection individually with detailed logging to help identify
+        any connection issues. It attempts to connect to each remote agent address,
+        retrieve its agent card, and store the connection for later use.
+
+        Raises:
+            No exceptions are raised, but errors are logged.
+        """
+
+        if not agent_config.agent.remote_agents:
+            return {}
+
+        remote_agent_configs = [
+            {"url": generate_url(host=remote_agent.host, port=remote_agent.port)}
+            for remote_agent in agent_config.agent.remote_agents
+        ]
+
+        async def _connect():
+            remote_agent_connections = {}
+            async with httpx.AsyncClient(timeout=60) as client:
+                for config in remote_agent_configs:
+                    address = config.get("url")
+                    logger.debug(f"--- Attempting connection to: {address} ---")
+                    try:
+                        card_resolver = A2ACardResolver(client, address)
+                        card = await card_resolver.get_agent_card()
+                        remote_connection = AgentClient(agent_card=card)
+                        remote_agent_connections[card.name] = remote_connection
+                    except Exception as e:
+                        logger.error(
+                            f"--- CRITICAL FAILURE for address: {address} ---",
+                            exc_info=True,
+                        )
+                        raise e
+            return remote_agent_connections
+
+        try:
+            return safe_async_run(_connect())
+        except Exception as e:
+            logger.error("--- CRITICAL FAILURE", exc_info=True)
+            raise e

{aigency-0.0.1rc136602178 → aigency-0.0.1rc163779164}/aigency/schemas/agent/agent.py

@@ -3,6 +3,7 @@ from typing import List, Optional
 from aigency.schemas.agent.model import AgentModel
 from aigency.schemas.agent.skills import Skill
 from aigency.schemas.agent.tools import Tool
+from aigency.schemas.agent.remote_agent import RemoteAgent
 
 
 class Agent(BaseModel):
@@ -12,3 +13,4 @@ class Agent(BaseModel):
     instruction: str
     skills: List[Skill]
     tools: Optional[List[Tool]] = []
+    remote_agents: Optional[List[RemoteAgent]] = []

aigency-0.0.1rc163779164/aigency/schemas/agent/remote_agent.py

@@ -0,0 +1,9 @@
+from pydantic import BaseModel, Field
+
+
+class RemoteAgent(BaseModel):
+    """Remote agent configuration."""
+
+    name: str
+    host: str
+    port: int = Field(..., ge=1, le=65535)

{aigency-0.0.1rc136602178 → aigency-0.0.1rc163779164}/aigency/utils/utils.py

@@ -11,6 +11,7 @@ from aigency.utils.logger import get_logger
 
 logger = get_logger()
 
+
 def convert_a2a_part_to_genai(part: Part) -> types.Part:
     """Convert a single A2A Part type into a Google Gen AI Part type.
 
@@ -54,6 +55,7 @@ def convert_genai_part_to_a2a(part: types.Part) -> Part:
         )
     raise ValueError(f"Unsupported part type: {part}")
 
+
 def expand_env_vars(env_dict):
     """
     Expande los valores del diccionario usando variables de entorno solo si el valor es una clave de entorno existente.
@@ -67,6 +69,21 @@ def expand_env_vars(env_dict):
             logger.warning(f"Environment variable {v} not found")
     return result
 
+
+def generate_url(host: str, port: int, path: str = "") -> str:
+    """Generate a URL from host, port, and path components.
+
+    Args:
+        host (str): Hostname or IP address.
+        port (int): Port number.
+        path (str, optional): URL path. Defaults to "".
+
+    Returns:
+        str: Complete URL in the format http://host:port/path.
+    """
+    return f"http://{host}:{port}{path}"
+
+
 def safe_async_run(coro):
     """Simple wrapper to safely run async code."""
     try:
@@ -93,4 +110,5 @@ def safe_async_run(coro):
         else:
             return loop.run_until_complete(coro)
     except RuntimeError:
-        return asyncio.run(coro)
+        return asyncio.run(coro)
+

{aigency-0.0.1rc136602178 → aigency-0.0.1rc163779164}/aigency.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aigency
-Version: 0.0.1rc136602178
+Version: 0.0.1rc163779164
 Summary: Add your description here
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown

{aigency-0.0.1rc136602178 → aigency-0.0.1rc163779164}/aigency.egg-info/SOURCES.txt

@@ -6,11 +6,14 @@ aigency.egg-info/SOURCES.txt
 aigency.egg-info/dependency_links.txt
 aigency.egg-info/requires.txt
 aigency.egg-info/top_level.txt
+aigency/agents/client.py
+aigency/agents/communicator.py
 aigency/agents/executor.py
 aigency/agents/generator.py
 aigency/schemas/aigency_config.py
 aigency/schemas/agent/agent.py
 aigency/schemas/agent/model.py
+aigency/schemas/agent/remote_agent.py
 aigency/schemas/agent/skills.py
 aigency/schemas/agent/tools.py
 aigency/schemas/metadata/metadata.py

{aigency-0.0.1rc136602178 → aigency-0.0.1rc163779164}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aigency"
-version = "0.0.1rc136602178"
+version = "0.0.1rc163779164"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"

aigency-0.0.1rc136602178/aigency/agents/generator.py

@@ -1,75 +0,0 @@
-"""Agent generator module for creating A2A agents."""
-
-from typing import Any, Dict, List
-
-from a2a.types import AgentCapabilities, AgentCard, AgentSkill
-from google.adk.agents import Agent
-from google.adk.artifacts import InMemoryArtifactService
-from google.adk.memory.in_memory_memory_service import InMemoryMemoryService
-from google.adk.runners import Runner
-from google.adk.sessions import InMemorySessionService
-
-from aigency.agents.executor import AgentA2AExecutor
-from aigency.schemas.aigency_config import AigencyConfig
-from aigency.tools.generator import ToolGenerator
-
-
-class AgentA2AGenerator:
-    """Generator for creating A2A agents and related components."""
-
-    @staticmethod
-    def create_agent(agent_config: AigencyConfig) -> Agent:
-
-        tools = [
-            ToolGenerator.create_tool(tool_cfg) for tool_cfg in agent_config.agent.tools
-        ]
-
-        return Agent(
-            name=agent_config.metadata.name,
-            model=agent_config.agent.model.name,
-            instruction=agent_config.agent.instruction,
-            tools=tools,
-        )
-
-    @staticmethod
-    def build_agent_card(agent_config: AigencyConfig) -> AgentCard:
-
-        # TODO: Parse properly
-        capabilities = AgentCapabilities(
-            streaming=agent_config.service.capabilities.streaming
-        )
-
-        skills = [
-            AgentSkill(
-                id=skill.id,
-                name=skill.name,
-                description=skill.description,
-                tags=skill.tags,
-                examples=skill.examples,
-            )
-            for skill in agent_config.agent.skills
-        ]
-
-        return AgentCard(
-            name=agent_config.metadata.name,
-            description=agent_config.metadata.description,
-            url=agent_config.service.url,
-            version=agent_config.metadata.version,
-            default_input_modes=agent_config.service.interface.default_input_modes,
-            default_output_modes=agent_config.service.interface.default_output_modes,
-            capabilities=capabilities,
-            skills=skills,
-        )
-
-    @staticmethod
-    def build_executor(agent: Agent, agent_card: AgentCard) -> AgentA2AExecutor:
-
-        runner = Runner(
-            app_name=agent.name,
-            agent=agent,
-            artifact_service=InMemoryArtifactService(),
-            session_service=InMemorySessionService(),
-            memory_service=InMemoryMemoryService(),
-        )
-
-        return AgentA2AExecutor(runner=runner, card=agent_card)