Something went wrong. Try again.
personal memory agent
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420# SPDX-License-Identifier: AGPL-3.0-only# Copyright (c) 2026 sol pbc
"""Tests for observe.hear.load_transcript() function."""
import jsonimport tempfilefrom pathlib import Path
from solstone.observe.hear import load_transcript
def test_load_transcript_native_with_metadata(): """Test loading native transcript with topics/setting metadata.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "120000_audio.jsonl"
lines = [ json.dumps({"topics": "meeting, standup", "setting": "work"}), json.dumps({"start": "12:00:01", "source": "mic", "text": "Hello"}), json.dumps({"start": "12:00:05", "source": "sys", "text": "Hi there"}), ] file_path.write_text("\n".join(lines) + "\n", encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is not None assert metadata["topics"] == "meeting, standup" assert metadata["setting"] == "work" assert len(entries) == 2 assert entries[0]["start"] == "12:00:01" assert entries[0]["text"] == "Hello" assert entries[1]["start"] == "12:00:05" assert entries[1]["text"] == "Hi there"
def test_load_transcript_native_empty_metadata(): """Test loading native transcript with empty metadata.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "120000_audio.jsonl"
lines = [ json.dumps({}), json.dumps({"start": "12:00:01", "source": "mic", "text": "Test"}), ] file_path.write_text("\n".join(lines) + "\n", encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is not None assert metadata == {} assert len(entries) == 1 assert entries[0]["text"] == "Test"
def test_load_transcript_imported(): """Test loading imported transcript with imported metadata.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "120000_imported_audio.jsonl"
lines = [ json.dumps({"imported": {"id": "20240101_120000", "facet": "personal"}}), json.dumps({"start": "12:00:01", "text": "Imported entry"}), ] file_path.write_text("\n".join(lines) + "\n", encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is not None assert "imported" in metadata assert metadata["imported"]["id"] == "20240101_120000" assert metadata["imported"]["facet"] == "personal" assert len(entries) == 1 assert entries[0]["text"] == "Imported entry"
def test_load_transcript_empty_file(): """Test loading an empty file returns error.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "empty.jsonl" file_path.write_text("", encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is None assert "error" in metadata assert "empty" in metadata["error"].lower()
def test_load_transcript_file_not_found(): """Test loading non-existent file returns error.""" metadata, entries, formatted_text = load_transcript("/nonexistent/file.jsonl")
assert entries is None assert "error" in metadata assert "not found" in metadata["error"].lower()
def test_load_transcript_invalid_metadata_json(): """Test loading file with invalid JSON in metadata line.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "bad_metadata.jsonl" file_path.write_text("not valid json\n", encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is None assert "error" in metadata assert "metadata" in metadata["error"].lower()
def test_load_transcript_invalid_entry_json(): """Test loading file with invalid JSON in entry line.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "bad_entry.jsonl"
lines = [ json.dumps({}), "not valid json", ] file_path.write_text("\n".join(lines) + "\n", encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is None assert "error" in metadata assert "line 2" in metadata["error"].lower()
def test_load_transcript_metadata_not_dict(): """Test loading file where metadata is not a dict.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "bad_metadata_type.jsonl" file_path.write_text('["not", "a", "dict"]\n', encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is None assert "error" in metadata assert "object" in metadata["error"].lower()
def test_load_transcript_entry_not_dict(): """Test loading file where entry is not a dict.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "bad_entry_type.jsonl"
lines = [ json.dumps({}), '"string entry"', ] file_path.write_text("\n".join(lines) + "\n", encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is None assert "error" in metadata assert "line 2" in metadata["error"].lower()
def test_load_transcript_blank_lines_ignored(): """Test that blank lines between entries are ignored.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "with_blanks.jsonl"
lines = [ json.dumps({}), "", json.dumps({"start": "12:00:01", "text": "First"}), "", "", json.dumps({"start": "12:00:02", "text": "Second"}), "", ] file_path.write_text("\n".join(lines), encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is not None assert len(entries) == 2 assert entries[0]["text"] == "First" assert entries[1]["text"] == "Second"
def test_load_transcript_only_metadata_no_entries(): """Test loading file with only metadata line and no entries.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "only_metadata.jsonl" file_path.write_text(json.dumps({"topics": "test"}) + "\n", encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is not None assert metadata["topics"] == "test" assert entries == []
def test_load_transcript_with_path_object(): """Test that function accepts Path objects.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "test.jsonl"
lines = [ json.dumps({}), json.dumps({"start": "12:00:01", "text": "Test"}), ] file_path.write_text("\n".join(lines) + "\n", encoding="utf-8")
# Pass as Path object metadata, entries, formatted_text = load_transcript(file_path)
assert entries is not None assert len(entries) == 1
def test_load_transcript_with_string_path(): """Test that function accepts string paths.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "test.jsonl"
lines = [ json.dumps({}), json.dumps({"start": "12:00:01", "text": "Test"}), ] file_path.write_text("\n".join(lines) + "\n", encoding="utf-8")
# Pass as string metadata, entries, formatted_text = load_transcript(str(file_path))
assert entries is not None assert len(entries) == 1
def test_load_transcript_all_fields(): """Test that all entry fields are preserved.""" with tempfile.TemporaryDirectory() as tmpdir: file_path = Path(tmpdir) / "complete.jsonl"
lines = [ json.dumps({}), json.dumps( { "start": "12:00:01", "source": "mic", "speaker": 1, "text": "Complete entry", "description": "confident", } ), ] file_path.write_text("\n".join(lines) + "\n", encoding="utf-8")
metadata, entries, formatted_text = load_transcript(file_path)
assert entries is not None assert len(entries) == 1 entry = entries[0] assert entry["start"] == "12:00:01" assert entry["source"] == "mic" assert entry["speaker"] == 1 assert entry["text"] == "Complete entry" assert entry["description"] == "confident"
def test_load_transcript_formatted_text_basic(): """Test formatted text output with metadata and entries.""" with tempfile.TemporaryDirectory() as tmpdir: # Create test transcript file in YYYYMMDD/HHMMSS_LEN segment day_dir = Path(tmpdir) / "20250615" day_dir.mkdir() segment_dir = day_dir / "100500_300" segment_dir.mkdir() file_path = segment_dir / "audio.jsonl"
# Write JSONL with metadata and entries metadata = {"topics": ["meeting", "planning"], "setting": "work"} entries = [ {"start": "00:01:23", "source": "mic", "speaker": 1, "text": "Hello world"}, {"start": "00:01:25", "source": "sys", "speaker": 2, "text": "Hi there"}, { "start": "00:01:30", "source": "mic", "speaker": 1, "text": "How are you?", }, ]
lines = [json.dumps(metadata)] lines.extend(json.dumps(entry) for entry in entries) file_path.write_text("\n".join(lines) + "\n")
# Load transcript metadata_out, entries_out, formatted_text = load_transcript(file_path)
# Verify formatted text output assert "Start: 2025-06-15 10:05am" in formatted_text assert "Topics: meeting, planning" in formatted_text assert "Setting: work" in formatted_text assert "[00:01:23] (mic) Speaker 1: Hello world" in formatted_text assert "[00:01:25] (sys) Speaker 2: Hi there" in formatted_text assert "[00:01:30] (mic) Speaker 1: How are you?" in formatted_text
def test_load_transcript_formatted_text_minimal(): """Test formatted text with minimal metadata.""" with tempfile.TemporaryDirectory() as tmpdir: day_dir = Path(tmpdir) / "20250615" day_dir.mkdir() segment_dir = day_dir / "100500_300" segment_dir.mkdir() file_path = segment_dir / "audio.jsonl"
# Minimal metadata (empty dict) metadata = {} entries = [ {"start": "00:00:01", "text": "Simple text"}, {"start": "00:00:05", "source": "mic", "text": "With source"}, ]
lines = [json.dumps(metadata)] lines.extend(json.dumps(entry) for entry in entries) file_path.write_text("\n".join(lines) + "\n")
metadata_out, entries_out, formatted_text = load_transcript(file_path)
assert "Start: 2025-06-15 10:05am" in formatted_text assert "[00:00:01] Simple text" in formatted_text assert "[00:00:05] (mic) With source" in formatted_text
def test_load_transcript_formatted_text_imported(): """Test formatted text of imported transcript with facet metadata.""" with tempfile.TemporaryDirectory() as tmpdir: day_dir = Path(tmpdir) / "20250615" day_dir.mkdir() segment_dir = day_dir / "100500_300" segment_dir.mkdir() file_path = segment_dir / "imported_audio.jsonl"
metadata = { "imported": {"id": "abc123", "facet": "uavionix"}, "topics": ["discussion"], } entries = [ { "start": "00:00:10", "source": "mic", "speaker": 1, "text": "Imported content", } ]
lines = [json.dumps(metadata)] lines.extend(json.dumps(entry) for entry in entries) file_path.write_text("\n".join(lines) + "\n")
metadata_out, entries_out, formatted_text = load_transcript(file_path)
assert "Start: 2025-06-15 10:05am" in formatted_text assert "Topics: discussion" in formatted_text assert "Facet: uavionix" in formatted_text assert "Import ID: abc123" in formatted_text assert "[00:00:10] (mic) Speaker 1: Imported content" in formatted_text
def test_load_transcript_formatted_text_no_speaker(): """Test formatted text for entries without speaker information.""" with tempfile.TemporaryDirectory() as tmpdir: day_dir = Path(tmpdir) / "20250615" day_dir.mkdir() segment_dir = day_dir / "100500_300" segment_dir.mkdir() file_path = segment_dir / "audio.jsonl"
metadata = {"setting": "personal"} entries = [ {"start": "00:00:01", "source": "mic", "text": "No speaker here"}, {"start": "00:00:05", "text": "No source or speaker"}, ]
lines = [json.dumps(metadata)] lines.extend(json.dumps(entry) for entry in entries) file_path.write_text("\n".join(lines) + "\n")
metadata_out, entries_out, formatted_text = load_transcript(file_path)
assert "[00:00:01] (mic) No speaker here" in formatted_text assert "[00:00:05] No source or speaker" in formatted_text
def test_load_transcript_formatted_text_error(): """Test formatted text for error cases.""" metadata, entries, formatted_text = load_transcript("/nonexistent/file.jsonl")
assert entries is None assert "Error loading transcript:" in formatted_text assert "not found" in formatted_text.lower()
def test_load_transcript_formatted_text_no_day_in_path(): """Test formatted text when day can't be parsed from path.""" with tempfile.TemporaryDirectory() as tmpdir: # File not in a YYYYMMDD directory file_path = Path(tmpdir) / "100500_audio.jsonl"
metadata = {"setting": "test"} entries = [{"start": "00:00:01", "text": "Test"}]
lines = [json.dumps(metadata)] lines.extend(json.dumps(entry) for entry in entries) file_path.write_text("\n".join(lines) + "\n")
metadata_out, entries_out, formatted_text = load_transcript(file_path)
# Should still work, just without "Start:" header assert "Setting: test" in formatted_text assert "[00:00:01] Test" in formatted_text # Start header should not be present since we couldn't parse the day assert "Start:" not in formatted_text