From aade1524867e98bdf43b9d1b7244e7aee3d3f2e1 Mon Sep 17 00:00:00 2001
From: Paul Harrison <paul@harrison.sh>
Date: Sat, 17 Feb 2024 16:13:54 +0000
Subject: [PATCH] Add GPT-4-turbo-preview to enabled models

---
 pyproject.toml           | 2 +-
 src/llm_chat/chat.py     | 4 ++++
 src/llm_chat/settings.py | 1 +
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index f542a1a..6f4146a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "llm-chat"
-version = "1.1.1"
+version = "1.1.2"
 description = "A general CLI interface for large language models."
 authors = ["Paul Harrison <paul@harrison.sh>"]
 readme = "README.md"
diff --git a/src/llm_chat/chat.py b/src/llm_chat/chat.py
index 8ee06a0..5247b90 100644
--- a/src/llm_chat/chat.py
+++ b/src/llm_chat/chat.py
@@ -89,6 +89,10 @@ class Chat:
             Token.COMPLETION: 0.06,
             Token.PROMPT: 0.03,
         },
+        Model.GPT4_TURBO: {
+            Token.COMPLETION: 0.03,
+            Token.PROMPT: 0.01,
+        },
     }
 
     def __init__(
diff --git a/src/llm_chat/settings.py b/src/llm_chat/settings.py
index dd54961..6973f10 100644
--- a/src/llm_chat/settings.py
+++ b/src/llm_chat/settings.py
@@ -10,6 +10,7 @@ class Model(StrEnum):
 
     GPT3 = "gpt-3.5-turbo"
     GPT4 = "gpt-4"
+    GPT4_TURBO = "gpt-4-turbo-preview"
 
 
 DEFAULT_MODEL = Model.GPT3