o i@s\ddlmZddlZddlZddlZddlmZmZm Z m Z ddl m Z ddl mZddlmZddlmZmZmZddlZddlZddlmZmZejd d ZeeejGd d d Z e Z! ed Z"d+ddZ#d,ddZ$edZ%edZ&ddZ'd-d d!Z(zdd"l)m*Z*Wne+yGd#d$d$e,Z*Ynwerej-Z.dSe d.d)d*Z.dS)/) annotationsN) AwaitableCallable GeneratorSequence)contextmanager)partial)Path) TYPE_CHECKINGAnyTypeVar) ParamSpecTypeIszpydantic-evals) otel_scopec@seZdZdZdS)UnsetzA singleton to represent an unset value. Used to distinguish between explicitly set `None` values and values that were never set. Copied from pydantic_ai/_utils.py. N)__name__ __module__ __qualname____doc__rr[/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_evals/_utils.pyrsrT t_or_unset T | Unsetreturn TypeIs[T]cCs|tuS)aCheck if a value is set (not the UNSET singleton). Args: t_or_unset: The value to check, which may be the UNSET singleton or a regular value. Returns: True if the value is not UNSET, narrowing the type to T in a type-aware way. )UNSET)rrrris_set%s rfuncCallable[..., Any]strc sfd fdd z|jWSty2}ztt|ddr,t|jdWYd}~S|d}~ww) aGet the name of a function, unwrapping partials and decorators. Args: func: The function to get the name of. Returns: The name of the function. Raises: AttributeError: If the function doesn't have a __name__ attribute and isn't a method. frrcst|tr |jSt|S)zFUnwraps f, also unwrapping partials, for the sake of getting f's name.) isinstancerrinspectunwrap)r!_unwraprrr&>s   z,get_unwrapped_function_name.._unwrap__call__Nz .__call__)r!rrr)rAttributeErrorr#ismethodgetattrtyper)rerr%rget_unwrapped_function_name1s  r-_P_RcCs6zt}W|Styt}t|Y|Sw)N)asyncioget_event_loop RuntimeErrornew_event_loopset_event_loop) event_looprrrr1Rs   r1tasks$Sequence[Callable[[], Awaitable[T]]]list[T]c sdgt|d fdd }t4IdH}t|D] \}}||||qWdIdHS1IdHs9wYS) zRun multiple awaitable callables concurrently using an AnyIO task group. Args: tasks: A list of no-argument callables that return awaitable objects. Returns: A list of results in the same order as the input tasks. NtskCallable[[], Awaitable[T]]indexintrNonecs|IdH|<dS)zHHelper function to run a task and store the result in the correct index.Nr)r9r;resultsrr _run_taskfsz$task_group_gather.._run_task)r9r:r;r<rr=)lenanyiocreate_task_group enumerate start_soon)r6r@tgitaskrr>rtask_group_gather[s rI)LogfireNotConfiguredWarningc@s eZdZdS)rJN)rrrrrrrrJxsrJargsr kwargs.Generator[logfire_api.LogfireSpan, None, None]c ost3tjdtdtj|i| }|VWdn1s#wYWddSWddS1s;wYdS)zCCreate a Logfire span without warning if logfire is not configured.ignore)categoryN)warningscatch_warningsfilterwarningsrJ_logfirespan)rKrLrTrrr logfire_spans "rU)rrrr)rrrr )r6r7rr8)rKr rLr rrM)/ __future__r _annotationsr0r#rPcollections.abcrrrr contextlibr functoolsrpathlibr typingr r r rB logfire_apityping_extensionsr rLogfirerSadd_non_user_code_prefix__file__parentabsoluterrrrr-r.r/r1rIlogfire._internal.configrJ ImportError UserWarningrTrUrrrrsB