adjustments to pypi

docs/docs/installation/locally.md

@@ -27,12 +27,8 @@ def main():
     get_settings().set("openai.key", openai_key)
     get_settings().set("github.user_token", user_token)
 
-    # Preparing the command
-    run_command = f"--pr_url={pr_url} {command.lstrip('/')}"
-    args = cli.set_parser().parse_args(run_command.split())
-
     # Run the command. Feedback will appear in GitHub PR comments
-    cli.run(args=args)
+    cli.run_command(pr_url, command)
 
 
 if __name__ == '__main__':

pr_agent/cli.py

@@ -50,6 +50,13 @@ def set_parser():
     parser.add_argument('rest', nargs=argparse.REMAINDER, default=[])
     return parser
 
+def run_command(pr_url, command):
+    # Preparing the command
+    run_command = f"--pr_url={pr_url} {command.lstrip('/')}"
+    args = set_parser().parse_args(run_command.split())
+
+    # Run the command. Feedback will appear in GitHub PR comments
+    run(args=args)
 
 def run(inargs=None, args=None):
     parser = set_parser()

pr_agent/cli_pip.py

@@ -1,8 +1,5 @@
 from pr_agent import cli
 from pr_agent.config_loader import get_settings
-from pr_agent.log import setup_logger
-
-setup_logger()
 
 
 def main():
@@ -18,12 +15,8 @@ def main():
     get_settings().set("openai.key", openai_key)
     get_settings().set("github.user_token", user_token)
 
-    # Preparing the command
-    run_command = f"--pr_url={pr_url} {command.lstrip('/')}"
-    args = cli.set_parser().parse_args(run_command.split())
-
     # Run the command. Feedback will appear in GitHub PR comments
-    cli.run(args=args)
+    cli.run_command(pr_url, command)
 
 
 if __name__ == '__main__':

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pr-agent"
-version = "0.2.0"
+version = "0.2.1"
 
 authors = [{name= "CodiumAI", email = "[email protected]"}]