rustchain-community 1.0.0

Open-source AI agent framework with core functionality and plugin system
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

use crate::core::error::RustChainError;
use std::collections::hash_map::DefaultHasher;
use std::fs;
use std::hash::{Hash, Hasher};
use std::path::PathBuf;

pub fn hash_prompt(prompt: &str) -> u64 {
    let mut hasher = DefaultHasher::new();
    prompt.hash(&mut hasher);
    hasher.finish()
}

pub fn cache_lookup(prompt: &str) -> Option<String> {
    let key = hash_prompt(prompt);
    let path = PathBuf::from(format!("cache/prompts/{}.txt", key));
    fs::read_to_string(path).ok()
}

pub fn cache_store(prompt: &str, result: &str) -> Result<(), RustChainError> {
    let key = hash_prompt(prompt);
    let path = PathBuf::from("cache/prompts");
    fs::create_dir_all(&path)?;
    fs::write(path.join(format!("{}.txt", key)), result)?;
    Ok(())
}