personal memory agent
formatting
+9
-8
+5
-1
tests/test_dream_segment.py
+5
-1
tests/test_dream_segment.py
···
116
116
def mock_get_muse_configs(schedule=None, **kwargs):
117
117
return {
118
118
"low_priority": {"priority": 10, "output": "md", "schedule": "segment"},
119
-
"high_priority": {"priority": 90, "output": "md", "schedule": "segment"},
119
+
"high_priority": {
120
+
"priority": 90,
121
+
"output": "md",
122
+
"schedule": "segment",
123
+
},
120
124
"mid_priority": {"priority": 50, "output": "md", "schedule": "segment"},
121
125
}
122
126
+1
-2
tests/test_generators.py
+1
-2
tests/test_generators.py
···
81
81
82
82
# Unknown schedule returns empty dict
83
83
assert (
84
-
muse.get_muse_configs(has_tools=False, has_output=True, schedule="hourly")
85
-
== {}
84
+
muse.get_muse_configs(has_tools=False, has_output=True, schedule="hourly") == {}
86
85
)
87
86
assert muse.get_muse_configs(has_tools=False, has_output=True, schedule="") == {}
88
87
+1
-3
tests/test_journal_index.py
+1
-3
tests/test_journal_index.py
···
639
639
from think.indexer.journal import index_file, search_journal
640
640
641
641
# Index a specific file
642
-
result = index_file(
643
-
str(journal_fixture), "20240101/agents/flow.md", verbose=True
644
-
)
642
+
result = index_file(str(journal_fixture), "20240101/agents/flow.md", verbose=True)
645
643
assert result is True
646
644
647
645
# Should be searchable
+1
-1
think/agents.py
+1
-1
think/agents.py
···
28
28
from google.genai import types
29
29
30
30
from think.cluster import cluster, cluster_period, cluster_span
31
-
from think.providers.shared import Event, GenerateResult
32
31
from think.muse import (
33
32
compose_instructions,
34
33
get_agent_filter,
···
40
39
source_is_enabled,
41
40
source_is_required,
42
41
)
42
+
from think.providers.shared import Event, GenerateResult
43
43
from think.utils import (
44
44
day_log,
45
45
day_path,
+1
-1
think/dream.py
+1
-1
think/dream.py
···
19
19
from think.callosum import CallosumConnection
20
20
from think.cortex_client import cortex_request, get_agent_end_state, wait_for_agents
21
21
from think.facets import get_active_facets, get_enabled_facets, get_facets
22
-
from think.runner import run_task
23
22
from think.muse import get_muse_configs, get_output_path
23
+
from think.runner import run_task
24
24
from think.utils import (
25
25
day_input_summary,
26
26
day_log,