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

//! Storage builder for fluent API

use crate::core::{IngestionOutcome, ScopeIdentifiers};
use crate::engine::NeomemxEngine;
use crate::error::{NeomemxError, Result};
use std::collections::HashMap;

/// Metadata map type
pub type MetadataMap = HashMap<String, serde_json::Value>;

/// Builder for storage operations
pub struct StorageBuilder {
    engine: NeomemxEngine,
    content: String,
    scope: Option<ScopeIdentifiers>,
    metadata: Option<MetadataMap>,
    enable_extraction: bool,
    enable_graph: bool,
}

impl StorageBuilder {
    /// Create a new storage builder
    pub(crate) fn new(engine: NeomemxEngine, content: impl Into<String>) -> Self {
        Self {
            engine,
            content: content.into(),
            scope: None,
            metadata: None,
            enable_extraction: true,
            enable_graph: false,
        }
    }

    /// Set scoping identifiers
    pub fn with_scope(mut self, scope: ScopeIdentifiers) -> Self {
        self.scope = Some(scope);
        self
    }

    /// Add metadata
    pub fn with_metadata(mut self, metadata: MetadataMap) -> Self {
        self.metadata = Some(metadata);
        self
    }

    /// Enable/disable fact extraction
    pub fn enable_extraction(mut self, enabled: bool) -> Self {
        self.enable_extraction = enabled;
        self
    }

    /// Enable/disable graph relationship building
    pub fn enable_graph(mut self, enabled: bool) -> Self {
        self.enable_graph = enabled;
        self
    }

    /// Execute the storage operation
    pub async fn execute(self) -> Result<IngestionOutcome> {
        let scope = self.scope.ok_or_else(|| NeomemxError::required("scope"))?;

        scope.validate()?;

        self.engine
            .store_internal(
                &self.content,
                scope,
                self.metadata,
                self.enable_extraction,
                self.enable_graph,
            )
            .await
    }
}