mhR"dZddlZddlmZmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlm Z ddl!m"Z"m#Z#m$Z$m%Z%Gdde Z&Gdde Z'GddZ(GddeZ)Gdde)Z*Gdde)Z+e e*e+fZ,dS)z@privateN)ABCabstractmethod)AnyDictListOptionalTuple TypedDictUnion)Dataset) DatasetItem) DatasetRun) DatasetStatus)MapValue) Observation)TraceWithFullDetails)CreateDatasetItemRequest)CreateDatasetRunItemRequest)CreateDatasetRequest) ChatMessagePrompt Prompt_Chat Prompt_TextceZdZUeeed<eeed<eeed<eeed<eeed<eeed<eeed<dS) ModelUsageunitinputoutputtotal input_cost output_cost total_costN)__name__ __module__ __qualname__rstr__annotations__intfloat9/usr/local/lib/python3.11/site-packages/langfuse/model.pyrr*s 3- C= SM C=%   r+rc$eZdZUeed<eed<dS)ChatMessageDictrolecontentN)r#r$r%r&r'r*r+r,r.r.4s" III LLLLLr+r.c eZdZdZdZedededee eeeffdZ edede efdZ ed ded ee eefdefd ZdS) TemplateParserz{{z}}r0 start_idxreturncL|tj|}|dkrdS|tj|}|dkrdS||t tjz|}|||t tjzfS)zHReturns (variable_name, start_pos, end_pos) or None if no variable foundN)findr2OPENINGCLOSINGlenstrip)r0r3 var_startvar_end variable_names r,_parse_next_variablez#TemplateParser._parse_next_variable=s LL!7CC ??4,,~5yAA b==4 N233 3g = %'' y'C8N4O4O*OPPr+cg}d}|t|krTt||}|sn6||d|d}|t|kT|S)Nr)r:r2r?append)r0namescurr_idxresults r,find_variable_namesz"TemplateParser.find_variable_namesOsvW%%#88(KKF  LL # # #ayH W%% r+Ndatac||Sg}d}|t|krt||}|s|||dn|\}}}||||||vr3|||t ||ndn|||||}|t|kd|S)Nr)r:r2r?rBr&join)r0rG result_listrDrEr>r<r=s r,compile_templatezTemplateParser.compile_template]s" <N W%%#88(KKF ""7899#566606 -M9g   wx '9: ; ; ;$$""04]0C0OC]+,,,UW""79W+<#=>>>H#W%%&ww{###r+N)r#r$r%r8r9 staticmethodr&r(rr r?rrFrrrLr*r+r,r2r29sGGQQ!$Q %S# & 'QQQ\Q" S T#Y   \ $$#$Xd38n-E$QT$$$\$$$r+r2cJeZdZUeed<eed<eeefed<eeed<eeed<e eed<dde d e fd Z e d eeeeffd Zee d eefd Ze dZe dZedefdZdS)BasePromptClientnameversionconfiglabelstagscommit_messageFprompt is_fallbackc|j|_|j|_|j|_|j|_|j|_|j|_||_dSrM)rQrRrSrTrUrVrX)selfrWrXs r,__init__zBasePromptClient.__init__sHK ~ m m K $3&r+r4c dSrMr*rZkwargss r,compilezBasePromptClient.compile r+cdSrMr*rZs r, variableszBasePromptClient.variabless  r+cdSrMr*rZothers r,__eq__zBasePromptClient.__eq__r`r+cdSrMr*rbs r,get_langchain_promptz%BasePromptClient.get_langchain_promptr`r+r0c.tjdd|S)Nz{{\s*(\w+)\s*}}z{\g<1>})resubr0s r,_get_langchain_prompt_stringz-BasePromptClient._get_langchain_prompt_stringsv(*g>>>r+NF)r#r$r%r&r'r(rrrrrboolr[rr rr_propertyrcrgrirNrnr*r+r,rPrP{sX III LLL cN I s)OOOSM!!!''v'D'''' 5d;.?)?#@   ^  49   ^X   ^   ^ ?c???\???r+rPcpeZdZd dedeffd ZdefdZede efdZ dZ defd Z xZ S) TextPromptClientFrWrXcdt|||j|_dSrMsuperr[rWrZrWrX __class__s r,r[zTextPromptClient.__init__s* ---m r+r4c Bt|j|SrM)r2rLrWr]s r,r_zTextPromptClient.compiles..t{FCCCr+c@t|jS)z5Return all the variable names in the prompt template.)r2rFrWrbs r,rczTextPromptClient.variabless11$+>>>r+ct||jr@|j|jko/|j|jko|j|jko|j|jkSdS)NF) isinstancerxrQrRrWrSres r,rgzTextPromptClient.__eq__sc dEO , ,  UZ'0LEM10K5</0K5</  ur+c ~|r t|j|n|j}||S)afConvert Langfuse prompt into string compatible with Langchain PromptTemplate. This method adapts the mustache-style double curly braces {{variable}} used in Langfuse to the single curly brace {variable} format expected by Langchain. kwargs: Optional keyword arguments to precompile the template string. Variables that match the provided keyword arguments will be precompiled. Remaining variables must then be handled by Langchain's prompt template. Returns: str: The string that can be plugged into Langchain's PromptTemplate. )r2rLrWrn)rZr^rWs r,riz%TextPromptClient.get_langchain_promptsC N + +DK @ @ @  00888r+ro)r#r$r%rrpr[r&r_rqrrcrgri __classcell__rxs@r,rsrss$${$$$$$$$D3DDDD?49???X?   999999999r+rscveZdZd dedeffd ZdeefdZe dee fdZ dZ d Z xZS) ChatPromptClientFrWrXcxt||d|jD|_dS)NcDg|]}t|j|jS)r/r0)r.r/r0).0ps r, z-ChatPromptClient.__init__..s6   @AO ; ; ;   r+rurws r,r[zChatPromptClient.__init__sC ---  EK]    r+r4c *fd|jDS)Nc|g|]8}tt|d|d9S)r0r/)r0r/)r.r2rL)r chat_messager^s r,rz,ChatPromptClient.compile..sa    &77 +V"&)       r+rWr]s `r,r_zChatPromptClient.compiles2    !%     r+c$d|jDS)z:Return all the variable names in the chat prompt template.cXg|]'}t|dD]}|(Srm)r2rF)rrvariables r,rz.ChatPromptClient.variables..sR   *>>|I?VWW       r+rrbs r,rczChatPromptClient.variabless%  $     r+ct||jra|j|jkoP|j|jko@t dt |j|jDo|j|jkSdS)Nc3dK|]+\}}|d|dko|d|dkV,dS)r/r0Nr*)rm1m2s r, z*ChatPromptClient.__eq__..sYBvJ"V*,OI"Y-1Or+F)r|rxrQrRallziprWrSres r,rgzChatPromptClient.__eq__s dEO , ,  UZ'0LEM10"%dk5<"@"@0 K5</ ur+c .fdjDS)aConvert Langfuse prompt into string compatible with Langchain ChatPromptTemplate. It specifically adapts the mustache-style double curly braces {{variable}} used in Langfuse to the single curly brace {variable} format expected by Langchain. kwargs: Optional keyword arguments to precompile the template string. Variables that match the provided keyword arguments will be precompiled. Remaining variables must then be handled by Langchain's prompt template. Returns: List of messages in the format expected by Langchain's ChatPromptTemplate: (role, content) tuple. c g|]H}|dr!t|dn|dfISr)rnr2rL)rmsgr^rZs r,rz9ChatPromptClient.get_langchain_prompt..so    F 11(N33C NFKKKY     r+rr]s``r,riz%ChatPromptClient.get_langchain_prompts8      {    r+ro)r#r$r%rrpr[rr.r_rqr&rcrgrir~rs@r,rrs  {        4#8      49   X           r+r)-__doc__rkabcrrtypingrrrrr r r ,langfuse.api.resources.commons.types.datasetr 1langfuse.api.resources.commons.types.dataset_itemr 0langfuse.api.resources.commons.types.dataset_runr3langfuse.api.resources.commons.types.dataset_statusr.langfuse.api.resources.commons.types.map_valuer0langfuse.api.resources.commons.types.observationrrs ########EEEEEEEEEEEEEEEEEE JIIIIIHGGGGGDCCCCCHHHHHH  YXXXXXXXXXXX        i ?$?$?$?$?$?$?$?$D$?$?$?$?$?s$?$?$?N+9+9+9+9+9'+9+9+9\@ @ @ @ @ '@ @ @ F%'778 r+