j" UdZddlmZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZddlmZddlmZerdd lmZdd lmZdd lmZmZmZm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&dd l'm(Z(m)Z)ddl*m+Z+ddl,m-Z-iZ.ee/e dfe0d<GddZ1Gdde1Z2Gdde1Z3Gdde1Z4Gdde1Z5Gdde1Z6Gdde1Z7Gdd e1Z8Gd!d"e1Z9Gd#d$e1Z:Gd%d&e1Z;e.<e2e3e5e6e7e8e:e9e;d' d(S))a{OTEL span wrapper for Langfuse. This module defines custom span classes that extend OpenTelemetry spans with Langfuse-specific functionality. These wrapper classes provide methods for creating, updating, and scoring various types of spans used in AI application tracing. Classes: - LangfuseObservationWrapper: Abstract base class for all Langfuse spans - LangfuseSpan: Implementation for general-purpose spans - LangfuseGeneration: Specialized span implementation for LLM generations All span classes provide methods for media processing, attribute management, and scoring integration specific to Langfuse's observability platform. )datetime)time_ns) TYPE_CHECKINGAnyDictLiteralOptionalTypeUnioncastoverload)trace)Status StatusCode)_AgnosticContextManager) PromptClient)Langfuse) deprecated)LangfuseOtelSpanAttributescreate_generation_attributescreate_span_attributescreate_trace_attributes)ObservationTypeGenerationLikeObservationTypeLiteralObservationTypeLiteralNoEventObservationTypeSpanLikeget_observation_types_list)MapValue ScoreDataType)langfuse_logger) SpanLevelLangfuseObservationWrapper_OBSERVATION_CLASS_MAPc$VeZdZdZddddddddddddddddejdddedeed eed eed ee d ee d ee dee dee dee dee dee e e fdee e efdee e efdeef"dZdddeeddfdZedddddeed eeddfdZdidZedddddddd e d!ed"ee d#eeejejfd$ee d%ee d&ee d eeddfd'Zedejdddddd e d!e d"ee d#eeejejfd$ee d%ee d&ee d eeddfd(Zdddddddd e d!eee fd"ee d#eed$ee d%ee d&ee d eeddfd)Zedddddddd e d!ed"ee d#eeejejfd$ee d%ee d&ee d eeddfd*Zedejdddddd e d!e d"ee d#eeejejfd$ee d%ee d&ee d eeddfd+Zdddddddd e d!eee fd"ee d#eed$ee d%ee d&ee d eeddfd,Zddddd-d.ejdeedeed eed eeddf d/Z dd0d1eed.ejd2eeded ed fdeefd3Z!d1edefd4Z"dd0d1eed2eeded ed fdeefd5Z#ddd6dee dee ddfd7Z$dddddddddddddd8 d ee deed eed eed ee dee dee dee dee dee e e fdee e efdee e efdeed9eddfd:Z%eddddddd;d e ded.deed eed eed ee dee dee dd d e ded?deed eed eed ee dee dee dee dee dee e e fdee e efdee e efdeedd@fdAZ&eddddddd;d e dedBdeed eed eed ee dee dee ddCfdDZ&eddddddd;d e dedEdeed eed eed ee dee dee ddFfdGZ&eddddddd;d e dedHdeed eed eed ee dee dee ddIfdJZ&eddddddd;d e dedKdeed eed eed ee dee dee ddLfdMZ&eddddddd;d e dedNdeed eed eed ee dee dee ddOfdPZ&eddddddddddddd> d e dedQdeed eed eed ee dee dee dee dee dee e e fdee e efdee e efdeeddRfdSZ&eddddddd;d e dedTdeed eed eed ee dee dee ddUfdVZ&eddddddd;d e dedWdeed eed eed ee dee dee ddXfdYZ&d.dddddddddddddZ d e dedeed eed eed ee dee dee dee dee dee e e fdee e efdee e efdeeded[fd\Z&eddddddd;d e ded.deed eed eed ee dee dee de'd<fd]Z(eddddddddddddd> d e ded?deed eed eed ee dee dee dee dee dee e e fdee e efdee e efdeede'd@fd^Z(eddddddddddddd> d e dedQdeed eed eed ee dee dee dee dee dee e e fdee e efdee e efdeede'dRfd_Z(eddddddd;d e dedBdeed eed eed ee dee dee de'dCfd`Z(eddddddd;d e dedEdeed eed eed ee dee dee de'dFfdaZ(eddddddd;d e dedHdeed eed eed ee dee dee de'dIfdbZ(eddddddd;d e dedKdeed eed eed ee dee dee de'dLfdcZ(eddddddd;d e dedNdeed eed eed ee dee dee de'dOfddZ(eddddddd;d e dedTdeed eed eed ee dee dee de'dUfdeZ(d.dddddddddddddZ d e de)deed eed eed ee dee dee dee dee dee e e fdee e efdee e efdeede'edffdgZ(ddddddd;d e deed eed eed ee dee dee ddXfdhZ*dS)jr"aAbstract base class for all Langfuse span types. This class provides common functionality for all Langfuse span types, including media processing, attribute management, and scoring. It wraps an OpenTelemetry span and extends it with Langfuse-specific features. Attributes: _otel_span: The underlying OpenTelemetry span _langfuse_client: Reference to the parent Langfuse client trace_id: The trace ID for this span observation_id: The observation ID (span ID) for this span Ninputoutputmetadata environmentreleaseversionlevelstatus_messagecompletion_start_timemodelmodel_parameters usage_details cost_detailsprompt otel_spanlangfuse_clientras_typer&r'r(r)r*r+r,r-r.r/r0r1r2r3c*||_|jtj|||_||_|j||_|j||_ |p |jj |_ |j *|jtj |j |p |jj |_ |j *|jtj |j |jrx||d|j}||d|j}||d|j}i}|t!t"vr0t%|||| | | | | ||||t't"| }njt)|||| | | t't*t,t.t0df|t!t.vs|dkr|nd}|tjd|jd |D|| | dSdS) aInitialize a new Langfuse span wrapper. Args: otel_span: The OpenTelemetry span to wrap langfuse_client: Reference to the parent Langfuse client as_type: The type of span ("span" or "generation") input: Input data for the span (any JSON-serializable object) output: Output data from the span (any JSON-serializable object) metadata: Additional metadata to associate with the span environment: The tracing environment release: Release identifier for the application version: Version identifier for the code or component level: Importance level of the span (info, warning, error) status_message: Optional status message for the span completion_start_time: When the model started generating the response model: Name/identifier of the AI model used (e.g., "gpt-4") model_parameters: Parameters used for the model (e.g., temperature, max_tokens) usage_details: Token usage information (e.g., prompt_tokens, completion_tokens) cost_details: Cost information for the model call prompt: Associated prompt template from Langfuse prompt management Nr&datafieldspanr'r() r&r'r(r+r,r-r.r/r0r1r2r3observation_typeeventr&r'r(r+r,r-r<ci|] \}}||| SN).0kvs e/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/langfuse/_client/span.py z7LangfuseObservationWrapper.__init__..sFFF$!Q A r,r-) _otel_span set_attributerOBSERVATION_TYPE_langfuse_client_observation_type_get_otel_trace_idtrace_id_get_otel_span_idid _environment ENVIRONMENT_releaseRELEASE is_recording_process_media_and_apply_maskrrrr rr r rrpopset_attributesitems_set_otel_span_status_if_error)selfr4r5r6r&r'r(r)r*r+r,r-r.r/r0r1r2r3media_processed_inputmedia_processed_outputmedia_processed_metadata attributess rE__init__z#LangfuseObservationWrapper.__init__NsT$ %% & 7   !0!(-@@KK '99)DD'M4+@+M   ( O ) )*68I    A4#8#A = $ O ) )*2DM    ? ' ' ) )= $($F$F'%G%% !&*%G%G8$/&H&& "(,'I'IZdo(J(( $J45RSSSS9/15##1*?%5"/!-!%)5&& *4/15##1%) '>@P'P!QR"56MNNOO"g-- " && $ NN5F M M M O * *FF*"2"2"4"4FFF     / /N 0     w= = rGend_timercreturnc<|j||S)a_End the span, marking it as completed. This method ends the wrapped OpenTelemetry span, marking the end of the operation being traced. After this method is called, the span is considered complete and can no longer be modified. Args: end_time: Optional explicit end time in nanoseconds since epoch rb)rIend)r\rcs rErfzLangfuseObservationWrapper.ends" X... rGzTrace-level input/output is deprecated. For trace attributes (user_id, session_id, tags, etc.), use propagate_attributes() instead. This method will be removed in a future major version.r&r'c|js|S||d|j}||d|j}t||}|j||S)aSet trace-level input and output for the trace this span belongs to. .. deprecated:: This is a legacy method for backward compatibility with Langfuse platform features that still rely on trace-level input/output (e.g., legacy LLM-as-a-judge evaluators). It will be removed in a future major version. For setting other trace attributes (user_id, session_id, metadata, tags, version), use :meth:`Langfuse.propagate_attributes` instead. Args: input: Input data to associate with the trace. output: Output data to associate with the trace. Returns: The span instance for method chaining. r&r8r'rg)rIrVrWrrY)r\r&r'r]r^r`s rE set_trace_ioz'LangfuseObservationWrapper.set_trace_ios8++-- K $ B BgDO!C! ! "&!C!Cxdo"D" " -')   &&z222 rGc|js|Std}|j||S)aMake this trace publicly accessible via its URL. When a trace is published, anyone with the trace link can view the full trace without needing to be logged in to Langfuse. This action cannot be undone programmatically - once any span in a trace is published, the entire trace becomes public. Returns: The span instance for method chaining. T)public)rIrVrrY)r\r`s rEset_trace_as_publicz.LangfuseObservationWrapper.set_trace_as_public sJ++-- K,D999  &&z222 rG)score_id data_typecomment config_id timestampr(namevaluermrnrorprqcdSr@rA r\rrrsrmrnrorprqr(s rEscorez LangfuseObservationWrapper.score srGcdSr@rArus rErvz LangfuseObservationWrapper.score/rwrGc |j|tt||j|j|tt d||||| dS)aCreate a score for this specific span. This method creates a score associated with this specific span (observation). Scores can represent any kind of evaluation, feedback, or quality metric. Args: name: Name of the score (e.g., "relevance", "accuracy") value: Score value (numeric for NUMERIC/BOOLEAN, string for CATEGORICAL/TEXT) score_id: Optional custom ID for the score (auto-generated if not provided) data_type: Type of score (NUMERIC, BOOLEAN, CATEGORICAL, or TEXT) comment: Optional comment or explanation for the score config_id: Optional ID of a score config defined in Langfuse timestamp: Optional timestamp for the score (defaults to current UTC time) metadata: Optional metadata to be attached to the score Example: ```python with langfuse.start_as_current_observation(name="process-query") as span: # Do work result = process_data() # Score the span span.score( name="accuracy", value=0.95, data_type="NUMERIC", comment="High accuracy result" ) ```  CATEGORICALTEXT) rrrsrOobservation_idrmrnrorprqr(N)rL create_scorer strrOrQrrus rErvz LangfuseObservationWrapper.score?skT **sE""]77#899EE + rGcdSr@rArus rE score_tracez&LangfuseObservationWrapper.score_tracevrwrGcdSr@rArus rErz&LangfuseObservationWrapper.score_tracerwrGc |j|tt||j|tt d||||| dS)aCreate a score for the entire trace that this span belongs to. This method creates a score associated with the entire trace that this span belongs to, rather than the specific span. This is useful for overall evaluations that apply to the complete trace. Args: name: Name of the score (e.g., "user_satisfaction", "overall_quality") value: Score value (numeric for NUMERIC/BOOLEAN, string for CATEGORICAL/TEXT) score_id: Optional custom ID for the score (auto-generated if not provided) data_type: Type of score (NUMERIC, BOOLEAN, CATEGORICAL, or TEXT) comment: Optional comment or explanation for the score config_id: Optional ID of a score config defined in Langfuse timestamp: Optional timestamp for the score (defaults to current UTC time) metadata: Optional metadata to be attached to the score Example: ```python with langfuse.start_as_current_observation(name="handle-request") as span: # Process the complete request result = process_request() # Score the entire trace (not just this span) span.score_trace( name="overall_quality", value=0.9, data_type="NUMERIC", comment="Good overall experience" ) ``` rz) rrrsrOrmrnrorprqr(N)rLr~r rrOrrus rErz&LangfuseObservationWrapper.score_tracesfV **sE""]7#899EE + rG)r6r&r'r(r;c|||d}|||d}|||d}|dkrt|||nt|||} || dS)a Set span attributes after processing media and applying masks. Internal method that processes media in the input, output, and metadata and applies any configured masking before setting them as span attributes. Args: span: The OpenTelemetry span to set attributes on as_type: The type of span ("span" or "generation") input: Input data to process and set output: Output data to process and set metadata: Metadata to process and set r&)r;r9r:r'r( generation)r&r'r(N)rWrrrY) r\r;r6r&r'r(processed_inputprocessed_outputprocessed_metadatamedia_processed_attributess rE_set_processed_span_attributesz9LangfuseObservationWrapper._set_processed_span_attributess*<<=    ==>   "??@  ,&& )%'+     (%'+ # 677777rGr9r9r:cX||||S)aProcess media in an attribute and apply masking. Internal method that processes any media content in the data and applies the configured masking function to the result. Args: data: The data to process span: The OpenTelemetry span context field: Which field this data represents (input, output, or metadata) Returns: The processed and masked data )r9r:r)_mask_attribute_process_media_in_attribute)r\r9r;r:s rErWz8LangfuseObservationWrapper._process_media_and_apply_masks7(##11t51II$   rGc|jjs|S |j|S#t$r"}tjd|Yd}~dSd}~wwxYw)aRApply the configured mask function to data. Internal method that applies the client's configured masking function to the provided data, with error handling and fallback. Args: data: The data to mask Returns: The masked data, or the original data if no mask is configured rziMasking error: Custom mask function threw exception when processing data. Using fallback masking. Error: Nz*)rL_mask Exceptionr error)r\r9es rErz*LangfuseObservationWrapper._mask_attributes$* K @(..D.99 9 @ @ @  !|}   @?????  @s+ AAAc|jj2|jjj|||j|jS|S)aProcess any media content in the attribute data. Internal method that identifies and processes any media content in the provided data, using the client's media manager. Args: data: The data to process for media content span: The OpenTelemetry span context field: Which field this data represents (input, output, or metadata) Returns: The data with any media content processed N)r9r:rOr})rL _resources_media_manager_find_and_process_mediarOrQ)r\r9r:s rErz6LangfuseObservationWrapper._process_media_in_attribute2sN&  + 7%0?WW!]#'7 X  rGrHc|dkr`|jrI |jttj|dS#t $rYdSwxYwdSdS)aSet OpenTelemetry span status to ERROR if level is ERROR. This method sets the underlying OpenTelemetry span status to ERROR when the Langfuse observation level is set to ERROR, ensuring consistency between Langfuse and OpenTelemetry error states. Args: level: The span level to check status_message: Optional status message to include as description ERROR) descriptionN)rIrV set_statusrrrr)r\r,r-s rEr[z9LangfuseObservationWrapper._set_otel_span_status_if_errorQs G   < < > >  **:+HHH        s3A A$#A$ rrr&r'r(r+r,r-r.r/r0r1r2r3kwargsc  |js|S||d|j}||d|j}||d|j}|r|j||jt t vr5t||||||tt |j|| | | | |  }nyt||||||tttttdf|jt tvs |jdkr|jnd}|j| ||| |S) aUpdate this observation with new information. This method updates the observation with new information that becomes available during execution, such as outputs, metadata, or status changes. Args: name: Observation name input: Updated input data for the operation output: Output data from the operation metadata: Additional metadata to associate with the observation version: Version identifier for the code or component level: Importance level of the observation (info, warning, error) status_message: Optional status message for the observation completion_start_time: When the generation started (for generation types) model: Model identifier used (for generation types) model_parameters: Parameters passed to the model (for generation types) usage_details: Token or other usage statistics (for generation types) cost_details: Cost breakdown for the operation (for generation types) prompt: Reference to the prompt used (for generation types) **kwargs: Additional keyword arguments (ignored) r&r8r'r() r&r'r(r+r,r-r<r.r/r0r1r2r3r=Nr>)r`rH)rIrVrW update_namerMrrrr rr r rrrYr[)r\rrr&r'r(r+r,r-r.r/r0r1r2r3rrrrr`s rEupdatez!LangfuseObservationWrapper.updategsN++-- K<<gDO=   ==xdo>  "??$/@    . O ' ' - - -  !%? )& &   6%'+-!%1*""'<!1+)!JJ(0%'+-!%U#:GGIItITTM U U U U U U U U U U U U U U U'#4,} ,    01NOO O O   -B"(8%2$0$    ! //;///s!DD  D cdSr@rArs rEstart_as_current_observationz7LangfuseObservationWrapper.start_as_current_observation 36#rGcdSr@rArs rErz7LangfuseObservationWrapper.start_as_current_observations $9<rGcdSr@rArs rErz7LangfuseObservationWrapper.start_as_current_observations $8;srGcdSr@rArs rErz7LangfuseObservationWrapper.start_as_current_observation  473rGcdSr@rArs rErz7LangfuseObservationWrapper.start_as_current_observationrrGcdSr@rArs rErz7LangfuseObservationWrapper.start_as_current_observation&rrGcdSr@rArs rErz7LangfuseObservationWrapper.start_as_current_observation4 8;srGcdSr@rArs rErz7LangfuseObservationWrapper.start_as_current_observationBrrGcdSr@rArs rErz7LangfuseObservationWrapper.start_as_current_observationPrrG) rrrrrrrrrc`|j||d|j||||||| | | | | |S)a@Create a new child observation and set it as the current observation in a context manager. This is the generic method for creating any type of child observation with context management. It delegates to the client's _create_span_with_parent_context method. Args: name: Name of the observation as_type: Type of observation to create input: Input data for the operation output: Output data from the operation metadata: Additional metadata to associate with the observation version: Version identifier for the code or component level: Importance level of the observation (info, warning, error) status_message: Optional status message for the observation completion_start_time: When the model started generating (for generation types) model: Name/identifier of the AI model used (for generation types) model_parameters: Parameters used for the model (for generation types) usage_details: Token usage information (for generation types) cost_details: Cost information (for generation types) prompt: Associated prompt template (for generation types) Returns: A context manager that yields a new observation of the specified type N)rrr6remote_parent_spanparentr&r'r(r+r,r-r.r/r0r1r2r3)rL _create_span_with_parent_contextrIrs rErz7LangfuseObservationWrapper.start_as_current_observation^sXn$EE#?)"7-'%!F   rGcVt}tj|j5|jj||} dddn #1swxYwYtdt| |j||||j |j ||| |S)aCreate a new Langfuse observation of type 'EVENT'. Args: name: Name of the span (e.g., function or operation name) input: Input data for the operation (can be any JSON-serializable object) output: Output data from the operation (can be any JSON-serializable object) metadata: Additional metadata to associate with the span version: Version identifier for the code or component level: Importance level of the span (info, warning, error) status_message: Optional status message for the span Returns: The LangfuseEvent object Example: ```python event = langfuse.create_event(name="process-event") ``` rNrrrb) rrrrIrLrrr rrRrTrf) r\rrr&r'r(r+r,r-rqrs rE create_eventz'LangfuseObservationWrapper.create_events<II  $T_ 5 5   1>IIiJM                  ' $ 5! - -   c9c%%   s"AAA)rdr")+__name__ __module__ __qualname____doc__rSpanrr rrr!rrrintfloatrrarfrrirlr rrNUMERICBOOLEANrvr{r|r rrrWrrr[rrrrrrrArGrEr"r"@s  & $ $"&%)!%!%%)(,48#:>2637)-'~~~"&~$ ~ ( ~ } ~ ~3-~c]~#~#~ "~! ~ (1~}~ #4X #67!~" S#X/#~$tCJ/0%~&&'~~~~@04   x} 8T    Z A $ $ (((}( ( & (((  (T( #' !%#'(,"&      3-   M)=+@@ A # C= H% 3-     X  #'  %!%#'(,"&      3-   M-}/AA B # C= H% 3-     X (#'-1!%#'(,"&5 5 5 5 UCZ 5 3- 5 M* 5 #5 C=5 H%5 3-5  5 5 5 5 n #' !%#'(,"&      3-   M)=+@@ A # C= H% 3-     X  #'  %!%#'(,"&      3-   M-}/AA B # C= H% 3-     X (#'-1!%#'(,"&5 5 5 5 UCZ 5 3- 5 M* 5 #5 C=5 H%5 3-5  5 5 5 5 v59# $"&383838!3801 38 } 38  383-38 38383838p#   sm !  WW%wx'8'*:MMN  #     0@s@s@@@@6#smWW%wx'8'*:MMN  # @/3TX +DLSM 2## $"&!%%)(,48#:>2637)-cccsmc} c  c 3- c#c "c! c (1c}c#4X #67c S#X/ctCJ/0c&c !c" &#ccccJ $ $"&!%%)(,      }   3- #  " !      X  $ $"&!%%)(,48#:>2637)-!####& # } #  #3-### "#! # (1#}##4X #67# S#X/#tCJ/0# &!#" ####X#& $ $"&!%%)(,    !  }   3- #  " !      X  $ $"&!%%)(,      }   3- #  " !      X  $ $"&!%%)(,    !  }   3- #  " !      X  $ $"&!%%)(, " " " "% " } "  "3- "# " " "!  "  " " "X " $ $"&!%%)(, " " " "% " } "  "3- "# " " "!  "  " " "X " $ $"&!%%)(,48#:>2637)-!""""% " } "  "3-"#" ""! " (1"}"#4X #67" S#X/"tCJ/0" &!"" #"""X"& $ $"&!%%)(, " " " "% " } "  "3- "# " " "!  "  " " "X " $ $"&!%%)(,    !  }   3- #  " !      X "+1# $"&!%%)(,48#:>2637)-!n0n0n0n0( n0 } n0  n03-n0#n0 "n0! n0 (1n0}n0#4X #67n0 S#X/n0tCJ/0n0 &!n0"   #n0n0n0n0` $ $"&!%%)(, 6 6 6 6 6 } 6  63- 6# 6 " 6!  6 ! 0 6 6 6X 6 $ $"&!%%)(,48#:>2637)-!<<<<& < } <  <3-<#< "<! < (1<}<#4X #67< S#X/<tCJ/0< &!<" !!5 6#<<<X<& $ $"&!%%)(,48#:>2637)-!;;;;% ; } ;  ;3-;#; ";! ; (1;};#4X #67; S#X/;tCJ/0; &!;" !!4 5#;;;X;& $ $"&!%%)(, 7 7 7 7! 7 } 7  73- 7# 7 " 7!  7 ! 1 7 7 7X 7 $ $"&!%%)(, 6 6 6 6 6 } 6  63- 6# 6 " 6!  6 ! 0 6 6 6X 6 $ $"&!%%)(, 7 7 7 7! 7 } 7  73- 7# 7 " 7!  7 ! 1 7 7 7X 7 $ $"&!%%)(, ; ; ; ;% ; } ;  ;3- ;# ; " ;!  ; !!4 5 ; ; ;X ; $ $"&!%%)(, ; ; ; ;% ; } ;  ;3- ;# ; " ;!  ; !!4 5 ; ; ;X ; $ $"&!%%)(, ; ; ; ;% ; } ;  ;3- ;# ; " ;!  ; !!4 5 ; ; ;X ;"28# $"&!%%)(,48#:>2637)-!H H H H / H } H  H 3-H #H  "H ! H  (1H }H #4X #67H  S#X/H tCJ/0H &!H $ !  !  %H H H H \ $ $"&!%%)(,3 3 3 3 } 3  3 3- 3 #3  "3 ! 3  3 3 3 3 3 3 rGceZdZdZddddddddddejdddeedeed eed eed eed eed ee deeffdZ xZ S)raStandard span implementation for general operations in Langfuse. This class represents a general-purpose span that can be used to trace any operation in your application. It extends the base LangfuseObservationWrapper with specific methods for creating child spans, generations, and updating span-specific attributes. If possible, use a more specific type for better observability and insights. Nr&r'r(r)r*r+r,r-r4r5rr&r'r(r)r*r+r,r-c `t|d|||||||| |  dS)aInitialize a new LangfuseSpan. Args: otel_span: The OpenTelemetry span to wrap langfuse_client: Reference to the parent Langfuse client input: Input data for the span (any JSON-serializable object) output: Output data from the span (any JSON-serializable object) metadata: Additional metadata to associate with the span environment: The tracing environment release: Release identifier for the application version: Version identifier for the code or component level: Importance level of the span (info, warning, error) status_message: Optional status message for the span r; r4r6r5r&r'r(r)r*r+r,r-Nsuperra r\r4r5r&r'r(r)r*r+r,r- __class__s rErazLangfuseSpan.__init__sQ8 +#)  rG) rrrrrrr rrr!ra __classcell__rs@rErrs $ $"&%)!%!%%)(,( ( ( "&( $ ( } (  ( 3-( c]( #( #(  "( ! ( ( ( ( ( ( ( ( ( ( rGrc"feZdZdZddddddddddddddddejdddeedeed eed eed eed eed ee deedee deedee ee fdee ee fdee eefdeef fdZxZS)ra+Specialized span implementation for AI model generations in Langfuse. This class represents a generation span specifically designed for tracking AI/LLM operations. It extends the base LangfuseObservationWrapper with specialized attributes for model details, token usage, and costs. Nr%r4r5rr&r'r(r)r*r+r,r-r.r/r0r1r2r3cltd||||||||| | | | | |||dS)aInitialize a new LangfuseGeneration span. Args: otel_span: The OpenTelemetry span to wrap langfuse_client: Reference to the parent Langfuse client input: Input data for the generation (e.g., prompts) output: Output from the generation (e.g., completions) metadata: Additional metadata to associate with the generation environment: The tracing environment release: Release identifier for the application version: Version identifier for the model or component level: Importance level of the generation (info, warning, error) status_message: Optional status message for the generation completion_start_time: When the model started generating the response model: Name/identifier of the AI model used (e.g., "gpt-4") model_parameters: Parameters used for the model (e.g., temperature, max_tokens) usage_details: Token usage information (e.g., prompt_tokens, completion_tokens) cost_details: Cost information for the model call prompt: Associated prompt template from Langfuse prompt management r)r6r4r5r&r'r(r)r*r+r,r-r.r/r0r1r2r3Nr)r\r4r5r&r'r(r)r*r+r,r-r.r/r0r1r2r3rs rErazLangfuseGeneration.__init__sdP  +#)"7-'%#      rG)rrrrrrr rrr!rrrrrrrarrs@rErrsx $ $"&%)!%!%%)(,48#:>2637)-%: : : "&: $ : } :  : 3-: c]: #: #:  ": ! :  (1: }: #4X #67:  S#X/!: "tCJ/0#: $&%: : : : : : : : : : rGrceZdZdZddddddddddejdddeedeed eed eed eed eed ee deeffdZ dddddddddddddd deedeedeed eed eed ee deedee deedee ee fdee eefdee eefdeededdfdZxZS)rz4Specialized span implementation for Langfuse Events.Nrr4r5rr&r'r(r)r*r+r,r-c `t|d|||||||| |  dS)aInitialize a new LangfuseEvent span. Args: otel_span: The OpenTelemetry span to wrap langfuse_client: Reference to the parent Langfuse client input: Input data for the event output: Output from the event metadata: Additional metadata to associate with the generation environment: The tracing environment release: Release identifier for the application version: Version identifier for the model or component level: Importance level of the generation (info, warning, error) status_message: Optional status message for the generation r=rNrrs rErazLangfuseEvent.__init__[sQ8 +#)  rGrrrr.r/r0r1r2r3rrdc  .tjd|S)zUpdate is not allowed for LangfuseEvent because events cannot be updated. This method logs a warning and returns self without making changes. Returns: self: Returns the unchanged LangfuseEvent instance zWAttempted to update LangfuseEvent observation. Events cannot be updated after creation.)r r)r\rrr&r'r(r+r,r-r.r/r0r1r2r3rs rErzLangfuseEvent.updates#2  e    rG)rrrrrrr rrr!rarrrrrrrrrs@rErrXs!>> $ $"&%)!%!%%)(,( ( ( "&( $ ( } (  ( 3-( c]( #( #(  "( ! ( ( ( ( ( ( Z## $"&!%%)(,48#:>2637)-sm}    3- # "!  (1}#4X #67 S#X/tCJ/0& !" #rGrc,eZdZdZdeddffd ZxZS)rzLAgent observation for reasoning blocks that act on tools using LLM guidance.rrdNc Dd|d<tjdi|dS)z$Initialize a new LangfuseAgent span.rr6NrArr\rrs rErazLangfuseAgent.__init__0#y""6"""""rGrrrrrrarrs@rErrsQVV############rGrc,eZdZdZdeddffd ZxZS)rzOTool observation representing external tool calls, e.g., calling a weather API.rrdNc Dd|d<tjdi|dS)z#Initialize a new LangfuseTool span.rr6NrArrs rErazLangfuseTool.__init__s0"y""6"""""rGrrs@rErrQYY############rGrc,eZdZdZdeddffd ZxZS)rzcChain observation for connecting LLM application steps, e.g. passing context from retriever to LLM.rrdNc Dd|d<tjdi|dS)z$Initialize a new LangfuseChain span.rr6NrArrs rErazLangfuseChain.__init__rrGrrs@rErrsQmm############rGrc,eZdZdZdeddffd ZxZS)rzVRetriever observation for data retrieval steps, e.g. vector store or database queries.rrdNc Dd|d<tjdi|dS)z(Initialize a new LangfuseRetriever span.rr6NrArrs rErazLangfuseRetriever.__init__0'y""6"""""rGrrs@rErrsQ``############rGrc,eZdZdZdeddffd ZxZS)rzOEmbedding observation for LLM embedding calls, typically used before retrieval.rrdNc Dd|d<tjdi|dS)z(Initialize a new LangfuseEmbedding span.rr6NrArrs rErazLangfuseEmbedding.__init__rrGrrs@rErrrrGrc,eZdZdZdeddffd ZxZS)rzZEvaluator observation for assessing relevance, correctness, or helpfulness of LLM outputs.rrdNc Dd|d<tjdi|dS)z(Initialize a new LangfuseEvaluator span.rr6NrArrs rErazLangfuseEvaluator.__init__rrGrrs@rErrsQdd############rGrc,eZdZdZdeddffd ZxZS)rzRGuardrail observation for protection e.g. against jailbreaks or offensive content.rrdNc Dd|d<tjdi|dS)z(Initialize a new LangfuseGuardrail span.rr6NrArrs rErazLangfuseGuardrail.__init__rrGrrs@rErrsQ\\############rGr) r;rrrrrrrrN)=rrtimertypingrrrrr r r r r opentelemetryrropentelemetry.trace.statusrropentelemetry.util._decoratorrlangfuse.modelrlangfuse._client.clientrtyping_extensionsrlangfuse._client.attributesrrrrlangfuse._client.constantsrrrrr langfuse.apirrlangfuse.loggerr langfuse.typesr!r#r__annotations__r"rrrrrrrrrrrrArGrErsD                         21111199999999AAAAAA''''''1000000(((((( 10000000++++++$$$$$$ IKS$'C"DDEJJJ[ [ [ [ [ [ [ [ |$2 2 2 2 2 -2 2 2 jB B B B B 3B B B JIIIII.IIIX#####.########-########.########2########2########2########2###(&&&&       rG