jUddlmZmZerddlmZmZddlmZmZddl m Z ddlm Z ddl m Z dZeeed <dZed ed <d ed dfdZ 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 /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/litellm/integrations/dotprompt/__init__.pyset_global_prompt_directoryrs&/G#ct|dd}t|dd}t|dd}|r tdt|dd} 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_file)rrrr)getattr ValueErrorr Exception)litellm_params prompt_specrrrrdot_prompt_manageres rprompt_initializerr"s ~/A4H.->K T:I u  .->K --##  "! sA A$AA$)rr rrr r)rrrr rr )typingrrprompt_managerrr"litellm.types.prompts.init_promptsrr -litellm.integrations.custom_prompt_managementr r dotprompt_managerr r str__annotations__rrr" DOT_PROMPTvalueprompt_initializer_registry__all__rrr/s**=RTJ/*.#-37x07 03 04 0)8D> **002D  r