Skip to content

Commit

Permalink
fix linting
Browse files Browse the repository at this point in the history
  • Loading branch information
zackees committed Dec 21, 2023
1 parent 507ab62 commit 7580b4d
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 10 deletions.
22 changes: 17 additions & 5 deletions src/zcmds/cmds/common/aicode.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
import shutil
import subprocess
import sys
from typing import Tuple

from zcmds.cmds.common.openaicfg import create_or_load_config, save_config

Expand All @@ -24,13 +25,15 @@ def install_aider_if_missing() -> None:
assert shutil.which("aider") is not None, "aider not found after install"


def parse_args() -> argparse.Namespace:
def parse_args() -> Tuple[argparse.Namespace, list]:
argparser = argparse.ArgumentParser(usage="Ask OpenAI for help with code")
argparser.add_argument(
"prompt", nargs="*", help="Args to pass onto aider"
) # Changed nargs to '*'
argparser.add_argument("--set-key", help="Set OpenAI key")
argparser.add_argument("--upgrade", action="store_true", help="Upgrade aider using pipx")
argparser.add_argument(
"--upgrade", action="store_true", help="Upgrade aider using pipx"
)

model_group = argparser.add_mutually_exclusive_group()
model_group.add_argument(
Expand All @@ -48,6 +51,11 @@ def parse_args() -> argparse.Namespace:
default=False,
help=f"bleeding edge model: {ADVANCED_MODEL}",
)
model_group.add_argument(
"--auto-commit",
"-a",
action="store_true",
)
model_group.add_argument("--model", default=None)
args, unknown_args = argparser.parse_known_args()
return args, unknown_args
Expand Down Expand Up @@ -90,9 +98,13 @@ def cli() -> int:
os.environ["AIDER_MODEL"] = model
print(f"Starting aider with model {os.environ['AIDER_MODEL']}")
os.environ["OPENAI_API_KEY"] = openai_key
return subprocess.call(
["aider", "--no-auto-commits"] + args.prompt + unknown_args
) # args.prompt and unknown_args are now lists
cmd_list = ["aider"]
if args.auto_commit:
cmd_list.append("--auto-commit")
else:
cmd_list.append("--no-auto-commit")
cmd_list += args.prompt + unknown_args
return subprocess.call(cmd_list)


def main() -> int:
Expand Down
5 changes: 0 additions & 5 deletions src/zcmds/cmds/common/audnorm.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,9 +118,4 @@ def main():


if __name__ == "__main__":
file = r"E:\wetransfer_epstein-1-mov_2023-12-15_2009\THE NEED_RealESR_Gx4_60%_Interpolatation-Low_small.mp4"
out = r"E:\wetransfer_epstein-1-mov_2023-12-15_2009\THE NEED_RealESR_Gx4_60%_Interpolatation-Low_small_norm.mp4"
import sys

sys.argv.extend([file, out])
main()

0 comments on commit 7580b4d

Please sign in to comment.