mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-02-12 11:01:38 +00:00
e9485ca8ea
Some checks are pending
Build and Release / Read metadata (push) Waiting to run
Build and Release / Build app (${{ matrix.dotnet_runtime }}) (-aarch64-apple-darwin, osx-arm64, macos-latest, aarch64-apple-darwin, dmg updater) (push) Blocked by required conditions
Build and Release / Build app (${{ matrix.dotnet_runtime }}) (-aarch64-pc-windows-msvc.exe, win-arm64, windows-latest, aarch64-pc-windows-msvc, nsis updater) (push) Blocked by required conditions
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) Blocked by required conditions
Build and Release / Build app (${{ matrix.dotnet_runtime }}) (-x86_64-apple-darwin, osx-x64, macos-latest, x86_64-apple-darwin, dmg updater) (push) Blocked by required conditions
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) Blocked by required conditions
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) Blocked by required conditions
Build and Release / Prepare & create release (push) Blocked by required conditions
Build and Release / Publish release (push) Blocked by required conditions
250 lines
8.3 KiB
C#
250 lines
8.3 KiB
C#
using AIStudio.Provider;
|
|
|
|
namespace AIStudio.Settings;
|
|
|
|
public static partial class ProviderExtensions
|
|
{
|
|
private static List<Capability> GetModelCapabilitiesOpenRouter(Model model)
|
|
{
|
|
var modelName = model.Id.ToLowerInvariant().AsSpan();
|
|
|
|
//
|
|
// OpenRouter model IDs follow the pattern: "provider/model-name"
|
|
// Examples:
|
|
// - openai/gpt-4o
|
|
// - anthropic/claude-3-5-sonnet
|
|
// - google/gemini-pro-1.5
|
|
// - meta-llama/llama-3.1-405b-instruct
|
|
//
|
|
// We need to detect capabilities based on both provider and model name.
|
|
//
|
|
|
|
//
|
|
// OpenAI models via OpenRouter:
|
|
//
|
|
if (modelName.IndexOf("openai/") is not -1)
|
|
{
|
|
// Reasoning models (o1, o3, o4 series)
|
|
if (modelName.IndexOf("/o1") is not -1 ||
|
|
modelName.IndexOf("/o3") is not -1 ||
|
|
modelName.IndexOf("/o4") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.ALWAYS_REASONING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// GPT-4o and GPT-5 series with multimodal
|
|
if (modelName.IndexOf("/gpt-4o") is not -1 ||
|
|
modelName.IndexOf("/gpt-5") is not -1 ||
|
|
modelName.IndexOf("/chatgpt-4o") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Standard GPT-4
|
|
if (modelName.IndexOf("/gpt-4") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// GPT-3.5
|
|
if (modelName.IndexOf("/gpt-3.5") is not -1 ||
|
|
modelName.IndexOf("/gpt-3") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
}
|
|
|
|
//
|
|
// Anthropic models via OpenRouter:
|
|
//
|
|
if (modelName.IndexOf("anthropic/") is not -1)
|
|
{
|
|
// Claude 3.5 and newer with vision
|
|
if (modelName.IndexOf("/claude-3.5") is not -1 ||
|
|
modelName.IndexOf("/claude-3-5") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Claude 3 Opus/Sonnet with vision
|
|
if (modelName.IndexOf("/claude-3-opus") is not -1 ||
|
|
modelName.IndexOf("/claude-3-sonnet") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Other Claude 3 models
|
|
if (modelName.IndexOf("/claude-3") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
}
|
|
|
|
//
|
|
// Google models via OpenRouter:
|
|
//
|
|
if (modelName.IndexOf("google/") is not -1)
|
|
{
|
|
// Gemini models with multimodal
|
|
if (modelName.IndexOf("/gemini") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
}
|
|
|
|
//
|
|
// xAI Grok models via OpenRouter:
|
|
//
|
|
if (modelName.IndexOf("x-ai/") is not -1 || modelName.IndexOf("/grok") is not -1)
|
|
{
|
|
if (modelName.IndexOf("-vision") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
}
|
|
|
|
//
|
|
// DeepSeek models via OpenRouter:
|
|
//
|
|
if (modelName.IndexOf("/deepseek") is not -1)
|
|
{
|
|
if (modelName.IndexOf("-r1") is not -1 || modelName.IndexOf(" r1") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.ALWAYS_REASONING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
}
|
|
|
|
//
|
|
// Mistral models via OpenRouter:
|
|
//
|
|
if (modelName.IndexOf("/mistral") is not -1 || modelName.IndexOf("/pixtral") is not -1)
|
|
{
|
|
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,
|
|
];
|
|
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
}
|
|
|
|
//
|
|
// Meta Llama models via OpenRouter:
|
|
//
|
|
if (modelName.IndexOf("/llama") is not -1)
|
|
{
|
|
// Llama 4 with vision
|
|
if (modelName.IndexOf("/llama-4") is not -1 ||
|
|
modelName.IndexOf("/llama4") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Vision models
|
|
if (modelName.IndexOf("-vision") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT, Capability.MULTIPLE_IMAGE_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Llama 3.1+ with function calling
|
|
if (modelName.IndexOf("/llama-3.") is not -1 ||
|
|
modelName.IndexOf("/llama3.") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.FUNCTION_CALLING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
// Default Llama
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
}
|
|
|
|
//
|
|
// Qwen models via OpenRouter:
|
|
//
|
|
if (modelName.IndexOf("/qwen") is not -1 || modelName.IndexOf("/qwq") is not -1)
|
|
{
|
|
if (modelName.IndexOf("/qwq") is not -1)
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.ALWAYS_REASONING,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
}
|
|
|
|
//
|
|
// Default for unknown models:
|
|
// Assume basic text input/output with chat completion
|
|
//
|
|
return [
|
|
Capability.TEXT_INPUT,
|
|
Capability.TEXT_OUTPUT,
|
|
Capability.CHAT_COMPLETION_API,
|
|
];
|
|
}
|
|
}
|