o ưiS@sddlZddlZddlmZddlmZmZmZddlm Z ddl m Z ddl m Z mZmZiZddZeZGd d d ZeZdS) N)Path)CallableDictOptional)verbose_proxy_logger)CustomPromptManagement) PromptInfoPromptLiteLLMParams PromptSpecc Csi}zttjjj}tj|d}tj|std|WSt |D]}tj||}tj |r8| dr9q$tj|d}tj|sGq$d|}z2td|t |}t|dr}t|d}t|tr}||td|d t|Wq$ty} ztd |d | WYd } ~ q$d } ~ wty} ztd |d | WYd } ~ q$d } ~ wwtd t|dt|W|Sty} ztd| WYd } ~ |Sd } ~ ww)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 Z integrationsz integrations directory not found__z __init__.pyzlitellm.integrations.z$Discovering prompt integrations in: prompt_initializer_registryz%Found prompt_initializer_registry in z: zCould not import NzError processing z Discovered z prompt initializers: z'Error discovering prompt initializers: )r__file__parentospathjoinexistsrdebuglistdirisdir startswith importlib import_modulehasattrgetattr isinstancedictupdatelistkeys ImportErrorerror Exceptionlen) Zdiscovered_initializersZ current_dirZintegrations_diritemZ item_pathZ init_file module_pathmoduleregistryer)\/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/prompts/prompt_registry.py(get_prompt_initializer_from_integrationss\            r+c@s|eZdZdZddZ ddedeedeefdd Zd edeefd d Z d edee fd dZ dede efddZ dS)InMemoryPromptRegistryzM Class that handles adding prompt callbacks to the CallbacksManager. cCsi|_ i|_dSN)IN_MEMORY_PROMPTSprompt_id_to_custom_prompt)selfr)r)r*__init__^s zInMemoryPromptRegistry.__init__Npromptconfig_file_pathreturnc Csddl}|j}||jvrtd|j|Sd}|j}td|t|tr.td i|}n|}|j }|dur;t dt |} | r\| ||}t|t sUt dt||j|nt d|t|||jpmtdd |j|jd } | |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_PROMPTSzlitellm_params= %szprompt_integration is requiredz(CustomPromptManagement is required, got zUnsupported prompt: config)Z prompt_type) prompt_idlitellm_params prompt_info created_at updated_atr))litellmr6r.rrr7rrr prompt_integration ValueErrorr getrtypeZlogging_callback_managerZadd_litellm_callbackr r8rr9r:r/) r0r2r3r;r6Zcustom_prompt_callbackZlitellm_params_datar7r<Z initializerZ parsed_promptr)r)r*initialize_promptksB           z(InMemoryPromptRegistry.initialize_promptr6cC |j|S)z4 Get a prompt by its ID from memory )r.r>r0r6r)r)r*get_prompt_by_ids z'InMemoryPromptRegistry.get_prompt_by_idcCrA)z= Get a prompt callback by its ID from memory )r/r>rBr)r)r*get_prompt_callback_by_ids z0InMemoryPromptRegistry.get_prompt_callback_by_idbase_prompt_idcsNddlmfdd|jD}|D]}|j|=||jvr$|j|=q|S)z Delete all prompts matching the given base prompt ID from memory. Args: base_prompt_id: The base prompt ID (without version suffix) Returns: List of prompt IDs that were deleted r)get_base_prompt_idcsg|] }|dkr|qS))r6r)).0pidrErFr)r* s zDInMemoryPromptRegistry.delete_prompts_by_base_id..)Z&litellm.proxy.prompts.prompt_endpointsrFr.rr/)r0rEZprompts_to_deleterHr)rIr*delete_prompts_by_base_ids  z0InMemoryPromptRegistry.delete_prompts_by_base_idr-)__name__ __module__ __qualname____doc__r1r rstrr@rCrrDrrKr)r)r)r*r,Ys$ 8 r,)rrpathlibrtypingrrrZlitellm._loggingrZ-litellm.integrations.custom_prompt_managementrZ"litellm.types.prompts.init_promptsrr r r r+r,ZIN_MEMORY_PROMPT_REGISTRYr)r)r)r*s   E r