ai-parrot 0.17.2__cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

parrot/agents/__init__.py

@@ -0,0 +1,29 @@
+"""
+Parrot Agents - Core and Plugin Agents
+"""
+from importlib import import_module
+from parrot.plugins import setup_plugin_importer, dynamic_import_helper
+# Setup the plugin importer for agents
+setup_plugin_importer('parrot.agents', 'agents')
+
+
+# Enable dynamic imports
+def __getattr__(name):
+    """
+    Getattr with dynamic import.
+    First, search in PLUGINS_DIR, if doesn't exists,
+    then look on parrot.bots package.
+    """
+    try:
+        return dynamic_import_helper(__name__, name)
+    except AttributeError:
+        # try to import from parrot.bots
+        try:
+            module = import_module(f"parrot.bots.{name.lower()}")
+            if hasattr(module, name):
+                return getattr(module, name)
+        except (ImportError, ModuleNotFoundError):
+            pass
+    raise AttributeError(
+        f"module '{__name__}' has no attribute '{name}'"
+    )

parrot/bots/__init__.py

@@ -0,0 +1,12 @@
+
+from .abstract import AbstractBot
+from .basic import BasicBot
+from .chatbot import Chatbot
+from .agent import Agent, BasicAgent
+__all__ = (
+    "AbstractBot",
+    "BasicBot",
+    "Agent",
+    "BasicAgent",
+    "Chatbot",
+)

parrot/bots/a2a_agent.py

@@ -0,0 +1,19 @@
+from aiohttp import web
+from .agent import BasicAgent
+from ..a2a.server import A2AEnabledMixin
+
+
+class A2AAgent(BasicAgent, A2AEnabledMixin):
+    """An AI-Parrot Agent with A2A capabilities."""
+
+    def __init__(self, app: web.Application, url: str, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.app = app
+        self.url = url
+        # add the "configure" into the startup signal
+        self.app.middlewares.append(self.configure)
+
+    async def configure(self):
+        """Configure the agent and initialize A2A server."""
+        await super().configure()
+        self.setup_a2a(self.app, self.url)