jddlZddlZddlZddlmZmZmZmZmZddl m Z ddl m Z ddlmZddlm ZddlmZdd lmZej(eZGd d ej.Zy) N)AnyDictOptionalUnionList)override)types)prompt_template)client) base_promptceZdZdZdej j dddddfdededee ee fdej de d eed eed eed ee ed dfdZ ddZeed efdZed efdZeed eefdZeed efdZeed ee ee ffdZeed ej fdZeed eefdZeed eefdZeed eefdZeed ee efdZede d eee e ee fffdZed e ee ffdZededej@d dfdZ!y) Promptz] Prompt class represents a prompt with a name, prompt text/template and commit hash. NTnamepromptmetadatatypevalidate_placeholdersid descriptionchange_descriptiontagsreturnc tj||||_||_||_||_||_||_||_| rtj| ng|_ |jy)a Initializes a new instance of the class with the given parameters. Creates a new text prompt using the opik client and sets the initial state of the instance attributes based on the created prompt. Parameters: name: The name for the prompt. prompt: The template for the prompt. metadata: Optional metadata for the prompt. type: The template type (MUSTACHE or JINJA2). validate_placeholders: Whether to validate template placeholders. id: Optional unique identifier (UUID) for the prompt. description: Optional description of the prompt (up to 255 characters). change_description: Optional description of changes in this version. tags: Optional list of tags to associate with the prompt. Raises: PromptTemplateStructureMismatch: If a chat prompt with the same name already exists (template structure is immutable). )templaterrN) r PromptTemplate _template_name _metadata_type_id _description_change_descriptioncopy_tags_sync_with_backend) selfrrrrrrrrrs |/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/api_objects/prompt/text/prompt.py__init__zPrompt.__init__sk>)77$>S  ! '#5 (,TYYt_"  !c  ddlm}|j}tj|j }|j |j|jj|j|j|j|j|j|j}|j |_|j$|_|j(|_|j(|_ |j,|_|j.|_y)Nr) opik_client)rrrrrrrr)opik.api_objectsr-get_client_cached prompt_client PromptClient rest_client create_promptrrtextr r!r"r#r$r&commit_commit prompt_id__internal_api__prompt_id__r__internal_api__version_id__rr)r(r- opik_client_prompt_client_prompt_versions r)r'zPrompt._sync_with_backend@s0"446 &33L4L4LM'55>>&&^^xx))#776  &,, +9+C+C(,:,=,=)!$$#1#D#D #(( r+c|jS)zThe name of the prompt.)rr(s r)rz Prompt.nameXzzr+c,t|jS)z"The latest template of the prompt.)strrr>s r)rz Prompt.prompt^s4>>""r+c|jS)zThe commit hash of the prompt.)r6r>s r)r5z Prompt.commitcs||r+c|jS)z,The unique identifier of the prompt version.)r9r>s r) version_idzPrompt.version_idis000r+c@tj|jS)z2The metadata dictionary associated with the prompt)r%deepcopyr r>s r)rzPrompt.metadataos}}T^^,,r+c|jS)zThe prompt type of the prompt.)r!r>s r)rz Prompt.typeur?r+c|jS)z+The unique identifier (UUID) of the prompt.)r"r>s r)rz Prompt.id{s xxr+c|jS)zThe description of the prompt.)r#r>s r)rzPrompt.descriptions   r+c|jS)z+The description of changes in this version.)r$r>s r)rzPrompt.change_descriptions'''r+c\|jrtj|jSgS)z,The list of tags associated with the prompt.)r&r%r>s r)rz Prompt.tagss")- tyy$::r+kwargsc r|jduxr>|jjddk(xr|jjddk(}|jjdi|}|r t j |S|S#tj $rtjd|d|cYSwxYw) a Replaces placeholders in the template with provided keyword arguments. Args: **kwargs: Arbitrary keyword arguments where the key represents the placeholder in the template and the value is the value to replace the placeholder with. Returns: A string with all placeholders replaced by their corresponding values from kwargs. N created_fromopik_uir messages_jsonzFailed to parse JSON string: z@. Make sure chat prompt is valid JSON. Returning the raw string.) r getrformatjsonloadsJSONDecodeErrorLOGGERerror)r(rLis_playground_chat_promptformatted_strings r)rSz Prompt.formats NN$ & >"">2i? >""6*o= " 14>>00:6: $ (zz"233  '' ( 34D3EFFG('  (s.B.B65B6c(|jdd|jid}|j|j|d<|j|j|dd<|j|j|dd<|j |j |dd<|S)z Convert the prompt to an info dictionary for serialization. Returns: Dictionary containing prompt metadata and version information. r4r)rtemplate_structureversionrr]r5r)rrr8r5r9r )r( info_dicts r)__internal_api__to_info_dict__z%Prompt.__internal_api__to_info_dict__sII"(DKK%   + + 7">>IdO ;; "-1[[Ii  *  , , 8)-)J)JIi  & >> %/3~~Ii  ,r+r<cb|j|}|j|_|j|_||_t j|jtj|jxstjj|_ |j|_|j |_|j|_|j|_d|_|j*|_|j.r&t1j0|j.|_|Sg|_|S)N)rr)__new__rr9r7r8rr rr prompt_types PromptTyperMUSTACHErr5r6rr r!r"r#rr$rr%r&)clsrr<rs r)from_fern_prompt_versionzPrompt.from_fern_prompt_versionsS!.<.?.?+-;-E-E* *99#,,(()<)<=0&&//  (..)22%** #&&  &4%F%F"9G9L9Ltyy!4!45  SU  r+)rN)"__name__ __module__ __qualname____doc__rbrcrdrArrrboolrr*r'propertyrrrr5rDrrrrrrrrSr_ classmethodrest_api_typesPromptVersionDetailrfrQr+r)rrs.2(4(?(?(H(H&* %),0$(*"*"*"4S>* *" %% *" $ *" SM*"c]*"%SM*"tCy!*" *"X)0 c###   1C11 -(4S>2-- l-- HSM !Xc]!! (HSM(( ;htCy);; s uS$tCH~2F-F'G  :S#X:'::  r+r)r%rTloggingtypingrrrrrtyping_extensionsr opik.rest_apir rnr rbr r0r getLoggerrgrW BasePromptrrQr+r)rwsL 33&1$&   8 $[[ # #[r+