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