Implement debater

pyproject.toml

@@ -1,10 +1,12 @@
 [project]
-name = "agent-template"
+name = "simple-agent"
 version = "0.1.0"
-description = "A template for A2A agents"
+description = "Simple A2A agent"
 readme = "README.md"
 requires-python = ">=3.13"
 dependencies = [
     "a2a-sdk[http-server]>=0.3.20",
+    "litellm>=1.80.9",
+    "python-dotenv>=1.2.1",
     "uvicorn>=0.38.0",
 ]

src/agent.py

@@ -1,19 +1,29 @@
+from dotenv import load_dotenv
+import litellm
 from a2a.server.tasks import TaskUpdater
+from a2a.types import Part, TextPart
+
 from messenger import Messenger
 
+load_dotenv()
+
 
 class Agent:
     def __init__(self):
-        self.messenger = Messenger()
-        # initialize other state here
+        self.messages = []
 
     async def run(self, input_text: str, updater: TaskUpdater) -> None:
-        """Implement your agent logic here.
-
-        Args:
-            input_text: The incoming message text
-            updater: Report progress (update_status) and results (add_artifact)
+        print(f"> {input_text}")
+        self.messages.append({"content": input_text, "role": "user"})
+        completion = litellm.completion(
+            model="gpt-4o",
+            messages=self.messages
+        )
+        response = completion.choices[0].message.content
+        self.messages.append({"content": response, "role": "assistant"})
+        print(response)
 
-        Use self.messenger.talk_to_agent(message, url) to call other agents.
-        """
-        raise NotImplementedError("Agent not implemented.")
+        await updater.add_artifact(
+            parts=[Part(root=TextPart(text=response))],
+            name="Result",
+        )

src/server.py

@@ -20,26 +20,15 @@ def main():
     parser.add_argument("--card-url", type=str, help="URL to advertise in the agent card")
     args = parser.parse_args()
 
-    # Fill in your agent card
-    # See: https://a2a-protocol.org/latest/tutorials/python/3-agent-skills-and-card/
-
-    skill = AgentSkill(
-        id="",
-        name="",
-        description="",
-        tags=[""],
-        examples=[""]
-    )
-
     agent_card = AgentCard(
-        name="",
-        description="",
+        name="SimpleAgent",
+        description='Responds to messages.',
         url=args.card_url or f"http://{args.host}:{args.port}/",
         version='1.0.0',
         default_input_modes=['text'],
         default_output_modes=['text'],
         capabilities=AgentCapabilities(streaming=True),
-        skills=[skill]
+        skills=[]
     )
 
     request_handler = DefaultRequestHandler(