mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-23 00:43:18 +00:00
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from pydantic import BaseModel, Field, ConfigDict
|
|
|
|
|
|
class FileEdge(BaseModel):
|
|
""" Format:
|
|
[[X]] bare wikilink → predicate=None
|
|
extends:: [[X]] line-level Dataview → predicate="extends"
|
|
[extends:: [[X]]] inline-bracketed → predicate="extends"
|
|
extends:: [[A]], [[B]] multi-target → 2 edges
|
|
"""
|
|
link: str = Field(default=...)
|
|
predicate: str | None = Field(default=None)
|
|
|
|
@property
|
|
def link_path(self) -> str:
|
|
return self.link.split("#", 1)[0]
|
|
|
|
@property
|
|
def link_anchor(self) -> str:
|
|
link_split = self.link.split("#", 1)
|
|
return link_split[1] if len(link_split) > 1 else ""
|
|
|
|
|
|
class FileFrontMatter(BaseModel):
|
|
model_config = ConfigDict(extra="allow")
|
|
|
|
title: str = Field(default="")
|
|
description: str = Field(default="")
|
|
tags: list[str] | None = Field(default=None)
|
|
|
|
@property
|
|
def metadata(self) -> dict:
|
|
return dict(self.__pydantic_extra__ or {})
|
|
|
|
|
|
class FileNode(BaseModel):
|
|
path: str = Field(default=...)
|
|
st_mtime: float = Field(default=...)
|
|
edges: list[FileEdge] = Field(default_factory=list)
|
|
chunk_ids: list[str] = Field(default_factory=list)
|
|
front_matter: FileFrontMatter = Field(default_factory=FileFrontMatter)
|