adjustments to pypi

docs/docs/installation/locally.md

@@ -1,3 +1,46 @@
+## Using Pip package
+
+1. Install the package:
+
+```
+pip install pr-agent
+```
+
+2. Run the relevant tool. Make sure to fill in the required parameters (`user_token`, `openai_key`, `pr_url`, `command`):
+
+```python
+from pr_agent import cli
+from pr_agent.config_loader import get_settings
+from pr_agent.log import setup_logger
+
+setup_logger()
+
+
+def main():
+    # Fill in the following values
+    provider = "github" # GitHub provider
+    user_token = "..."  # GitHub user token
+    openai_key = "..."  # OpenAI key
+    pr_url = "..."      # PR URL, for example 'https://github.com/Codium-ai/pr-agent/pull/809'
+    command = "/review" # Command to run (e.g. '/review', '/describe', '/ask="What is the purpose of this PR?"', ...)
+
+    # Setting the configurations
+    get_settings().set("CONFIG.git_provider", provider)
+    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)
+
+
+if __name__ == '__main__':
+    main()
+```
+
 ## Use Docker image (no installation required)
 
 A list of the relevant tools can be found in the [tools guide](../tools/ask.md).

docs/docs/usage-guide/automations_and_usage.md

@@ -1,5 +1,5 @@
 ## Local repo (CLI)
-When running from your local repo (CLI), your local configuration file will be used.
+When running from your locally cloned PR-Agent repo (CLI), your local configuration file will be used.
 Examples of invoking the different tools via the CLI:
 
 - **Review**:       `python -m pr_agent.cli --pr_url=<pr_url>  review`

pr_agent/cli.py

@@ -9,9 +9,7 @@ from pr_agent.log import setup_logger
 log_level = os.environ.get("LOG_LEVEL", "INFO")
 setup_logger(log_level)
 
-
-
-def run(inargs=None):
+def set_parser():
     parser = argparse.ArgumentParser(description='AI based pull request analyzer', usage=
     """\
     Usage: cli.py --pr-url=<URL on supported git hosting service> <command> [<args>].
@@ -50,6 +48,12 @@ For example: 'python cli.py --pr_url=... review --pr_reviewer.extra_instructions
     parser.add_argument('--issue_url', type=str, help='The URL of the Issue to review', default=None)
     parser.add_argument('command', type=str, help='The', choices=commands, default='review')
     parser.add_argument('rest', nargs=argparse.REMAINDER, default=[])
+    return parser
+
+
+def run(inargs=None, args=None):
+    parser = set_parser()
+    if not args:
         args = parser.parse_args(inargs)
     if not args.pr_url and not args.issue_url:
         parser.print_help()

pr_agent/cli_pip.py

@@ -0,0 +1,30 @@
+from pr_agent import cli
+from pr_agent.config_loader import get_settings
+from pr_agent.log import setup_logger
+
+setup_logger()
+
+
+def main():
+    # Fill in the following values
+    provider = "github"  # GitHub provider
+    user_token = "..."  # GitHub user token
+    openai_key = "..."  # OpenAI key
+    pr_url = "..."  # PR URL, for example 'https://github.com/Codium-ai/pr-agent/pull/809'
+    command = "/review"  # Command to run (e.g. '/review', '/describe', '/ask="What is the purpose of this PR?"')
+
+    # Setting the configurations
+    get_settings().set("CONFIG.git_provider", provider)
+    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)
+
+
+if __name__ == '__main__':
+    main()

pr_agent/tools/pr_questions.py

@@ -17,6 +17,7 @@ from pr_agent.servers.help import HelpMessage
 class PRQuestions:
     def __init__(self, pr_url: str, args=None, ai_handler: partial[BaseAiHandler,] = LiteLLMAIHandler):
         question_str = self.parse_args(args)
+        self.pr_url = pr_url
         self.git_provider = get_git_provider()(pr_url)
         self.main_pr_language = get_main_pr_language(
             self.git_provider.get_languages(), self.git_provider.get_files()
@@ -49,7 +50,7 @@ class PRQuestions:
         return question_str
 
     async def run(self):
-        get_logger().info('Answering a PR question...')
+        get_logger().info(f'Answering a PR question about the PR {self.pr_url} ')
         relevant_configs = {'pr_questions': dict(get_settings().pr_questions),
                             'config': dict(get_settings().config)}
         get_logger().debug("Relevant configs", artifacts=relevant_configs)

pr_agent/tools/pr_similar_issue.py

@@ -3,9 +3,6 @@ from enum import Enum
 from typing import List
 
 import openai
-import pandas as pd
-import pinecone
-from pinecone_datasets import Dataset, DatasetMetadata
 from pydantic import BaseModel, Field
 
 from pr_agent.algo import MAX_TOKENS
@@ -36,6 +33,12 @@ class PRSimilarIssue:
         index_name = self.index_name = "codium-ai-pr-agent-issues"
 
         if get_settings().pr_similar_issue.vectordb == "pinecone":
+            try:
+                import pinecone
+                from pinecone_datasets import Dataset, DatasetMetadata
+                import pandas as pd
+            except:
+                raise Exception("Please install 'pinecone' and 'pinecone_datasets' to use pinecone as vectordb")
             # assuming pinecone api key and environment are set in secrets file
             try:
                 api_key = get_settings().pinecone.api_key
@@ -107,7 +110,10 @@ class PRSimilarIssue:
                     get_logger().info('No new issues to update')
         
         elif get_settings().pr_similar_issue.vectordb == "lancedb":
+            try:
                 import lancedb # import lancedb only if needed
+            except:
+                raise Exception("Please install lancedb to use lancedb as vectordb")
             self.db = lancedb.connect(get_settings().lancedb.uri)
             self.table = None