update: samples

Author: kimtth

.gitignore

@@ -8,3 +8,4 @@ template.html
 git_cmp_needs_update.csv
 git_cmp_index_incremental.csv
 /__pycache__
+/.vscode

agent/a2a_server_client/samples/python/.gitignore

@@ -9,4 +9,6 @@ wheels/
 # Virtual environments
 .venv
 
-.DS_Store
+.DS_Store
+
+.env

agent/a2a_server_client/samples/python/agents/google_adk/agent.py

@@ -126,9 +126,9 @@ def invoke(self, query, session_id) -> str:
           state={},
           session_id=session_id,
       )
-    events = self._runner.run(
+    events = list(self._runner.run(
         user_id=self._user_id, session_id=session.id, new_message=content
-    )
+    ))
     if not events or not events[-1].content or not events[-1].content.parts:
       return ""
     return "\n".join([p.text for p in events[-1].content.parts if p.text])
@@ -183,7 +183,7 @@ def _build_agent(self) -> LlmAgent:
             " given the amount and purpose of the reimbursement."
         ),
         instruction="""
-    You are an agent who handle the reimbursement process for employees.
+    You are an agent who handles the reimbursement process for employees.
 
     When you receive an reimbursement request, you should first create a new request form using create_request_form(). Only provide default values if they are provided by the user, otherwise use an empty string as the default value.
       1. 'Date': the date of the transaction.

agent/a2a_server_client/samples/python/agents/marvin/__init__.py

@@ -1,10 +1,14 @@
-from common.client import A2AClient, A2ACardResolver
-from common.types import TaskState, Task
-from common.utils.push_notification_auth import PushNotificationReceiverAuth
 import asyncclick as click
 import asyncio
-from uuid import uuid4
+import base64
+import os
 import urllib
+from uuid import uuid4
+
+from common.client import A2AClient, A2ACardResolver
+from common.types import TaskState, Task, TextPart, FilePart, FileContent
+from common.utils.push_notification_auth import PushNotificationReceiverAuth
+
 
 @click.command()
 @click.option("--agent", default="http://localhost:10000")
@@ -60,20 +64,42 @@ async def completeTask(client: A2AClient, streaming, use_push_notifications: boo
     )
     if prompt == ":q" or prompt == "quit":
         return False
-
+    
+    message = {
+        "role": "user",
+        "parts": [
+            {
+                "type": "text",
+                "text": prompt,
+            }
+        ]
+    }
+    
+    file_path = click.prompt(
+        "Select a file path to attach? (press enter to skip)",
+        default="",
+        show_default=False,
+    )
+    if file_path and file_path.strip() != "":
+        with open(file_path, "rb") as f:
+            file_content = base64.b64encode(f.read()).decode('utf-8')
+            file_name = os.path.basename(file_path)
+        
+        message["parts"].append(
+            {
+                "type": "file",
+                "file": {
+                    "name": file_name,
+                    "bytes": file_content,
+                }
+            }
+        )
+ 
     payload = {
         "id": taskId,
         "sessionId": sessionId,
         "acceptedOutputModes": ["text"],
-        "message": {
-            "role": "user",
-            "parts": [
-                {
-                    "type": "text",
-                    "text": prompt,
-                }
-            ],
-        },
+        "message": message,
     }
 
     if use_push_notifications:

agent/a2a_server_client/samples/python/hosts/cli/__main__.py

@@ -23,7 +23,7 @@ def __init__(self, host, port, notification_receiver_auth: PushNotificationRecei
 
     def start(self):
         try:
-            # Need to start server is separate thread as current thread
+            # Need to start server in separate thread as current thread
             # will be blocked when it is waiting on user prompt.
             asyncio.run_coroutine_threadsafe(
                 self.start_server(),

agent/a2a_server_client/samples/python/hosts/cli/push_notification_listener.py

@@ -85,7 +85,7 @@ def create_agent(self) -> Agent:
 
   def root_instruction(self, context: ReadonlyContext) -> str:
     current_agent = self.check_state(context)
-    return f"""You are a expert delegator that can delegate the user request to the
+    return f"""You are an expert delegator that can delegate the user request to the
 appropriate remote agents.
 
 Discovery:
@@ -94,12 +94,12 @@ def root_instruction(self, context: ReadonlyContext) -> str:
 
 Execution:
 - For actionable tasks, you can use `create_task` to assign tasks to remote agents to perform.
-Be sure to include the remote agent name when you response to the user.
+Be sure to include the remote agent name when you respond to the user.
 
 You can use `check_pending_task_states` to check the states of the pending
 tasks.
 
-Please rely on tools to address the request, don't make up the response. If you are not sure, please ask the user for more details.
+Please rely on tools to address the request, and don't make up the response. If you are not sure, please ask the user for more details.
 Focus on the most recent parts of the conversation primarily.
 
 If there is an active agent, send the request to that agent with the update task tool.

agent/a2a_server_client/samples/python/hosts/multiagent/host_agent.py

@@ -12,7 +12,7 @@
 from common.client import A2AClient
 
 TaskCallbackArg = Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent
-TaskUpdateCallback = Callable[[TaskCallbackArg], Task]
+TaskUpdateCallback = Callable[[TaskCallbackArg, AgentCard], Task]
 
 class RemoteAgentConnections:
   """A class to hold the connections to the remote agents."""
@@ -44,7 +44,7 @@ async def send_task(
                 message=request.message,
             ),
             history=[request.message],
-        ))
+        ), self.card)
       async for response in self.agent_client.send_task_streaming(request.model_dump()):
         merge_metadata(response.result, request)
         # For task status updates, we need to propagate metadata and provide
@@ -60,7 +60,7 @@ async def send_task(
             m.metadata['last_message_id'] = m.metadata['message_id']
           m.metadata['message_id'] = str(uuid.uuid4())
         if task_callback:
-          task = task_callback(response.result)
+          task = task_callback(response.result, self.card)
         if hasattr(response.result, 'final') and response.result.final:
           break
       return task
@@ -81,7 +81,7 @@ async def send_task(
         m.metadata['message_id'] = str(uuid.uuid4())
 
       if task_callback:
-        task_callback(response.result)
+        task_callback(response.result, self.card)
       return response.result
 
 def merge_metadata(target, source):

agent/a2a_server_client/samples/python/hosts/multiagent/remote_agent_connection.py

@@ -5,37 +5,35 @@ description = "Agent2Agent samples"
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
-    "asyncclick>=8.1.8",
-    "asyncio>=3.4.3",
-    "click>=8.1.8",
-    "google-genai>=1.9.0",
     "httpx>=0.28.1",
     "httpx-sse>=0.4.0",
-    "langchain-google-genai>=2.0.10",
-    "langgraph>=0.3.18",
+    "jwcrypto>=1.5.6",
     "pydantic>=2.10.6",
+    "pyjwt>=2.10.1",
     "sse-starlette>=2.2.1",
     "starlette>=0.46.1",
-    "streamlit>=1.44.0",
+    "typing-extensions>=4.12.2",
     "uvicorn>=0.34.0",
-    "google-adk>=0.0.3",
-    "jwcrypto>=1.5.6",
-    "pyjwt>=2.10.1",
 ]
 
 [tool.hatch.build.targets.wheel]
 packages = ["common", "hosts"]
 
 [tool.uv.workspace]
-members = ["agents/crewai"]
+members = [
+    "agents/crewai",
+    "agents/google_adk",
+    "agents/langgraph",
+    "agents/marvin",
+    "hosts/cli",
+    "hosts/multiagent",
+    "agents/llama_index_file_chat",
+    "agents/semantickernel",
+]
 
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
 [dependency-groups]
-dev = [
-    "pytest>=8.3.5",
-    "pytest-mock>=3.14.0",
-    "ruff>=0.11.2",
-]
+dev = ["pytest>=8.3.5", "pytest-mock>=3.14.0", "ruff>=0.11.2"]