o ȳgwJ@sxdZddlmZddlZddlmZmZmZmZm Z ddl m Z ddl m Z mZmZer2ddlmZGdd d e ZdS) zUtil that calls gitlab.) annotationsN) TYPE_CHECKINGAnyDictListOptional)get_from_dict_or_env) BaseModel ConfigDictmodel_validator)Issuec@sJeZdZUdZdZded<dZded<dZded< dZded< dZ ded < dZ ded < dZ ded < e d d Z eddedLddZdMddZdNddZdOd d!ZdPd#d$ZdQd&d'ZdRd)d*ZdSd,d-ZdRd.d/ZdSd0d1ZdNd2d3ZdNd4d5ZdTd7d8ZdUdVd;d<ZdUdWd>d?ZdXdAdBZdNdCdDZ dYdFdGZ!dZdJdKZ"dS)[GitLabAPIWrapperzWrapper for GitLab API.Nrgitlabgitlab_repo_instancez Optional[str] gitlab_urlgitlab_repositorygitlab_personal_access_token gitlab_branchgitlab_base_branchforbid)extrabefore)modevaluesrreturnc Cst|dddd}t|dd}t|dd}t|d d d d}t|d d d d}zddl}Wn ty5tdw|j||dd}|||d<|j||d<||d<||d<||d<||d <||d <|S)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)rr ImportErrorGitlabauthprojectsget) clsrrrrrrrgr,`/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/gitlab.pyvalidate_environment'sH  z%GitLabAPIWrapper.validate_environmentissues List[Issue] List[dict]cCs.g}|D]}|j}|j}|||dq|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)r2iidappend)selfr/parsedissuer2r3r,r,r- parse_issuesYs zGitLabAPIWrapper.parse_issuesstrcCsJ|jjjdd}t|dkr#||}dtt|dt|}|SdS)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)rr/listlenr9r:)r6r/ parsed_issuesparsed_issues_strr,r,r- get_issueshs  zGitLabAPIWrapper.get_issues issue_numberintDict[str, Any]cCs|jj|}d}g}t|dkrA|jj|d}t|dkrn"|D]}|j|j}||j|j ddq!|d7}t|dks|j |j t |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)r2rIcomments) rr/r)r?notesr>idr5rIauthorr2 descriptionr:)r6rCr8rGrL comments_pagecommentr,r,r- get_issuezs$     zGitLabAPIWrapper.get_issuepr_queryc Cs|j|jkrdSz(|dd}|t|dd}|jj|j|j||dgd}d|jWStyH}z d t |WYd}~Sd}~ww) 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_branchr2rPlabelszSuccessfully created PR number z*Unable to make pull request due to error: ) rrsplitr?r mergerequestscreater4 Exceptionr:)r6rTr2rIprer,r,r-create_pull_requests$  z$GitLabAPIWrapper.create_pull_request comment_queryc Cst|dd}|tt|dd}z|jj|}|jd|idt|WSt yC}z dt|WYd}~Sd}~ww)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 rrVNrIzCommented on issue z%Unable to make comment due to error: ) rDrZr?r:rr/r)rMr\r])r6rarCrRr8r_r,r,r-comment_on_issues z!GitLabAPIWrapper.comment_on_issue file_querycCs|j|jkr d|jdS|dd}|t|dd}z|jj||jd|dWStyL|jd |||d }|jj|d |YSw) 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.rUrrVNzFile already exists at z. Use update_file insteadCreate )branchcommit_message file_pathcontentz Created file ) rrrZr?rfilesr)r]r\)r6rcri file_contentsdatar,r,r- create_files&   zGitLabAPIWrapper.create_filericCs |jj||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)rrkr)rdecode)r6rifiler,r,r- read_fileszGitLabAPIWrapper.read_filec Cs|j|jkr d|jdSzP|dd}|dddd}|ddd d}||}|||}||krD Wd S|jd |d ||d gd}|jj|d|WSt yu}z dt |WYd}~Sd}~ww)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 rdrerUrzOLD <<<>>> 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.rfupdate)actionrirj)rgrhactionsz Updated file z$Unable to update file due to error: N) rrrZstriprqreplacercommitsr\r]r:) r6rcriold_file_contentsnew_file_contents file_contentupdated_file_contentcommitr_r,r,r- update_files@   zGitLabAPIWrapper.update_filec Csn|j|jkr d|jdSz|jj||jd|d|WSty6}z dt|WYd}~Sd}~ww)z Deletes a file from the repo Parameters: file_path(str): Where the file is Returns: str: Success or failure message rdrezDelete z Deleted file z$Unable to delete file due to error: N)rrrrkdeleter]r:)r6rir_r,r,r- delete_file5s   zGitLabAPIWrapper.delete_filecC|jdurdS||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 Nz-No base branch set. Please set a base branch.)r _list_filesr6r,r,r-list_files_in_main_branchK  z*GitLabAPIWrapper.list_files_in_main_branchcCr)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 N*No active branch set. Please set a branch.rrrr,r,r-list_files_in_bot_branchWrz)GitLabAPIWrapper.list_files_in_bot_branchpathcCs|jdurdS|j|j|dS)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 Nrrgrr)r6rr,r,r-list_files_from_directorycs z*GitLabAPIWrapper.list_files_from_directoryrgc Csvz#|j||d}|rd|}dt|d|d|WSd|dWSty:}z d|WYd}~Sd}~ww) NrrUr=z files in branch `z`: zNo files found in branch: ``zError: )_get_repository_filesjoinr?r])r6rgrrk files_strr_r,r,r-rss zGitLabAPIWrapper._list_files List[str]cCsT|jj||d}g}|D]}|ddkr ||||dq ||dq |S)N)refrtypetreer)rrepository_treeextendrr5)r6rgr repo_contentsrkrjr,r,r-rs z&GitLabAPIWrapper._get_repository_filesproposed_branch_namec Csddlm}d}|}t|D]Q}z|jj||jd}|j|_d|jdWS|y_}z*|jdkrGd|j vrG|d 7}|d |}nt d |t d |WYd }~qd }~wwd|d|dS)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)rgrzBranch '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])r6rr max_attemptsnew_branch_nameiresponser_r,r,r- create_branchs>      zGitLabAPIWrapper.create_branchcCsDdd|jjjddD}|r d|}dtt|d|SdS) z Get the list of branches in the repository Returns: str: A plaintext report containing the number of branches and each branch name cSg|]}|jqSr,r.0rgr,r,r- sz:GitLabAPIWrapper.list_branches_in_repo..TallrUr=z branches in the repository: z#No branches found in the repository)rrr>rr:r?)r6r branches_strr,r,r-list_branches_in_repos z&GitLabAPIWrapper.list_branches_in_repo branch_namecCsHdd|jjjddD}||vr||_d|dSd|dt|S) zEquivalent to `git checkout branch_name` for this Agent. Clones formatting from Gitlab. Returns an Error (as a string) if branch doesn't exist. cSrr,rrr,r,r-rsz6GitLabAPIWrapper.set_active_branch..TrzSwitched to branch `rzError z/ does not exist,in repo with current branches: )rrr>rr:)r6r curr_branchesr,r,r-set_active_branchs z"GitLabAPIWrapper.set_active_branchrquerycCs |dkr|S|dkrt|t|S|dkr||S|dkr(||S|dkr1||S|dkr:||S|dkrC| |S|dkrL| |S|d krU| |S|d kr]| S|d krf| |S|d krn|S|d krv|S|dkr||Std|)NrBrSrbrnr`rqr}rrrrrrrz Invalid mode)rBjsondumpsrSrDrbrnr`rqr}rrrrrrr ValueError)r6rrr,r,r-runs:          zGitLabAPIWrapper.run)rrrr)r/r0rr1)rr:)rCrDrrE)rTr:rr:)rar:rr:)rcr:rr:)rir:rr:)rr:rr:)r)rgr:rr:rr:)rgr:rr:rr)rr:rr:)rr:rr:)rr:rr:rr:)#__name__ __module__ __qualname____doc__r__annotations__rrrrrrr model_configr classmethodr.r9rBrSr`rbrnrqr}rrrrrrrrrrr,r,r,r-r sL         0      " >    1 r )r __future__rrtypingrrrrrlangchain_core.utilsrpydanticr r r gitlab.v4.objectsr r r,r,r,r-s