BAML Workflow Rendering

engine/Cargo.toml

@@ -163,8 +163,8 @@ tokio = { version = "1", default-features = false, features = [
   "time",
 ] }
 
-js-sys = "=0.3.78"
-wasm-bindgen = "=0.2.101"
+js-sys = "=0.3.82"
+wasm-bindgen = "=0.2.105"
 wasm-bindgen-futures = "0.4.42"
 wasm-bindgen-test = "0.3.45"
 web-sys = { version = "0.3.78", features = [
@@ -179,6 +179,8 @@ web-sys = { version = "0.3.78", features = [
   "Window",
 ] }
 web-time = "1.1.0"
+tsify = { version = "0.5.6", features = ["js"] }
+
 
 [workspace.package]
 version = "0.214.0"

engine/baml-compiler/src/baml_src/foo.baml

@@ -0,0 +1,6 @@
+function Foo() -> string {
+    //# lorem
+    //## ipsum
+    "ipsum"
+    //# dolor
+}

engine/baml-compiler/src/watch/watch_event.rs

@@ -89,6 +89,10 @@ pub fn shared_noop_handler() -> SharedWatchHandler {
 pub enum WatchBamlValue {
     Value(BamlValueWithMeta<WatchValueMetadata>),
     Header(HeaderContext),
+    /// HACK: Emitted synthetically when a new header comes in at the same or shallower level.
+    /// This signals that the previous header's scope has ended. This is a workaround until
+    /// proper exit events are emitted from the interpreter.
+    HeaderStopped(HeaderContext),
     StreamStart(StreamId),
     StreamUpdate(StreamId, BamlValueWithMeta<WatchValueMetadata>),
     StreamEnd(StreamId),
@@ -145,6 +149,15 @@ impl fmt::Display for WatchNotification {
                     title = header.title
                 )
             }
+            WatchBamlValue::HeaderStopped(header) => {
+                write!(
+                    f,
+                    "(header stopped L{level}) {function}.{title}",
+                    level = header.level,
+                    function = self.function_name,
+                    title = header.title
+                )
+            }
             WatchBamlValue::StreamStart(stream_id) => {
                 write!(f, "(stream start) {stream_id}")
             }
@@ -203,6 +216,18 @@ impl WatchNotification {
         }
     }
 
+    /// HACK: Create a synthetic "stopped" notification for a header block.
+    /// Used when a new header at the same or shallower level comes in.
+    pub fn new_block_stopped(header: HeaderContext, function_name: String) -> Self {
+        Self {
+            value: WatchBamlValue::HeaderStopped(header),
+            variable_name: None,
+            channel_name: None,
+            function_name,
+            is_stream: false,
+        }
+    }
+
     pub fn new_stream_start(
         variable_name: String,
         stream_id: StreamId,

engine/baml-lib/baml-types/src/lib.rs

@@ -1,10 +1,19 @@
+use serde::{Deserialize, Serialize};
+
 mod constraint;
 pub mod expr;
 mod map;
 mod media;
 mod minijinja;
 pub mod tracing;
 
+#[derive(Clone, Copy, Debug, PartialEq, Eq, Serialize, Deserialize)]
+#[serde(rename_all = "snake_case")]
+pub enum FunctionFlavor {
+    Llm,
+    Expr,
+}
+
 pub mod baml_value;
 mod generator;
 pub mod ir_type;

engine/baml-lsp-types/src/baml_lsp_types.rs

@@ -3,6 +3,7 @@
 // but without the wasm-bindgen dependency.
 // Wasm-bindgen causes strange issues with ruby builds so we'd rather not add that dependency if we can.
 
+use baml_types::FunctionFlavor;
 use serde::{Deserialize, Serialize};
 
 #[derive(Clone, Debug, Serialize, Deserialize)]
@@ -57,6 +58,7 @@ pub struct BamlFunctionTestCasePair {
 pub struct BamlFunction {
     pub name: String,
     pub span: BamlSpan,
+    pub function_type: FunctionFlavor,
     pub test_cases: Vec<BamlFunctionTestCasePair>,
     pub test_snippet: String,
     pub signature: String,

engine/baml-runtime/src/lib.rs

@@ -956,7 +956,7 @@ impl BamlRuntime {
         baml_log::set_from_env(&env_vars).unwrap();
 
         log::info!(
-            "[Runtime] run_test_with_expr_events start function={function_name} test={test_name}"
+            "[Runtime] run_test_with_expr_events start function={function_name} test={test_name}...."
         );
 
         let call = self

engine/baml-runtime/src/runtime/mod.rs

@@ -43,9 +43,20 @@ impl CachedClient {
     }
 
     pub fn has_env_vars_changed(&self, new_env_vars: &HashMap<String, String>) -> bool {
-        self.env_vars
+        // Check if any existing env vars have different values
+        let values_changed = self.env_vars
             .iter()
-            .any(|(k, v)| new_env_vars.get(k).is_some_and(|v2| v2 != v))
+            .any(|(k, v)| new_env_vars.get(k).is_some_and(|v2| v2 != v));
+
+        if values_changed {
+            return true;
+        }
+
+        // Check if BOUNDARY_PROXY_URL was added or removed (affects client configuration)
+        let had_proxy = self.env_vars.contains_key("BOUNDARY_PROXY_URL");
+        let has_proxy = new_env_vars.contains_key("BOUNDARY_PROXY_URL");
+
+        had_proxy != has_proxy
     }
 }
 

engine/baml-schema-wasm/Cargo.toml

@@ -58,6 +58,7 @@ either = "1.8.1"
 itertools = "0.13.0"
 once_cell.workspace = true
 stream-cancel = "0.8"
+tsify.workspace = true
 
 [dev-dependencies]
 dissimilar = "1.0.4"