Merge pull request #815 from Codium-ai/tr/pypi

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]"}]