feat: support gpt-3.5-turbo-instruct model (#1195)

2023-09-19 02:05:04 +08:00 · 2023-09-19 02:05:04 +08:00 · ae3f1ac0a9
commit ae3f1ac0a9
parent 269a465fc4
7 changed files with 25 additions and 9 deletions
--- a/api/core/model_providers/providers/openai_provider.py
+++ b/api/core/model_providers/providers/openai_provider.py
@ -40,6 +40,10 @@ class OpenAIProvider(BaseModelProvider):
                        ModelFeature.AGENT_THOUGHT.value
                    ]
                },
+                {
+                    'id': 'gpt-3.5-turbo-instruct',
+                    'name': 'GPT-3.5-Turbo-Instruct',
+                },
                {
                    'id': 'gpt-3.5-turbo-16k',
                    'name': 'gpt-3.5-turbo-16k',
@ -128,6 +132,7 @@ class OpenAIProvider(BaseModelProvider):
            'gpt-4': 8192,
            'gpt-4-32k': 32768,
            'gpt-3.5-turbo': 4096,
+            'gpt-3.5-turbo-instruct': 4096,
            'gpt-3.5-turbo-16k': 16384,
            'text-davinci-003': 4097,
        }