ruchat 0.1.2

ollama/chroma command-line AI chat tool
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

use crate::error::RuChatError;
use ollama_rs::models::ModelOptions;
use serde_json::Value;

/// Reads a JSON file containing model options.
///
/// This function reads the specified JSON file and parses it into a `Value`.
///
/// # Parameters
///
/// - `options_path`: The path to the JSON file containing model options.
///
/// # Returns
///
/// A `Result` containing the parsed `Value` or a `RuChatError`.
async fn read_options_file(options_path: &str) -> Result<Value, RuChatError> {
    let options = std::fs::read_to_string(options_path)?;
    let options = serde_json::from_str(&options)?;
    Ok(options)
}

/// Get model options for prompt handling from a JSON file.
///
/// This function retrieves model options from a specified JSON configuration
/// file. If no configuration file is provided, it returns the default model
/// options.
///
/// # Parameters
///
/// - `config`: An optional path to the JSON configuration file.
///
/// # Returns
///
/// A `Result` containing the `ModelOptions` or a `RuChatError`.
pub(crate) async fn get_options(config: &Option<String>) -> Result<ModelOptions, RuChatError> {
    if let Some(options_path) = config {
        let mut defaults = serde_json::to_value(ModelOptions::default())?;

        if let Value::Object(ref mut defaults) = defaults {
            let updates = read_options_file(options_path).await?;
            if let Value::Object(config_updates) = updates {
                for (k, v) in config_updates.into_iter() {
                    if defaults.contains_key(&k) && !v.is_null() {
                        defaults[&k] = v.clone();
                    }
                }
            }
        }
        serde_json::from_value(defaults).map_err(RuChatError::SerdeError)
    } else {
        Ok(ModelOptions::default())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::fs;
    use std::path::Path;

    #[tokio::test]
    async fn test_read_options_file() {
        let path = "test_options.json";
        fs::write(path, r#"{\"option1\": \"value1\"}"#).unwrap();
        let result = read_options_file(path).await;
        assert!(result.is_ok());
        let value = result.unwrap();
        assert_eq!(value["option1"], "value1");
        fs::remove_file(path).unwrap();
    }

    #[tokio::test]
    async fn test_get_options_with_file() {
        let path = "test_options.json";
        fs::write(path, r#"{\"option1\": \"value1\"}"#).unwrap();
        let config = Some(path.to_string());
        let result = get_options(&config).await;
        assert!(result.is_ok());
        fs::remove_file(path).unwrap();
    }

    #[tokio::test]
    async fn test_get_options_without_file() {
        let config = None;
        let result = get_options(&config).await;
        assert!(result.is_ok());
    }
}