Skip to content

Commit 035eed3

Browse files
fix(example): correct ParsedTaskGoal usage and add unrelated memory for testing (#23)
Co-authored-by: CaralHsi <[email protected]>
1 parent 6148813 commit 035eed3

File tree

1 file changed

+52
-7
lines changed

1 file changed

+52
-7
lines changed

examples/basic_modules/tree_textual_memory_reasoner.py

Lines changed: 52 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,11 @@
3737

3838
# Step 1: Prepare a mock ParsedTaskGoal
3939
parsed_goal = ParsedTaskGoal(
40-
topic_level=["Multi-UAV Long-Term Coverage"],
41-
concept_level=["Coverage Metrics", "Reward Function Design", "Energy Model"],
42-
fact_level=["CT and FT Definition", "Reward Components", "Energy Cost Components"],
43-
goal_type="explanation",
44-
graph_suggestion="Use all relevant knowledge from previous paper review",
45-
retrieval_keywords=["UAV", "coverage", "energy", "reward"],
40+
memories=["Multi-UAV Long-Term Coverage", "Coverage Metrics", "Reward Function Design", "Energy Model",
41+
"CT and FT Definition", "Reward Components", "Energy Cost Components"],
42+
keys=["UAV", "coverage", "energy", "reward"],
43+
tags=[],
44+
goal_type="explanation"
4645
)
4746

4847
query = "How can multiple UAVs coordinate to maximize coverage while saving energy?"
@@ -85,7 +84,53 @@
8584
sources=["paper://multi-uav-coverage/metrics"],
8685
embedding=[0.01] * 768,
8786
),
88-
)
87+
),
88+
TextualMemoryItem(
89+
id="c34f5e6b-2d34-4e6f-8c9b-abcdef123456",
90+
memory="The capital of France is Paris, which is known for the Eiffel Tower.",
91+
metadata=TreeNodeTextualMemoryMetadata(
92+
user_id=None,
93+
session_id=None,
94+
status="activated",
95+
type="fact",
96+
memory_time="2024-01-01",
97+
source="file",
98+
confidence=90.0,
99+
entities=["France", "Paris", "Eiffel Tower"],
100+
tags=["geography", "city", "landmark"],
101+
visibility="public",
102+
updated_at="2025-06-11T11:51:24.438001",
103+
memory_type="LongTermMemory",
104+
key="Geography Fact",
105+
value="Paris is the capital of France",
106+
hierarchy_level="concept",
107+
sources=["wikipedia://paris"],
108+
embedding=[0.03] * 768,
109+
),
110+
),
111+
TextualMemoryItem(
112+
id="d56a7b8c-3e45-4f7a-9dab-fedcba654321",
113+
memory="Total energy cost is calculated from both mechanical movement and communication transmission.",
114+
metadata=TreeNodeTextualMemoryMetadata(
115+
user_id=None,
116+
session_id=None,
117+
status="activated",
118+
type="fact",
119+
memory_time="2024-01-01",
120+
source="file",
121+
confidence=89.0,
122+
entities=["movement power", "transmission power"],
123+
tags=["energy", "movement", "transmission"],
124+
visibility="public",
125+
updated_at="2025-06-11T11:51:24.438001",
126+
memory_type="LongTermMemory",
127+
key="Energy Cost Components",
128+
value="Includes movement and communication energy",
129+
hierarchy_level="fact",
130+
sources=["paper://multi-uav-coverage/energy-detail"],
131+
embedding=[0.04] * 768,
132+
),
133+
),
89134
]
90135

91136
# Step 7: Init memory retriever

0 commit comments

Comments
 (0)