File tree 1 file changed +7
-3
lines changed
1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change 4
4
5
5
import asyncio
6
6
import contextvars
7
- import dataclasses
8
7
import inspect
9
8
import logging
10
9
import threading
43
42
)
44
43
45
44
import nexusrpc
45
+ import nexusrpc .handler
46
46
from typing_extensions import (
47
47
Concatenate ,
48
48
Literal ,
@@ -5158,7 +5158,7 @@ def __init__(
5158
5158
# TODO(dan): overloads: no-input, operation name, ret type
5159
5159
async def start_operation (
5160
5160
self ,
5161
- operation : Union [nexusrpc .Operation [I , O ], str ],
5161
+ operation : Union [nexusrpc .interface . Operation [I , O ], str ],
5162
5162
input : I ,
5163
5163
* ,
5164
5164
schedule_to_close_timeout : Optional [timedelta ] = None ,
@@ -5178,7 +5178,11 @@ async def start_operation(
5178
5178
# TODO(dan): overloads: no-input, operation name, ret type
5179
5179
async def execute_operation (
5180
5180
self ,
5181
- operation : nexusrpc .Operation [I , O ],
5181
+ operation : Union [
5182
+ nexusrpc .interface .Operation [I , O ],
5183
+ Callable [..., nexusrpc .handler .Operation [I , O ]],
5184
+ str ,
5185
+ ],
5182
5186
input : I ,
5183
5187
* ,
5184
5188
schedule_to_close_timeout : Optional [timedelta ] = None ,
You can’t perform that action at this time.
0 commit comments