Initial standalone memabra release

tests/test_trajectory_summary.py

@@ -0,0 +1,58 @@
+from memabra.trajectory_summary import TrajectorySummarizer
+
+
+def test_summarize_direct_answer_success():
+    summarizer = TrajectorySummarizer()
+    trajectory = {
+        "task": {"input": "What is 2+2?"},
+        "decisions": [{"decision_type": "direct_answer"}],
+        "outcome": {"status": "success", "steps": 1, "tool_errors": 0, "user_corrections": 0},
+        "reward": {"total": 1.0},
+    }
+    summary = summarizer.summarize(trajectory)
+    assert "Task: 'What is 2+2?'" in summary
+    assert "Actions: direct_answer" in summary
+    assert "Outcome: success (reward=1.0, steps=1)" in summary
+
+
+def test_summarize_multi_step_with_tool_errors():
+    summarizer = TrajectorySummarizer()
+    trajectory = {
+        "task": {"input": "Run analysis"},
+        "decisions": [
+            {"decision_type": "clarify"},
+            {"decision_type": "call_tool"},
+            {"decision_type": "direct_answer"},
+        ],
+        "outcome": {"status": "partial_success", "steps": 3, "tool_errors": 1, "user_corrections": 1},
+        "reward": {"total": 0.5},
+    }
+    summary = summarizer.summarize(trajectory)
+    assert "Actions: clarify -> call_tool -> direct_answer" in summary
+    assert "Outcome: partial_success (reward=0.5, steps=3)" in summary
+    assert "Tool errors: 1" in summary
+    assert "User corrections: 1" in summary
+
+
+def test_summarize_truncates_long_input():
+    summarizer = TrajectorySummarizer()
+    long_input = "a" * 100
+    trajectory = {
+        "task": {"input": long_input},
+        "decisions": [{"decision_type": "direct_answer"}],
+        "outcome": {"status": "success", "steps": 1, "tool_errors": 0, "user_corrections": 0},
+        "reward": {"total": 0.9},
+    }
+    summary = summarizer.summarize(trajectory)
+    assert "Task: '" in summary
+    assert "..." in summary
+    assert len(summary) < 300
+
+
+def test_summarize_handles_missing_fields_gracefully():
+    summarizer = TrajectorySummarizer()
+    trajectory = {}
+    summary = summarizer.summarize(trajectory)
+    assert "Task: ''" in summary
+    assert "Actions: none" in summary
+    assert "Outcome: unknown (reward=0.0, steps=0)" in summary