o i_@sddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd l m Z dd l mZdd l mZed dGdddZdS)) dataclass)replace)Any)Literal)Optional)Union)extract_template) render_chat)safe_substitute)Message)Prompt)PromptFallbackT)frozenc@seZdZUdZeed<eed<eeed<eded<eee e fed<dZ eeed <dZ eeed <d ed eee e ffd dZ d ed efddZd efddZd eeeffddZedeeefd dfddZdedd dfddZe ddeded dfddZdS) ManagedPrompta An immutable prompt template retrieved from the Datadog Prompt Registry. STABLE API: - format(**vars) -> str | list[dict] - to_annotation_dict(**vars) -> dict[str, Any] INTERNAL (may change): - All fields (id, version, label, source, template) idversionlabel)registrycachefallbacksourcetemplateN_uuid _version_uuid variablesreturncKs$t|jtr t|j|St|j|S)a Render the template with variables. For text templates, returns the rendered string. For chat templates, returns the rendered message list. Uses safe substitution: missing variables are left as placeholders. Args: **variables: Template variables to substitute Returns: str (for text templates) or list[Message] (for chat templates) ) isinstancerstrr r )selfrrR/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/llmobs/_prompts/prompt.pyformat%s   zManagedPrompt.formatcKsr|j|jd}|r ||d<|jr|j|d<|jr|j|d<|jr%|j|d<t|jtr2|j|d<|S|j|d<|S)aL Convert to the format expected by annotation_context. Args: **variables: Variable names and values used in the prompt template. Pass the same kwargs here that you use with format(). Example: prompt.to_annotation_dict(name="Alice", topic="weather") )rrrr prompt_uuidprompt_version_uuidr chat_template)rrrrrrrr)rrresultrrr to_annotation_dict8s        z ManagedPrompt.to_annotation_dictc Cs&d|jd|jd|jd|jd S)NzManagedPrompt(id=z , version=z, label=z , source=))rrrrrrrr __repr__Ws&zManagedPrompt.__repr__cCs"|j|j|j|j|j|j|jdS)z6Serialize to a JSON-compatible dict for cache storage.rrrrrrrr*r(rrr _serializeZszManagedPrompt._serializedatac Cs:||d|d|d|dd|d|d|dd S) z&Deserialize from a cache storage dict.rrrrrrrrr*)get)clsr,rrr _deserializefs zManagedPrompt._deserializecCs|j|kr|St||dS)zCCreate a copy with a different source. Used internally for caching.)r)rr)rrrrr _with_sourcess  zManagedPrompt._with_source prompt_idrcCsZd}d}|dur$t|r|n|}t|tr"t|}|dp d}n|}|||dd|dS)a'Create a ManagedPrompt from a fallback value. Args: prompt_id: The prompt identifier. fallback: A string, message list, Prompt dict, or callable returning any of those. Returns: A ManagedPrompt with source="fallback" and label=None. rNr)rrrrr)callablerdictrr-)r.r1rrrvaluerrr from_fallbackys zManagedPrompt.from_fallback)N)__name__ __module__ __qualname____doc__r__annotations__rrrlistr rrr!rr r&r)r4r+ classmethodr/r0r r6rrrr rs2      rN) dataclassesrrtypingrrrrddtrace.llmobs._prompts.utilsrr r ddtrace.llmobs.typesr r r rrrrr s