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

Merge pull request #373 from promptmetheus/fix-openai-completion-model-names

Browse source 
fix OpenAI completion model names
This commit is contained in:
Krish Dholakia 2023-09-14 11:20:19 -07:00 • committed by GitHub
commit 8c2248abba
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-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 = [
"command-nightly",
@ -256,4 +263,4 @@ from .exceptions import (
ContextWindowExceededError,
)
from .budget_manager import BudgetManager
from .budget_manager import BudgetManager