Adding optional remote_name command argument to ai tool

packages/component_code_gen/code_gen/generate_for_github_issue.py

@@ -57,7 +57,7 @@ def generate_app_file_prompt(requirements, app_file_content):
 {requirements}"""
 
 
-def generate(issue_number, output_dir, generate_pr=True, clean=False, verbose=False, tries=3):
+def generate(issue_number, output_dir, generate_pr=True, clean=False, verbose=False, tries=3, remote_name="origin"):
     repo_path = os.path.abspath(os.path.join("..", ".."))
     output_dir = os.path.abspath(output_dir)
 
@@ -71,7 +71,7 @@ def generate(issue_number, output_dir, generate_pr=True, clean=False, verbose=Fa
             return
 
         branch_name = f"issue-{issue_number}"
-        run_command("git fetch origin")
+        run_command(f"git fetch {remote_name}")
 
         if any(reference.name == branch_name for reference in repo.references):
             # branch name already exists
@@ -80,7 +80,7 @@ def generate(issue_number, output_dir, generate_pr=True, clean=False, verbose=Fa
             # create new branch
             run_command(f"git checkout -b {branch_name}")
 
-        run_command("git reset --hard origin/master")
+        run_command(f"git reset --hard {remote_name}/master")
 
     # parse github issue description
     md = requests.get(
@@ -184,6 +184,6 @@ def generate(issue_number, output_dir, generate_pr=True, clean=False, verbose=Fa
         run_command(["npx pnpm i"])
         run_command(f"git add -f {app_base_path} {repo_path}/pnpm-lock.yaml")
         run_command(f"git commit --no-verify -m '{app} init'")
-        run_command(f"git push -f --set-upstream origin {branch_name}")
+        run_command(f"git push -f --set-upstream {remote_name} {branch_name}")
         run_command(
             f"gh pr create -d -l ai-assisted -t 'New Components - {app}' -b 'Resolves #{issue_number}.'")

packages/component_code_gen/main.py

@@ -27,11 +27,12 @@
                         required=False, default=os.path.join("..", "..", "components"))
     parser.add_argument('--verbose', dest='verbose', help='Set the logging to debug',
                         default=False, action='store_true')
+    parser.add_argument('--remote', dest='remote_name', help='The Git remote name', default='origin')
     args = parser.parse_args()
 
     if args.issue:
         generate(args.issue, output_dir=args.output_dir, generate_pr=not args.skip_pr,
-                 clean=args.clean, verbose=args.verbose, tries=args.tries)
+                 clean=args.clean, verbose=args.verbose, tries=args.tries, remote_name=args.remote_name)
     else:
         if not args.type:
             raise argparse.ArgumentTypeError("--type is required")