Add gpt-4-1106-preview (gpt-4-turbo) to model list

jepler · Nov 9, 2023 · 1e856e4 · 1e856e4
1 parent 4454071
commit 1e856e4
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/chap/backends/openai_chatgpt.py b/src/chap/backends/openai_chatgpt.py
@@ -4,6 +4,7 @@
 
 import functools
 import json
+import warnings
 from dataclasses import dataclass
 from typing import AsyncGenerator, cast
 
@@ -36,16 +37,20 @@ def num_tokens_for_messages(self, messages: Session) -> int:
     @functools.cache
     def from_model(cls, model: str) -> "EncodingMeta":
         if model == "gpt-3.5-turbo":
-            # print("Warning: gpt-3.5-turbo may update over time. Returning num tokens assuming gpt-3.5-turbo-0613.")
+            warnings.warn(
+                "Warning: gpt-3.5-turbo may update over time. Returning num tokens assuming gpt-3.5-turbo-0613."
+            )
             model = "gpt-3.5-turbo-0613"
         if model == "gpt-4":
-            # print("Warning: gpt-4 may update over time. Returning num tokens assuming gpt-4-0613.")
+            warnings.warn(
+                "Warning: gpt-4 may update over time. Returning num tokens assuming gpt-4-0613."
+            )
             model = "gpt-4-0613"
 
         try:
             encoding = tiktoken.encoding_for_model(model)
         except KeyError:
-            print("Warning: model not found. Using cl100k_base encoding.")
+            warnings.warn("Warning: model not found. Using cl100k_base encoding.")
             encoding = tiktoken.get_encoding("cl100k_base")
 
         if model in {
@@ -55,6 +60,7 @@ def from_model(cls, model: str) -> "EncodingMeta":
             "gpt-4-32k-0314",
             "gpt-4-0613",
             "gpt-4-32k-0613",
+            "gpt-4-1106-preview",
         }:
             tokens_per_message = 3
             tokens_per_name = 1