tandem-memory 0.6.9

Memory storage and embedding utilities for Tandem
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

impl MemoryManager {
    async fn repair_store(&self) -> bool {
        self.store
            .recover_backend(MemoryBackendRecoveryRequest {
                action: MemoryBackendRecoveryAction::RepairIndexes,
                confirm_data_loss: false,
            })
            .await
            .map(|result| result.changed)
            .unwrap_or(false)
    }

    async fn reset_store(&self) -> MemoryResult<()> {
        self.store
            .recover_backend(MemoryBackendRecoveryRequest {
                action: MemoryBackendRecoveryAction::ResetAllData,
                confirm_data_loss: true,
            })
            .await
            .map(|_| ())
            .map_err(MemoryError::from)
    }

    async fn write_cleanup_log(
        &self,
        cleanup_type: &str,
        tier: MemoryTier,
        project_id: Option<&str>,
        session_id: Option<&str>,
        chunks_deleted: i64,
        tenant_scope: &MemoryTenantScope,
    ) -> MemoryResult<()> {
        match self
            .store
            .write(MemoryStoreWriteRequest::CleanupLog {
                scope: MemoryWriteScope::tenant(tenant_scope.clone()),
                entry: MemoryCleanupLogWrite {
                    cleanup_type: cleanup_type.to_string(),
                    tier,
                    project_id: project_id.map(ToString::to_string),
                    session_id: session_id.map(ToString::to_string),
                    chunks_deleted,
                    bytes_reclaimed: 0,
                },
            })
            .await
            .map_err(MemoryError::from)?
        {
            MemoryStoreWriteResult::Stored => Ok(()),
            _ => Err(Self::unexpected_store_result("write cleanup log")),
        }
    }

    async fn write_context_layer(
        &self,
        node_id: &str,
        layer_type: LayerType,
        content: &str,
        token_count: i64,
        source_chunk_id: Option<&str>,
        tenant_scope: &MemoryTenantScope,
    ) -> MemoryResult<String> {
        match self
            .store
            .write(MemoryStoreWriteRequest::ContextLayer {
                scope: MemoryWriteScope::tenant(tenant_scope.clone()),
                node_id: node_id.to_string(),
                layer_type,
                content: content.to_string(),
                token_count,
                source_chunk_id: source_chunk_id.map(ToString::to_string),
            })
            .await
            .map_err(MemoryError::from)?
        {
            MemoryStoreWriteResult::ContextLayerCreated(id) => Ok(id),
            _ => Err(Self::unexpected_store_result("create context layer")),
        }
    }
}