Add Capability enum to represent AI model capabilities

2025-08-20 23:32:55 +00:00 · 2025-05-06 08:27:23 +02:00 · 2025-05-06 08:27:23 +02:00 · 9d2b2a0a5e
commit 9d2b2a0a5e
parent 07b858fd72
1 changed files with 87 additions and 0 deletions
--- a/Studio/Provider/Capability.cs
+++ b/Studio/Provider/Capability.cs
@ -0,0 +1,87 @@
+namespace AIStudio.Provider;
+
+/// <summary>
+/// Represents the capabilities of an AI model.
+/// </summary>
+public enum Capability
+{
+    /// <summary>
+    /// No capabilities specified.
+    /// </summary>
+    NONE,
+    
+    /// <summary>
+    /// We don't know what the AI model can do.
+    /// </summary>
+    UNKNOWN,
+    
+    /// <summary>
+    /// The AI model can perform text input.
+    /// </summary>
+    TEXT_INPUT,
+    
+    /// <summary>
+    /// The AI model can perform audio input, such as music or sound.
+    /// </summary>
+    AUDIO_INPUT,
+    
+    /// <summary>
+    /// The AI model can perform image input, such as photos or drawings.
+    /// </summary>
+    IMAGE_INPUT,
+    
+    /// <summary>
+    /// The AI model can perform speech input.
+    /// </summary>
+    SPEECH_INPUT,
+    
+    /// <summary>
+    /// The AI model can perform video input, such as video files or streams.
+    /// </summary>
+    VIDEO_INPUT,
+    
+    /// <summary>
+    /// The AI model can generate text output.
+    /// </summary>
+    TEXT_OUTPUT,
+    
+    /// <summary>
+    /// The AI model can generate audio output, such as music or sound.
+    /// </summary>
+    AUDIO_OUTPUT,
+    
+    /// <summary>
+    /// The AI model can generate image output, such as photos or drawings.
+    /// </summary>
+    IMAGE_OUTPUT,
+    
+    /// <summary>
+    /// The AI model can generate speech output.
+    /// </summary>
+    SPEECH_OUTPUT,
+    
+    /// <summary>
+    /// The AI model can generate video output.
+    /// </summary>
+    VIDEO_OUTPUT,
+    
+    /// <summary>
+    /// The AI model can perform reasoning tasks.
+    /// </summary>
+    REASONING,
+    
+    /// <summary>
+    /// The AI model can embed information or data.
+    /// </summary>
+    EMBEDDING,
+    
+    /// <summary>
+    /// The AI model can perform in real-time.
+    /// </summary>
+    REALTIME,
+    
+    /// <summary>
+    /// The AI model can perform function calling, such as invoking APIs or executing functions.
+    /// </summary>
+    FUNCTION_CALLING,
+}