Add 16k models (#38)

also added a `3.5` alias
simonw · Jun 17, 2023 · b12e28c · b12e28c
1 parent 5e21195
commit b12e28c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/llm/cli.py b/llm/cli.py
@@ -20,7 +20,14 @@
 
 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: "