From b12e28c4ec418399b4ea23073a4bfcd5def92251 Mon Sep 17 00:00:00 2001
From: Benjamin Kirkbride <benjaminkirkbride@gmail.com>
Date: Sat, 17 Jun 2023 04:22:52 -0400
Subject: [PATCH] Add 16k models (#38)

also added a `3.5` alias
---
 llm/cli.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

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: "