ollama-oxide 0.2.0

A Rust library for integrating with Ollama's native API, providing low-level inference and high-level conveniences.
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

//! Format setting primitive type

use serde::{Deserialize, Serialize};

/// Format setting for generate requests
///
/// Controls the output format of the model response.
/// Can be a string (like "json") or a JSON schema object.
///
/// # Examples
///
/// ```no_run
/// use ollama_oxide::FormatSetting;
///
/// let json = FormatSetting::json();
/// let schema = FormatSetting::schema(serde_json::json!({
///     "type": "object",
///     "properties": { "name": { "type": "string" } }
/// }));
/// ```
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(untagged)]
pub enum FormatSetting {
    /// Simple format string (e.g., "json")
    String(String),
    /// JSON schema object for structured output
    Schema(serde_json::Value),
}

impl FormatSetting {
    /// Create JSON format
    pub fn json() -> Self {
        Self::String("json".to_string())
    }

    /// Create custom format string
    pub fn string(format: impl Into<String>) -> Self {
        Self::String(format.into())
    }

    /// Create schema-based format
    pub fn schema(schema: serde_json::Value) -> Self {
        Self::Schema(schema)
    }
}

impl From<&str> for FormatSetting {
    fn from(s: &str) -> Self {
        Self::String(s.to_string())
    }
}

impl From<serde_json::Value> for FormatSetting {
    fn from(v: serde_json::Value) -> Self {
        Self::Schema(v)
    }
}