diff --git a/lib/crewai/src/crewai/a2a/updates/polling/handler.py b/lib/crewai/src/crewai/a2a/updates/polling/handler.py index 453affe21..84d41afcf 100644 --- a/lib/crewai/src/crewai/a2a/updates/polling/handler.py +++ b/lib/crewai/src/crewai/a2a/updates/polling/handler.py @@ -4,7 +4,7 @@ from __future__ import annotations import asyncio import time -from typing import TYPE_CHECKING, Any, Unpack +from typing import TYPE_CHECKING, Any from a2a.client import Client from a2a.types import ( @@ -13,6 +13,7 @@ from a2a.types import ( TaskQueryParams, TaskState, ) +from typing_extensions import Unpack from crewai.a2a.errors import A2APollingTimeoutError from crewai.a2a.task_helpers import ( diff --git a/lib/crewai/src/crewai/a2a/updates/streaming/handler.py b/lib/crewai/src/crewai/a2a/updates/streaming/handler.py index 74930e6a0..eb05724e7 100644 --- a/lib/crewai/src/crewai/a2a/updates/streaming/handler.py +++ b/lib/crewai/src/crewai/a2a/updates/streaming/handler.py @@ -2,7 +2,6 @@ from __future__ import annotations -from typing import Unpack import uuid from a2a.client import Client @@ -17,6 +16,7 @@ from a2a.types import ( TaskStatusUpdateEvent, TextPart, ) +from typing_extensions import Unpack from crewai.a2a.task_helpers import ( ACTIONABLE_STATES,