tPjSdZddlmZmZmZddlmZmZddlm Z ddl m Z erddl m ZneZddlmZmZddlmZdd lmZdd lmZd d lmZGd dZGddZGdde Zy)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_CHECKINGAnyFinal) DictLoaderselect_autoescape)ImmutableSandboxedEnvironment)CustomPromptManagement)Logging)PromptManagementBasePromptManagementClient)AllMessageValues) PromptSpec)StandardCallbackDynamicParams)BitBucketClientc BeZdZdZ d dededeeefdedzfdZdZy) BitBucketPromptTemplatez= Represents a prompt template loaded from BitBucket. N template_idcontentmetadatamodelcp||_||_||_|xs|jd|_|jd|_|jd|_|jdijdi|_|jDcic] \}}|dvs ||c}}|_ ycc}}w)Nr temperature max_tokensinputschema)rrr) rrrgetrrr input_schemaitemsoptional_params)selfrrrrkvs |/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/bitbucket/bitbucket_prompt_manager.py__init__z BitBucketPromptTemplate.__init__!s'   3hll73 #<< 6",,|4$LL"599(BG191AlAQNkEk1lls  B2#B2c<d|jd|jdS)NzBitBucketPromptTemplate(id='z ', model='z'))rrr!s r$__repr__z BitBucketPromptTemplate.__repr__1s#-d.>.>-?z$**UWXXN) __name__ __module__ __qualname____doc__strdictrr%r(r)r$rrsN! mmmsCx. m Tz m Yr)rceZdZdZ ddeeefdedzfdZdeddfdZdedede fd Z d edeeeffd Z dd ed eeefdzdefdZ d ede dzfdZ deefdZy)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 ||_||_i|_t||_t t itddgdddddd |_|jr|j|jyy) Nhtmlxmlz{{z}}z{%z%}z{#z#})loader autoescapevariable_start_stringvariable_end_stringblock_start_stringblock_end_stringcomment_start_stringcomment_end_string) r4r5promptsrbitbucket_clientrrr jinja_env_load_prompt_from_bitbucketr!r4r5s r$r%z!BitBucketTemplateManager.__init__Bs{ !1";= /0@ A7b>(&%9"& $#!!%#   >>  , ,T^^ < r)returnc |jj|d}|r"|j||}||j|<yy#t$r}t d|d|d}~wwxYw)z,Load a specific .prompt file from BitBucket.z.promptzFailed to load prompt 'z' from BitBucket: N)rBget_file_content_parse_prompt_filerA Exception)r!r5prompt_contenttemplatees r$rDz4BitBucketTemplateManager._load_prompt_from_bitbucketbs~ X$($9$9$J$Ji[X_K`$aN"&"9"9.)"T*2 Y' X5i[@RSTRUVW W XsAA A$AA$rct|jdrL|jdd}t|dk\r'|dj}|dj}n d}|}nd}|}i}|r ddl}|j |xsi}t|||S#t $r|j|}Y*t$ri}Y6wxYw) z?Parse a .prompt file content and extract metadata and template.z---rrN)rrr) startswithsplitlenstripyaml safe_load ImportError_parse_yaml_basicrJr)r!rr5partsfrontmatter_strtemplate_contentrrVs r$rIz+BitBucketTemplateManager._parse_prompt_filens   e $"==2E5zQ"'(.."2#(8>>#3 "$#*  O& $&  >>/:@b'!$   C11/B  s'BB7* B76B7yaml_strci}|jdD]}|j}d|vs|jdr*|jdd\}}|j}|j}|jdvr|jdk(||<|j rt |||<|j ddj rt|||<|jd ||<|S) z@Basic YAML parser for simple cases when PyYAML is not available. :#r)truefalserb.rQz"')rSrUrRlowerisdigitintreplacefloat)r!r]resultlinekeyvalues r$rYz*BitBucketTemplateManager._parse_yaml_basics(*NN4( 5D::D;P:\_:--0G$0N-)S )r)r3c eZdZdZ d&deeefdedzfdZedefdZ ede fdZ d&ded eeefdzde eeeefffd Z d'd edzd eed eeefezdzdeeefdzdedzd eeefdzde eeeeefdzffdZdedeefdZ d'd edzdedeed eeefezdzdeeefdzdedzd eeefdzdefdZdeefdZd(dZdedzdedzdedefdZ d)dedzdedzd edzdededzdedzdefdZ d*dedzd edzdededzdedzdedzdefdZ d+ded eedededzd edzdededzdedzdedzd edzd!edzde eeeeffd"Z d,ded eedededzd edzded#ededzd$eedzdedzdedzd edzd!edzde eeeeffd%Zy)-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"}] ) Nr4r5c.||_||_d|_yr*)r4r5_prompt_managerrEs r$r%zBitBucketPromptManager.__init__s !1"@Dr)rFcy)z1 1 ,     1 1 7 78egh i^+ +  ,s$AA1# A11 B2:-B-'B2-B2rKcg}|jjd}d}g}|D]i}|j}|s|jjdrL|r3|r1|j |dj |jdd}|ddjg}|jjdrL|r3|r1|j |dj |jdd}|d djg}|jjd rM|r3|r1|j |dj |jdd }|d djg}Y|j |l|r3|r1|j |dj |jd|s$|jrd|jdg}|S) z Parse prompt content into a list of messages. Handles both simple prompts and multi-role conversations. r_Nzsystem:rsystemzuser:rz assistant: assistant )rUrSrerRappendjoin)r!rKrlines current_rolecurrent_contentrks r$rz0BitBucketPromptManager._parse_prompt_to_messagesDs %++-33D9 & -D::>#3"4((1OOO$0'+yy'A'G'G'I & #'8>>#3"4((6OOO$0'+yy'A'G'G'I + #'9??#4"5 &&t,M& -R O OO\dii>X>^>^>`a bN002!'N4H4H4JKLHr)responseinput_messagesc |S)zL Post-call hook for any post-processing after the LLM call. r1) r!rrrrrr5rrs r$post_call_hookz%BitBucketPromptManager.post_call_hooks r)c6|jjS)z!Get list of available prompt IDs.)rr{r's r$get_available_promptsz,BitBucketPromptManager.get_available_promptss""1133r)cD|jrd|_|jyy)zReload prompts from BitBucket.N)r5rrr's r$reload_promptsz%BitBucketPromptManager.reload_promptss >>#'D    r) prompt_specdynamic_callback_paramsc |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. Nr1)r!r5rrs r$should_run_prompt_managementz3BitBucketPromptManager.should_run_prompt_managements$$r) prompt_labelprompt_versionc| td ||jjvr|jj||j ||\}}|j |} |j d} i} dD]} | |vs|| | | <t|| | | dS#t$r} td|d| d} ~ wwxYw)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 managerrr)r5prompt_templateprompt_template_modelprompt_template_optional_paramscompleted_messageszError compiling prompt 'z': ) rqrrArDrrrr rJ)r!r5rrrrrrrrtemplate_modelr rrMs r$_compile_prompt_helperz-BitBucketPromptManager._compile_prompt_helpers$  QR R# K 3 3 ; ;;##?? J04/G/G Sc/d ,O_#<<_MH%4$7$7$@N&(O DO+-(B99B>cRK| td|j||||||Sw)z Async version of compile prompt helper. Since BitBucket operations use sync client, this simply delegates to the sync version. r)r5rrrrr)rqr)r!r5rrrrrs r$async_compile_prompt_helperz2BitBucketPromptManager.async_compile_prompt_helpersD  QR R**#-$;%) +  s%'rnon_default_paramsignore_prompt_manager_model%ignore_prompt_manager_optional_paramsc @tj||||||||||  S)zq Get chat completion prompt from BitBucket and return processed model, messages, and parameters. )rrr)r get_chat_completion_prompt) r!rrrr5rrrrrrrs r$rz1BitBucketPromptManager.get_chat_completion_prompts5"$>>       ##%)  r)litellm_logging_objtoolscdKtj|||||||||| | | | | d{S7w)zY Async version - delegates to PromptManagementBase async implementation. ) r5rrrrrrrrrN)r async_get_chat_completion_prompt)r!rrrr5rrrrrrrrrs r$rz7BitBucketPromptManager.async_get_chat_completion_promptsO&*JJ    - 3$;#%)(C2W    s '0.0r*)NNNN)rFN)NN)NNN)NNNFF)NNNNFF) r+r,r-r.r0r/rr%propertyrr3rtuplerryr rrrrrrrboolrrgr rrrLiteLLMLoggingObjrr1r)r$r}r}s8!%EsCx.E:E#$ 8$$37))sCx.4/) sDcN" # )H6:04 $268,t8,'(8,CH~+d2 8, S#X- 8, : 8,sCx.4/8, t$%tCH~'<< =8,t::EU@V:B6:04 $26t-.  CH~+d2  S#X- :sCx.4/  4tCy4 %: % $& %"? %  %($(%)8K:8K $&8K+ 8K "? 8K Dj 8Kd 8K 8K~*.#'%) : + "?   $&  Dj  d    B*.#'%)38=B  '( !  :  +  "?  $& Dj d  &*D[ 04d{  sD)*D0 1 N*.#'#'%)38=B" " '(" ! " : " + " "?" /"  $&" DzD " Dj" d " &*D[" 04d{"  sD)*D0 1" r)r}N)r.typingrrrjinja2rrjinja2.sandboxr-litellm.integrations.custom_prompt_managementr *litellm.litellm_core_utils.litellm_loggingr r+litellm.integrations.prompt_management_baser r litellm.types.llms.openair "litellm.types.prompts.init_promptsrlitellm.types.utilsrrBrrr3r}r1r)r$rsb -,08PW79=-YY2@)@)FC 3C r)