Compare commits
2 Commits
2764777c0b
...
aade152486
Author | SHA1 | Date |
---|---|---|
Paul Harrison | aade152486 | |
Paul Harrison | 50fa0cc5ae |
|
@ -1,6 +1,6 @@
|
|||
[tool.poetry]
|
||||
name = "llm-chat"
|
||||
version = "1.1.0"
|
||||
version = "1.1.2"
|
||||
description = "A general CLI interface for large language models."
|
||||
authors = ["Paul Harrison <paul@harrison.sh>"]
|
||||
readme = "README.md"
|
||||
|
|
|
@ -82,13 +82,17 @@ class Chat:
|
|||
|
||||
_pricing: dict[Model, dict[Token, float]] = {
|
||||
Model.GPT3: {
|
||||
Token.COMPLETION: 0.002,
|
||||
Token.COMPLETION: 0.0005,
|
||||
Token.PROMPT: 0.0015,
|
||||
},
|
||||
Model.GPT4: {
|
||||
Token.COMPLETION: 0.06,
|
||||
Token.PROMPT: 0.03,
|
||||
},
|
||||
Model.GPT4_TURBO: {
|
||||
Token.COMPLETION: 0.03,
|
||||
Token.PROMPT: 0.01,
|
||||
},
|
||||
}
|
||||
|
||||
def __init__(
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue