wesichain-core 0.3.0

Rust-native LLM agents & chains with resumable ReAct workflows
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

use serde::{Deserialize, Serialize};
use serde_json::Value;
use std::collections::HashMap;

/// A serializable representation of a Runnable.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum SerializableRunnable {
    Chain {
        steps: Vec<SerializableRunnable>,
    },
    Parallel {
        steps: HashMap<String, SerializableRunnable>,
    },
    Fallbacks {
        primary: Box<SerializableRunnable>,
        fallbacks: Vec<SerializableRunnable>,
    },
    Llm {
        model: String,
        #[serde(default)]
        params: HashMap<String, Value>,
    },
    Parser {
        kind: String, // "str", "json", "structured"
        #[serde(default)]
        target_type: Option<String>,
    },
    Prompt {
        template: String,
        input_variables: Vec<String>,
    },
    Tool {
        name: String,
        description: Option<String>,
        schema: Option<Value>,
    },
    Passthrough,
}

impl SerializableRunnable {
    pub fn to_json(&self) -> serde_json::Result<String> {
        serde_json::to_string(self)
    }

    pub fn from_json(json: &str) -> serde_json::Result<Self> {
        serde_json::from_str(json)
    }
}

// Conversion to actual Runnable trait objects is handled in `src/persistence.rs` via `reconstruct`.