Skip to content

Commit 98d4414

Browse files
feat: Add custom id generators option to simple_request_context_builder
1 parent f666bec commit 98d4414

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/a2a/server/agent_execution/simple_request_context_builder.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from a2a.server.agent_execution import RequestContext, RequestContextBuilder
44
from a2a.server.context import ServerCallContext
5-
from a2a.server.id_generator import IdGenerator
5+
from a2a.server.id_generator import IDGenerator
66
from a2a.server.tasks import TaskStore
77
from a2a.types import MessageSendParams, Task
88

@@ -14,8 +14,8 @@ def __init__(
1414
self,
1515
should_populate_referred_tasks: bool = False,
1616
task_store: TaskStore | None = None,
17-
task_id_generator: IdGenerator | None = None,
18-
context_id_generator: IdGenerator | None = None,
17+
task_id_generator: IDGenerator | None = None,
18+
context_id_generator: IDGenerator | None = None,
1919
) -> None:
2020
"""Initializes the SimpleRequestContextBuilder.
2121

0 commit comments

Comments
 (0)