Add Agent2Agent demo code

Author: ChawlaAvi

agent2agent-demo/agent1.py

@@ -0,0 +1,44 @@
+from python_a2a import A2AServer, skill, agent, run_server, TaskStatus, TaskState
+import math
+import re
+
+@agent(
+    name="Sine Agent",
+    description="Provides the sine of a number",
+    version="1.0.0"
+)
+class SineAgent(A2AServer):
+    
+    @skill(
+        name="Get Sine",
+        description="Get the sine of a number",
+        tags=["sine", "sin"]
+    )
+    def get_sine(self, number):
+        """Get the sine of a number."""
+        # Mock implementation
+        return f"The sine of {number} is {math.sin(number)}"
+    
+    def handle_task(self, task):
+        # Extract location from message
+
+        input_message = task.message["content"]["text"]
+
+        # regex to extract the number from the text
+        match = re.search(r"([-+]?[0-9]*\.?[0-9]+)", input_message)
+
+        number = float(match.group(1))
+        print("number", number)
+        # Get weather and create response
+        sine_output = self.get_sine(number)
+        task.artifacts = [{
+                "parts": [{"type": "text", "text": sine_output}]
+            }]
+        task.status = TaskStatus(state=TaskState.COMPLETED)
+
+        return task
+
+# Run the server
+if __name__ == "__main__":
+    agent = SineAgent()
+    run_server(agent, port=4737)

agent2agent-demo/agent2.py

@@ -0,0 +1,44 @@
+from python_a2a import A2AServer, skill, agent, run_server, TaskStatus, TaskState
+import math
+import re
+
+@agent(
+    name="Cosine Agent",
+    description="Provides the cosine of a number",
+    version="1.0.0"
+)
+class CosineAgent(A2AServer):
+    
+    @skill(
+        name="Get Cos",
+        description="Get the cosine of a number",
+        tags=["cos", "cosine"]
+    )
+    def get_cosine(self, number):
+        """Get the cosine of a number."""
+        # Mock implementation
+        return f"The cosine of {number} is {math.cos(number)}"
+    
+    def handle_task(self, task):
+        # Extract location from message
+
+        input_message = task.message["content"]["text"]
+
+        # regex to extract the number from the text
+        match = re.search(r"([-+]?[0-9]*\.?[0-9]+)", input_message)
+
+        number = float(match.group(1))
+        print("number", number)
+        # Get weather and create response
+        cosine_output = self.get_cosine(number)
+        task.artifacts = [{
+                "parts": [{"type": "text", "text": cosine_output}]
+            }]
+        task.status = TaskStatus(state=TaskState.COMPLETED)
+
+        return task
+
+# Run the server
+if __name__ == "__main__":
+    agent = CosineAgent()
+    run_server(agent, port=4738)

agent2agent-demo/agent3.py

@@ -0,0 +1,44 @@
+from python_a2a import A2AServer, skill, agent, run_server, TaskStatus, TaskState
+import math
+import re
+
+@agent(
+    name="Tangent Agent",
+    description="Provides the tangent of a number",
+    version="1.0.0"
+)
+class TangentAgent(A2AServer):
+    
+    @skill(
+        name="Get Tangent",
+        description="Get the tangent of a number",
+        tags=["tangent", "tan"]
+    )
+    def get_tangent(self, number):
+        """Get the tangent of a number."""
+        # Mock implementation
+        return f"The tangent of {number} is {math.tan(number)}"
+    
+    def handle_task(self, task):
+        # Extract location from message
+
+        input_message = task.message["content"]["text"]
+
+        # regex to extract the number from the text
+        match = re.search(r"([-+]?[0-9]*\.?[0-9]+)", input_message)
+
+        number = float(match.group(1))
+        print("number", number)
+        # Get weather and create response
+        tangent_output = self.get_tangent(number)
+        task.artifacts = [{
+                "parts": [{"type": "text", "text": tangent_output}]
+            }]
+        task.status = TaskStatus(state=TaskState.COMPLETED)
+
+        return task
+
+# Run the server
+if __name__ == "__main__":
+    agent = TangentAgent()
+    run_server(agent, port=4739)

agent2agent-demo/notebook.ipynb

@@ -0,0 +1,112 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!curl -LsSf https://astral.sh/uv/install.sh | sh\n",
+    "!uv pip install python-a2a"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# open a terminal and run: python agent1.py\n",
+    "# open another terminal and run: python agent2.py\n",
+    "# open another terminal and run: python agent3.py\n",
+    "\n",
+    "# after that, run the following cells"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from python_a2a import AgentNetwork, A2AClient, AIAgentRouter\n",
+    "\n",
+    "# Create an agent network\n",
+    "network = AgentNetwork(name=\"Math Assistant Network\")\n",
+    "\n",
+    "# Add agents to the network\n",
+    "network.add(\"Sine\", \"http://localhost:4737\")\n",
+    "network.add(\"Cosine\", \"http://localhost:4738\")\n",
+    "network.add(\"Tangent\", \"http://localhost:4739\")\n",
+    "\n",
+    "# Create a router to intelligently direct queries to the best agent\n",
+    "router = AIAgentRouter(\n",
+    "    llm_client=A2AClient(\"http://localhost:5000/openai\"),  # LLM for making routing decisions\n",
+    "    agent_network=network\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Route a query to the appropriate agent\n",
+    "query = \"Tan of 0.78545\"\n",
+    "agent_name, confidence = router.route_query(query)\n",
+    "print(f\"Routing to {agent_name} with {confidence:.2f} confidence\")\n",
+    "\n",
+    "# Get the selected agent and ask the question\n",
+    "agent = network.get_agent(agent_name)\n",
+    "response = agent.ask(query)\n",
+    "print(f\"Response: {response}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Route a query to the appropriate agent\n",
+    "query = \"Sine of 3.14159\"\n",
+    "agent_name, confidence = router.route_query(query)\n",
+    "print(f\"Routing to {agent_name} with {confidence:.2f} confidence\")\n",
+    "\n",
+    "# Get the selected agent and ask the question\n",
+    "agent = network.get_agent(agent_name)\n",
+    "response = agent.ask(query)\n",
+    "print(f\"Response: {response}\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "base",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.12.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}