xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-04 21:00:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #446 from Codium-ai/tr/fix_cli_args

Browse Source 
Handling CLI Arguments with Quotes in pr_agent
This commit is contained in:
Ori Kotek
2023-11-12 17:29:38 +02:00
committed by GitHub
parent 37e6608e68 9e5e9afe92
commit 1df36c6a44
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pr_agent/agent/pr_agent.py
View File

@ -46,10 +46,13 @@ class PRAgent:
apply_repo_settings(pr_url) apply_repo_settings(pr_url)
# Then, apply user specific settings if exists # Then, apply user specific settings if exists
if isinstance(request, str):
request = request.replace("'", "\\'") request = request.replace("'", "\\'")
lexer = shlex.shlex(request, posix=True) lexer = shlex.shlex(request, posix=True)
lexer.whitespace_split = True lexer.whitespace_split = True
action, *args = list(lexer) action, *args = list(lexer)
else:
action, *args = request
args = update_settings_from_args(args) args = update_settings_from_args(args)
action = action.lstrip("/").lower() action = action.lstrip("/").lower()

6
pr_agent/cli.py
View File

@ -8,6 +8,8 @@ from pr_agent.log import setup_logger
setup_logger() setup_logger()
def run(inargs=None): def run(inargs=None):
parser = argparse.ArgumentParser(description='AI based pull request analyzer', usage= parser = argparse.ArgumentParser(description='AI based pull request analyzer', usage=
"""\ """\
@ -51,9 +53,9 @@ For example: 'python cli.py --pr_url=... review --pr_reviewer.extra_instructions
command = args.command.lower() command = args.command.lower()
get_settings().set("CONFIG.CLI_MODE", True) get_settings().set("CONFIG.CLI_MODE", True)
if args.issue_url: if args.issue_url:
result = asyncio.run(PRAgent().handle_request(args.issue_url, command + " " + " ".join(args.rest))) result = asyncio.run(PRAgent().handle_request(args.issue_url, [command] + args.rest))
else: else:
result = asyncio.run(PRAgent().handle_request(args.pr_url, command + " " + " ".join(args.rest))) result = asyncio.run(PRAgent().handle_request(args.pr_url, [command] + args.rest))
if not result: if not result:
parser.print_help() parser.print_help()