Implement simple agent

pyproject.toml

@@ -1,11 +1,13 @@
 [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,32 +1,35 @@
+from dotenv import load_dotenv
+import litellm
+
 from a2a.server.tasks import TaskUpdater
 from a2a.types import Message, TaskState, Part, TextPart
 from a2a.utils import get_message_text, new_agent_text_message
 
-from messenger import Messenger
+load_dotenv()
 
 
 class Agent:
     def __init__(self):
-        self.messenger = Messenger()
-        # Initialize other state here
+        self.messages = []
 
     async def run(self, message: Message, updater: TaskUpdater) -> None:
-        """Implement your agent logic here.
-
-        Args:
-            message: The incoming message
-            updater: Report progress (update_status) and results (add_artifact)
-
-        Use self.messenger.talk_to_agent(message, url) to call other agents.
-        """
         input_text = get_message_text(message)
-
-        # Replace this example code with your agent logic
+        print(f"> {input_text}")
 
         await updater.update_status(
             TaskState.working, new_agent_text_message("Thinking...")
         )
+
+        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)
+
         await updater.add_artifact(
-            parts=[Part(root=TextPart(text=input_text))],
-            name="Echo",
+            parts=[Part(root=TextPart(text=response))],
+            name="Response",
         )

src/server.py

@@ -20,20 +20,17 @@ 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=[]
+        id="chat",
+        name="Chat",
+        description="Answer questions and have conversations",
+        tags=["conversation"],
+        examples=["Hello!", "Tell me a joke"]
     )
 
     agent_card = AgentCard(
-        name="",
-        description="",
+        name="Simple Agent",
+        description="A conversational LLM agent",
         url=args.card_url or f"http://{args.host}:{args.port}/",
         version='1.0.0',
         default_input_modes=['text'],