initial tests

Signed-off-by: Nick Mitchell <[email protected]>

Author: starpit

pdl-live-react/src-tauri/src/pdl/interpreter.rs

@@ -1,9 +1,9 @@
 // use ::std::cell::LazyCell;
 use ::std::collections::HashMap;
-use ::std::env::current_dir;
+// use ::std::env::current_dir;
 use ::std::error::Error;
 use ::std::fs::File;
-use ::std::path::PathBuf;
+// use ::std::path::PathBuf;
 
 use minijinja::value::ValueKind;
 use minijinja::Environment;
@@ -30,8 +30,8 @@ type Interpretation = Result<(Context, PdlBlock), Box<dyn Error>>;
 struct Interpreter<'a> {
     // batch: u32,
     // role: Role,
-    cwd: Box<PathBuf>,
-    id_stack: Vec<String>,
+    // cwd: Box<PathBuf>,
+    // id_stack: Vec<String>,
     jinja_env: Environment<'a>,
     rt: Runtime,
     scope: Vec<Scope>,
@@ -43,8 +43,8 @@ impl<'a> Interpreter<'a> {
         Self {
             // batch: 0,
             // role: Role::User,
-            cwd: Box::new(current_dir().unwrap_or(PathBuf::from("/"))),
-            id_stack: vec![],
+            // cwd: Box::new(current_dir().unwrap_or(PathBuf::from("/"))),
+            // id_stack: vec![],
             jinja_env: Environment::new(),
             rt: Runtime::new().unwrap(),
             scope: vec![Scope::new()],
@@ -119,7 +119,7 @@ impl<'a> Interpreter<'a> {
                 0.0
             };
 
-            let tools = if let Some(Value::Array(tools)) = parameters.get(&"tools".to_string()) {
+            let tools = if let Some(Value::Array(_tools)) = parameters.get(&"tools".to_string()) {
                 // TODO
                 //tools.into_iter().map(|tool| function!()).collect()
                 vec![]

pdl-live-react/src-tauri/src/pdl/interpreter_tests.rs

@@ -0,0 +1,48 @@
+#[cfg(test)]
+mod tests {
+    //    use super::*;
+    use ::std::error::Error;
+
+    use crate::pdl::{
+        ast::{
+            PdlBlock,
+            /*PdlCallBlock,*/
+            PdlModelBlock, /*PdlRepeatBlock, PdlTextBlock, PdlUsage, Role,*/
+        },
+        interpreter::run,
+    };
+
+    use ollama_rs::generation::chat::MessageRole;
+
+    const DEFAULT_MODEL: &'static str = "ollama/granite3.2:2b";
+
+    #[test]
+    fn string() -> Result<(), Box<dyn Error>> {
+        let (messages, _) = run(&PdlBlock::String("hello".into()), false)?;
+        assert_eq!(messages.len(), 1);
+        assert_eq!(messages[0].role, MessageRole::User);
+        assert_eq!(messages[0].content, "hello");
+        Ok(())
+    }
+
+    #[test]
+    fn single_model() -> Result<(), Box<dyn Error>> {
+        let (messages, _) = run(
+            &PdlBlock::Model(PdlModelBlock {
+                def: None,
+                description: None,
+                model_response: None,
+                parameters: None,
+                pdl_result: None,
+                pdl_usage: None,
+                model: DEFAULT_MODEL.into(),
+                input: Some(Box::new(PdlBlock::String("hello".into()))),
+            }),
+            false,
+        )?;
+        assert_eq!(messages.len(), 1);
+        assert_eq!(messages[0].role, MessageRole::Assistant);
+        assert!(messages[0].content.contains("Hello!"));
+        Ok(())
+    }
+}

pdl-live-react/src-tauri/src/pdl/mod.rs

@@ -1,6 +1,7 @@
 pub mod ast;
 pub mod extract;
 pub mod interpreter;
+mod interpreter_tests;
 pub mod pip;
 pub mod pull;
 pub mod requirements;