o i@sddlmZddlZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl Z ddl m Z mZddlmZddlmZe dZd#d dZd#ddZd$ddZd%ddZd&ddZd'd!d"ZdS)() annotationsN)Any AwaitableCallableOptionalTypeTypeVar)InputTOutputT)WorkflowRunOperationContext)WorkflowHandleServiceHandlerTstartdCallable[[ServiceHandlerT, WorkflowRunOperationContext, InputT], Awaitable[WorkflowHandle[OutputT]]]return6tuple[Optional[Type[InputT]], Optional[Type[OutputT]]]c Cst|\}}t|}|sd}nt|ts#td|jd|d}|rVt|}t |dkrS|r7d|nd}td|d|jd t ||d d}||fS|\}||fS) zReturn operation input and output types. `start` must be a type-annotated start method that returns a :py:class:`temporalio.nexus.WorkflowHandle`. NzExpected return type of z, to be a subclass of WorkflowHandle, but is r : zExpected return type z of z- to have exactly one type parameter, but has .) 3_get_start_method_input_and_output_type_annotationstyping get_origin issubclassr warningswarn__name__get_argslen)r input_type output_type origin_typeargssuffixr$\/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/temporalio/nexus/_util.py?get_workflow_run_start_method_input_and_output_type_annotationss4      r&cCszt|}Wntytd|dYdSw|dd}t|dkrF|r.d|nd}td|dt||d d}||fS|\}}t|t s_td |d |d d}||fS) Nz Expected decorated start method z to have type annotationsNNrrrzI to have exactly 2 type-annotated parameters (ctx and input), but it has rzExpected first parameter of z: to be an instance of WorkflowRunOperationContext, but is ) rget_type_hints TypeErrorrrpoprvaluesrr )rtype_annotationsr r#rctx_typer$r$r%rFs:      rfnCallable[..., Any]strcCsBt|dd}|st|rt|dr|jj}|std|d|S)z%Return the name of a callable object.rN__call__z,Could not determine callable name: expected z' to be a function or callable instance.)getattrcallablehasattr __class__rr*)r/ method_namer$r$r%get_callable_namels r8objrMtuple[Optional[Callable[[Any], Any]], Optional[nexusrpc.Operation[Any, Any]]]cCsFt|}|r |}n t|dd}rt|}t|tjsdS||fS)zReturn the :py:class:`Operation` for the object along with the factory function. ``obj`` should be a decorated operation start method. __nexus_operation_factory__Nr')nexusrpcget_operation_definitionr3 isinstance Operation)r9op_defnfactoryr$r$r%get_operation_factoryzs   rBoperation_factoryCallable[[Any], Any]NonecCst|d|dS)ztSet the :py:class:`OperationHandler` factory for this object. ``obj`` should be an operation start method. r;N)setattr)r9rCr$r$r%set_operation_factorysrGboolcCsBt|tjr|j}t|tjst|p t|o tt|ddS)ziReturn True if `obj` is an async callable. Supports partials of async callable class instances. r2N)r> functoolspartialfuncinspectiscoroutinefunctionr4r3)r9r$r$r%is_async_callables  rN)rrrr)r/r0rr1)r9rrr:)r9rrCrDrrE)r9rrrH) __future__rrIrLrrrrrrrrr<r r #temporalio.nexus._operation_contextr _tokenr rr&rr8rBrGrNr$r$r$r%s"       ' &