feat: Integrate DSRs for automatic SYSTEM prompt optimization (#45)

tilekit/Cargo.toml

@@ -5,3 +5,10 @@ edition = "2024"
 
 [dependencies]
 nom = "8"
+serde = { version = "1.0", features = ["derive"] }
+serde_json = "1.0"
+anyhow = "1.0"
+tokio = { version = "1" , features = ["macros", "rt-multi-thread"]}
+dspy-rs = "0.7.3"
+bon = "3.7.0"
+indexmap = "2.2.0"

tilekit/src/lib.rs

@@ -1 +1,2 @@
 pub mod modelfile;
+pub mod optimize;

tilekit/src/modelfile.rs

@@ -167,16 +167,22 @@ impl Modelfile {
     }
 
     pub fn add_system(&mut self, value: &str) -> Result<(), String> {
-        if self.system.is_some() {
-            let error = "Modelfile can only have one SYSTEM instruction".to_owned();
-            self.errors.push(error.clone());
-
-            self.data.push(format!("SYSTEM {}", value));
-            Err(error)
-        } else {
-            self.system = Some(value.to_owned());
-            Ok(())
+        self.system = Some(value.to_owned());
+        let formatted = format!("SYSTEM {}", value);
+
+        // Find and replace or add to data
+        let mut found = false;
+        for line in self.data.iter_mut() {
+            if line.to_uppercase().starts_with("SYSTEM") {
+                *line = formatted.clone();
+                found = true;
+                break;
+            }
         }
+        if !found {
+            self.data.push(formatted);
+        }
+        Ok(())
     }
 
     pub fn add_comment(&mut self, value: &str) -> Result<(), String> {