Dhee/dhee/world_memory/__init__.py at c7258c16247fa232696310d8f21351123f7d4405 · Sankhya-AI/Dhee · GitHub

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
"""World-memory primitives for screen-aware desktop memory.

This subsystem keeps transition-oriented screen/app memory separate from
the main Dhee text-memory tables while mirroring compact summaries back
into Dhee for durable retrieval and cognition.
"""

from .capture_store import CaptureStore
from .encoder import (
    ContentAwareFrameEncoder,
    DeterministicFrameEncoder,
    NvidiaVLFrameEncoder,
    create_default_encoder,
)
from .predictor import ActionConditionedPredictor, compute_surprise
from .causal_graph import CausalGraphProjection
from .gem_extractor import (
    GEM_SCHEMA_VERSION,
    MemoryGem,
    extract_memory_gems,
    score_memory_gem,
    submit_gem_learning_candidates,
    submit_projected_gem_learning_candidate,
    summarize_gems,
    write_gem_raw_events,
)
from .schema import (
    CAUSAL_PROJECTION_VERSION,
    CAUSAL_SCHEMA_VERSION,
    ActionTransition,
    CausalEdge,
    CaptureAction,
    CaptureEvent,
    CaptureLink,
    CapturePolicy,
    CaptureSession,
    CapturedArtifact,
    CapturedObservation,
    CapturedSurface,
    CheckpointReport,
    EvidenceChunk,
    EventFrame,
    RawEvent,
    RetrievalTrace,
    TransitionMatch,
    WorldState,
)
from .session_graph import SessionGraphStore
from .service import MemoryOSService
from .store import WorldMemoryStore

__all__ = [
    "ActionConditionedPredictor",
    "ActionTransition",
    "CAUSAL_PROJECTION_VERSION",
    "CAUSAL_SCHEMA_VERSION",
    "CausalEdge",
    "CausalGraphProjection",
    "GEM_SCHEMA_VERSION",
    "CaptureAction",
    "CaptureEvent",
    "CaptureLink",
    "CapturePolicy",
    "CaptureSession",
    "CaptureStore",
    "CapturedArtifact",
    "CapturedObservation",
    "CapturedSurface",
    "CheckpointReport",
    "ContentAwareFrameEncoder",
    "DeterministicFrameEncoder",
    "EvidenceChunk",
    "EventFrame",
    "MemoryGem",
    "MemoryOSService",
    "NvidiaVLFrameEncoder",
    "SessionGraphStore",
    "RawEvent",
    "RetrievalTrace",
    "TransitionMatch",
    "WorldMemoryStore",
    "WorldState",
    "compute_surprise",
    "create_default_encoder",
    "extract_memory_gems",
    "score_memory_gem",
    "submit_gem_learning_candidates",
    "submit_projected_gem_learning_candidate",
    "summarize_gems",
    "write_gem_raw_events",
]