o ȳg@sdZddlmZddlZddlmZmZmZmZm Z ddl Z ddl m Z ddl mZmZmZerd?ZdmdAdBZdndDdEZ dodGdHZ!dpdJdKZ"dodLdMZ#dpdNdOZ$dqdQdRZ%dqdSdTZ&drdVdWZ'dedXdYZ(dedZd[Z)dsd]d^Z*dtd`daZ+dS)uGitHubAPIWrapperzWrapper for GitHub API.Nrgithubgithub_repo_instancez Optional[str]github_repository github_app_idgithub_app_private_key active_branchgithub_base_branchforbid)extrabefore)modevaluesrrc Cs|t|dd}t|dd}t|dd}z ddlm}m}Wn ty'td wzt|d  }|}Wd n1s=wYWn tyN|}Ynw|||} || d } | } | skt d |d|dz| d} Wnt y} zt d| d } ~ ww| } | |}t|dd|j d}t|dd|j d}| |d<||d<||d<||d<||d<||d<||d<|S)z?Validate that api key and python package exists in environment.rGITHUB_REPOSITORYr GITHUB_APP_IDrGITHUB_APP_PRIVATE_KEYr)AuthGithubIntegrationzHPyGithub is not installed. Please install it with `pip install PyGithub`rN)authz;Please make sure to install the created github app with id z on the repo: zrMore instructions can be found at https://docs.github.com/en/apps/using-github-apps/installing-your-own-github-appzBPlease make sure to give correct github parameters Error message: rGITHUB_BASE_BRANCH)defaultr ACTIVE_BRANCHrr)rrr$r%ropenread ExceptionAppAuthget_installations ValueErrorget_github_for_installationget_repodefault_branch)clsr rrrr$r%f private_keyr'gi installationegreporrrrrvalidate_environment,s        z%GitHubAPIWrapper.validate_environmentissues List[Issue] List[dict]cCsTg}|D]#}|j}|j}|jr|jjnd}||d}|dur"||d<||q|S) Extracts title and number from each Issue and puts them in a dictionary Parameters: issues(List[Issue]): A list of Github Issue objects Returns: List[dict]: A dictionary of issue titles and numbers N)titlenumber opened_by)rArBuserloginappend)selfr=parsedissuerArBrC issue_dictrrr parse_issues|s  zGitHubAPIWrapper.parse_issues pull_requestsList[PullRequest]cCs6g}|D]}||j|jt|jt|jdq|S)r@)rArBcommitscomments)rFrArBstrrNrO)rGrLrHprrrrparse_pull_requestssz$GitHubAPIWrapper.parse_pull_requestsrPcCsN|jjdd}dd|D}|r%||}dtt|dt|}|SdS)z Fetches all open issues from the repo excluding pull requests Returns: str: A plaintext report containing the number of issues and each issue's title and number. r+statecSsg|]}|js|qSr) pull_request).0rIrrr sz/GitHubAPIWrapper.get_issues..Found z issues: zNo open issues available)r get_issuesrKrPlen)rGr= parsed_issuesparsed_issues_strrrrrYs zGitHubAPIWrapper.get_issuescCsF|jjdd}|jdkr!||}dtt|dt|}|SdS)z Fetches all open PRs from the repo Returns: str: A plaintext report containing the number of PRs and each PR's title and number. r+rSrrXz pull requests: zNo open pull requests available)r get_pulls totalCountrRrPrZ)rGrL parsed_prsparsed_prs_strrrrlist_open_pull_requestss  z(GitHubAPIWrapper.list_open_pull_requestsc Csg}z8|jjd|jd}|D]}|jdkr|||jq||jq|r8d|}dt |d|WSWdSt yP}z t |WYd}~Sd}~ww) z Fetches all files in the main branch of the repo. Returns: str: A plaintext report containing the paths and names of the files. refdir rXz files in the main branch: z!No files found in the main branchN) r get_contentsrtypeextend _list_filespathrFjoinrZr-rPrGfilescontentscontent files_strr9rrrlist_files_in_main_branchs"  z*GitHubAPIWrapper.list_files_in_main_branch branch_namecCsBdd|jD}||vr||_d|dSd|dt|S)zEquivalent to `git checkout branch_name` for this Agent. Clones formatting from Github. Returns an Error (as a string) if branch doesn't exist. cSg|]}|jqSrnamerVbranchrrrrWz6GitHubAPIWrapper.set_active_branch..zSwitched to branch ``zError z/ does not exist,in repo with current branches: )r get_branchesrrP)rGrs curr_branchesrrrset_active_branchs z"GitHubAPIWrapper.set_active_branchc Cslzdd|jD}|rd|}dt|d|WSWdSty5}z t|WYd}~Sd}~ww)z Fetches a list of all branches in the repository. Returns: str: A plaintext report containing the names of the branches. cSrtrrurwrrrrWryz:GitHubAPIWrapper.list_branches_in_repo..rfrXz branches in the repository: z#No branches found in the repositoryN)rr{rlrZr-rP)rGbranches branches_strr9rrrlist_branches_in_repos  z&GitHubAPIWrapper.list_branches_in_repoproposed_branch_namec Csddlm}d}|}|j|jj}tdD]S}z|jjd||jjd||_ d|dWS|yi}z,|j dkrQd |j d vrQ|d 7}|d |}nt d |t d|WYd}~qd}~wwd|dS)a5 Create a new branch, and set it as the active bot branch. Equivalent to `git switch -c proposed_branch_name` If the proposed branch already exists, we append _v1 then _v2... until a unique name is found. Returns: str: A plaintext success message. rGithubExceptioniz refs/heads/)rdshazBranch 'z9' created successfully, and set as current active branch.izReference already existsmessage_vz Failed to create branch. Error: z8Unable to create branch name from proposed_branch_name: NzeUnable to create branch. At least 1000 branches exist with named derived from proposed_branch_name: `rz)rrr get_branchr3rangecreate_git_refcommitrrstatusdataprintr-)rGrrinew_branch_name base_branchr9rrr create_branch s>     zGitHubAPIWrapper.create_branchc Csg}zA|jjd|jd}|D]}|jdkr|||jq||jq|rt | j d} n td| j dWq't| } | j dkrW| j } n td | j d Wq't|d | | jd }|||kr|| j| | j| jd ||7}Wq'ty}z td|WYd}~q'd}~ww|d7}q)avFetches the full text of all files in a PR. Truncates after first 3k tokens. # TODO: Enhancement to summarize files with ctags if they're getting long. Args: pr_number(int): The number of the pull request on Github Returns: dict: A dictionary containing the issue's title, body, and comments as a string i rrT download_urlzFailed to download file: z , skippingz'Failed downloading file content (Error z ). Skipping cl100k_basezfile_name file_contents)filenamero additions deletionsz.Error when reading files from a PR on github. Nr)rrget_pullr get_filesrrZrequestsget contents_url status_codejsonloadstextr get_encodingencoderrFrrr-)rGrrMAX_TOKENS_FOR_FILESpr_filesrQ total_tokensr files_pagefilefile_metadata_responserfile_content_response file_content file_tokensr9rrrlist_pull_request_filessd  )        z(GitHubAPIWrapper.list_pull_request_filescsd|jj|d}dddd dfdd }i}||d|j||dt|||d|jr3|jndg}d}t|dkr|||}t|dkrNn.|D]!}t|j|jj d} | krfn | | | 7qP|d7}t|dks@||dt|g} d}t| dkr| |} t| dkrn,| D]} td| j j i} | krn | | | 7q|d7}t| dks||dt| |S)a} Fetches a specific pull request and its first 10 comments, limited by max_tokens. Parameters: pr_number(int): The number for the Github pull max_tokens(int): The maximum number of tokens in the response Returns: dict: A dictionary containing the pull's title, body, and comments as a string irrrrPrrcSst}t|d|S)Nr)rrZrr)rrrrr get_tokenssz5GitHubAPIWrapper.get_pull_request..get_tokens data_dictrkeyvalueNonecs,|}|kr|||<|7dSdSNr)rrrtokensr max_tokensrrr add_to_dicts   z6GitHubAPIWrapper.get_pull_request..add_to_dictrArBrrbrrrrOrrNN)rrPrr)rrrrPrrPrr)rrrArPrrZget_issue_commentsrrDrErF get_commitsrr)rGrpullr response_dictrOrrr comment_strrN commits_pager commit_strrrrget_pull_requestsP            z!GitHubAPIWrapper.get_pull_requestpr_queryc Cs|j|jkrdSz$|dd}|t|dd}|jj|||j|jd}d|jWStyD}z dt|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 zaCannot make a pull request because commits are already in the main or master branch.rfrN)rArheadbasezSuccessfully created PR number z*Unable to make pull request due to error: ) rrsplitrZr create_pullrBr-rP)rGrrArrQr9rrrcreate_pull_requests z$GitHubAPIWrapper.create_pull_request comment_queryc Cst|dd}|tt|dd}z|jj|d}||dt|WSty@}z dt|WYd}~Sd}~ww)af Adds a comment to a github 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 rrNrzCommented on issue z%Unable to make comment due to error: )rrrZrPrrcreate_commentr-)rGrrrrIr9rrrcomment_on_issue0s  z!GitHubAPIWrapper.comment_on_issue file_queryc Cs|j|jkr d|jdS|dd}|t|dd}z5z|jj||jd}|r7d|d |jd WWSWn tyAYnw|jj|d |||jd d |WStyk}z dt|WYd}~Sd}~ww)a Creates a new file on the Github 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 z2You're attempting to commit to the directly to theF branch, which is protected. Please create a new branch and try again.rfrrNrczFile already exists at `z ` on branch `z+`. You must use `update_file` to modify it.zCreate )rkrrprxz Created file z"Unable to make file due to error: ) rrrrZrrgr- create_filerP)rGr file_path file_contentsrr9rrrrDs@   zGitHubAPIWrapper.create_filerc Csbz|jj||jd}|jdWSty0}zd|d|jdt|WYd}~Sd}~ww)a Read a file from this agent's branch, defined by self.active_branch, which supports PR branches. Parameters: file_path(str): the file path Returns: str: The file decoded as a string, or an error message if not found rczutf-8zFile not found `z ` on branch`z `. Error: N)rrgrdecoded_contentdecoder-rP)rGrrr9rrr read_filers zGitHubAPIWrapper.read_filec Cs|j|jkr d|jdSzU|dd}|dddd}|ddd d}||}|||}||krD Wd S|jj|d t|||j|jj ||jd j d dt|WSt yz}z dt|WYd}~Sd}~ww)a9 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 2You're attempting to commit to the directlyto the rrfrzOLD <<<>>> 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.zUpdate rc)rkrrprxrz Updated file z$Unable to update file due to error: N) rrrstriprreplacer update_filerPrgrr-)rGrrold_file_contentsnew_file_contentsrupdated_file_contentr9rrrrsB    zGitHubAPIWrapper.update_filec Cs|j|jkr d|jdSz|jj|d||j|jj||jdjdd|WSty?}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 rrzDelete rc)rkrrxrz Deleted file z$Unable to delete file due to error: N)rrr delete_filergrr-rP)rGrr9rrrrs(  zGitHubAPIWrapper.delete_filequeryc Csj|jj||jd}td|j}d|dg}|d|D]}|d|jd|jd|jqd |S) z Searches issues and pull requests in the repository. Parameters: query(str): The search query Returns: str: A string containing the first 5 issues and pull requests )r;Top results:NTitle: z , Number: z , State: rf) r search_issuesrminr^rFrArBrTrl)rGr search_result max_itemsresultsrIrrrsearch_issues_and_prss  z&GitHubAPIWrapper.search_issues_and_prscCs|jj||jd}|jdkrdStd|j}d|d|jdg}d}|D]&}||kr-n|jj|j|jdj }| d |jd |d |d 7}q%d |S)z Searches code in the repository. # Todo: limit total tokens returned... Parameters: query(str): The search query Returns: str: A string containing, at most, the top 5 search results )rr;rz0 results found.rz Showing top z of rrcz Filepath: `z` File contents: z rrf) r search_coderr^rrrgrkrrrrFrl)rGrr max_resultsrcountcoderrrrrs.      zGitHubAPIWrapper.search_codereviewer_usernamec sjjddd}tfdd|Dd}|durdjdSz|j|gd d |d |jWStyG}z d |WYd}~Sd}~ww) a; Creates a review request on *THE* open pull request that matches the current active_branch. Parameters: reviewer_username(str): The username of the person who is being requested Returns: str: A message confirming the creation of the review request r+created)rTsortc3s"|] }|jjjkr|VqdSr)rrdr)rVrQrGrr s z9GitHubAPIWrapper.create_review_request..Nz3No open pull request found for the current branch `rz) reviewersz Review request created for user z on PR #z-Failed to create a review request with error )rr]nextrcreate_review_requestrBr-)rGrrLrQr9rrrrs* z&GitHubAPIWrapper.create_review_requestcCs&|j}d|jd|jd|jS)zu Fetches the latest release of the repository. Returns: str: The latest release zLatest title:  tag:  body: )rget_latest_releaserAtag_namer)rGreleaserrrr0s  z#GitHubAPIWrapper.get_latest_releasec Csb|j}td|j}d|dg}|d|D]}|d|jd|jd|jqd|S) zi Fetches all releases of the repository. Returns: str: The releases rrrNrz, Tag: z, Body: rf) r get_releasesrr^rFrArrrl)rGreleasesrrrrrrr >s    zGitHubAPIWrapper.get_releasesrcCs(|j|}d|jd|jd|jS)z Fetches a specific release of the repository. Parameters: tag_name(str): The tag name of the release Returns: str: The release z Release: rr)r get_releaserArr)rGrrrrrr Qs zGitHubAPIWrapper.get_releasercCs|dkrt|t|S|dkrt|t|S|dkr*t|t|S|dkr2|S|dkr;||S|dkrD||S|dkrM| |S|dkrV| |S|d kr_| |S|d krh| |S|d krp| S|d krx|S|d kr|S|dkr|S|dkr||S|dkr||S|dkr||S|dkr||S|dkr||S|dkr||S|dkr|S|dkr|S|dkr||Std|)NrrrrYrrrrrrrarrrrr}rrrrrrr r z Invalid mode)rdumpsrrrrrYrrrrrrrarrrrr}rrrrrrr r r0)rGrrrrrrunbs^              zGitHubAPIWrapper.run)r rrr)r=r>rr?)rLrMrr?)rrP)rsrPrrP)rrPrrP)rrPrrP)rrPrr)rrrr)rrrr)rrrr)rrPrrP)rrPrrP)rrPrrP)rrPrrP)rrPrrP)rrPrrP)rrPrrP)rrPrrPrrP),__name__ __module__ __qualname____doc__r__annotations__rrrrrrr model_configr classmethodr<rKrRrYrarrr}rrrrrjrrrrrrrrrrrrrr r r rrrrrsT         N        ,    ! ? ?   .  :   ! !  r)rr)r __future__rrtypingrrrrrrlangchain_core.utilsrpydanticr r r github.Issuer github.PullRequestr rrrrrrs