From bcec2a0870d0ef649961b6c91ec866b612680b9e Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Sun, 13 Aug 2023 10:05:06 -0700
Subject: fix: :bug: fix for Azure OpenAI model names

---
 continuedev/src/continuedev/libs/llm/openai.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'continuedev/src')

diff --git a/continuedev/src/continuedev/libs/llm/openai.py b/continuedev/src/continuedev/libs/llm/openai.py
index 09cfa4fa..9c553354 100644
--- a/continuedev/src/continuedev/libs/llm/openai.py
+++ b/continuedev/src/continuedev/libs/llm/openai.py
@@ -24,6 +24,9 @@ MAX_TOKENS_FOR_MODEL = {
     "gpt-3.5-turbo-0613": 4096,
     "gpt-3.5-turbo-16k": 16384,
     "gpt-4": 8192,
+    "gpt-35-turbo-16k": 16384,
+    "gpt-35-turbo-0613": 4096,
+    "gpt-35-turbo": 4096,
 }
 
 
@@ -151,3 +154,6 @@ class OpenAI(LLM):
             self.write_log(f"Completion:\n\n{resp}")
 
         return resp
+
+
+openai.Completion.create(prompt="HI", model="text-davinci-003")
-- 
cgit v1.2.3-70-g09d2