o 3i* @sdZddlZddlZddlZddlZddlmZmZm Z m Z ddl m Z m Z ddlmZedZdZdZeee eeeeffZeee efZGd d d Zd ed e ed efddZddd ed e eded efddZded efddZGdddZdS)zv Prompt management for PostHog AI SDK. Fetch and compile LLM prompts from PostHog with caching and fallback support. N)AnyDictOptionalUnion) USER_AGENT _get_session)remove_trailing_slashposthogzhttps://us.posthog.comi,c@s"eZdZdZdedefddZdS) CachedPromptzCached prompt with metadata.prompt fetched_atcCs||_||_dSNr r )selfr r r`/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/posthog/ai/prompts.py__init__s zCachedPrompt.__init__N)__name__ __module__ __qualname____doc__strfloatrrrrrr sr nameversionreturncCs||fS)z9Build a cache key for latest or versioned prompt fetches.r)rrrrr _cache_key!srF capitalizercCs6|rdnd}|d|d}|dur|d|S|S)z.Format a prompt reference for logs and errors.Promptr z ""Nz version r)rrrprefixlabelrrr_prompt_reference&s r#datacCs"t|tod|vot|dtS)z5Check if the response is a valid prompt API response.r ) isinstancedictgetr)r$rrr_is_prompt_api_response1s r(c @seZdZdZ dddddddeedeedeedeedeef d d Zdddd d ed eedeedeedef ddZ dede defddZ dddd eedeeddfddZ dd edeedefddZ dS)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', ) # 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', }) ``` N)personal_api_keyproject_api_keyhostdefault_cache_ttl_secondsr r*r+r,r-cCs||pt|_i|_|dur+t|ddpd|_t|ddpd|_tt|ddp&t|_dS|p.d|_|p3d|_t|p9t|_dS)a 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) Nr*api_keyraw_host) DEFAULT_CACHE_TTL_SECONDS_default_cache_ttl_seconds_cachegetattr_personal_api_key_project_api_keyr APP_ENDPOINT_host)rr r*r+r,r-rrrr^s  zPrompts.__init__)cache_ttl_secondsfallbackrrr9r:rrc Cs|dur|n|j}t||}|j|}t}|dur(||j|k} | r(|jSz|||} t} t| | d|j|<| WSt yw} z-t ||} |dur_t d| | |jWYd} ~ S|durrt d| | |WYd} ~ Sd} ~ ww)aV Fetch a prompt by name from the PostHog API. Caching behavior: 1. If cache is fresh, return cached value 2. If fetch fails and cache exists (stale), return stale cache with warning 3. If fetch fails and fallback provided, return fallback with warning 4. If fetch fails with no cache/fallback, raise exception Args: name: The name of the prompt to fetch 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: The prompt string Raises: Exception: If the prompt cannot be fetched and no fallback is available Nrz;[PostHog Prompts] Failed to fetch %s, using stale cache: %sz8[PostHog Prompts] Failed to fetch %s, using fallback: %s) r2rr3r'timer r _fetch_prompt_from_apir Exceptionr#logwarning)rrr9r:rttl cache_keycachednowis_freshr r errorprompt_referencerrrr'sF     z Prompts.getr variablescs&dtjdtffdd }td||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 matchrcs(|d}|vrt|S|dS)Nr)groupr)rH variable_namerGrrreplace_variables   z)Prompts.compile..replace_variablez\{\{([\w.-]+)\}\})reMatchrsub)rr rGrMrrLrcompileszPrompts.compile)rcs~|dur dur tddur|jdS|dur'|jt|ddSfdd|jD}|D] }|j|dq3dS)z Clear cached prompts. Args: name: Specific prompt name to clear. If None, clears all cached prompts. version: Specific prompt version to clear. Requires name. Nz('version' requires 'name' to be providedcsg|] }|dkr|qS)rr).0keyrrr sz'Prompts.clear_cache..) ValueErrorr3clearpopr)rrr keys_to_clearrSrrTr clear_caches  zPrompts.clear_cachec Cs<|jstd|jstdtjj|dd}d|ji}|dur#||d<tj|}|jd|d |}t||}t||d d }d |jt d } t j || dd} | j sz| j dkrbtd|d| j dkrotd|dtd|d| j z| } Wntytd|wt| std|| dS)a Fetch prompt from PostHog API. Endpoint: {host}/api/environments/@current/llm_prompts/name/{encoded_name}/ ?token={encoded_project_api_key}[&version={version}] Auth: Bearer {personal_api_key} Args: name: The name of the prompt to fetch version: Specific prompt version to fetch. If None, fetches the latest Returns: The prompt string Raises: Exception: If the prompt cannot be fetched zz[PostHog Prompts] personal_api_key is required to fetch prompts. Please provide it when initializing the Prompts instance.zy[PostHog Prompts] project_api_key is required to fetch prompts. Please provide it when initializing the Prompts instance.r.)safetokenNrz,/api/environments/@current/llm_prompts/name/z/?TrzBearer ) Authorizationz User-Agent )headerstimeoutiz[PostHog Prompts] z not foundiz$[PostHog Prompts] Access denied for zf. Check that your personal_api_key has the correct permissions and the LLM prompts feature is enabled.z"[PostHog Prompts] Failed to fetch z: HTTP z.[PostHog Prompts] Invalid response format for r )r5r=r6urllibparsequote urlencoder8r#rrr'ok status_codejsonr() rrr encoded_name query_params encoded_queryurlrF prompt_labelr_responser$rrrr<sT         zPrompts._fetch_prompt_from_apir )rrrrrrrintrr'PromptVariablesrQrZr<rrrrr):sZ% ' Q r)) rloggingrNr; urllib.parseratypingrrrrposthog.requestrr posthog.utilsr getLoggerr>r7r1rrnrboolrotuplePromptCacheKeyr rr#r(r)rrrrs6