jJ~dZddlmZddlZddlmZmZmZmZm Z ddl m Z ddl m Z mZmZerddlmZGdd e Zy) zUtil that calls gitlab.) annotationsN) TYPE_CHECKINGAnyDictListOptional)get_from_dict_or_env) BaseModel ConfigDictmodel_validator)IssuecNeZdZUdZdZded<dZded<dZded< dZded< dZ ded < dZ ded < dZ ded < e d Z eded#dZd$dZd%dZd&dZd'dZd(dZd)dZd*dZd)dZd*dZd%dZd%dZd+dZd,d-dZd,d.dZd/dZd%d Z d0d!Z!d1d"Z"y)2GitLabAPIWrapperzWrapper for GitLab API.Nrgitlabgitlab_repo_instancez Optional[str] gitlab_urlgitlab_repositorygitlab_personal_access_token gitlab_branchgitlab_base_branchforbid)extrabefore)modect|ddd}t|dd}t|dd}t|d d d }t|d d d } ddl}|j||d}|j ||d<|j j ||d<||d<||d<||d<||d <||d <|S#t$r tdwxYw)z?Validate that api key and python package exists in environment.r GITLAB_URLzhttps://gitlab.com)defaultrGITLAB_REPOSITORYrGITLAB_PERSONAL_ACCESS_TOKENr GITLAB_BRANCHmainrGITLAB_BASE_BRANCHrNzRpython-gitlab is not installed. Please install it with `pip install python-gitlab`T)url private_token keep_base_urlrr)r r ImportErrorGitlabauthprojectsget) clsvaluesrrrrrrgs n/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/utilities/gitlab.pyvalidate_environmentz%GitLabAPIWrapper.validate_environment's# * L,8L 1 ')< (< 24R( $- O_f 2 (*>    MM6   x)*8I)J%&)|&7"#1M-."/'9#$ - E  s B//Ccpg}|D].}|j}|j}|j||d0|S)z Extracts title and number from each Issue and puts them in a dictionary Parameters: issues(List[Issue]): A list of gitlab Issue objects Returns: List[dict]: A dictionary of issue titles and numbers )titlenumber)r1iidappend)selfissuesparsedissuer1r2s r. parse_issueszGitLabAPIWrapper.parse_issuesYs=EKKEYYF MMEV< = c|jjjd}t|dkDr9|j |}dt t|zdzt |z}|Sy)z Fetches all open issues from the repo Returns: str: A plaintext report containing the number of issues and each issue's title and number. opened)staterFound z issues: zNo open issues available)rr6listlenr9str)r5r6 parsed_issuesparsed_issues_strs r. get_issueszGitLabAPIWrapper.get_issueshso**1166X6F v;? --f5M3s=122\AC DVV % $-r:c|jjj|}d}g}t|dkr|jj |}t|dk(rnk|D]R}|jj|j }|j|j|jddT|dz }t|dkr|j|jt|dS)a Fetches a specific issue and its first 10 comments Parameters: issue_number(int): The number for the gitlab issue Returns: dict: A dictionary containing the issue's title, body, and comments as a string r )pageusername)bodyuser)r1rIcomments) rr6r*r@notesr?idr4rIauthorr1 descriptionrA)r5 issue_numberr8rGrL comments_pagecomments r. get_issuezGitLabAPIWrapper.get_issuezs))0044\B!(mr!!KK,,$,7M=!Q&(++//'**5 ' 'z :) AID(mr![[%%H   r:ch|j|jk(ry |jdd}|t|dzd}|jj j |j|j||dgd}d|jS#t$r}d t|zcYd}~Sd}~wwxYw) a Makes a pull request from the bot's branch to the base branch Parameters: pr_query(str): a string which contains the PR title and the PR body. The title is the first line in the string, and the body are the rest of the string. For example, "Updated README made changes to add info" Returns: str: A success or failure message zXCannot make a pull request because commits are already in the master branch rNzcreated-by-agent) source_branch target_branchr1rPlabelszSuccessfully created PR number z*Unable to make pull request due to error: ) rrsplitr@r mergerequestscreater3 ExceptionrA)r5pr_queryr1rIpres r.create_pull_requestz$GitLabAPIWrapper.create_pull_requests  " "d&8&8 88 N t,Q/E Q 01..<<CC)-););)-)@)@!&'+#5"6 9AA NDs1vMM NsA3B B1 B,&B1,B1cZt|jdd}|tt|dzd} |jj j |}|jjd|idt|zS#t$r}dt|zcYd}~Sd}~wwxYw)af Adds a comment to a gitlab issue Parameters: comment_query(str): a string which contains the issue number, two newlines, and the comment. for example: "1 Working on it now" adds the comment "working on it now" to issue 1 Returns: str: A success or failure message z rrWNrIzCommented on issue z%Unable to make comment due to error: ) intr[r@rArr6r*rMr]r^)r5 comment_queryrQrSr8ras r.comment_on_issuez!GitLabAPIWrapper.comment_on_issues=..v6q9: C $5 6 : <= E--4488FE KK  0 1(3|+<< < E;c!fD D EsAB B* B%B*%B*c|j|jk(rd|jdS|jdd}|t|dzd} |jj j ||jd|dS#t$rA|jd |z||d }|jj j|d |zcYSwxYw) a Creates a new file on the gitlab repo Parameters: file_query(str): a string which contains the file path and the file contents. The file path is the first line in the string, and the contents are the rest of the string. For example, "hello_world.md # Hello World!" Returns: str: A success or failure message +You're attempting to commit directlyto the F branch, which is protected. Please create a new branch and try again.rVrrWNzFile already exists at z. Use update_file insteadCreate )branchcommit_message file_pathcontentz Created file ) rrr[r@rfilesr*r^r])r5 file_queryrm file_contentsdatas r. create_filezGitLabAPIWrapper.create_files   !8!8 81123<<  $$T*1- "3y>A#5#78  /  % % + + / / 4;M;M N,YK7PQ Q /,,"+i"7&( D  % % + + 2 24 8"Y. . /s5BACCc|jjj||j}|j j dS)z Reads a file from the gitlab repo Parameters: file_path(str): the file path Returns: str: The file decoded as a string zutf-8)rror*rdecode)r5rmfiles r. read_filezGitLabAPIWrapper.read_files?((..229d>P>PQ{{}##G,,r:cn|j|jk(rd|jdS |jdd}|jddjddj}|jddjd dj}|j |}|j ||}||k(r y |jd |zd ||d gd}|j jj|d|zS#t$r}dt|zcYd}~Sd}~wwxYw)a8 Updates a file with new content. Parameters: file_query(str): Contains the file path and the file contents. The old file contents is wrapped in OLD <<<< and >>>> OLD The new file contents is wrapped in NEW <<<< and >>>> NEW For example: test/hello.txt OLD <<<< Hello Earth! >>>> OLD NEW <<<< Hello Mars! >>>> NEW Returns: A success or failure message rhrirVrzOLD <<<>>> OLDzNEW <<<>>> NEWzFile content was not updated because old content was not found.It may be helpful to use the read_file action to get the current file contents.rjupdate)actionrmrn)rkrlactionsz Updated file z$Unable to update file due to error: N) rrr[striprwreplacercommitsr]r^rA) r5rprmold_file_contentsnew_file_contents file_contentupdated_file_contentcommitras r. update_filezGitLabAPIWrapper.update_filesj$   !8!8 81123<<  $ D"((.q1I  ,Q/55jA!DJJL   ,Q/55jA!DJJL  >>)4L#/#7#7!#4$ 331,,"+i"7#+%.#7 F  % % - - 4 4V <"Y. . D:SVC C Ds%B%DAD D4 D/)D4/D4c|j|jk(rd|jdS |jjj ||jd|zd|zS#t $r}dt |zcYd}~Sd}~wwxYw)z Deletes a file from the repo Parameters: file_path(str): Where the file is Returns: str: Success or failure message rhrizDelete z Deleted file z$Unable to delete file due to error: N)rrrrodeleter^rA)r5rmras r. delete_filezGitLabAPIWrapper.delete_file8s   !8!8 81123<<   D  % % + + 2 24--y9/D #Y. . D:SVC C Ds8A$$ B- B:BBcR|jy|j|jS)z Get the list of files in the main branch of the repository Returns: str: A plaintext report containing the list of files in the repository in the main branch z-No base branch set. Please set a base branch.)r _list_filesr5s r.list_files_in_main_branchz*GitLabAPIWrapper.list_files_in_main_branchNs*  " " *B 7 788r:cR|jy|j|jS)z Get the list of files in the active branch of the repository Returns: str: A plaintext report containing the list of files in the repository in the active branch *No active branch set. Please set a branch.rrrs r.list_files_in_bot_branchz)GitLabAPIWrapper.list_files_in_bot_branchZs*    %? 2 233r:cV|jy|j|j|S)a Get the list of files in the active branch of the repository from a specific directory Returns: str: A plaintext report containing the list of files in the repository in the active branch from the specified directory rrkpathr)r5rs r.list_files_from_directoryz*GitLabAPIWrapper.list_files_from_directoryfs7    %?%%   r:c |j||}|r%dj|}dt|d|d|Sd|dS#t$r}d|cYd}~Sd}~wwxYw) NrrVr>z files in branch `z`: zNo files found in branch: ``zError: )_get_repository_filesjoinr@r^)r5rkrro files_strras r.rzGitLabAPIWrapper._list_filesvs~ !../E IIe, E |+=fXT)UU4VHA>> !QC=  !s"9AA A AAAc|jj||}g}|D]C}|ddk(r%|j|j||d0|j |dE|S)N)refrtypetreer)rrepository_treeextendrr4)r5rkr repo_contentsrorns r.rz&GitLabAPIWrapper._get_repository_filessm11AAfSWAX $Gv&( T77PQ WV_- %  r:cddlm}d}|}t|D]W} |jjj ||j d}|j|_d|jdcSd|d|dS#|$rP}|jdk(rd|jvr |d z }|d |}ntd |td |Yd }~d }~wwxYw)z Create a new branch in the repository and set it as the active branch Parameters: proposed_branch_name (str): The name of the new branch to be created Returns: str: A success or failure message r)GitlabCreateErrord)rkrzBranch 'z9' created successfully, and set as current active branch.izBranch already existsrK_vz Failed to create branch. Error: z8Unable to create branch name from proposed_branch_name: Nz"Unable to create branch. At least z? branches exist with named derived from proposed_branch_name: `r) rrrangerbranchesr]rname response_code error_messageprintr^)r5proposed_branch_namer max_attemptsnew_branch_nameiresponseras r. create_branchzGitLabAPIWrapper.create_branchs - .|$A 44==DD"1#11&.]]"x}}o.NN%>1?&&:%;1 > % OOs*/1??BFA)=(>b&DOAC  Cc|jjjdDcgc]}|j}}|r+dj |}dt t |d|Sycc}w)z Get the list of branches in the repository Returns: str: A plaintext report containing the number of branches and each branch name TallrVr>z branches in the repository: z#No branches found in the repository)rrr?rrrAr@)r5rkr branches_strs r.list_branches_in_repoz&GitLabAPIWrapper.list_branches_in_repos'+&?&?&H&H&M&MRV&M&W &WFFKK&W   99X.LS]+,-!N$ 5 sA,c|jjjdDcgc]}|j}}||vr ||_d|dSd|dt |Scc}w)zEquivalent to `git checkout branch_name` for this Agent. Clones formatting from Gitlab. Returns an Error (as a string) if branch doesn't exist. TrzSwitched to branch `rzError z/ does not exist,in repo with current branches: )rrr?rrrA)r5 branch_namerk curr_branchess r.set_active_branchz"GitLabAPIWrapper.set_active_branchs33<<AAB  KK  - '!,D )+a8 8 &225m2D1EG  sA c|dk(r|jS|dk(r-tj|jt |S|dk(r|j |S|dk(r|j |S|dk(r|j|S|dk(r|j|S|dk(r|j|S|dk(r|j|S|d k(r|j|S|d k(r|jS|d k(r|j|S|d k(r|jS|d k(r|jS|dk(r|j!|St#d|z)NrDrTrfrsrbrwrrrrrrrrz Invalid mode)rDjsondumpsrTrdrfrsrbrwrrrrrrrr ValueError)r5rquerys r.runzGitLabAPIWrapper.runsv < ??$ $ [ ::dnnSZ89 9 ' '((/ / ] "##E* * * *++E2 2 [ >>%( ( ] "##E* * ] "##E* * _ $%%e, , , ,--/ / ( ())%0 0 0 0113 3 / /002 2 0 011%8 8^d23 3r:)r,rreturnr)r6z List[Issue]rz List[dict])rrA)rQrdrzDict[str, Any])r_rArrA)rerArrA)rprArrA)rmrArrA)rrArrA))rkrArrArrA)rkrArrArz List[str])rrArrA)rrArrA)rrArrArrA)#__name__ __module__ __qualname____doc__r__annotations__rrrrrrr model_configr classmethodr/r9rDrTrbrfrsrwrrrrrrrrrrrr:r.rrs!FC $#$ $J $)'+}+P26 -6P#'M=')- ,L(#.$.` .$ @N>E( /D -<D|D, 9 4 ! / b5&*4r:r)r __future__rrtypingrrrrrlangchain_core.utilsr pydanticr r r gitlab.v4.objectsr rrr:r.rs1" ;;5;;'x4yx4r: