cognis-core 0.2.1

Core traits and types for the Cognis LLM framework
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

//! Type mappings for serialization.
//!
//! Mirrors Python `langchain_core.load.mapping`.

use std::collections::HashMap;

/// Get the default mapping of class paths to their modules.
/// This is used during deserialization to validate allowed types.
pub fn get_default_mappings() -> HashMap<String, String> {
    let mut m = HashMap::new();
    // Core types
    m.insert(
        "cognis_core.prompts.base.PromptTemplate".to_string(),
        "prompts".to_string(),
    );
    m.insert(
        "cognis_core.prompts.chat.ChatPromptTemplate".to_string(),
        "prompts".to_string(),
    );
    m.insert(
        "cognis_core.prompts.few_shot.FewShotPromptTemplate".to_string(),
        "prompts".to_string(),
    );
    m.insert(
        "cognis_core.output_parsers.json.JsonOutputParser".to_string(),
        "output_parsers".to_string(),
    );
    m.insert(
        "cognis_core.output_parsers.string.StrOutputParser".to_string(),
        "output_parsers".to_string(),
    );
    m.insert(
        "cognis_core.output_parsers.list.CommaSeparatedListOutputParser".to_string(),
        "output_parsers".to_string(),
    );
    m.insert(
        "cognis_core.documents.Document".to_string(),
        "documents".to_string(),
    );
    m.insert(
        "cognis_core.messages.HumanMessage".to_string(),
        "messages".to_string(),
    );
    m.insert(
        "cognis_core.messages.AIMessage".to_string(),
        "messages".to_string(),
    );
    m.insert(
        "cognis_core.messages.SystemMessage".to_string(),
        "messages".to_string(),
    );
    m.insert(
        "cognis_core.messages.ToolMessage".to_string(),
        "messages".to_string(),
    );
    // LangChain compatibility
    m.insert(
        "langchain_core.prompts.prompt.PromptTemplate".to_string(),
        "prompts".to_string(),
    );
    m.insert(
        "langchain_core.prompts.chat.ChatPromptTemplate".to_string(),
        "prompts".to_string(),
    );
    m.insert(
        "langchain_core.output_parsers.json.JsonOutputParser".to_string(),
        "output_parsers".to_string(),
    );
    m.insert(
        "langchain_core.output_parsers.string.StrOutputParser".to_string(),
        "output_parsers".to_string(),
    );
    m.insert(
        "langchain_core.documents.base.Document".to_string(),
        "documents".to_string(),
    );
    m
}

/// Check if a class path is in the default allowed mappings.
pub fn is_allowed_class_path(path: &str) -> bool {
    get_default_mappings().contains_key(path)
}