j"DdZddlZddlmZmZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZmZmZmZmZmZmZmZmZmZmZmZddlmZGdde ZGd d e ZGd d e Z Gd de Z!ee e!fZ"GddZ#GddeZ$Gdde$Z%Gdde$Z&ee%e&fZ'dS)z@privateN)ABCabstractmethod) AnyDictListLiteralOptionalSequenceTuple TypedDictUnion) CreateDatasetItemRequestCreateDatasetRequestDataset DatasetItem DatasetRun DatasetStatusMapValue ObservationPrompt Prompt_Chat Prompt_TextTraceWithFullDetails)langfuse_loggerc$eZdZUeed<eed<dS)ChatMessageDictrolecontentN__name__ __module__ __qualname__str__annotations__X/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/langfuse/model.pyrr" III LLLLLr&rc$eZdZUeed<eed<dS)ChatMessagePlaceholderDictrrNrr%r&r'r*r*r(r&r*c:eZdZUeded<eed<eed<dS)'ChatMessageWithPlaceholdersDict_MessagemessagetyperrNr r!r"rr$r#r%r&r'r,r,"s3 )  III LLLLLr&r,c0eZdZUeded<eed<dS)+ChatMessageWithPlaceholdersDict_Placeholder placeholderr.nameNr/r%r&r'r1r1(s* -    IIIIIr&r1c eZdZdZdZedededee eeeffdZ edede efdZ ed ded ee eefdefd ZdS) TemplateParser{{}}r start_idxreturncL|tj|}|dkrdS|tj|}|dkrdS||t tjz|}|||t tjzfS)zHReturns (variable_name, start_pos, end_pos) or None if no variable foundN)findr5OPENINGCLOSINGlenstrip)rr8 var_startvar_end variable_names r'_parse_next_variablez#TemplateParser._parse_next_variable7s LL!7CC ??4,,~5yAA b==4 N233 3g = %'' y'C8N4O4O*OPPr&cg}d}|t|krTt||}|sn6||d|d}|t|kT|S)Nr)r?r5rDappend)rnamescurr_idxresults r'find_variable_namesz"TemplateParser.find_variable_namesIsvW%%#88(KKF  LL # # #ayH W%% r&Ndatac||Sg}d}|t|krt||}|s|||dn|\}}}||||||vr3|||t ||ndn|||||}|t|kd|S)Nr)r?r5rDrGr#join)rrL result_listrIrJrCrArBs r'compile_templatezTemplateParser.compile_templateWs" <N W%%#88(KKF ""7899#566606 -M9g   wx '9: ; ; ;$$""04]0C0OC]+,,,UW""79W+<#=>>>H#W%%&ww{###r&N)r r!r"r=r> staticmethodr#intr r rDrrKrrrQr%r&r'r5r53sGGQQ!$Q %S# & 'QQQ\Q" S T#Y   \ $$#$Xd38n-E$QT$$$\$$$r&r5c eZdZUeed<eed<eeefed<eeed<eeed<e eed<dde d e fd Z e d eeefd eeeeeeefeefffd Zee d eefdZe ded e fdZe d efdZeded efdZeded efdZdS)BasePromptClientr3versionconfiglabelstagscommit_messageFprompt is_fallbackc|j|_|j|_|j|_|j|_|j|_|j|_||_dSrR)r3rWrXrYrZr[r])selfr\r]s r'__init__zBasePromptClient.__init__}sHK ~ m m K $3&r&kwargsr9c dSrRr%r_ras r'compilezBasePromptClient.compiles  r&cdSrRr%r_s r' variableszBasePromptClient.variabless  r&othercdSrRr%r_rhs r'__eq__zBasePromptClient.__eq__ r&cdSrRr%rfs r'get_langchain_promptz%BasePromptClient.get_langchain_promptrlr&rcbt|}tjdd|S)Nz{{\s*(\w+)\s*}}z{\g<1>})rV_escape_json_for_langchainresub)rjson_escaped_contents r'_get_langchain_prompt_stringz-BasePromptClient._get_langchain_prompt_strings,/JJ7SSv(*6JKKKr&textc(g}g}dt|}}||krb||}|dkr|dz|kr*||dzdkr|d|dz }H|dz}||kr?||r%|dz }||kr||%||ko ||dv}||rdnd|||dz }|dkrl|dz|kr+||dzdkr|d|dz }|r|nd }||rdnd|dz }H|||dz }||kbd |S) uJEscapes every curly-brace that is part of a JSON object by doubling it. A curly brace is considered “JSON-related” when, after skipping any immediate whitespace, the next non-whitespace character is a single or double quote. Braces that are already doubled (e.g. {{variable}} placeholders) are left untouched. Parameters ---------- text : str The input string that may contain JSON snippets. Returns: ------- str The string with JSON-related braces doubled. r{r6rF>"'}r7FrN)r?rGisspacepoprO)ruoutstackinchjis_jsons r'rpz+BasePromptClient._escape_json_for_langchains*#d))1!eeaBSyyq5199a!e!3!3JJt$$$FAE!eeQ 1 1eFA!eeQ 1 1ea%9DGz$9 7344444 W%%%Qsq5199a!e!3!3JJt$$$FA).9%))+++E 7344444Q JJrNNN FAK!eeNwws||r&NF)r r!r"r#r$rTrrrr rboolr`rr r rr1rdpropertyrgobjectrkrnrSrtrpr%r&r'rVrVus III LLL cN I s)OOOSM!!!''v'D''''  c3h    S#X;=        ^   49   ^X  F t   ^  c   ^ LcLcLLL\L ?????\???r&rVceZdZd dedeffd ZdeeefdefdZ e de efdZ d e defd Zdeeefdefd ZxZS) TextPromptClientFr\r]cdt|||j|_dSrR)superr`r\)r_r\r] __class__s r'r`zTextPromptClient.__init__s* ---m r&rar9c Bt|j|SrR)r5rQr\rcs r'rdzTextPromptClient.compiles..t{FCCCr&c@t|jS)z5Return all the variable names in the prompt template.)r5rKr\rfs r'rgzTextPromptClient.variabless11$+>>>r&rhct||jr@|j|jko/|j|jko|j|jko|j|jkSdS)NF) isinstancerr3rWr\rXrjs r'rkzTextPromptClient.__eq__sc eT^ , ,  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. )r5rQr\rt)r_rar\s r'rnz%TextPromptClient.get_langchain_promptsC N + +DK @ @ @  00888r&r)r r!r"rrr`r r#rrdrrrgrrkrn __classcell__rs@r'rrs$${$$$$$$$Dc3hDCDDDD?49???X? F t    9U38_999999999r&rceZdZd dedeffd Zdeeefde ee eefe e ffdZ edeefdZd edefd Zdeeefdeeeeefeffd ZxZS) ChatPromptClientFr\r]ct||g|_|jD]}t|drJt|dr:|jdkr/|jt d|j\t|drDt|dr4|jtd|j |j dS) Nr.r3r2)r.r3rrr-)r.rr) rr`r\hasattrr.rGr1r3r,rr)r_r\r]prs r'r`zChatPromptClient.__init__s ---=?   Aq&!! ga&8&8 QV}=T=T ""?*V F## 9(=(=  "";&V !   r&rar9c Bg}g}|jD]}|ddkrL|}|t|dt|d|[|ddkr|d}||vrP||}t |t r|D]}t |trpt|} |dd| d<t|dd || d<|| |tdt|d |d } tj | Ӑi|tdt|d |d t|} tj | ||||d|rd |d} tj | |S)aCompile the prompt with placeholders and variables. Args: **kwargs: Can contain both placeholder values (list of chat messages) and variable values. Placeholders are resolved first, then variables are substituted. Returns: List of compiled chat messages as plain dictionaries, with unresolved placeholders kept as-is. r.r-rr)rrr2r3 NOT_GIVENrNz Placeholder 'z^' should contain a list of chat messages with 'role' and 'content' fields. Appended as string.z,' must contain a list of chat messages, got z Placeholders zE have not been resolved. Pass them as keyword arguments to compile().) r\rGrr5rQrlistdictgetr#rwarningr.) r_racompiled_messagesunresolved_placeholders chat_message message_objplaceholder_nameplaceholder_valuemsg compiled_msgno_role_content_in_placeholderplaceholder_not_a_listunresolved_placeholders_messages r'rdzChatPromptClient.compile0s.  VX K4 I4 ILF#y00* !((#(0 . ? ?' 2"!!f%66#/#7 #v--(./?(@%!"3T::H#4XXC)#t44X/3Cyy 7:wwv{7S7S V 4$2$C$C(+ 2(>(>(.%&%&!-Y 7 !2 8 8 F F F F 1 8 8$3-8034E0F0F%&%&%&!"!"!" BBQaBBBBBB > / 78V W W W W+X.*00+%0(+,=(>(> 2YAQ2Y2Y@DEV@W@W2Y2Y.'/0FGGGG&,,\:::+22<3GHHH " E/]>U/]/]/] +  #$C D D D  r&cg}|jD]A}|ddkr3|t|dB|S)z:Return all the variable names in the chat prompt template.r.r-r)r\extendr5rK)r_rgrs r'rgzChatPromptClient.variablessa  K  LF#y00  "66|I7NOOr&rhcFt||jr|j|jkoz|j|jkojt |jt |jko@t dt|j|jDo|j|jkSdS)Nc3K|]m\}}|ddko/|ddko#|d|dko|d|dkp)|ddko|ddko|d|dkVndS)r.r-rrr2r3Nr%).0m1m2s r' z*ChatPromptClient.__eq__..sB6 i/;vJ)3;vJ"V*4;yMR ]: 6 m35vJ-75vJ"V*4r&F) rrr3rWr?r\allziprXrjs r'rkzChatPromptClient.__eq__s eT^ , ,  UZ'0LEM10 $$EL(9(990#&dk5<"@"@0(K5</+ 0ur&c |jd i|}g}|D]}t|trcd|vr_|ddkrS|d} ddlm}|||V#t $r}d}t ||d}~wwxYwt|tr>d |vr:d |vr6||d ||d f|S) 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. Placeholders are filled-in from kwargs and unresolved placeholders are returned as Langchain MessagesPlaceholder. 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. Can also contain placeholders (list of chat messages) which will be resolved prior to variable compilation. Returns: List of messages in the format expected by Langchain's ChatPromptTemplate: (role, content) tuples for regular messages or MessagesPlaceholder objects for unresolved placeholders. r.r2r3r)MessagesPlaceholder)rCzVlangchain_core is required to use get_langchain_prompt() with unresolved placeholders.Nrrr%)rdrrlangchain_core.prompts.chatrrG ImportErrorrt) r_rarlangchain_messagesrrre import_errors r'rnz%ChatPromptClient.get_langchain_promptsG&)DL22622@B$  C#t$$ 33v;-;W;W#&v; ;OOOOOO&--++:JKKK#;;;#{L%l33:;c4((Vs]]yC?O?O&--K ==c)nMM"!s%A(( B 2BB r)r r!r"rrr`r r#rr rrr1rdrrrgrrkr rnrrs@r'rrs5{,T!S/T!  cNO-X X  T!T!T!T!l 49   X Ft8,"c3h," eE#s(OS() *,",",",",",",","r&r)(__doc__rqabcrrtypingrrrrr r r r r langfuse.apirrrrrrrrrrrrlangfuse.loggerrrr*r,r1ChatMessageWithPlaceholdersDictr5rVrr PromptClientr%r&r'rs ########XXXXXXXXXXXXXXXXXXXXXX                            ,+++++i  i ) #(+/1# ?$?$?$?$?$?$?$?$Dsssssssssl+9+9+9+9+9'+9+9+9\A"A"A"A"A"'A"A"A"H%'778 r&