openfunctions-rs 0.1.0

A universal framework for creating and managing LLM tools and agents
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

//! Manages the environment variables for tool execution.

use std::collections::HashMap;

/// Environment manager for tool execution.
pub struct Environment {
    variables: HashMap<String, String>,
}

impl Environment {
    /// Create a new environment manager.
    pub fn new() -> Self {
        Self {
            variables: HashMap::new(),
        }
    }

    /// Get the base environment variables.
    pub fn get_base_env(&self) -> HashMap<String, String> {
        self.variables.clone()
    }

    /// Set an environment variable.
    pub fn set(&mut self, key: String, value: String) {
        self.variables.insert(key, value);
    }
}

impl Default for Environment {
    fn default() -> Self {
        Self::new()
    }
}