tPjqUdZddlmZmZmZddlmZmZddlm Z ddl m Z erddl m ZneZddlmZddlmZmZdd lmZdd lmZdd lmZd Zeed <dedefdZdedefdZGddZGddZ Gdde Z!GddZ"y)z9 GitLab prompt manager with configurable prompts folder. ) TYPE_CHECKINGAnyFinal) DictLoaderselect_autoescape)ImmutableSandboxedEnvironment)CustomPromptManagement)Logging) GitLabClient)PromptManagementBasePromptManagementClient)AllMessageValues) PromptSpec)StandardCallbackDynamicParamszgitlab:: GITLAB_PREFIXraw_idreturncd|jtr|St|jddS)uMConvert GitLab path IDs like 'invoice/extract' → 'gitlab::invoice::extract'/::) startswithrreplace)rs v/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/gitlab/gitlab_prompt_manager.pyencode_prompt_idrs0 ' _V^^C67 88 encoded_idct|jts|S|ttdjddS)u8Convert 'gitlab::invoice::extract' → 'invoice/extract'Nrr)rrlenr)rs rdecode_prompt_idr#s6   / c-(* + 3 3D# >>rc >eZdZ ddededeeefdedzfdZdZy) GitLabPromptTemplateN template_idcontentmetadatamodelcp||_||_||_|xs|jd|_|jd|_|jd|_|jdijdi|_|jDcic] \}}|dvs ||c}}|_ ycc}}w)Nr% temperature max_tokensinputschema)r%r)r#) r"r#r$getr%r'r( input_schemaitemsoptional_params)selfr"r#r$r%kvs r__init__zGitLabPromptTemplate.__init__+s'   3hll73 #<< 6",,|4$LL"599(BG191AlAQNkEk1lls  B2#B2c<d|jd|jdS)NzGitLabPromptTemplate(id='z ', model='z'))r"r%r/s r__repr__zGitLabPromptTemplate.__repr__;s#*4+;+;* 2 \d6H6H6L6LX6V \Z\ %* 7b>(&%9"& $#!!%#   >>  ) )$.. 9 rrc^t|}|jr|jd|dS|dS)zHMap a prompt_id to a repo path (respects prompts_path and adds .prompt).r.prompt)rrD)r/r@s r_id_to_repo_pathz&GitLabTemplateManager._id_to_repo_pathps<$Y/   ''()G< <G$$r repo_pathc"|jd}|jrW|j|jjddzr*|t|jjddzd}|j d}t |S)z Map a repo path like 'prompts/chat/greeting.prompt' to an ID relative to prompts_path without the extension (e.g., 'chat/greeting'). rNrX)rSrDrr removesuffixr)r/rZpaths r_repo_path_to_idz&GitLabTemplateManager._repo_path_to_idws s#   1B1B1H1H1MPS1S!TD--33C89A=?@D  +%%rrAc |j|}|jj||}|r"|j||}||j|<yy#t $r}t dt |d|d}~wwxYw)zLLoad a specific .prompt file from GitLab (scoped under prompts_path if set).r`zFailed to load prompt 'z' from GitLab: N)rYrBget_file_content_parse_prompt_filerQ Exceptionr)r/r@rA file_pathprompt_contenttemplatees rrUz.GitLabTemplateManager._load_prompt_from_gitlabs g#44Y?I$($6$6$G$G WZ$G$[N"&"9"9.)"T*2 Y' g56Fy6Q5RRabcadef f gsAA A=A88A=T recursiverjc|j|}g}|D]2}||jvr|j||j|4|S)zW Eagerly load all .prompt files from prompts_path. Returns loaded IDs. ri)list_templatesrQrUappend)r/rjfilesloadedpids rload_all_promptsz&GitLabTemplateManager.load_all_promptssX**Y*?#% C$,,&--c2 MM#   rr#ct|jdrL|jdd}t|dk\r'|dj}|dj}n d}|}nd}|}i}|r ddl}|j |xsi}t|||S#t $r|j|}Y*t$ri}Y6wxYw)Nz---r\rFr)r"r#r$) rsplitrrSyaml safe_load ImportError_parse_yaml_basicrdr!)r/r#r@partsfrontmatter_strtemplate_contentr$rvs rrcz(GitLabTemplateManager._parse_prompt_files   e $"==2E5zQ"'(.."2#(8>>#3 "$#*  O& #%  >>/:@b $!$   C11/B  s'BB7* B76B7yaml_strc.i}|jdD]}|j}d|vs|jdr*|jdd\}}|j}|j}|jdvr|jdk(||<|j rt |||<|j ddj r t|||<|jd ||<|S#t$r|||<YwxYw) N :#r\)truefalser.rFz"') rurSrlowerisdigitintrfloatrd)r/r}resultlinekeyvalues rryz'GitLabTemplateManager._parse_yaml_basics(*NN4( 5D::r>?sR!%-1 &:CH~&::&:4Z &: $d* &:T%#%#% &# &# &MQ g# gsTz gUY g59 T T#Y  # # BV <#$sCx.*:3:4S>D;P:\_:--0Dt0K-37+=4+=49+=rr>c eZdZdZ d)deeefdedzdedzdedzfdZe defd Z e de fd Z d*dd ded eeefdzdedzde eeeefffd Z d+dedzdeedeeefezdzdeeefdzdedzd eeefdzdedzde eeeeefdzffdZdedeefdZ d,dedzdedeedeeefezdzdeeefdzdedzd eeefdzdefdZdeefdZd-dZdedzdedzdedefdZ d.dedzdedzd edzdededzdedzdefdZ d)dedzd edzdededzdedzdedzdefd Z d/d!edeed"ededzd edzdededzdedzdedzd#edzd$edzde eeeeffd%Z d0d!edeed"ededzd edzded&ededzd'eedzdedzdedzd#edzd$edzde eeeeffd(Z y)1GitLabPromptManagerak GitLab prompt manager with folder support. Example config: gitlab_config = { "project": "group/subgroup/repo", "access_token": "glpat_***", "tag": "v1.2.3", # optional; takes precedence "branch": "main", # default fallback "prompts_path": "prompts/chat" } Nr?r@rArBc||_||_d|_||_||_|jr2t |j|j|j|_yy)N)r?r@rA)r?r@_prompt_manager _ref_override_injected_gitlab_clientr>rVs rr2zGitLabPromptManager.__init__s]+"=A '4$ >>#8"00..&&$D  rrcy)Ngitlabr<r4s rintegration_namez$GitLabPromptManager.integration_name*src|jr?r@rrr4s rprompt_managerz"GitLabPromptManager.prompt_manager.sK    '#8"00..&&":: $D ###rr`prompt_variablescl||jjvr|jj|||jj|}|st d|d|jj ||xsi}|j |j|jd|j}||fS)Nr`zPrompt template 'r)r%r'r() rrQrUrrrr%r'r(r.)r/r@rrArgrendered_promptr$s rget_prompt_templatez'GitLabPromptManager.get_prompt_template9s D//77 7    8 8 8 L--::9E0 ;GH H!%!4!4!D!DYP`Pfdf!g^^#//"-- &&   ((ruser_idmessages function_calllitellm_paramsprompt_versionc |s||fS |xs|jdxs |j} |j||| \} } |j| } | r| } n d| dg|z} |i}| jdr| d|d<dD]}|| vs| |||<| |fS#t$r8}ddl}|j jjd|||fcYd}~Sd}~wwxYw) Ngit_refr`userroler#r%r'r(top_pfrequency_penaltypresence_penaltyrz(Error in GitLab prompt pre_call_hook: %s) r+rr_parse_prompt_to_messagesrdlitellm_loggingverbose_proxy_loggererror)r/rrrrr@rrkwargsrrprompt_metadataparsed_messagesfinal_messagesparamrhrs r pre_call_hookz!GitLabPromptManager.pre_call_hookQs^+ +! ,+Zvzz)/DZHZHZG/3/G/G Scip/G/q ,O_%)%C%CO%TO9H+1o"N!ORZ!Z%!#""7+*9'*Bw' CO+,;E,BN5) C">1 1 ,     1 1 7 78bde f^+ +  ,s$A?B B C-C CCrfcg}|jjd}d}g}|D]O}|j}|s|j}|jdrL|r3|r1|j |dj |jdd}|ddjg}|jdrL|r3|r1|j |dj |jdd}|ddjg}|jd rM|r3|r1|j |dj |jdd }|d djg}?|j |R|r3|r1|j |dj |jd|s$|jrd|jdg}|S) Nrzsystem:rsystemzuser:rz assistant: assistant )rSrurrrmjoin) r/rfrlines current_rolecurrent_contentrrlows rrz-GitLabPromptManager._parse_prompt_to_messagess+-%++-33D9#' %'$ -C99;D**,C~~i(OOO$0'+yy'A'G'G'I ( #'8>>#3"4(OOO$0'+yy'A'G'G'I & #'8>>#3"4 -OOO$0'+yy'A'G'G'I + #'9??#4"5&&t,I$ -L O OO\dii>X>^>^>`a bN002!'N4H4H4JKLHrresponseinput_messagesc |Sr6r<) r/rrrrrr@rrs rpost_call_hookz"GitLabPromptManager.post_call_hooks rct|jjj} |j |jj t|S#t $r Yt|SwxYw)z Return prompt IDs. Prefer already-loaded templates in memory to avoid unnecessary network calls (and to make tests deterministic). )setrrQkeysupdaterlrdsorted)r/idss rget_available_promptsz)GitLabPromptManager.get_available_promptssm ,,4499;<  JJt**99; <c{  c{ s)A## A98A9cD|jrd|_|j}yyr6)r@rr)r/_s rreload_promptsz"GitLabPromptManager.reload_promptss! >>#'D ##A r prompt_specdynamic_callback_paramsc |duSr6r<)r/r@rrs rshould_run_prompt_managementz0GitLabPromptManager.should_run_prompt_managements $$r prompt_labelc| td t|}||jjvrGt |drt |dij dnd}|jj|||j||\} } |j| } | j d} i} dD]}|| vs| || |<t|| | | dS#t$r}td|d |d}~wwxYw) N/prompt_id is required for GitLab prompt managerextrarr`r%r)r@prompt_templateprompt_template_modelprompt_template_optional_paramscompleted_messageszError compiling prompt 'z': ) rrrrQhasattrgetattrr+rUrrr rd)r/r@rrrrr decoded_idrrrrtemplate_modelr.rrhs r_compile_prompt_helperz*GitLabPromptManager._compile_prompt_helpers8  NO O" K 0 ;J!4!4!>       ##%)  rlitellm_logging_objtoolscdKtj|||||||||| | | | | d{S7w)zY Async version - delegates to PromptManagementBase async implementation. ) r@rrrrrrrrrN)r async_get_chat_completion_prompt)r/r%rrr@rrrrrrrrrs rrz4GitLabPromptManager.async_get_chat_completion_prompt>sO&*JJ    - 3$;#%)(C2W    s '0.0rr6)NNNNN)NNNN)rN)NN)NNNFF)NNNNFF)!r7r8r9rr;r:rr r2propertyrr>rtuplerrrrrrrrrrrrrr rrrLiteLLMLoggingObjrr<rrrr s   !%-1 CH~:4Z  $d* &#$ 5$$37)  ))sCx.4/) 4Z ) sDcN" # )86:04 $26%).,t.,'(.,CH~+d2 ., S#X- ., : .,sCx.4/.,d ., t$%tCH~'<< =.,`00EU@V0n6:04 $26 t  -.  CH~+d2  S#X- : sCx.4/    tCy $ %:% $&%"? %  %$(%).K:.K $&.K+ .K "? .K Dj .Kd .K .Kj*.#'%) : + "?   $&  Dj  d    B*.#'%)38=B  '( !  :  +  "?  $& Dj d  &*D[ 04d{  sD)*D0 1 H*.#'#'%)38=B" " '(" ! " : " + " "?" /"  $&" DzD " Dj" d " &*D[" 04d{"  sD)*D0 1" rrc eZdZdZddddeeefdedzdedzddfdZd d d e deeeeefffd Z d d d e deeeeefffd Z de efdZ de efdZdedeeefdzfdZdedeeefdzfdZdededeeeffdZy)GitLabPromptCachea Cache all .prompt files from a GitLab repo into memory. - Keys are the *repo file paths* (e.g. "prompts/chat/greet/hi.prompt") mapped to JSON-like dicts containing content + metadata. - Also exposes a by-ID view (ID == path relative to prompts_path without ".prompt", e.g. "greet/hi"). Usage: cfg = { "project": "group/subgroup/repo", "access_token": "glpat_***", "prompts_path": "prompts/chat", # optional, can be empty for repo root # "branch": "main", # default is "main" # "tag": "v1.2.3", # takes precedence over branch # "base_url": "https://gitlab.com/api/v4" # default } cache = GitLabPromptCache(cfg) cache.load_all() # fetch + parse all .prompt files print(cache.list_files()) # repo file paths print(cache.list_ids()) # template IDs relative to prompts_path prompt_json = cache.get_by_file("prompts/chat/greet/hi.prompt") prompt_json2 = cache.get_by_id("greet/hi") # If GitLab content changes and you want to refresh: cache.reload() # re-scan and refresh all N)rArBr?rArBrc~t|d|||_|jj|_i|_i|_yr)rrtemplate_manager_by_file_by_id)r/r?rArBs rr2zGitLabPromptCache.__init__sC2''   8<7J7J7Y7Y46 13 rTrirjc|jj|}|D]}||jjvr|jj||jj |}|6|jj||jj |}||jj |}|j ||}||j|<t|}||j|<|jS)z Scan GitLab for all .prompt files under prompts_path, load and parse each, and return the mapping of repo file path -> JSON-like dict. ri) r rlrQrUrrY_template_to_jsonrrr)r/rjrrptmplreentryrs rload_allzGitLabPromptCache.load_alls **99I9N ,C$//777%%>>sC((55c:D|%%>>sC,,99#>|-->>sCI**35E',DMM) $)#.J&+DKK #' ,,{{rc|jj|jj|j|S)z(Clear the cache and re-load from GitLab.ri)rclearrr)r/rjs rreloadzGitLabPromptCache.reloads3  }}y}11rcHt|jjS)z,Return the repo file paths currently cached.)rrrr4s rrzGitLabPromptCache.list_filessDMM&&())rcHt|jjS)zWReturn the template IDs (relative to prompts_path, without extension) currently cached.)rrrr4s rlist_idszGitLabPromptCache.list_idssDKK$$&''rrec8|jj|S)z+Get a cached prompt JSON by repo file path.)rr+)r/res r get_by_filezGitLabPromptCache.get_by_files}}  ++rr@c||jvr|j|St|}t|}|jj|xs|jj|S)zAGet a cached prompt JSON by prompt ID (relative to prompts_path).)rrrr+)r/r@decodedencodeds r get_by_idzGitLabPromptCache.get_by_idsX  #;;y) )*)4)'2{{w'C4;;??7+CCrrc  t|jxsi}|j}|j}|j}t|j xsi}||j j||j|||||dS)zc Normalize a GitLabPromptTemplate into a JSON-like dict that is easy to serialize. )idr^r#r$r%r'r(r.) r;r$r%r'r(r.r rYr#)r/r@rmdr%r'r(r.s rrz#GitLabPromptCache._template_to_jsons ,"-zz!--  OO !%d&:&:&@b!A))::9E||&$.  r)r7r8r9rr;r:rr r2rrrrrrrr r!rr<rrr r cs)H-1 4CH~44Z 4 $d* 4  40-1TT#tCH~:M5N<+/2424T#s(^8K3L2 *DI*($s)(,S,T#s(^d-B, D3 D4S>D+@ D 3 6J tTWY\T\~ rr N)#rtypingrrrjinja2rrjinja2.sandboxr-litellm.integrations.custom_prompt_managementr *litellm.litellm_core_utils.litellm_loggingr r )litellm.integrations.gitlab.gitlab_clientr +litellm.integrations.prompt_management_baser r litellm.types.llms.openair"litellm.types.prompts.init_promptsrlitellm.types.utilsrr__annotations__r:rrr!r>rr r<rrr/s-,08PWB79=! u!9S9S9???VV*G=G=TW 0W t N N r