tPjUddlmZmZerddlmZddlmZmZddlm Z ddlm Z ddlm Z dZ ee dze d<d e d dfd Zdd Ze j j"eiZee d <gdZy)) TYPE_CHECKINGFinal)CustomPromptManagement)PromptLiteLLMParams PromptSpec)BitBucketPromptManager)SupportedPromptIntegrationsNglobal_bitbucket_configconfigreturncddl}||_y)ar Set the global BitBucket configuration for prompt management. Args: config: Dictionary containing BitBucket configuration - workspace: BitBucket workspace name - repository: Repository name - access_token: BitBucket access token - branch: Branch to fetch prompts from (default: main) rN)litellmr )r rs l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/bitbucket/__init__.pyset_global_bitbucket_configrs&,G#ct|dd}t|dd}|s td t||}|S#t$r}|d}~wwxYw)z: Initialize a prompt from a BitBucket repository. bitbucket_configN prompt_idz=bitbucket_config is required for BitBucket prompt integration)rr)getattr ValueErrorr Exception)litellm_params prompt_specrrbitbucket_prompt_manageres rprompt_initializerr!sd&n6H$O~{DAI XYY*@-+  (' s8 AAAprompt_initializer_registry)r r r)rrrrr r)typingrr-litellm.integrations.custom_prompt_managementr"litellm.types.prompts.init_promptsrrrr r r dict__annotations__rr BITBUCKETvaluer__all__rrr(su''TR@J</3td{+2 - - - , ))//1C&U  r