MindWorkAI/AI-Studio
MindWorkAI/AI-Studio
1
0
Fork 0
mirror of https://github.com/MindWorkAI/AI-Studio.git synced 2025-08-21 01:52:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update GetModelCapabilities to include multiple image input and optional reasoning for Claude models

Browse Source
This commit is contained in:
Thorsten Sommer 2025-05-07 19:57:43 +02:00
parent e503811ffd
commit 45dbbdf299
Signed by: tsommer
GPG Key ID: 371BBA77A02C0108
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
app/MindWork AI Studio/Provider/Anthropic/ProviderAnthropic.cs
View File

@ -110,26 +110,33 @@ public sealed class ProviderAnthropic(ILogger logger) : BaseProvider("https://ap
return Task.FromResult(Enumerable.Empty<Model>());
}
#region Overrides of BaseProvider
public override IReadOnlyCollection<Capability> GetModelCapabilities(Model model)
{
var modelName = model.Id.ToLowerInvariant().AsSpan();
// Claude 3.7 is able to do reasoning:
if(modelName.StartsWith("claude-3-7"))
return [Capability.TEXT_INPUT, Capability.IMAGE_INPUT, Capability.TEXT_OUTPUT, Capability.REASONING, Capability.FUNCTION_CALLING];
return [
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
Capability.TEXT_OUTPUT,
Capability.OPTIONAL_REASONING, Capability.FUNCTION_CALLING];
// All other 3.x models are able to process text and images as input:
if(modelName.StartsWith("claude-3-"))
return [Capability.TEXT_INPUT, Capability.IMAGE_INPUT, Capability.TEXT_OUTPUT, Capability.FUNCTION_CALLING];
return [
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
Capability.TEXT_OUTPUT,
Capability.FUNCTION_CALLING];
// Any other model is able to process text only:
return [Capability.TEXT_INPUT, Capability.TEXT_OUTPUT, Capability.FUNCTION_CALLING];
return [
Capability.TEXT_INPUT,
Capability.TEXT_OUTPUT,
Capability.FUNCTION_CALLING];
}
#endregion
#endregion
private async Task<IEnumerable<Model>> LoadModels(CancellationToken token, string? apiKeyProvisional = null)