i UddlmZmZerddlmZmZddlmZmZddl m Z ddlm Z ddl m Z dZeeed <dZed ed <d ed dfdZded efdZ ddZe j,j.eiZgdZy)) TYPE_CHECKINGOptional) PromptManagerPromptTemplate)PromptLiteLLMParams PromptSpec)CustomPromptManagement)SupportedPromptIntegrations)DotpromptManagerNglobal_prompt_directoryrglobal_prompt_manager directoryreturncddl}||_y)z Set the global prompt directory for dotprompt files. Args: directory: Path to directory containing .prompt files rN)litellmr )rrs x/Users/manta/Documents/Projects/TheRoad-I1/.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|} 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_filer)r r!r#r")getattr ValueErrorrr Exception) litellm_params prompt_specr r!r"r#rdot_prompt_manageres rprompt_initializerr+-s ~/A4H.->K T:I u  .->K 0CTJ[=>OP  --##  "! s!A22 B;A==B)rr rrr r)r'rr(r rr )typingrrrrr"litellm.types.prompts.init_promptsrr -litellm.integrations.custom_prompt_managementr r dotprompt_managerr r str__annotations__rrdictrr+ DOT_PROMPTvalueprompt_initializer_registry__all__rrr8s**=RTJ/*.#-37x07 03 04 0 >s >t > )8DF **002D  r