neomemx 0.1.2

A high-performance memory library for AI agents with semantic search
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

//! Change tracking and history

use crate::core::scope::ScopeIdentifiers;
use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};

/// Type of modification to a fact
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum ChangeType {
    /// Fact was created
    Created,
    /// Fact was updated
    Updated,
    /// Fact was deleted
    Deleted,
    /// Fact was consolidated with another
    Consolidated,
}

impl ChangeType {
    /// Convert to string representation
    pub fn as_str(&self) -> &'static str {
        match self {
            Self::Created => "created",
            Self::Updated => "updated",
            Self::Deleted => "deleted",
            Self::Consolidated => "consolidated",
        }
    }
}

impl std::fmt::Display for ChangeType {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.as_str())
    }
}

/// Record of a change to a fact
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ChangeLog {
    /// ID of the fact that changed
    pub fact_id: String,

    /// Type of change
    pub change_type: ChangeType,

    /// Timestamp of the change
    pub timestamp: DateTime<Utc>,

    /// Previous content (if applicable)
    #[serde(skip_serializing_if = "Option::is_none")]
    pub previous_content: Option<String>,

    /// New content (if applicable)
    #[serde(skip_serializing_if = "Option::is_none")]
    pub new_content: Option<String>,

    /// Scope identifiers
    pub scope: ScopeIdentifiers,

    /// Actor who made the change
    #[serde(skip_serializing_if = "Option::is_none")]
    pub actor: Option<String>,
}

impl ChangeLog {
    /// Create a new change log entry
    pub fn new(
        fact_id: impl Into<String>,
        change_type: ChangeType,
        scope: ScopeIdentifiers,
    ) -> Self {
        Self {
            fact_id: fact_id.into(),
            change_type,
            timestamp: Utc::now(),
            previous_content: None,
            new_content: None,
            scope,
            actor: None,
        }
    }

    /// Set previous content
    pub fn with_previous_content(mut self, content: impl Into<String>) -> Self {
        self.previous_content = Some(content.into());
        self
    }

    /// Set new content
    pub fn with_new_content(mut self, content: impl Into<String>) -> Self {
        self.new_content = Some(content.into());
        self
    }

    /// Set actor
    pub fn with_actor(mut self, actor: impl Into<String>) -> Self {
        self.actor = Some(actor.into());
        self
    }
}