Skip to content

Commit 8cf9bae

Browse files
committed
Simplify context_id creation
1 parent 409e3a6 commit 8cf9bae

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

src/a2a/utils/task.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -28,16 +28,10 @@ def new_task(request: Message) -> Task:
2828
if isinstance(part.root, TextPart) and not part.root.text:
2929
raise ValueError('TextPart content cannot be empty')
3030

31-
context_id_str = request.context_id
32-
if not context_id_str:
33-
context_id = str(uuid.uuid4())
34-
else:
35-
context_id = context_id_str
36-
3731
return Task(
3832
status=TaskStatus(state=TaskState.submitted),
3933
id=(request.task_id if request.task_id else str(uuid.uuid4())),
40-
context_id=context_id,
34+
context_id=(request.context_id if request.context_id else str(uuid.uuid4())),
4135
history=[request],
4236
)
4337

0 commit comments

Comments
 (0)