Initial commit

src/agent/purpose.rs

@@ -0,0 +1,67 @@
+#[derive(Debug, Clone, Copy, PartialEq)]
+pub enum AgentPurpose {
+    CatchAll,
+    ImageGeneration,
+    TextGeneration,
+    TextToSpeech,
+    SpeechToText,
+}
+
+impl AgentPurpose {
+    pub fn from_str(s: &str) -> Option<Self> {
+        match s {
+            "catch-all" => Some(Self::CatchAll),
+            "image-generation" => Some(Self::ImageGeneration),
+            "text-generation" => Some(Self::TextGeneration),
+            "text-to-speech" => Some(Self::TextToSpeech),
+            "speech-to-text" => Some(Self::SpeechToText),
+            _ => None,
+        }
+    }
+
+    pub fn as_str(&self) -> &'static str {
+        match self {
+            Self::CatchAll => "catch-all",
+            Self::ImageGeneration => "image-generation",
+            Self::TextGeneration => "text-generation",
+            Self::TextToSpeech => "text-to-speech",
+            Self::SpeechToText => "speech-to-text",
+        }
+    }
+
+    pub fn choices() -> Vec<&'static Self> {
+        vec![
+            &Self::TextGeneration,
+            &Self::SpeechToText,
+            &Self::TextToSpeech,
+            &Self::ImageGeneration,
+            &Self::CatchAll,
+        ]
+    }
+
+    pub fn emoji(&self) -> &'static str {
+        match self {
+            Self::CatchAll => "❓",
+            Self::TextGeneration => "💬",
+            Self::SpeechToText => "🦻",
+            Self::TextToSpeech => "🗣️",
+            Self::ImageGeneration => "🖌️",
+        }
+    }
+
+    pub fn heading(&self) -> &'static str {
+        match self {
+            Self::CatchAll => "Catch-All",
+            Self::TextGeneration => "Text Generation",
+            Self::SpeechToText => "Speech-to-Text",
+            Self::TextToSpeech => "Text-to-Speech",
+            Self::ImageGeneration => "Image Generation",
+        }
+    }
+}
+
+impl std::fmt::Display for AgentPurpose {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        f.write_str(self.as_str())
+    }
+}