Fix openai.py linting errors

This commit is contained in:
Reinier van der Leer
2023-06-15 18:50:26 +02:00
parent 426cfef882
commit 12588b6483

View File

@@ -49,10 +49,10 @@ OPEN_AI_CHAT_MODELS = {
}
# Set aliases for rolling model IDs
chat_model_mapping = {
"gpt-3.5-turbo": "gpt-3.5-turbo-0301",
"gpt-3.5-turbo": "gpt-3.5-turbo-0301",
"gpt-3.5-turbo-16k": "gpt-3.5-turbo-16k-0613",
"gpt-4": "gpt-4-0314",
"gpt-4-32k": "gpt-4-32k-0314",
"gpt-4": "gpt-4-0314",
"gpt-4-32k": "gpt-4-32k-0314",
}
for alias, target in chat_model_mapping.items():
alias_info = ChatModelInfo(**OPEN_AI_CHAT_MODELS[target].__dict__)