File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
src/a2a/server/apps/jsonrpc Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 55
66from abc import ABC , abstractmethod
77from collections .abc import AsyncGenerator , Callable
8- from typing import TYPE_CHECKING , Any
8+ from typing import TYPE_CHECKING , Any , Type
99
10- from pydantic import ValidationError
10+ from pydantic import BaseModel , ValidationError
1111
1212from a2a .auth .user import UnauthenticatedUser
1313from a2a .auth .user import User as A2AUser
@@ -156,7 +156,7 @@ class JSONRPCApplication(ABC):
156156 """
157157
158158 # Method-to-model mapping for centralized routing
159- METHOD_TO_MODEL = {
159+ METHOD_TO_MODEL : dict [ str , Type [ BaseModel ]] = {
160160 'message/send' : SendMessageRequest ,
161161 'message/stream' : SendStreamingMessageRequest ,
162162 'tasks/get' : GetTaskRequest ,
You can’t perform that action at this time.
0 commit comments