diff options
| author | Nate Sesti <sestinj@gmail.com> | 2023-08-13 10:05:06 -0700 |
|---|---|---|
| committer | Nate Sesti <sestinj@gmail.com> | 2023-08-13 10:05:06 -0700 |
| commit | bcec2a0870d0ef649961b6c91ec866b612680b9e (patch) | |
| tree | ab8e9107f99b72c453b9099f6a6f144d0e01b9b4 /continuedev | |
| parent | 73bb5f5bb9e582841f452942c25f923a6b32e21f (diff) | |
| download | sncontinue-bcec2a0870d0ef649961b6c91ec866b612680b9e.tar.gz sncontinue-bcec2a0870d0ef649961b6c91ec866b612680b9e.tar.bz2 sncontinue-bcec2a0870d0ef649961b6c91ec866b612680b9e.zip | |
fix: :bug: fix for Azure OpenAI model names
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") |