o ưi@sUddlmZmZerddlmZddlmZmZddlm Z ddlm Z ddlmZdZ ee e d<d e d dfd d Z    dddZe jjeiZgdZdS)) TYPE_CHECKINGOptional)BitBucketPromptManager)PromptLiteLLMParams PromptSpec)CustomPromptManagement)SupportedPromptIntegrationsNglobal_bitbucket_configconfigreturncCsddl}||_dS)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 r r^/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/bitbucket/__init__.pyset_global_bitbucket_configs rlitellm_paramsr prompt_specrrc CsTt|dd}t|dd}|stdz t||d}|WSty)}z|d}~ww)z: Initialize a prompt from a BitBucket repository. bitbucket_configN prompt_idz=bitbucket_config is required for BitBucket prompt integration)rr)getattr ValueErrorr Exception)rrrrbitbucket_prompt_managererrrprompt_initializer s  r)rrr )rrrrr r)typingrrrrZ"litellm.types.prompts.init_promptsrrZ-litellm.integrations.custom_prompt_managementrr r dict__annotations__rrZ BITBUCKETvalueZprompt_initializer_registry__all__rrrrs&