PR feedback

src/dispatch/__init__.py

@@ -4,7 +4,7 @@
 
 import dispatch.integrations
 from dispatch.client import DEFAULT_API_URL, Client
-from dispatch.directive import exit, poll
+from dispatch.coroutine import exit, poll
 from dispatch.id import DispatchID
 from dispatch.proto import Call, Error, Input, Output
 from dispatch.status import Status

src/dispatch/function.py

@@ -8,7 +8,7 @@
 from typing import Any, Callable, Dict, TypeAlias
 
 from dispatch.client import Client
-from dispatch.directive import Directive
+from dispatch.coroutine import Directive
 from dispatch.experimental.durable import durable
 from dispatch.experimental.multicolor import (
     CustomYield,