Skip to content

Commit

Permalink
fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
JoelKronander committed Nov 10, 2023
1 parent c124a00 commit ea97fad
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 3 deletions.
3 changes: 2 additions & 1 deletion copilot/conversation.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,14 @@
class Model(Enum):
GPT_4 = "gpt-4"
GPT_35_TURBO = "gpt-3.5-turbo"
GPT_4_TURBO = "gpt-4-1106-preview"


def argparse_model_type(model_str):
try:
return Model[model_str.upper().replace("-", "_")]
except KeyError:
raise argparse.ArgumentTypeError(f"Invalid model: {model_str}. Allowed values are 'gpt-4' and 'gpt-3.5-turbo'.")
raise argparse.ArgumentTypeError(f"Invalid model: {model_str}. Allowed values are 'gpt-4', 'gpt-4-turbo' and 'gpt-3.5-turbo'.")


@dataclass
Expand Down
2 changes: 1 addition & 1 deletion copilot/parse_args.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def parse_terminal_copilot_args():
"-c", "--count", help="The number of commands to output when JSON output is specified."
)
parser.add_argument(
"-m", "--model", type=argparse_model_type, default=Model.GPT_35_TURBO,
"-m", "--model", type=argparse_model_type, default=Model.GPT_4_TURBO,
help="The model to use. Defaults to gpt-3.5-turbo."
)
parser.add_argument(
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

setup(
name ='terminal-copilot',
version ='1.5.0',
version ='1.5.3',
author ='Methexis',
author_email ='[email protected]',
url ='https://github.com/Methexis-Inc/terminal-copilot',
Expand Down

0 comments on commit ea97fad

Please sign in to comment.