mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-02-12 16:01:37 +00:00
250 lines
8.3 KiB
C#
250 lines
8.3 KiB
C#
|
using AIStudio.Provider;
|
||
|
|
|
||
|
|
namespace AIStudio.Settings;
|
||
|
|
|
||
|
|
public static partial class ProviderExtensions
|
||
|
|
{
|
||
|
|
public 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,
|
||
|
|
];
|
||
|
|
}
|
||
|
|
}
|