baml 0.220.0

BAML runtime for Rust - type-safe LLM function calls
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

//! HTTP types for request/response introspection
//!
//! These wrap FFI pointers to HTTP objects managed by the BAML runtime.

use std::{collections::HashMap, ffi::c_void};

use super::{RawObject, RawObjectTrait};
use crate::{baml_unreachable, error::BamlError, proto::baml_cffi_v1::BamlObjectType};

// =============================================================================
// HTTPBody
// =============================================================================

define_raw_object_wrapper! {
    /// HTTP request or response body
    HTTPBody => ObjectHttpBody
}

impl HTTPBody {
    /// Get body as text (UTF-8 decoded)
    pub fn text(&self) -> Result<String, BamlError> {
        self.raw.try_call_method("text", ())
    }

    /// Get body as JSON value
    ///
    /// Parses the body text as JSON. Returns an error if the body is not valid
    /// JSON.
    pub fn json(&self) -> Result<serde_json::Value, BamlError> {
        let text: String = self.raw.try_call_method("text", ())?;
        serde_json::from_str(&text)
            .map_err(|e| BamlError::internal(format!("failed to parse JSON: {e}")))
    }
}

// =============================================================================
// HTTPRequest
// =============================================================================

define_raw_object_wrapper! {
    /// HTTP request details
    HTTPRequest => ObjectHttpRequest
}

impl HTTPRequest {
    /// Get the request ID
    pub fn id(&self) -> String {
        self.raw.call_method("id", ())
    }

    /// Get the request URL
    pub fn url(&self) -> String {
        self.raw.call_method("url", ())
    }

    /// Get the HTTP method (GET, POST, etc.)
    pub fn method(&self) -> String {
        self.raw.call_method("method", ())
    }

    /// Get request headers
    pub fn headers(&self) -> HashMap<String, String> {
        self.raw.call_method("headers", ())
    }

    /// Get the request body
    pub fn body(&self) -> HTTPBody {
        self.raw
            .call_method_for_object("body", ())
            .unwrap_or_else(|e| baml_unreachable!("Failed to get body: {e}"))
    }
}

// =============================================================================
// HTTPResponse
// =============================================================================

define_raw_object_wrapper! {
    /// HTTP response details
    HTTPResponse => ObjectHttpResponse
}

impl HTTPResponse {
    /// Get the request ID this response corresponds to
    pub fn id(&self) -> String {
        self.raw.call_method("id", ())
    }

    /// Get the HTTP status code
    pub fn status(&self) -> i64 {
        self.raw.call_method("status", ())
    }

    /// Get response headers
    pub fn headers(&self) -> HashMap<String, String> {
        self.raw.call_method("headers", ())
    }

    /// Get the response body
    pub fn body(&self) -> HTTPBody {
        self.raw
            .call_method_for_object("body", ())
            .unwrap_or_else(|e| baml_unreachable!("Failed to get body: {e}"))
    }
}

// =============================================================================
// SSEResponse
// =============================================================================

define_raw_object_wrapper! {
    /// Server-Sent Event response chunk
    SSEResponse => ObjectSseResponse
}

impl SSEResponse {
    /// Get the SSE data as text
    pub fn text(&self) -> String {
        self.raw.call_method("text", ())
    }

    /// Try to parse the SSE data as JSON
    pub fn json(&self) -> Option<serde_json::Value> {
        let text: String = self.raw.call_method("text", ());
        serde_json::from_str(&text).ok()
    }
}