tPji/LdZddlZddlmZmZddlmZddlmZGddZ y)z GitLab API client for fetching files from GitLab repositories. Now supports selecting a tag via `config["tag"]`; falls back to branch ("main"). N)AnyFinal)quote) HTTPHandlerc eZdZdZdeeeffdZdddededzdefd Zdddededzdefd Z dddd ed e dedzdefd Z deddfdZ dddededzdedzfdZ dddededzdedzfdZ dddd eded e dedzdeef dZdeeeffdZde fdZdeeeeffdZdddededzdeeefdzfdZdZy) GitLabClienta Client for interacting with the GitLab API to fetch files. Supports: - Authentication with personal/access tokens or OAuth bearer tokens - Fetching file contents from repositories (raw endpoint with JSON fallback) - Namespace/project path or numeric project ID addressing - Ref selection via tag (preferred) or branch (default "main") - Directory listing via the repository tree API configc|jd}|jd}|| td||_t||_|jdd|_|jdd|_|j sd|_|jd |_|jd d |_t|j|jgs tdt|jxs |j |_ d d d |_ |j dk(rd|j|jd<n|j|jd<tt|jd|_ t|_y)a Initialize the GitLab client. Args: config: Dictionary containing: - project: Project path ("group/subgroup/repo") or numeric project ID (str|int) [required] - access_token: GitLab personal/access token or OAuth token [required] (str) - auth_method: 'token' (default; sends Private-Token) or 'oauth' (Authorization: Bearer) - tag: Tag name to fetch from (takes precedence over branch if provided) - branch: Branch to fetch from (default: "main") - base_url: Base GitLab API URL (default: "https://gitlab.com/api/v4") project access_tokenNz%project and access_token are required auth_methodtokenbranchmaintagbase_urlzhttps://gitlab.com/api/v4application/json)Acceptz Content-TypeoauthzBearer Authorizationz Private-Tokensafe)get ValueErrorr strr r rrrallrefheadersr _project_encr http_handler)selfr r r s n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/gitlab/gitlab_client.py__init__zGitLabClient.__init__sW I.$jj8 ?l2DE E") !$\!2!::mW=jj40 {{ DK::e$ :/JK DLL$"3"345DE Etxx.4;;/).     w &.5d6G6G5H,IDLL )-1,=,=DLL )"#dll"3"=(MNr file_pathrreturnct|d}t|xs |jd}|jd|jd|d|S)Nrr /projects//repository/files/z /raw?ref=rrrr r"r'rfile_encref_qs r# _file_raw_urlzGitLabClient._file_raw_urlOsO 3S_DHH26-- 4+<+<*==OPXzYbchbijjr%ct|d}t|xs |jd}|jd|jd|d|S)Nrrr*r+z?ref=r,r-s r#_file_json_urlzGitLabClient._file_json_urlTsO 3S_DHH26-- 4+<+<*==OPXzY^_d^effr%directory_path recursivec|rdt|dnd}|rdnd}t|xs |jd}|jd|jd|||S)Nz&path=rrz&recursive=truer*z/repository/tree?ref=r,)r"r3r4rpath_qrec_qr/s r# _tree_urlzGitLabClient._tree_urlYsnFT&~B!? @AY[,5(2S_DHH26-- 4+<+<*==RSXRYZ`Yabgahiir%c,|s td||_y)z;Override the default ref (tag/branch) for subsequent calls.zref must be a non-empty stringN)rr)r"rs r#set_refzGitLabClient.set_refis=> >r%c|j||} |jj||j}|jdk(r|j ||S|j |jjdxsdj}|jdsd|vs|jdr |jS |jjd S#t$r |jjd d cYSwxYw#t$ri}tt|d d dd }|dk(rYd }~y |dk(rtd|d|jd|dk(r tdtd|d|d }~wwxYw)a Fetch the content of a file from the GitLab repository at the given ref (tag, branch, or commit SHA). If `ref` is None, uses self.ref. Strategy: 1) Try the RAW endpoint (returns bytes of the file) 2) Fallback to the JSON endpoint (returns base64-encoded content) Returns: File content as UTF-8 string, or None if file not found. r&r content-typerztext/zcharset=rutf-8replaceerrorsresponseN status_codeAccess denied to file '.'. Check your GitLab permissions for project ''.?Authentication failed. Check your GitLab token and auth_method.Failed to fetch file '': )r0r!rrrD_get_file_content_via_jsonraise_for_statuslower startswithtextcontentdecode Exceptiongetattrr )r"r'rraw_urlrespctypeestatuss r#get_file_contentzGitLabClient.get_file_contentos++I3+? H++///ND3&66yc6JJ  ! ! # LL,,^<BIIKE(J%,?5CSCSTfCgyy  F||**733 F||**79*EE F H#GAz4$@-QUVF}}-i[8fgkgsgsfttvw} abb4YKs1#FG G HsDADA.DC))&DDDD FFAFFc|j||} |jj||j}|jdk(ry|j |j }|jd}|jdd}|r*|dk(r% tj|jd S|S#t$r)tj|jd d cYSwxYw#t$ri}tt|d dd d} | dk(rYd}~y| dk(rtd|d|jd| dk(r tdtd|d|d}~wwxYw)zf Fallback for get_file_content(): use the JSON file API which returns base64 content. r&r<r=NrRencodingrbase64r?r@rArCrDrErFrGrHrIrJrKz' via JSON endpoint: ) r2r!rrrDrNjsonr^ b64decoderSrTrUr ) r"r'rjson_urlrWdatarRr]rYrZs r#rMz'GitLabClient._get_file_content_via_jsons--iS-A Z++//$,,/OD3&  ! ! #))+D!XXi0G"hhz26H8x/W!++G4;;GDDN!W!++G4;;GI;VVW Z#GAz4$@-QUVF}}-i[8fgkgsgsfttvw} abb4YK?TUVTWXY Y ZsI6C2 A C2#B=;C2=/C/,C2.C//C22 E$;EAEE$file_extensionc|j|||} |jj||j}|jdk(rgS|j |j xsg}g}|D]N} | jddk(s| jdd} |r| j|s>|j| P|S#t$rk} tt| dd d d } | dk(rgcYd } ~ S| d k(rtd |d |jd| dk(r tdtd|d| d } ~ wwxYw)a List files in a directory with a specific extension using the repository tree API. Args: directory_path: Directory path in the repository (empty for repo root) file_extension: File extension to filter by (default: .prompt) recursive: If True, traverses subdirectories ref: Optional override (tag/branch/SHA). Defaults to self.ref. Returns: List of file paths (relative to repo root) )r4rr<r=typeblobpathrrCNrDrEzAccess denied to directory 'rGrHrIrJzFailed to list files in 'rL) r8r!rrrDrNr_endswithappendrTrUr ) r"r3rcr4rurlrWrbfilesitemr'rYrZs r# list_fileszGitLabClient.list_filessa(^^NiS^Q P++//T\\/JD3&  ! ! #))++D&(E 088F#v- $ 4I)Y-?-?-O Y/  0 L P#GAz4$@-QUVF} }2>2BBpquq}q}p~AB} abb77Gs1#NO O Ps67C ?C %C 4C D=D80D=6AD88D=c|jd|j} |jj||j}|j |j S#t$r}td|d}~wwxYw)z-Get information about the project/repository.r*r<zFailed to get repository info: N)rr r!rrrNr_rT)r"rjrWrYs r#get_repository_infoz GitLabClient.get_repository_infos j1B1B0CD C++//T\\/JD  ! ! #99;  C=aSAB B CsAA$$ B-A;;BcD |jy#t$rYywxYw)z*Test the connection to the GitLab project.TF)rorTr"s r#test_connectionzGitLabClient.test_connections(   $ $ &  s  c0|jd|jd} |jj||j}|j |j }t|tr|SgS#t$r}td|d}~wwxYw)z'Get list of branches in the repository.r*z/repository/branchesr<zFailed to get branches: N) rr r!rrrNr_ isinstancelistrT)r"rjrWrbrYs r# get_brancheszGitLabClient.get_branchess j1B1B0CCWX <++//T\\/JD  ! ! #))+D%dD14 9r 9 <6qc:; ; zcontent-lengthz last-modified) content_typecontent_length last_modifiedrCrDz!Failed to get file metadata for 'rL) r0dictrr!rrDrNrTrU)r"r'rrjrrWrYrZs r#get_file_metadatazGitLabClient.get_file_metadatas'' s'; S!$,,/G*GG ++//W/ED3&  ! ! # $ 0 0 @"&,,"2"23C"D!%!1!1/!B   S#GAz4$@-QUVF}? {#aSQR R  Ss%AB=A B== C>C9(C99C>cRt|dr|jjyy)z)Close the HTTP handler to free resources.r!N)hasattrr!closerqs r#rzGitLabClient.close s# 4 (    # # % )r%)rF)rz.promptF)__name__ __module__ __qualname____doc__r|rrr$r0r2boolr8r:r[rMrurmrorrrvr}rr%r#rr s 0*tCH~0*lBFkskC$Jk#k CGggS4Zg3g! j  j j j 4Z j j 34 EI'H#'HsTz'HSSWZ'HROSZCZtZWZ]aWaZB!' /P  /P/P/P /P 4Z /P c/PbCT#s(^C 2 <FJS3Sd SdSVX[S[n_cNcS6&r%r) rr^typingrr urllib.parser&litellm.llms.custom_httpx.http_handlerrrrr%r#rs$ >V&V&r%