diff options
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/src/continuedev/libs/llm/openai.py | 6 |
1 files changed, 6 insertions, 0 deletions
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") |