From 6d998ea541022596ef33f5d507fcd61d62a76be5 Mon Sep 17 00:00:00 2001
From: Toni Engelhardt <toni88x@gmail.com>
Date: Thu, 14 Sep 2023 18:57:48 +0100
Subject: [PATCH] fix openai completion model names

The model names are wrong, they need to be prefixed with `text-`.
---
 litellm/__init__.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/litellm/__init__.py b/litellm/__init__.py
index 882994184..fe0cbe49e 100644
--- a/litellm/__init__.py
+++ b/litellm/__init__.py
@@ -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
\ No newline at end of file
+from .budget_manager import BudgetManager