uPj.$UddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZiZdZeZGd d ZeZeed <y) N)Callable)Path)Final)verbose_proxy_logger)CustomPromptManagement) PromptInfoPromptLiteLLMParams PromptSpecc i} ttjjj}tjj |d}tjj |stjd|Stj|D],}tjj ||}tjj|r|jdrUtjj |d}tjj |sd|} tjd|tj|}t|dr[t|d}t!|t"r?|j%|tjd|t'|j)/tjd t1|t'|j)|S#t*$r"} tj,d || Yd } ~ d } ~ wt.$r"} tj,d || Yd } ~ d } ~ wwxYw#t.$r!} tj,d | Yd } ~ |Sd } ~ wwxYw)ax Get prompt initializers by discovering them from the prompt_integrations directory structure. Scans the integrations directory for subdirectories containing __init__.py files with either prompt_initializer_registry or initialize_prompt functions. Returns: Dict[str, Callable]: A dictionary mapping guardrail types to their initializer functions integrationsz integrations directory not found__z __init__.pyzlitellm.integrations.z&Discovering prompt integrations in: %sprompt_initializer_registryz+Found prompt_initializer_registry in %s: %szCould not import %s: %sNzError processing %s: %sz%Discovered %s prompt initializers: %sz)Error discovering prompt initializers: %s)r__file__parentospathjoinexistsrdebuglistdirisdir startswith importlib import_modulehasattrgetattr isinstancedictupdatelistkeys ImportErrorerror Exceptionlen) discovered_initializers current_dirintegrations_diritem item_path init_file module_pathmoduleregistryes j/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/prompts/prompt_registry.py(get_prompt_initializer_from_integrationsr1s;=2S!(^2299@@ "$'',,{N"Kww~~./ & &'I J* *JJ/0! D %5tI77>>),1$8K $**+SU`a"00=6#@A&v/LMH!(D1/66x@,22I;X\]e]j]j]lXm1! F "" 3S9P5QSWXoXtXtXvSw  #" $**+DkSTU $**+DkSTU  S""#NPQRR ""Ss\BIB/I7BH :I IH+%I+ I7IIII J%JJceZdZdZdZ ddededzdedzfdZdedeee ffdZ dededzfd Z dededzfd Z d ededzfd Z d ede dzfd Zd eddfdZddededzdeefdZy)InMemoryPromptRegistryzM Class that handles adding prompt callbacks to the CallbacksManager. c"i|_ i|_yN)IN_MEMORY_PROMPTSprompt_id_to_custom_prompt)selfs r0__init__zInMemoryPromptRegistry.__init__]s 8: UW' Npromptconfig_file_pathreturnc&ddl}|j}||jvr$tjd|j|S|j |\}}|j j|||j|<||j|<|S)z Initialize a guardrail from a dictionary and add it to the litellm callback manager Returns a Guardrail object if the guardrail is initialized successfully rNz-prompt_id already exists in IN_MEMORY_PROMPTSr;) litellm prompt_idr6rr_build_prompt_callbacklogging_callback_manageradd_litellm_callbackr7)r8r;r<r@rA parsed_promptcustom_prompt_callbacks r0initialize_promptz(InMemoryPromptRegistry.initialize_prompths !++ .. . & &'V W)))4 4040K0KSY0K0Z- -((==>TU-:y)5K'' 2r:c 4|j}tjd|t|tr t di|}n|}|j }| tdtj|}|td||||}t|tstdt|t|j||jxs td|j |j"|j$|j&|j(}||fS) Nzlitellm_params= %szprompt_integration is requiredzUnsupported prompt: z(CustomPromptManagement is required, got config) prompt_type)rAlitellm_params prompt_info created_at updated_atversion environment created_by)rKrrrrr prompt_integration ValueErrorrgetrtyper rArLrrMrNrOrPrQ)r8r;litellm_params_datarKrS initializerrFrEs r0rBz-InMemoryPromptRegistry._build_prompt_callbacks#%+%:%:""#79LM )4 00G3FGN0N$2$E$E  %=> >8<<=OP  34F3GHI I(3NF(K02HI:4@V;W:XY  *&&)**NjX.N((((NN**(( 444r:cddl}|j|\}}|jj|jd}|j j|jd||j j||j j|||j |j<||j|j<|S)Nrr?) r@rBr7poprAr6rCremove_callback_from_all_listsrD)r8r;r@rE new_callbackstale_callbacks r0 reload_promptz$InMemoryPromptRegistry.reload_prompts&*&A&A&A&P# | $ ? ? C CFDTDTVZ [ ""6#3#3T:  %  , , K KN [((==lK3@v//03>;QTX;X> \s\t\!!#PT*!`deh`i!r:r3IN_MEMORY_PROMPT_REGISTRY)rrcollections.abcrpathlibrtypingrlitellm._loggingr-litellm.integrations.custom_prompt_managementr"litellm.types.prompts.init_promptsrr r rr1r3rw__annotations__rRr:r0rs\ $1P !@#FGHN!N!b$:#;5;r: