diff --git a/libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py b/libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py index 8aec3ef9e..86812c258 100644 --- a/libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py +++ b/libs/scheduler-kafka/langgraph/scheduler/kafka/executor.py @@ -1,11 +1,12 @@ import asyncio from contextlib import AbstractAsyncContextManager, AsyncExitStack from functools import partial -from typing import Any, Optional, Self, Sequence +from typing import Any, Optional, Sequence import aiokafka import orjson from langchain_core.runnables import RunnableConfig +from typing_extensions import Self import langgraph.scheduler.kafka.serde as serde from langgraph.constants import CONFIG_KEY_DELEGATE, ERROR, NS_END, NS_SEP diff --git a/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py b/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py index a3da188e8..c13de9c08 100644 --- a/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py +++ b/libs/scheduler-kafka/langgraph/scheduler/kafka/orchestrator.py @@ -1,9 +1,10 @@ import asyncio from contextlib import AbstractAsyncContextManager, AsyncExitStack -from typing import Any, Optional, Self +from typing import Any, Optional import aiokafka from langchain_core.runnables import ensure_config +from typing_extensions import Self import langgraph.scheduler.kafka.serde as serde from langgraph.constants import (