o ưiS@sdZddlmZmZmZmZmZmZmZddl m Z m Z m Z ddl mZer-ddlmZneZddlmZmZddlmZddlmZdd lmZd d lmZGd d d ZGdddZGdddeZdS)z BitBucket prompt manager that integrates with LiteLLM's prompt management system. Fetches .prompt files from BitBucket repositories and provides team-based access control. ) TYPE_CHECKINGAnyDictListOptionalTupleUnion) DictLoader Environmentselect_autoescape)CustomPromptManagement)Logging)PromptManagementBasePromptManagementClient)AllMessageValues) PromptSpec)StandardCallbackDynamicParams)BitBucketClientc @sBeZdZdZ d dededeeefdeefddZd d Z dS) BitBucketPromptTemplatez= Represents a prompt template loaded from BitBucket. N template_idcontentmetadatamodelcCsh||_||_||_|p|d|_|d|_|d|_|didi|_dd|D|_ dS)Nr temperature max_tokensinputZschemacSsi|] \}}|dvr||qS))rrr).0kvrrn/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/bitbucket/bitbucket_prompt_manager.py .sz4BitBucketPromptTemplate.__init__..) rrrgetrrrZ input_schemaitemsoptional_params)selfrrrrrrr!__init__ s   z BitBucketPromptTemplate.__init__cCsd|jd|jdS)NzBitBucketPromptTemplate(id='z ', model='z'))rrr&rrr!__repr__2sz BitBucketPromptTemplate.__repr__N) __name__ __module__ __qualname____doc__strrrrr'r)rrrr!rs   rc@seZdZdZ ddeeefdeefddZdeddfdd Z d edede fd d Z d edeeeffddZ ddedeeeefdefddZ dedee fddZdeefddZdS)BitBucketTemplateManagerap Manager for loading and rendering .prompt files from BitBucket repositories. Supports: - Fetching .prompt files from BitBucket repositories - Team-based access control through BitBucket permissions - YAML frontmatter for metadata - Handlebars-style templating (using Jinja2) - Input/output schema validation - Model configuration Nbitbucket_config prompt_idc Cs\||_||_i|_t||_ttitddgddddddd |_|jr,| |jdSdS) Nhtmlxmlz{{z}}z{%z%}z{#z#})loaderZ autoescapeZvariable_start_stringZvariable_end_stringZblock_start_stringZblock_end_stringZcomment_start_stringZcomment_end_string) r1r2promptsrbitbucket_clientr r r jinja_env_load_prompt_from_bitbucketr&r1r2rrr!r'Cs"   z!BitBucketTemplateManager.__init__returnc Csfz|j|d}|r|||}||j|<WdSWdSty2}z td|d|d}~ww)z,Load a specific .prompt file from BitBucket.z.promptzFailed to load prompt 'z' from BitBucket: N)r7get_file_content_parse_prompt_filer6 Exception)r&r2prompt_contenttemplateerrr!r9]s z4BitBucketTemplateManager._load_prompt_from_bitbucketrcCs|dr#|dd}t|dkr|d}|d}n d}|}nd}|}i}|rQz ddl}||p6i}WntyF||}Yn tyPi}Ynwt |||dS) z?Parse a .prompt file content and extract metadata and template.z---rrN)rrr) startswithsplitlenstripyamlZ safe_load ImportError_parse_yaml_basicr>r)r&rr2partsZfrontmatter_strZtemplate_contentrrIrrr!r=ks0      z+BitBucketTemplateManager._parse_prompt_fileyaml_strcCsi}|dD]O}|}d|vrV|dsV|dd\}}|}|}|dvr5|dk||<q|r@t|||<q|ddrOt|||<q|d ||<q|S) z@Basic YAML parser for simple cases when PyYAML is not available. :#r)truefalserQ.rDz"')rFrHrElowerisdigitintreplacefloat)r&rMresultlinekeyvaluerrr!rKs  z*BitBucketTemplateManager._parse_yaml_basicr variablescCsF||jvr td|d|j|}|j|j}|jdi|p iS)z+Render a template with the given variables.z Template ' ' not foundNr)r6 ValueErrorr8Z from_stringrrender)r&rr]r@Zjinja_templaterrr!render_templates  z(BitBucketTemplateManager.render_templatecCs |j|S)zGet a template by ID.)r6r#)r&rrrr! get_templates z%BitBucketTemplateManager.get_templatecCst|jS)z List all available template IDs.)listr6keysr(rrr!list_templatessz'BitBucketTemplateManager.list_templatesr*)r+r,r-r.rr/rrr'r9rr=rKrarbrrerrrr!r06s6   $  r0c @sHeZdZdZ d5deeefdeefddZe defdd Z e de fd d Z d5ded eeeefde eeeefffd dZ    d6deedeedeeeeefefdeeeefdeed eeeefde eeeeeefffddZdedeefddZ    d6deededeedeeeeefefdeeeefdeed eeeefdefddZdeefddZd7ddZdeed eed!edefd"d#Z  d8deed eed eed!ed$eed%eedefd&d'Z   d9deed eed!ed eed$eed%eedefd(d)Z    * *d:d+edeed,edeed eed!ed eed$eed%eed-eed.eede eeeeffd/d0Z     * *d;d+edeed,edeed eed!ed1e!d eed2eeed$eed%eed-eed.eede eeeeffd3d4Z"dS)<BitBucketPromptManagerah BitBucket prompt manager that integrates with LiteLLM's prompt management system. This class enables using .prompt files from BitBucket repositories with the litellm completion() function by implementing the PromptManagementBase interface. Usage: # Configure BitBucket access bitbucket_config = { "workspace": "your-workspace", "repository": "your-repo", "access_token": "your-token", "branch": "main" # optional, defaults to main } # Use with completion response = litellm.completion( model="bitbucket/gpt-4", prompt_id="my_prompt", prompt_variables={"variable": "value"}, bitbucket_config=bitbucket_config, messages=[{"role": "user", "content": "This will be combined with the prompt"}] ) Nr1r2cCs||_||_d|_dSr*)r1r2_prompt_managerr:rrr!r's zBitBucketPromptManager.__init__r;cCsdS)zlitellmZ_loggingZverbose_proxy_loggererror)r&rmrnrorpr2rjkwargsrkprompt_metadataZparsed_messagesZfinal_messagesparamrArurrr! pre_call_hook s<      z$BitBucketPromptManager.pre_call_hookr?cCsVg}|d}d}g}|D]z}|}|sq|dr<|r0|r0||d|dd}|ddg}q|dr`|rT|rT||d|dd}|d dg}q|d r|rx|rx||d|dd }|d dg}q||q|r|r||d|d|s|rd|dg}|S) z Parse prompt content into a list of messages. Handles both simple prompts and multi-role conversations. rNNzsystem:rrsystemzuser:rqz assistant:Z assistant )rHrFrTrEappendjoin)r&r?rnlinesZ current_roleZcurrent_contentrZrrr!rtMsV     z0BitBucketPromptManager._parse_prompt_to_messagesresponseinput_messagesc Ks|S)zL Post-call hook for any post-processing after the LLM call. r) r&rmrrrorpr2rjrwrrr!post_call_hooksz%BitBucketPromptManager.post_call_hookcCs |jS)z!Get list of available prompt IDs.)rirer(rrr!get_available_promptss z,BitBucketPromptManager.get_available_promptscCs|jr d|_|jdSdS)zReload prompts from BitBucket.N)r2rgrir(rrr!reload_promptss z%BitBucketPromptManager.reload_prompts prompt_specdynamic_callback_paramscCs|duS)z Determine if prompt management should run based on the prompt_id. For BitBucket, we always return True and handle the prompt loading in the _compile_prompt_helper method. Nr)r&r2rrrrr!should_run_prompt_managements z3BitBucketPromptManager.should_run_prompt_management prompt_labelprompt_versionc Cs|durtdz9||jjvr|j||||\}}||} |d} i} dD] } | |vr7|| | | <q+t|| | | ddWStyW} z td|d| d} ~ ww)aS Compile a BitBucket prompt template into a PromptManagementClient structure. This method: 1. Loads the prompt template from BitBucket 2. Renders it with the provided variables 3. Converts the rendered text into chat messages 4. Extracts model and optional parameters from metadata N2prompt_id is required for BitBucket prompt managerrrs)r2Zprompt_templateZprompt_template_modelZprompt_template_optional_paramsZcompleted_messageszError compiling prompt 'z': ) r_rir6r9rlrtr#rr>)r&r2rrjrrrrkrxrnZtemplate_modelr%ryrArrr!_compile_prompt_helpers4     z-BitBucketPromptManager._compile_prompt_helpercs(|dur td|j||||||dS)z Async version of compile prompt helper. Since BitBucket operations use sync client, this simply delegates to the sync version. Nr)r2rrjrrr)r_r)r&r2rjrrrrrrr!async_compile_prompt_helpers z2BitBucketPromptManager.async_compile_prompt_helperFrnon_default_paramsignore_prompt_manager_model%ignore_prompt_manager_optional_paramsc Cstj|||||||||| d S)zq Get chat completion prompt from BitBucket and return processed model, messages, and parameters. )rrr)rget_chat_completion_prompt) r&rrnrr2rjrrrrrrrrr!rsz1BitBucketPromptManager.get_chat_completion_promptlitellm_logging_objtoolscs.tj|||||||||| | | | | dIdHS)zY Async version - delegates to PromptManagementBase async implementation. ) r2rjrrrrrrrrN)r async_get_chat_completion_prompt)r&rrnrr2rjrrrrrrrrrrr!r&s" z7BitBucketPromptManager.async_get_chat_completion_promptr*)NNNN)r;N)NN)NNN)NNNFF)NNNNFF)#r+r,r-r.rr/rrr'propertyrhr0rirrlrrrrzrtrrrrrboolrdictrVrrrrLiteLLMLoggingObjrrrrr!rfsX     & @C    A !     '     rfN) r.typingrrrrrrrZjinja2r r r Z-litellm.integrations.custom_prompt_managementr Z*litellm.litellm_core_utils.litellm_loggingr rZ+litellm.integrations.prompt_management_baserrZlitellm.types.llms.openairZ"litellm.types.prompts.init_promptsrZlitellm.types.utilsrr7rrr0rfrrrr!s $