diff --git a/app/MindWork AI Studio/Provider/OpenAI/ProviderOpenAI.cs b/app/MindWork AI Studio/Provider/OpenAI/ProviderOpenAI.cs
index 70b80f2e..3ca5ac0e 100644
--- a/app/MindWork AI Studio/Provider/OpenAI/ProviderOpenAI.cs
+++ b/app/MindWork AI Studio/Provider/OpenAI/ProviderOpenAI.cs
@@ -153,18 +153,18 @@ public sealed class ProviderOpenAI(ILogger logger) : BaseProvider("https://api.o
///
public Task> GetTextModels(string? apiKeyProvisional = null, CancellationToken token = default)
{
- return this.LoadModels("gpt-", token, apiKeyProvisional);
+ return this.LoadModels(["gpt-", "o1-"], token, apiKeyProvisional);
}
///
public Task> GetImageModels(string? apiKeyProvisional = null, CancellationToken token = default)
{
- return this.LoadModels("dall-e-", token, apiKeyProvisional);
+ return this.LoadModels(["dall-e-"], token, apiKeyProvisional);
}
#endregion
- private async Task> LoadModels(string prefix, CancellationToken token, string? apiKeyProvisional = null)
+ private async Task> LoadModels(string[] prefixes, CancellationToken token, string? apiKeyProvisional = null)
{
var secretKey = apiKeyProvisional switch
{
@@ -187,6 +187,6 @@ public sealed class ProviderOpenAI(ILogger logger) : BaseProvider("https://api.o
return [];
var modelResponse = await response.Content.ReadFromJsonAsync(token);
- return modelResponse.Data.Where(n => n.Id.StartsWith(prefix, StringComparison.InvariantCulture));
+ return modelResponse.Data.Where(model => prefixes.Any(prefix => model.Id.StartsWith(prefix, StringComparison.InvariantCulture)));
}
}
\ No newline at end of file