litellm-mirror/litellm/__init__.py
2023-08-01 20:32:16 -07:00

42 lines
783 B
Python

success_callback = []
failure_callback = []
set_verbose=False
####### COMPLETION MODELS ###################
open_ai_chat_completion_models = [
'gpt-3.5-turbo',
'gpt-4'
]
open_ai_text_completion_models = [
'text-davinci-003'
]
cohere_models = [
'command-nightly',
]
anthropic_models = [
"claude-2",
"claude-instant-1"
]
google_chat_models = [
"chat-bison-001"
]
google_completion_models = [
"text-bison-001",
]
####### EMBEDDING MODELS ###################
open_ai_embedding_models = [
'text-embedding-ada-002'
]
google_embedding_models = [
"embedding-gecko-001"
]
from .timeout import timeout
from .utils import client, logging, exception_type # Import all the symbols from main.py
from .main import * # Import all the symbols from main.py