We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8e655d1 commit 922ff10Copy full SHA for 922ff10
1 file changed
src/a2a/server/apps/jsonrpc/jsonrpc_app.py
@@ -5,9 +5,9 @@
5
6
from abc import ABC, abstractmethod
7
from collections.abc import AsyncGenerator, Callable
8
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, Type
9
10
-from pydantic import ValidationError
+from pydantic import BaseModel, ValidationError
11
12
from a2a.auth.user import UnauthenticatedUser
13
from a2a.auth.user import User as A2AUser
@@ -156,7 +156,7 @@ class JSONRPCApplication(ABC):
156
"""
157
158
# Method-to-model mapping for centralized routing
159
- METHOD_TO_MODEL = {
+ METHOD_TO_MODEL: dict[str, Type[BaseModel]] = {
160
'message/send': SendMessageRequest,
161
'message/stream': SendStreamingMessageRequest,
162
'tasks/get': GetTaskRequest,
0 commit comments