o iV=@slddlmZddlZddlZddlmZddlmZddlm Z ddlm Z ddl m Z m Z ddlZddlmZmZddlZd d lmZmZd d lmZmZmZd d lmZd d lmZGdddeZGdddeZGddde Z e Z!d6ddZ" d7d8d d!Z#ed9d'd(Z$Gd)d*d*e%Z&Gd+d,d,e%Z'Gd-d.d.e(Z)d:d0d1Z*d:d2d3Z+Gd4d5d5eZ,dS);) annotationsN)Iterator) lru_cache) Formatter)CodeType)AnyLiteral) NotRequired TypedDict)CallNodeFinderget_node_source_text) NOOP_SCRUBBER BaseScrubberMessageValueCleaner)warn_at_user_stacklevel)log_internal_errorc@seZdZUded<ded<dS) LiteralChunkzLiteral['lit']tstrvN__name__ __module__ __qualname____annotations__rra/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/_internal/formatter.pyrs  rc@s&eZdZUded<ded<ded<dS)ArgChunkzLiteral['arg']rrrzNotRequired[str]specNrrrrrrs  rc@s6eZdZdddd dZdddZdddddZdS)ChunksFormatterN) fstring_frame format_stringrkwargsdict[str, Any]scrubberrr!types.FrameType | Nonereturn9tuple[list[LiteralChunk | ArgChunk], dict[str, Any], str]cCs8|r ||||}|r |S|j|||d\}}|||fS)N)r#r%)_fstring_chunks_vformat_chunks)selfr"r#r%r!resultchunks extra_attrsrrrr-"s zChunksFormatter.chunksframetypes.FrameType@tuple[list[LiteralChunk | ArgChunk], dict[str, Any], str] | NonecCs|j}|j}|dus Jt|}|j}|durdS|tjjjkrEt|j dkr-|j d}n(|j D] }|j dkr<|j }nq0| ddSn|j rN|j d}n| ddSt|tjs]dS|j} i|j|} g} d} i} t|dd}|jD]U}t|tjr|j }| |d d | |7} qvt|tjsJt||j\}}}| d |d 7} t|| | }|| |<t|| i| d |i}|||}| |dd qv| || | | fS)Nr msg_templatez:Couldn't identify the `msg_template` argument in the call.rF check_keyslitrr{}@fvaluearg)f_codef_backFormattingCallNodeFindernodelogfireLogfirelog__code__lenargskeywordsr<valuewarn_inspect_arguments isinstanceast JoinedStr f_globalsf_localsrvaluesConstantappendFormattedValuecompile_formatted_valuesourceeval clean_valueupdater.)r+r#r%r/ called_code node_finder call_nodearg_nodekeyword global_vars local_varsr, new_templater. value_cleaner node_valuerHrT value_codeformatted_code formattedrrrr)=sZ               zChunksFormatter._fstring_chunksr2)recursion_depthreint4tuple[list[LiteralChunk | ArgChunk], dict[str, Any]]cCs|dkrtdg}d}t|dd}||D](\}} } } |r(||dd| dur@| d kr5td | d r]|rO|d d dkrO|d d| 7<n|| dd| dd } z || ||\} } Wnbtystdty}z6t|t | krtd| d|z|| } Wnty}z td|d|d|d}~wwWYd}~nd}~wt y}z td| d||d}~ww| durz| | | } Wnt y}z td| d||d}~ww|j | pd |t |dd\}}d dd|D} z|| | }Wnt y(}z td| d||d}~ww|| |}|dd}| r;| |d<||q||fS)z|Copied from `string.Formatter._vformat` https://github.com/python/cpython/blob/v3.11.4/Lib/string.py#L198-L247 then altered.rz"Max format spec recursion exceededrFr5r7r8Nr4zDEmpty curly brackets `{}` are not allowed. A field name is required.=rrz$Numeric field names are not allowed.z The field {z} is not defined.z The fields z and z are not defined.zError getting field {z}: zError converting field {r )r%recs|]}|dVqdSrNr.0chunkrrr z2ChunksFormatter._vformat_chunks..zError formatting field {r<r)KnownFormattingErrorrparserQendswith get_field IndexErrorKeyErrorrrepr Exception convert_fieldr*rjoin format_fieldrVr.)r+r"r#r%rer,rFr` literal_text field_name format_spec conversionobj _arg_usedexc1exc2excformat_spec_chunks_rHdrrrr*sv             zChunksFormatter._vformat_chunks) r"rr#r$r%rr!r&r'r()r#r$r%rr/r0r'r1) r"rr#r$r%rrerfr'rg)rrrr-r)r*rrrrr !s  ar r"rr#r$r%rr'cCst|||\}}}|SN)logfire_format_with_magic)r"r#r%r, _extra_attrs _new_templaterrrlogfire_formats  rr!r&tuple[str, dict[str, Any], str]c Csztj||||d\}}}ddd|D||fWSty7}ztt|p+t|jWYd}~n'd}~wtyN}z tt|WYd}~nd}~wt yYt Ynw|i|fS)N)r%r!r4csrjrkrrlrrrrorpz,logfire_format_with_magic..) chunks_formatterr-rzrqwarn_formattingr __cause__FStringAwaitErrorwarn_fstring_awaitrxr)r"r#r%r!r-r.r_errrrs$  "   rr@ast.FormattedValue ex_sourceexecuting.Sourcetuple[str, CodeType, CodeType]cCst|j|}t|jD] }t|tjrt|q t|dd}ttj tj tj dt d|j |jdgd}t|t|dd}|||fS)a Returns three things that can be expensive to compute. 1. Source code corresponding to the node value (excluding the format spec). 2. A compiled code object which can be evaluated to calculate the value. 3. Another code object which formats the value. z rUr;)idctx)rHrr~)rOz )r rHrKwalkrJAwaitrcompile ExpressionrLrRNameLoadrr~fix_missing_locations)r@rrTsub_noderbexprrcrrrrSs&       rSc@eZdZdZdS)rqa.An error raised when there's something wrong with a format string or the field values. In other words this should correspond to errors that would be raised when using `str.format`, and generally indicate a user error, most likely that they weren't trying to pass a template string at all. Nrrr__doc__rrrrrq=rqc@r)rzAn error raised when an await expression is found in an f-string. This is a specific case that can't be handled by f-string introspection and requires pre-evaluating the await expression before logging. NrrrrrrErrc@s eZdZdS)FormattingFailedWarningN)rrrrrrrrMsrmsgcCtd|tddS)NaO Ensure you are either: (1) passing an f-string directly, with inspect_arguments enabled and working, or (2) passing a literal `str.format`-style template, not a preformatted string. See https://logfire.pydantic.dev/docs/guides/onboarding-checklist/add-manual-tracing/#messages-and-span-names. The problem was: categoryrrrrrrrQs  rcCr)Na Cannot evaluate await expression in f-string. Pre-evaluate the expression before logging. For example, change: logfire.info(f"{await get_value()}") To: value = await get_value() logfire.info(f"{value}") The problematic f-string value was: rrrrrrr]s  rc@s,eZdZdZdddZdd d Zd d Zd S)r?zRFinds the call node corresponding to a call like `logfire.span` or `logfire.info`.r'Iterator[ast.AST]ccs4|jjD]}t|tjr|jEdHq|VqdSr)ex statementsrJrKWithitems)r+ statementrrrheuristic_main_nodesns   z-FormattingCallNodeFinder.heuristic_main_nodesr@ast.CallboolcCst|jp|jSr)rrFrG)r+r@rrrheuristic_call_node_filtervsz3FormattingCallNodeFinder.heuristic_call_node_filtercCsdS)NzgFalling back to normal message formatting which may result in loss of information if using an f-string.r)r+rrrwarn_inspect_arguments_middlezsz6FormattingCallNodeFinder.warn_inspect_arguments_middleN)r'r)r@rr'r)rrrrrrrrrrrr?ks    r?)r"rr#r$r%rr'rr) r"rr#r$r%rr!r&r'r)r@rrrr'r)rr)- __future__rrKtypescollections.abcr functoolsrstringrrtypingrr executingtyping_extensionsr r rA ast_utilsr r scrubbingrrr stack_inforutilsrrrr rrrrSrxrqr UserWarningrrrr?rrrrs>       J   &