44import click
55import uvicorn
66
7- from adk_agent import create_agent
8- from adk_agent_executor import ADKAgentExecutor
7+ from adk_agent import create_agent # type: ignore[import-not-found]
8+ from adk_agent_executor import ADKAgentExecutor # type: ignore[import-untyped]
99from dotenv import load_dotenv
10- from google .adk .artifacts import InMemoryArtifactService
11- from google .adk .memory .in_memory_memory_service import InMemoryMemoryService
12- from google .adk .runners import Runner
13- from google .adk .sessions import InMemorySessionService
10+ from google .adk .artifacts import (
11+ InMemoryArtifactService , # type: ignore[import-untyped]
12+ )
13+ from google .adk .memory .in_memory_memory_service import (
14+ InMemoryMemoryService , # type: ignore[import-untyped]
15+ )
16+ from google .adk .runners import Runner # type: ignore[import-untyped]
17+ from google .adk .sessions import (
18+ InMemorySessionService , # type: ignore[import-untyped]
19+ )
1420from starlette .applications import Starlette
1521from starlette .requests import Request
1622from starlette .responses import PlainTextResponse
1925from a2a .server .apps import A2AStarletteApplication
2026from a2a .server .request_handlers import DefaultRequestHandler
2127from a2a .server .tasks import InMemoryTaskStore
22- from a2a .types import (
23- AgentAuthentication ,
24- AgentCapabilities ,
25- AgentCard ,
26- AgentSkill ,
27- )
28+ from a2a .types import AgentCapabilities , AgentCard , AgentSkill
2829
2930
3031load_dotenv ()
@@ -63,7 +64,6 @@ def main(host: str, port: int):
6364 defaultOutputModes = ['text' ],
6465 capabilities = AgentCapabilities (streaming = True ),
6566 skills = [skill ],
66- authentication = AgentAuthentication (schemes = ['public' ]),
6767 )
6868
6969 adk_agent = create_agent (
@@ -81,7 +81,7 @@ def main(host: str, port: int):
8181
8282 async def handle_auth (request : Request ) -> PlainTextResponse :
8383 await agent_executor .on_auth_callback (
84- request .query_params .get ('state' ), str (request .url )
84+ str ( request .query_params .get ('state' ) ), str (request .url )
8585 )
8686 return PlainTextResponse ('Authentication successful.' )
8787
0 commit comments