dan/memory
1
0
Fork 0
mirror of https://github.com/mruwnik/memory.git synced 2026-01-02 17:22:58 +01:00
Code Issues 1 Packages Projects Releases Wiki Activity

proper modality for github items

Browse Source
This commit is contained in:
mruwnik 2025-12-23 22:13:11 +00:00
parent 526bfa5f6b
commit efbc469ee3
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/memory/common/db/models/source_items.py
View File

@ -894,6 +894,13 @@ class GithubItem(SourceItem):
merged_at=cast(datetime | None, self.merged_at), merged_at=cast(datetime | None, self.merged_at),
) )
def _chunk_contents(self) -> Sequence[extract.DataChunk]:
"""Override to use 'github' modality instead of default 'text'."""
content = cast(str | None, self.content)
if content:
return extract.extract_text(content, modality="github")
return []
class NotePayload(SourceItemPayload): class NotePayload(SourceItemPayload):
note_type: Annotated[str | None, "Category of the note"] note_type: Annotated[str | None, "Category of the note"]

2
src/memory/workers/tasks/github.py
View File

@ -58,7 +58,7 @@ def _create_github_item(
repo_tags = cast(list[str], repo.tags) or [] repo_tags = cast(list[str], repo.tags) or []
return GithubItem( return GithubItem(
modality="text", modality="github",
sha256=create_content_hash(content), sha256=create_content_hash(content),
content=content, content=content,
kind=issue_data["kind"], kind=issue_data["kind"],