mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-02-12 20:21:37 +00:00
48d762bc68
Some checks failed
Build and Release / Read metadata (push) Has been cancelled
Build and Release / Build app (${{ matrix.dotnet_runtime }}) (-aarch64-apple-darwin, osx-arm64, macos-latest, aarch64-apple-darwin, dmg updater) (push) Has been cancelled
Build and Release / Build app (${{ matrix.dotnet_runtime }}) (-aarch64-pc-windows-msvc.exe, win-arm64, windows-latest, aarch64-pc-windows-msvc, nsis updater) (push) Has been cancelled
Build and Release / Build app (${{ matrix.dotnet_runtime }}) (-aarch64-unknown-linux-gnu, linux-arm64, ubuntu-22.04-arm, aarch64-unknown-linux-gnu, appimage deb updater) (push) Has been cancelled
Build and Release / Build app (${{ matrix.dotnet_runtime }}) (-x86_64-apple-darwin, osx-x64, macos-latest, x86_64-apple-darwin, dmg updater) (push) Has been cancelled
Build and Release / Build app (${{ matrix.dotnet_runtime }}) (-x86_64-pc-windows-msvc.exe, win-x64, windows-latest, x86_64-pc-windows-msvc, nsis updater) (push) Has been cancelled
Build and Release / Build app (${{ matrix.dotnet_runtime }}) (-x86_64-unknown-linux-gnu, linux-x64, ubuntu-22.04, x86_64-unknown-linux-gnu, appimage deb updater) (push) Has been cancelled
Build and Release / Prepare & create release (push) Has been cancelled
Build and Release / Publish release (push) Has been cancelled
67 lines
2.0 KiB
C#
67 lines
2.0 KiB
C#
using AIStudio.Provider;
|
|
|
|
namespace AIStudio.Settings;
|
|
|
|
public static partial class ProviderExtensions
|
|
{
|
|
public static List<Capability> GetModelCapabilitiesMistral(Model model)
|
|
{
|
|
var modelName = model.Id.ToLowerInvariant().AsSpan();
|
|
|
|
// Pixtral models are able to do process images:
|
|
if (modelName.IndexOf("pixtral") is not -1)
|
|
return
|
|
[
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Mistral large:
|
|
if (modelName.IndexOf("mistral-large-") is not -1)
|
|
return
|
|
[
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Mistral medium:
|
|
if (modelName.IndexOf("mistral-medium-") is not -1)
|
|
return
|
|
[
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Mistral small:
|
|
if (modelName.IndexOf("mistral-small-") is not -1)
|
|
return
|
|
[
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Mistral saba:
|
|
if (modelName.IndexOf("mistral-saba-") is not -1)
|
|
return
|
|
[
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Default:
|
|
return GetModelCapabilitiesOpenSource(model);
|
|
}
|
|
} |