o iB@sUddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZmZddlmZmZddlmZmZmZddlmZmZddlmZddlmZdd lmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'dd l(m)Z)m*Z*dd l+m,Z,dd l-m.Z.m/Z/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8e6Z6erddl9m:Z:m;Z;ddlZ>ddl7m?Z@ddlAmBZBe.dZCe"dZDdd d!ZEdd%d&ZFdd*d+ZGdd.d/ZHe"d0ZIeGd1d2d2eeIZJeJeIdBZKd3eLd4< Gd5d6d6ZMeMZNdd:d;ZOeddAdBZPddEdFZQddIdJZRddLdMZSddQdRZTddSdTZUGdUdVdVeeIZVddYdZZWdd\d]ZXe*eZYdd_d`ZZe dae eIfZ[e%ddddeZ\e%ddgdeZ\ddhdeZ\ddndoZ]ddsdtZ^ddvdwZ_ddydzZ`dd|d}ZadddZbddZcdS)) annotationsN) AsyncIterable AsyncIterator AwaitableCallableIterableIterator)asynccontextmanagersuppress) dataclassfields is_dataclass)datetimetimezone)partial) GenericAlias) TYPE_CHECKINGAnyGeneric TypeAlias TypeGuardTypeVarget_args get_originoverload)run_sync) BaseModel TypeAdapter)JsonSchemaValue) ParamSpecTypeIs is_typeddict)typing_objects)is_union_origin) AbstractSpan) exceptions)AgentRunAgentRunResult)GraphRunGraphRunResult)messages)ObjectJsonSchema_P_RfuncCallable[_P, _R]args_P.argskwargs _P.kwargsreturncs&t|g|Ri|}t|IdHSN)rr)r/r1r3 wrapped_funcr8X/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/_utils.pyrun_in_executor-sr:type_rboolcCs<t|tot|t ot|tpt|pt|pt|ddS)zCheck if something is a pydantic model, dataclass or typedict. These should all generate a JSON Schema with `{"type": "object"}` and therefore be usable directly as function parameters. __is_model_like__F) isinstancetyper issubclassrr r!getattr)r;r8r8r9 is_model_like2s    rBschemarr,cCsddlm}|ddkr|S|d}r=d}||r;|di|t|d}r;|ddkr;t|s;|S|S|d) Nr%) UserErrorr?object$ref#/$defs/$defszSchema must be an object)r&rDget startswithlen _contains_ref)rCrDrefprefixresolvedr8r8r9check_object_json_schemaDs  rPobj'JsonSchemaValue | list[JsonSchemaValue]cCs6t|trd|vr dS|}n|}tdd|DS)z6Recursively check if an object contains any $ref keys.rFTcss&|]}t|ttBot|VqdSr6)r>dictlistrL).0itemr8r8r9 as$z _contains_ref..)r>rSvaluesany)rQitemsr8r8r9rLXs  rLTc@seZdZUdZded<dS)Somez(Analogous to Rust's `Option::Some` type.r[valueN)__name__ __module__ __qualname____doc____annotations__r8r8r8r9r\gs  r\rOptionc@seZdZdZdS)Unsetz(A singleton to represent an unset value.N)r^r_r`rar8r8r8r9rdrsrd t_or_unset T | Unset TypeGuard[T]cCs|tuSr6)UNSET)rer8r8r9is_set{sri aiterableAsyncIterable[T]soft_max_interval float | None%AsyncIterator[AsyncIterable[list[T]]]csdurd fdd }|VdSdd fdd }z+|VWrGdttjIdHWddS1s@wYdSdSrjdttj IdHWdw1sewYw) aGroup items from an async iterable into lists based on time interval between them. Effectively, this debounces the iterator. This returns a context manager usable as an iterator so any pending tasks can be cancelled if an error occurs during iteration. Usage: ```python async with group_by_temporal(yield_groups(), 0.1) as groups_iter: async for groups in groups_iter: print(groups) ``` Args: aiterable: The async iterable to group. soft_max_interval: Maximum interval over which to group items, this should avoid a trickle of items causing a group to never be yielded. It's a soft max in the sense that once we're over this time, we yield items as soon as `anext(aiter)` returns. If `None`, no grouping/debouncing is performed Returns: A context manager usable as an async iterable of lists of items produced by the input async iterable. Nr5AsyncIterator[list[T]]cs"2z 3dHW}|gVq6dSr6r8)rV)rjr8r9async_iter_groups_noops z1group_by_temporal..async_iter_groups_noopcsdur dks Jdg}t}t} |dur}nt|}dur2tt|tjf|dIdH\}}|rkz|}Wnt yZ|rU|VdYdSw| |d|durjt}n |rt|Vg}d}q)Nrz+soft_max_interval must be a positive numberT)timeout) time monotonicaiterasyncio create_taskanextwaitpopresultStopAsyncIterationappend)buffergroup_start_time aiterator wait_timedone_rVrjrltaskr8r9async_iter_groupss>  z,group_by_temporal..async_iter_groupsz#Cancelling due to error in iterator)r5ro)cancelr ruCancelledError)rjrlrprr8rr9group_by_temporals&1    "   riterator Iterator[T]c Cs,zt|WSty}zt|d}~ww)zGet the next item from a sync iterator, raising `StopAsyncIteration` if it's exhausted. Useful when iterating over a sync iterator in an async context. N)next StopIterationr{)rer8r8r9 sync_anexts  r async_iterAsyncIterator[T]ccs6t} z |t|VWn tyYdSwqr6)get_event_looprun_until_completerwr{)rloopr8r8r9sync_async_iterators rrcCstjtjdS)N)tz)rnowrutcr8r8r8r9now_utcsrt_messages.ToolCallPart | _messages.ToolReturnPart | _messages.RetryPromptPart | _messages.BuiltinToolCallPart | _messages.BuiltinToolReturnPartstrcCs |jptS)zTType guard that either returns the tool call id or generates a new one if it's None.) tool_call_idgenerate_tool_call_id)rr8r8r9guard_tool_call_ids rcCsdtjS)zJGenerate a tool call id. Ensure that the tool call id is unique. pyd_ai_)uuiduuid4hexr8r8r8r9rsrc@sBeZdZdZdddZddd Zdd d ZdddZdddZdS)PeekableAsyncStreamzWraps an async iterable of type T and allows peeking at the *next* item without consuming it. We only buffer one item at a time (the next item). Once that item is yielded, it is discarded. This is a single-pass stream. sourcerkcCs||_d|_t|_d|_dS)NF)_source _source_iterrh_buffer _exhausted)selfrr8r8r9__init__s zPeekableAsyncStream.__init__r5rfcsl|jrtSt|jts|jS|jdurt|j|_z t|jIdH|_W|jSt y5d|_tYSw)z|Returns the next item that would be yielded without consuming it. Returns None if the stream is exhausted. NT) rrhr>rrdrrtrrwr{rr8r8r9peeks    zPeekableAsyncStream.peekr<cst|IdHtS)z9Returns True if the stream is exhausted, False otherwise.N)r>rrdrr8r8r9 is_exhausted2sz PeekableAsyncStream.is_exhaustedrcCs|Sr6r8rr8r8r9 __aiter__6szPeekableAsyncStream.__aiter__r[csh|jrtt|jts|j}t|_|S|jdurt|j|_z t |jIdHWSty3d|_w)zYields the buffered item if present, otherwise fetches the next item from the underlying source. Raises StopAsyncIteration if the stream is exhausted. NT) rr{r>rrdrhrrtrrw)rrVr8r8r9 __anext__:s    zPeekableAsyncStream.__anext__N)rrk)r5rf)r5r<)r5r)r5r[) r^r_r`rarrrrrr8r8r8r9r s    rx5AgentRun | AgentRunResult | GraphRun | GraphRunResultcCs|jddpdS)NF)required) _traceparentrr8r8r9get_traceparentSsrrcs0fddtD}jjdd|dS)z6Exclude fields with values equal to the field default.c3s@|]}|jrt|j|jkr|jdt|jVqdS)=N)reprrAnamedefault)rUfrr8r9rWYs .z/dataclasses_no_defaults_repr..(, ))r __class__r`join)rkv_pairsr8rr9dataclasses_no_defaults_reprWs r int | floatcCs t|Sr6) _datetime_tavalidate_pythonrr8r8r9number_to_datetimebs r.AwaitableCallable[T]TypeIs[AwaitableCallable[T]]cCdSr6r8rQr8r8r9is_async_callableirTypeIs[AwaitableCallable[Any]]cCrr6r8rr8r8r9rmrcCs<t|tjr|j}t|tjst|pt|ot|jS)zCorrectly check if a callable is async. This function was copied from Starlette: https://github.com/encode/starlette/blob/78da9b9e218ab289117df7d62aee200ed4c59617/starlette/_utils.py#L36-L40 )r> functoolsrr/inspectiscoroutinefunctioncallable__call__rr8r8r9rqs  sdict[str, Any] name_mappingdict[str, str]Nonec Csd|vr |d}|dr |dd}|||}d||d<d|vr4|d}|D]}t||q,d|vrHt|dtrH|d}t||d|vrZ|d}|D]} t| |qRdD]} | |vrp|| } | D]} t| |qhq\dS) z@Update $refs in a schema to use the new names from name_mapping.rFrGN propertiesrZ prefixItems)anyOfoneOf)rJrIrX_update_mapped_json_schema_refsr>rS) rrrM original_namenew_namepropsproprZ prefix_itemsrV union_type union_itemsr8r8r9r}s0        rschemaslist[dict[str, Any]]6tuple[list[dict[str, Any]], dict[str, dict[str, Any]]]c Csi}g}|D]t}d|vr||q|}|dd}i}|D]M\}}||vr3|||<|||<q"|||kro|}|d} rI| d|}d} |} |d| }||vrg| d7} | d| }||vsX|||<|||<q"t||||q||fS)aMerges the `$defs` from different JSON schemas into a single deduplicated `$defs`, handling name collisions of `$defs` that are not the same, and rewrites `$ref`s to point to the new `$defs`. Returns a tuple of the rewritten schemas and a dictionary of the new `$defs`. rHNtitlerr%)r|copyryrZrIr) rall_defsrewritten_schemasrCdefsschema_name_mappingr def_schemarrioriginal_new_namer8r8r9merge_json_schema_defss<      r_kwargscCs0|rddd|D}td|dS)zValidate that no unknown kwargs remain after processing. Args: _kwargs: Dictionary of remaining kwargs after specific ones have been processed. Raises: UserError: If any unknown kwargs remain. rcss|] }d|dVqdS)`Nr8)rUkr8r8r9rWsz(validate_empty_kwargs..zUnknown keyword arguments: N)rkeysr&rD)runknown_kwargsr8r8r9validate_empty_kwargss rtextcCs4|dr|Sd}t||tj}|r|dS|S)N{z```(?:\w+)?\n(\{.*\})\n```r%)rJresearchDOTALLgroup)rregexmatchr8r8r9strip_markdown_fencess  rtpcCs.t|}t|r|j}t|}t|s |Sr6)rr" is_annotated __origin__roriginr8r8r9_unwrap_annotateds   rtuple[Any, ...]cCsBt|r|j}t|}t|}t|rtddt|DSdS)zZExtract the arguments of a Union type if `tp` is a union, otherwise return an empty tuple.css|]}t|VqdSr6)r)rUargr8r8r9rWsz!get_union_args..r8)r"is_typealiastype __value__rrr#tuplerrr8r8r9get_union_argss r cCs6zt}W|Styt}t|Y|Swr6)rur RuntimeErrornew_event_loopset_event_loop) event_loopr8r8r9rs   r)r/r0r1r2r3r4r5r.)r;rr5r<)rCrr5r,)rQrRr5r<)rerfr5rg)rjrkrlrmr5rn)rrr5r[)rrr5r)r5r)rrr5r)r5r)rrr5r)rrr5r)rrr5r)rQrr5r)rQrr5r)rQrr5r)rrrrr5r)rrr5r)rrr5r)rrr5r)rrr5r)rrr5r)d __future__r _annotationsrurrrrrrcollections.abcrrrrrr contextlibr r dataclassesr r r rrrtypesrtypingrrrrrrrrranyio.to_threadrpydanticrrpydantic.json_schemartyping_extensionsrr r!typing_inspectionr"typing_inspection.introspectionr#pydantic_graph._utilsr$rr&pydantic_ai.agentr'r(pydantic_graphr)r*r+ _messagestoolsr,r-r.r:rBrPrLr[r\rcrbrdrhrirrrrrrrrrrrAwaitableCallablerrrrrrr rr8r8r8r9s   ,                b   F      +