Add Prompts API

Cargo.toml

@@ -13,6 +13,7 @@ categories = ["api-bindings", "asynchronous", "development-tools"]
 [dependencies]
 anyhow = "1"
 base64 = "0.22"
+lingua = { git = "https://github.com/braintrustdata/lingua.git" }
 bon = "3"
 arc-swap = "1"
 async-trait = "0.1"

src/lib.rs

@@ -6,6 +6,7 @@ mod json_merge;
 mod log_queue;
 mod logger;
 mod logs3;
+mod prompt;
 mod span;
 mod span_components;
 mod stream;
@@ -21,12 +22,15 @@ pub use experiments::{
     GitMetadataSettings, MetricSummary, ProjectMetadata, RepoInfo, ScoreSummary,
 };
 pub use extractors::{extract_anthropic_usage, extract_openai_usage};
+pub use lingua::universal::{AssistantContent, TokenBudget, UserContent};
+pub use lingua::{Message, UniversalParams, UniversalRequest};
 pub use log_queue::LogQueueConfig;
 pub use logger::{
     BraintrustClient, BraintrustClientBuilder, LoginState, OrgInfo, DEFAULT_API_URL,
     DEFAULT_APP_URL,
 };
 pub use logs3::{Logs3BatchUploader, Logs3UploadResult};
+pub use prompt::{Prompt, PromptBuilder, PromptBuilderError};
 pub use span::{SpanBuilder, SpanHandle, SpanLog, SpanLogBuilder, SpanLogBuilderError};
 pub use span_components::SpanComponents;
 pub use stream::{

src/logger.rs

@@ -15,6 +15,7 @@ use crate::experiments::api::{
 };
 use crate::experiments::{BaseExperimentInfo, ExperimentBuilder};
 use crate::log_queue::{LogQueue, LogQueueConfig};
+use crate::prompt::{PromptBuilder, PromptFetchRequest, PromptFetchResponse, PromptFetcher};
 use crate::span::SpanSubmitter;
 use crate::types::{ParentSpanInfo, SpanPayload};
 
@@ -505,6 +506,13 @@ impl BraintrustClient {
         crate::span::SpanBuilder::new(submitter, token, org_id)
     }
 
+    /// Create a prompt builder with an explicit API token.
+    ///
+    /// Use this if you already have the token and don't want to use the login state.
+    pub fn prompt_builder_with_credentials(&self, token: impl Into<String>) -> PromptBuilder<Self> {
+        PromptBuilder::new(Arc::new(self.clone()), token)
+    }
+
     /// Perform login synchronously.
     async fn perform_login(&self, api_key: &str, org_name: Option<&str>) -> Result<()> {
         let login_url = self
@@ -910,6 +918,64 @@ impl ExperimentComparisonFetcher for BraintrustClient {
     }
 }
 
+#[async_trait::async_trait]
+impl PromptFetcher for BraintrustClient {
+    async fn fetch_prompt(
+        &self,
+        token: &str,
+        request: PromptFetchRequest,
+    ) -> Result<PromptFetchResponse> {
+        let mut url = self
+            .inner
+            .api_url
+            .join("v1/prompt")
+            .map_err(|e| BraintrustError::InvalidConfig(e.to_string()))?;
+
+        {
+            let mut query = url.query_pairs_mut();
+            query.append_pair("slug", &request.slug);
+            if let Some(project_id) = &request.project_id {
+                query.append_pair("project_id", project_id);
+            }
+            if let Some(project_name) = &request.project_name {
+                query.append_pair("project_name", project_name);
+            }
+            if let Some(version) = &request.version {
+                query.append_pair("version", version);
+            }
+            if let Some(environment) = &request.environment {
+                query.append_pair("environment", environment);
+            }
+        }
+
+        let response = self
+            .inner
+            .http_client
+            .get(url)
+            .bearer_auth(token)
+            .send()
+            .await
+            .map_err(|e| BraintrustError::Network(e.to_string()))?;
+
+        if !response.status().is_success() {
+            let status = response.status();
+            let body = response.text().await.unwrap_or_default();
+            return Err(BraintrustError::Api {
+                status: status.as_u16(),
+                message: body,
+            });
+        }
+
+        response
+            .json::<PromptFetchResponse>()
+            .await
+            .map_err(|e| BraintrustError::Api {
+                status: 200,
+                message: format!("Failed to parse prompt response: {}", e),
+            })
+    }
+}
+
 #[allow(dead_code)]
 #[derive(Debug, Default)]
 struct MutableSpanEvent {