Initial commit

src/agent/definition.rs

@@ -0,0 +1,55 @@
+use serde::de::Error as DeError;
+use serde::{Deserialize, Deserializer, Serialize, Serializer};
+
+use super::provider::AgentProvider;
+
+// Custom serialization for AgentProvider
+pub fn serialize_provider_to_string<S>(
+    value: &AgentProvider,
+    serializer: S,
+) -> Result<S::Ok, S::Error>
+where
+    S: Serializer,
+{
+    serializer.serialize_str(value.to_static_str())
+}
+
+// Custom deserialization for AgentProvider
+pub fn deserialize_provider_from_string<'de, D>(deserializer: D) -> Result<AgentProvider, D::Error>
+where
+    D: Deserializer<'de>,
+{
+    let s = String::deserialize(deserializer)?;
+    AgentProvider::from_string(&s).map_err(DeError::custom)
+}
+
+#[derive(Debug, Clone, Deserialize, Serialize)]
+pub struct AgentDefinition {
+    pub id: String,
+
+    #[serde(
+        serialize_with = "serialize_provider_to_string",
+        deserialize_with = "deserialize_provider_from_string"
+    )]
+    pub provider: AgentProvider,
+
+    pub config: serde_yaml::Value,
+}
+
+impl AgentDefinition {
+    pub fn new(id: String, provider: AgentProvider, config: serde_yaml::Value) -> Self {
+        Self {
+            id,
+            provider,
+            config,
+        }
+    }
+}
+
+impl PartialEq for AgentDefinition {
+    fn eq(&self, other: &Self) -> bool {
+        self.id == other.id
+    }
+}
+
+impl Eq for AgentDefinition {}