feat(ai): Update OpenAI models supporting structured output

eclipse-theia · Oct 3, 2024 · ac0c246 · ac0c246
1 parent ce5e13f
commit ac0c246
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/packages/ai-openai/src/node/openai-language-model.ts b/packages/ai-openai/src/node/openai-language-model.ts
@@ -134,9 +134,12 @@ export class OpenAiModel implements LanguageModel {
     }
 
     protected supportsStructuredOutput(): boolean {
-        // currently only the lastest 4o and 4o-mini models support structured output
-        // see https://platform.openai.com/docs/guides/structured-outputs
-        return this.model === 'gpt-4o-2024-08-06' || this.model === 'gpt-4o-mini';
+        // see https://platform.openai.com/docs/models/gpt-4o
+        return [
+            'gpt-4o',
+            'gpt-4o-2024-08-06',
+            'gpt-4o-mini'
+        ].includes(this.model);
     }
 
     protected async handleStructuredOutputRequest(openai: OpenAI, request: LanguageModelRequest): Promise<LanguageModelParsedResponse> {