MYPY fixes on Langraph example

Author: rajeshvelicheti

examples/langgraph/__main__.py

@@ -4,8 +4,8 @@
 import click
 import httpx
 
-from agent import CurrencyAgent
-from agent_executor import CurrencyAgentExecutor
+from agent import CurrencyAgent # type: ignore[import-untyped]
+from agent_executor import CurrencyAgentExecutor # type: ignore[import-untyped]
 from dotenv import load_dotenv
 
 from a2a.server.apps import A2AStarletteApplication

examples/langgraph/agent_executor.py

@@ -1,6 +1,5 @@
-from agent import CurrencyAgent
+from agent import CurrencyAgent # type: ignore[import-untyped]
 from typing_extensions import override
-
 from a2a.server.agent_execution import AgentExecutor, RequestContext
 from a2a.server.events.event_queue import EventQueue
 from a2a.types import (

examples/langgraph/test_client.py

@@ -72,7 +72,8 @@ async def run_single_turn_test(client: A2AClient) -> None:
     print('---Query Task---')
     # query the task
     task_id_payload = {'id': task_id}
-    get_request = GetTaskRequest(params=TaskQueryParams(**task_id_payload))
+    get_request = GetTaskRequest(params=TaskQueryParams(id=task_id))
+    #get_request = GetTaskRequest(params=TaskQueryParams(**task_id_payload))
     get_response: GetTaskResponse = await client.get_task(get_request)
     print_json_response(get_response, 'Query Task Response')