jKj> dZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZddlmZmZddlmZej*dZdZd Ze eeeeeeffZeee efZe d Z ed Gd dZ!GddZ"dede edefdZ#dddede ededefdZ$de defdZ%GddZ&y)zv Prompt management for PostHog AI SDK. Fetch and compile LLM prompts from PostHog with caching and fallback support. N) dataclass)AnyDictLiteralOptionalUnionoverload) USER_AGENT _get_session)remove_trailing_slashposthogzhttps://us.posthog.comi,)apicache stale_cache code_fallbackT)frozencNeZdZUdZeed<eed<dZeeed<dZ ee ed<y) PromptResultz8Result of a prompt fetch with metadata about its source.sourcepromptNnameversion) __name__ __module__ __qualname____doc__ PromptSource__annotations__strrrrintc/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/ai/prompts.pyrrs,B  KD(3-!GXc]!r"rc(eZdZdZdedededefdZy) CachedPromptzCached prompt with metadata.r fetched_atrrc<||_||_||_||_yNrr&rr)selfrr&rrs r#__init__zCachedPrompt.__init__*s $  r"N)rrrrrfloatr r+r!r"r#r%r%'s&&sS3r"r%rrreturnc ||fS)z9Build a cache key for latest or versioned prompt fetches.r!rrs r# _cache_keyr01s '?r"F capitalizer2c4|rdnd}|d|d}||d|S|S)z.Format a prompt reference for logs and errors.Promptrz ""z version r!)rrr2prefixlabels r#_prompt_referencer86s;$XFhba E '++ Lr"datact|txrbt|jdtxrAt|jdtxr t |jdt uS)z5Check if the response is a valid prompt API response.rrr) isinstancedictgetrtyper )r9s r#_is_prompt_api_responser?As` 4 - txx)3 / - txx' - - )$ % , r"c8eZdZdZ d$dddddddeedeedeedeed eed ef d Z e d d d d dede ddeedeedeede f dZ e d d d d dede ddeedeedeedef dZ e d d d d dedeedeedeedef dZ ddddddedeedeedeedeedeee ff dZ ddddedeedeede fdZdededefdZ d$dddeedeeddfd Zd!ededeeddfd"Z d$dedeedeeeffd#Zy)%Promptsa Fetch and compile LLM prompts from PostHog. Can be initialized with a PostHog client or with direct options. Examples: ```python from posthog import Posthog from posthog.ai.prompts import Prompts # With PostHog client posthog = Posthog('phc_xxx', host='https://us.posthog.com', personal_api_key='phx_xxx') prompts = Prompts(posthog) # Or with direct options (no PostHog client needed) prompts = Prompts( personal_api_key='phx_xxx', project_api_key='phc_xxx', host='https://us.posthog.com', ) # With error tracking: prompt fetch failures are reported to PostHog prompts = Prompts(posthog, capture_errors=True) # Fetch with caching and fallback template = prompts.get('support-system-prompt', fallback='You are a helpful assistant.') # Fetch a specific published version prompt_v1 = prompts.get('support-system-prompt', version=1) # Compile with variables system_prompt = prompts.compile(template, { 'company': 'Acme Corp', 'tier': 'premium', }) ``` NF)personal_api_keyproject_api_keyhostdefault_cache_ttl_secondscapture_errorsr rBrCrDrErFcZ|xst|_i|_d|_||_||_|Pt |ddxsd|_t |ddxsd|_tt |ddxst|_ y|xsd|_|xsd|_t|xst|_ y)ab Initialize Prompts. Args: posthog: PostHog client instance (optional if personal_api_key provided) personal_api_key: Direct personal API key (optional if posthog provided) project_api_key: Direct project API key (optional if posthog provided) host: PostHog host (defaults to app endpoint) default_cache_ttl_seconds: Default cache TTL (defaults to 300) capture_errors: If True and a PostHog client is provided, prompt fetch failures are reported to PostHog error tracking via capture_exception(). FNrBapi_keyraw_host) DEFAULT_CACHE_TTL_SECONDS_default_cache_ttl_seconds_cache_has_warned_deprecation_client_capture_errorsgetattr_personal_api_key_project_api_keyr APP_ENDPOINT_host)r*r rBrCrDrErFs r#r+zPrompts.__init__rs. & B)B ';= ',$ -  %,W6H$%O%USUD "$+GY$E$KD !.T2BlDJ&6%;D "$3$9rD !.t/C|DDJr".)cache_ttl_secondsfallbackrr with_metadataTrVrWrr-cyr(r!r*rrXrVrWrs r#r=z Prompts.getsr"cyr(r!rZs r#r=z Prompts.getsr"cyr(r!)r*rrVrWrs r#r=z Prompts.getsr")rXrVrWrcZ|/|js#d|_tjdtd |j |||}|dur|S|j S#t $rH}t||}|4tjd|||durtd| cYd}~S|cYd}~Sd}~wwxYw) a Fetch a prompt by name from the PostHog API. When ``with_metadata`` is ``True``, returns a :class:`PromptResult` with ``source``, ``name``, and ``version`` metadata. When omitted or ``False``, returns a plain string (deprecated -- will be removed in a future major version). Args: name: The name of the prompt to fetch with_metadata: If True, returns a PromptResult with source info. Omitting this parameter is deprecated. cache_ttl_seconds: Cache TTL in seconds (defaults to instance default) fallback: Fallback prompt to use if fetch fails and no cache available version: Specific prompt version to fetch. If None, fetches the latest version Returns: str if with_metadata is False/omitted, PromptResult if True Raises: Exception: If the prompt cannot be fetched and no fallback is available NTa`[PostHog Prompts] Calling get() without with_metadata=True is deprecated and will be removed in a future major version. Pass with_metadata=True to receive a PromptResult object with source, name, and version metadata. You can pass with_metadata=False to silence this warning, but the plain-string return will still be removed in the next major version.) stacklevelrVrz8[PostHog Prompts] Failed to fetch %s, using fallback: %sr)rr) rNwarningswarnDeprecationWarning _get_internalr Exceptionr8logwarningr) r*rrXrVrWrresulterrorprompt_references r#r=z Prompts.gets@  )E)E+/D ( MM#  ''(97(F$ ==  0w? # N$ !D('xPP  s5A A B*"5B%B*B%B*$B%%B*r`c||n |j}t||}|jj|}t j}|A||j z |k}|r-t d|j|j|jS |j||} t| dt j| d| d|j|<t d| d| d| dS#t$rq} |j| || t||} |Itj!d | | t d |j|j|jcYd} ~ Sd} ~ wwxYw) z Internal method that handles cache + fetch logic, returning full metadata. Does NOT handle the string ``fallback`` option -- the caller handles that. Nr)rrrrrrrr)rr/z;[PostHog Prompts] Failed to fetch %s, using stale cache: %sr)rLr0rMr=timer&rrrr_fetch_prompt_from_apir%re_maybe_capture_errorr8rfrg) r*rrVrttl cache_keycachednowis_freshr9rirjs r#rdzPrompts._get_internalsy!, 00 tW- +iik  f///36H#"!=="NN $ ..tW=D&2H~99;&\Y &DKK " H~&\Y     % %e$ % H0w? ! Q$ $(!=="NN  % s&A C11 E+:A%E&E+%E&&E+r variablescjdtjdtffd }tjd||S)az Replace {{variableName}} placeholders with values. Unmatched variables are left unchanged. Supports variable names with hyphens and dots (e.g., user-id, company.name). Args: prompt: The prompt template string variables: Object containing variable values Returns: The compiled prompt string matchr-cl|jd}|vrt|S|jdS)Nr)groupr)rv variable_namerts r#replace_variablez)Prompts.compile..replace_variableUs6!KKNM )9]344;;q> !r"z\{\{([\w.-]+)\}\})reMatchrsub)r*rrtr{s ` r#compilezPrompts.compileFs1 "BHH " "vv*,sm' '2YY),Y7?}Y Y,37JJ"*3-J c3hJr"rA)'rloggingr|rl urllib.parserra dataclassesrtypingrrrrrr posthog.requestr r posthog.utilsr getLoggerrfrTrKrr r,rrtuplePromptCacheKeyrrr%r0r8r?rAr!r"r#rs  !@@4/g "' sE#sE4"7889sHSM)*EF  $"""S8C=^ >C   6:#$LLr"