tPj UddlmZmZmZerddlmZddlmZmZddl m Z m Z ddlm Z ddl mZdZeedzed <dZeed ed <d ed dfdZded efdZddZe j.j0eiZeed<gdZy)) TYPE_CHECKINGFinalOptional)CustomPromptManagement)PromptLiteLLMParams PromptSpec) PromptManagerPromptTemplate)SupportedPromptIntegrations)DotpromptManagerNglobal_prompt_directoryr global_prompt_manager directoryreturncddl}||_y)z Set the global prompt directory for dotprompt files. Args: directory: Path to directory containing .prompt files rN)litellmr)rrs l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/dotprompt/__init__.pyset_global_prompt_directoryrs&/G#dotprompt_contentcjddlm}|}|j|\}}|j|dS)z Get the prompt data from the dotprompt content. The UI stores prompts under `dotprompt_content` in the database. This function parses the content and returns the prompt data in the format expected by the prompt manager. r )r )contentmetadata)prompt_managerr _parse_frontmatterstrip)rr temp_managerrrs r'_get_prompt_data_from_dotprompt_contentrs9 .(/L$778IJHg}}H ==rc^t|dd}t|dd}t|dd}|r tdt|dd}t|dd}|r|s |s t|}dd lm}|xs ||j xs |j } t |||| } | S#t$r} | d} ~ wwxYw) z2 Initialize a prompt from a .prompt file. prompt_directoryN prompt_data prompt_idzgCannot set prompt_directory when working with prompt_initializer. Needs to be a specific dotprompt file prompt_filerr )strip_version_suffix)r!r"r$r#)getattr ValueErrorrrr%r#r Exception) litellm_params prompt_specr!r"r#r$rr%registration_prompt_iddot_prompt_manageres rprompt_initializerr..s&n6H$O.->K~{DAI u  !EK '~7JDQ[=>OP 4$-$u1EkF[F[1\$u`k`u`u $4-##, % "! s B B,%B''B,prompt_initializer_registry)r r r rrr)r)rr*rrr)typingrrr-litellm.integrations.custom_prompt_managementr"litellm.types.prompts.init_promptsrrrr r r dotprompt_managerr rstr__annotations__rrdictrr. DOT_PROMPTvaluer/__all__rrr;s11TR=J/.2sTz*1:>uXo67> 03 04 0 >s >t > !J **002D&U  r