g ddlmZddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z ddlZddZddZddZej$dkrdd Zndd Zdd Zdd Zdd ZGddZeZe dZddZy)) annotationsN)datetimetimezone) AnnotatedAnyTypeVarUnionget_args get_originct|tjr |j}t |}t |r t |S|fS)zhExtract the arguments of a Union type if `response_type` is a union, otherwise return the original type.) isinstancetyping_extensions TypeAliasType __value__r origin_is_unionr )tporigins J/var/www/openai/venv/lib/python3.12/site-packages/pydantic_graph/_utils.pyget_union_argsr s@"'556 \\ ^Fv|u cvt|}|tus|tjurt|^}}||fS|gfS)zStrip `Annotated` from the type if present. Returns: `(tp argument, ())` if not annotated, otherwise `(stripped type, annotations)`. )r rrr )rrinner_tpargss runpack_annotatedrsC ^F f(9(C(CC"2,4~2v rcX|tjuryttddx}r||uSy)zCheck if a type is `Never`.TNeverNF)rrgetattr)r typing_nevers ris_neverr&s7  $ $$ !2GTB B B\!!r) c|tuSN)r rs rrr3s U{rc:|tuxs|tjuSr#)r types UnionTyper$s rrr8sU{3bEOO33rcbt|dk(r|dSdj|dddz|dzS)z.Join with a comma and 'and' for the last item.rz, Nz, and )lenjoin)itemss r comma_andr.<s; 5zQQxyys$x/%);;rc|K|jx}r<|jjjdr t |S|j Syy)zAttempt to get the namespace where the graph was defined. If the graph is defined with generics `Graph[a, b]` then another frame is inserted, and we have to skip that to get the correct namespace. Nz /typing.py)f_backf_code co_filenameendswithget_parent_namespacef_locals)framebacks rr4r4EsP  << 4 {{&&// =+D11}}$ rcJtjtjS)N)tz)rnowrutcrrnow_utcr=Ss <<8<< ((rceZdZdZy)UnsetzUA singleton to represent an unset value. Copied from pydantic_ai/_utils.py. N)__name__ __module__ __qualname____doc__r<rrr?r?Ws   rr?Tc|tuSr#)UNSET) t_or_unsets ris_setrHds U ""r)rrreturnztuple[Any, ...])rrrIztuple[Any, list[Any]])rrrIbool)rztype[Any] | NonerIrJ)r-z list[str]rIstr)r6ztypes.FrameType | NonerIzdict[str, Any] | None)rIr)rGz T | UnsetrIztyping_extensions.TypeGuard[T]) __future__r _annotationssysr&rrtypingrrrr r r rrrr version_inforr.r4r=r?rFrDrHr<rrrQss2 'GG  g 4< %)    CL#r