mirror of
https://github.com/Aider-AI/aider.git
synced 2025-06-02 02:34:59 +00:00
1475e0d0fd
# Context:
USER: You missed a bunch of explicit $CODER_xxx
ASSISTANT: I apologize for that. I will replace the remaining hardcoded "$CODER_" strings with f-strings that use the `env_prefix` variable.
```python
main.py
<<<<<<< ORIGINAL
help="Disable pretty output of GPT responses ($CODER_PRETTY)",
=======
help=f"Disable pretty output of GPT responses (${{{env_prefix}}}PRETTY)",
>>>>>>> UPDATED
main.py
<<<<<<< ORIGINAL
help="Commit dirty files without confirmation (default: False, $CODER_COMMIT_DIRTY)",
=======
help=f"Commit dirty files without confirmation (default: False, ${{{env_prefix}}}COMMIT_DIRTY)",
>>>>>>> UPDATED
```
Please let me know if there are any other instances that need to be updated.
81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
import os
|
|
import sys
|
|
import argparse
|
|
from dotenv import load_dotenv
|
|
from coder import Coder
|
|
|
|
|
|
def main():
|
|
load_dotenv()
|
|
env_prefix = "CODER_"
|
|
parser = argparse.ArgumentParser(description="Chat with GPT about code")
|
|
parser.add_argument(
|
|
"files",
|
|
metavar="FILE",
|
|
nargs="+",
|
|
help="a list of source code files",
|
|
)
|
|
parser.add_argument(
|
|
"--history-file",
|
|
metavar="HISTORY_FILE",
|
|
default=os.environ.get(f"{env_prefix}HISTORY_FILE", ".coder.history"),
|
|
help="Specify the history file (default: .coder.history, ${env_prefix}HISTORY_FILE)",
|
|
)
|
|
parser.add_argument(
|
|
"--model",
|
|
metavar="MODEL",
|
|
default="gpt-4",
|
|
help="Specify the model to use for the main chat (default: gpt-4)",
|
|
)
|
|
parser.add_argument(
|
|
"-3",
|
|
action="store_const",
|
|
dest="model",
|
|
const="gpt-3.5-turbo",
|
|
help="Use gpt-3.5-turbo model for the main chat",
|
|
)
|
|
parser.add_argument(
|
|
"-4",
|
|
action="store_const",
|
|
dest="model",
|
|
const="gpt-4",
|
|
help="Use gpt-4 model for the main chat",
|
|
)
|
|
parser.add_argument(
|
|
"--no-pretty",
|
|
action="store_false",
|
|
dest="pretty",
|
|
help=f"Disable pretty output of GPT responses (${{{env_prefix}}}PRETTY)",
|
|
default=bool(int(os.environ.get(f"{env_prefix}PRETTY", 1))),
|
|
)
|
|
parser.add_argument(
|
|
"--apply",
|
|
metavar="FILE",
|
|
help="Apply the changes from the given file instead of running the chat",
|
|
)
|
|
parser.add_argument(
|
|
"--commit-dirty",
|
|
action="store_true",
|
|
help=f"Commit dirty files without confirmation (default: False, ${{{env_prefix}}}COMMIT_DIRTY)",
|
|
default=bool(int(os.environ.get(f"{env_prefix}COMMIT_DIRTY", 0))),
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
fnames = args.files
|
|
pretty = args.pretty
|
|
|
|
coder = Coder(args.model, fnames, pretty, args.history_file)
|
|
coder.commit(ask=not args.commit_dirty, prefix="WIP: ")
|
|
|
|
if args.apply:
|
|
with open(args.apply, "r") as f:
|
|
content = f.read()
|
|
coder.update_files(content, inp="")
|
|
return
|
|
|
|
coder.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
status = main()
|
|
sys.exit(status)
|