o ưi @sUddlmZmZer!ddlmZmZddlmZmZddl m Z ddlm Z ddl m Z dZeeed <dZed ed <d ed dfddZded efddZ     dddZe jjeiZgdZdS)) TYPE_CHECKINGOptional) PromptManagerPromptTemplate)PromptLiteLLMParams PromptSpec)CustomPromptManagement)SupportedPromptIntegrations)DotpromptManagerNglobal_prompt_directoryrglobal_prompt_manager directoryreturncCsddl}||_dS)z Set the global prompt directory for dotprompt files. Args: directory: Path to directory containing .prompt files rN)litellmr )rrr^/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/dotprompt/__init__.pyset_global_prompt_directorys rdotprompt_contentcCs.ddlm}|}||\}}||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_managerrZ_parse_frontmatterstrip)rrZ temp_managerrrrrr'_get_prompt_data_from_dotprompt_contents rlitellm_paramsr prompt_specrr c Cst|dd}t|dd}t|dd}|rtdt|dd}t|dd}|r.|s.|s.t|}z t||||d}|WStyG}z|d}~ww) 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)rrrr)getattr ValueErrorrr Exception) rrrrrrrZdot_prompt_managererrrprompt_initializer.s.      r$)rr rrr r )rrrrrr )typingrrrrrZ"litellm.types.prompts.init_promptsrrZ-litellm.integrations.custom_prompt_managementr r Zdotprompt_managerr r str__annotations__r rdictrr$Z DOT_PROMPTvalueZprompt_initializer_registry__all__rrrrs*     $