ayda 1.1.1

Ask your Documents Anything. A tool for querying your documents with a large language model.
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

//! Administrative commands
//!
//! These commands are used to configure the application and perform other administrative tasks.
//!
//! # Examples
//! > aza admin configure
//!

use std::fs;
use std::io::{Error, Write};
use std::path::PathBuf;

use dialoguer::Editor;
use serde_json::{from_reader, from_str, to_string_pretty};
use tracing::{instrument, trace};

use crate::Config;

#[instrument]
pub fn configure(config_path: &PathBuf) -> Result<(), Error> {
    if !config_path.exists()
        || fs::File::open(config_path)
            .and_then(|f| from_reader::<_, Config>(f).map_err(std::io::Error::from))
            .is_err()
    {
        println!("Config file not found or invalid. Creating a new one.");
        let config = crate::get_config_parameters();
        let config_str = to_string_pretty(&config).unwrap();
        if let Some(parent_dir) = config_path.parent() {
            fs::create_dir_all(parent_dir)?;
        }
        let mut file = fs::File::create(config_path)?;
        file.write_all(config_str.as_bytes())?;

        trace!("Created config file: {:?}", &file);
        println!("Config file created at {:?}", config_path);

        Ok(())
    } else {
        println!("Editing config file.");
        let file = fs::File::open(config_path)?;
        let config: Config = from_reader(&file)?;
        let config_str = to_string_pretty(&config).unwrap();

        if let Some(rv) = Editor::new().edit(&config_str).unwrap() {
            println!("Your config:");
            println!("{}", rv);

            // round trip to ensure the edited config is valid
            let config: Config = from_str(&rv).unwrap();
            let config_str = to_string_pretty(&config).unwrap();

            let mut file = fs::File::create(config_path)?;
            file.write_all(config_str.as_bytes())?;
        } else {
            println!("Abort!");
        }

        Ok(())
    }
}