phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix openai completion model names

Browse source 
The model names are wrong, they need to be prefixed with `text-`.
This commit is contained in:
Toni Engelhardt 2023-09-14 18:57:48 +01:00 • committed by GitHub
parent 70a36740bc
commit 6d998ea541
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/__init__.py
View file

@ -91,7 +91,14 @@ open_ai_chat_completion_models = [
"gpt-3.5-turbo-16k", "gpt-3.5-turbo-16k",
"gpt-3.5-turbo-16k-0613", "gpt-3.5-turbo-16k-0613",
] ]
open_ai_text_completion_models = ["text-davinci-003", "curie-001", "babbage-001", "ada-001", "babbage-002", "davinci-002"] open_ai_text_completion_models = [
"text-davinci-003",
"text-curie-001",
"text-babbage-001",
"text-ada-001",
"text-babbage-002",
"text-davinci-002",
]
cohere_models = [ cohere_models = [
"command-nightly", "command-nightly",
@ -256,4 +263,4 @@ from .exceptions import (
ContextWindowExceededError, ContextWindowExceededError,
) )
from .budget_manager import BudgetManager from .budget_manager import BudgetManager