diff --git a/llm/cli.py b/llm/cli.py index 539f2ef..96f3443 100644 --- a/llm/cli.py +++ b/llm/cli.py @@ -20,7 +20,14 @@ warnings.simplefilter("ignore", ResourceWarning) DEFAULT_MODEL = "gpt-3.5-turbo" -MODEL_ALIASES = {"4": "gpt-4", "gpt4": "gpt-4", "chatgpt": "gpt-3.5-turbo"} +MODEL_ALIASES = { + "4": "gpt-4", + "gpt4": "gpt-4", + "chatgpt": "gpt-3.5-turbo", + "3.5": "gpt-3.5-turbo", + "chatgpt-16k": "gpt-3.5-turbo-16k", + "3.5-16k": "gpt-3.5-turbo-16k", +} DEFAULT_TEMPLATE = "prompt: "