Add GPT-4-turbo-preview to enabled models

This commit is contained in:
Paul Harrison 2024-02-17 16:13:54 +00:00
parent 50fa0cc5ae
commit aade152486
3 changed files with 6 additions and 1 deletions

View File

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

View File

@ -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__(

View File

@ -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