viz

engine/baml-rpc/src/ui/mod.rs

@@ -1,4 +1,5 @@
 pub mod ui_baml_src;
+pub mod ui_blobs;
 pub mod ui_control_plane_orgs;
 pub mod ui_control_plane_projects;
 pub mod ui_dashboard;

engine/baml-rpc/src/ui/ui_baml_src.rs

@@ -32,7 +32,7 @@ pub struct BamlSourceCode {
 pub struct AstNodeDefinition {
     pub project_id: String,
     pub ast_node_id: String,
-    #[ts(type = "any")]
+    #[ts(type = "unknown")]
     pub ast_node_definition: serde_json::Value,
     pub flattened_dependencies_ast_nodes: Vec<String>,
     pub baml_src_node_ids: Vec<String>,

engine/baml-rpc/src/ui/ui_blobs.rs

@@ -0,0 +1,42 @@
+use serde::{Deserialize, Serialize};
+use ts_rs::TS;
+
+use crate::ProjectId;
+
+/// Request to fetch blob content
+#[derive(Debug, Serialize, Deserialize, TS)]
+#[ts(export)]
+pub struct GetBlobRequest {
+    /// The 64-character BLAKE3 hash of the blob
+    pub blob_hash: String,
+    /// The project ID that owns this blob
+    #[ts(type = "string")]
+    pub project_id: ProjectId,
+    /// Response format: 'raw' returns binary, 'json' returns base64-encoded content
+    #[serde(default)]
+    pub format: BlobFormat,
+}
+
+/// Response format for blob content
+#[derive(Debug, Serialize, Deserialize, TS, Default)]
+#[ts(export)]
+#[serde(rename_all = "lowercase")]
+pub enum BlobFormat {
+    /// Return raw binary content with appropriate Content-Type header
+    #[default]
+    Raw,
+    /// Return JSON with base64-encoded content
+    Json,
+}
+
+/// JSON response format for blob content
+#[derive(Debug, Serialize, Deserialize, TS)]
+#[ts(export)]
+pub struct GetBlobResponse {
+    /// The blob hash
+    pub blob_hash: String,
+    /// The content type of the decoded blob (e.g., "image/png")
+    pub content_type: String,
+    /// Base64-encoded content (exact same as used in LLM request)
+    pub base64_content: String,
+}

engine/baml-rpc/src/ui/ui_types.rs

@@ -94,7 +94,7 @@ pub struct UiFunctionCall {
     #[ts(optional)]
     pub function_id: Option<UiFunctionIdString>,
 
-    #[ts(type = "Record<string, string>")]
+    #[ts(type = "Record<string, unknown>")]
     pub tags: serde_json::Map<String, serde_json::Value>,
 
     #[serde(rename = "start_epoch_ms")]
@@ -105,7 +105,7 @@ pub struct UiFunctionCall {
     pub end_time: Option<EpochMsTimestamp>,
     pub status: String,
 
-    #[ts(type = "any")]
+    #[ts(type = "unknown")]
     pub baml_options: serde_json::Value,
     pub inputs: Vec<UiFunctionInput>,
     #[ts(as = "Option<BamlValue>")]
@@ -130,9 +130,9 @@ pub struct UiLlmRequest {
     pub client_name: String,
     pub client_provider: String,
     // TODO: type this out properly.
-    #[ts(type = "any")]
+    #[ts(type = "unknown")]
     pub params: serde_json::Value,
-    #[ts(type = "any")]
+    #[ts(type = "unknown")]
     pub prompt: serde_json::Value,
 }
 
@@ -186,7 +186,7 @@ pub struct UiHttpRequest {
     pub start_time: EpochMsTimestamp,
     pub url: String,
     pub method: String,
-    #[ts(type = "Record<string, any> | undefined")]
+    #[ts(type = "Record<string, string> | undefined")]
     pub headers: Option<HashMap<String, String>>,
     pub body: String,
 }
@@ -197,7 +197,7 @@ pub struct UiHttpResponse {
     #[ts(type = "number")]
     pub end_time: EpochMsTimestamp,
     pub status_code: u16,
-    #[ts(type = "Record<string, any>")]
+    #[ts(type = "Record<string, unknown>")]
     pub headers: HashMap<String, serde_json::Value>,
     pub body: String,
 }

engine/baml-runtime/Cargo.toml

@@ -125,6 +125,7 @@ once_cell.workspace = true
 time.workspace = true
 tempfile = "3.19.0"
 sha2 = "0.10.9"
+blake3 = "1.8.2"
 
 
 [target.'cfg(target_arch = "wasm32")'.dependencies]

engine/baml-runtime/src/tracingv2/publisher/rpc_converters/blob_storage.rs

@@ -7,7 +7,7 @@ use std::{
 use baml_rpc::runtime_api::baml_value::{BamlValue, MediaValue, ValueContent};
 use base64::{engine::general_purpose, Engine as _};
 use serde::{Deserialize, Serialize};
-use sha2::{Digest, Sha256};
+use blake3;
 use tokio::sync::mpsc;
 
 use crate::tracingv2::publisher::publisher::BlobUploaderMessage;
@@ -82,11 +82,10 @@ impl BlobRefCache {
         }
     }
 
-    /// Generate a hash for a blob
+    /// Generate a hash for a blob using BLAKE3
     pub fn hash_blob(content: &[u8]) -> String {
-        let mut hasher = Sha256::new();
-        hasher.update(content);
-        format!("{:x}", hasher.finalize())
+        let hash = blake3::hash(content);
+        hash.to_hex().to_string()
     }
 
     /// Store a blob (as base64 string) and associate it with a function_call_id