tPj UddlmZmZmZerddlmZddlmZmZddl m Z ddlmZddlmZmZm Z ddl m Z m Z dZ eedzed <d ed dfd ZddZd eded efdZe j&j(eiZeed<gdZy)) TYPE_CHECKINGAnyFinal)CustomPromptManagement)PromptLiteLLMParams PromptSpec)GitLabPromptManager)rrSupportedPromptIntegrations)GitLabPromptCacher Nglobal_gitlab_configconfigreturncddl}||_y)af Set the global gitlab configuration for prompt management. Args: config: Dictionary containing gitlab configuration - workspace: gitlab workspace name - repository: Repository name - access_token: gitlab access token - branch: Branch to fetch prompts from (default: main) rN)litellmr )rrs i/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/gitlab/__init__.pyset_global_gitlab_configrs#)G litellm_paramsct|dd}t|dd}|s td t||}|S#t$r}|d}~wwxYw)z7 Initialize a prompt from a Gitlab repository. gitlab_configN prompt_id7gitlab_config is required for gitlab prompt integration)rr)getattr ValueErrorr Exception)r prompt_specrrgitlab_prompt_manageres rprompt_initializerr "sb#>?DIM~{DAI RSS':'(  %$ s8 AAApromptct|ddxsi}t|dd}|s tdt||j|S)a8 Build a GitLab-backed prompt manager for this prompt. Expected fields on litellm_params: - prompt_integration="gitlab" (handled by the caller) - gitlab_config: Dict[str, Any] (project/access_token/branch/prompts_path/etc.) - git_ref (optional): per-prompt tag/branch/SHA override rNgit_refr)rrref)rrr r)rr!rr#s r_gitlab_prompt_initializerr%7sT,3>?TX+Y+_]_M!(D!IG RSS #""  rprompt_initializer_registry)r r r r)rrrrrr)typingrrr-litellm.integrations.custom_prompt_managementr"litellm.types.prompts.init_promptsrrrr r r r dict__annotations__rr r%GITLABvaluer&__all__rrr0s,,TR:PkkI,0eD4K(/ *T *d * *' : &&,,.H&U  r