rename EntityParser to KnowledgeParser

This commit is contained in:
phernandez
2024-12-22 19:57:37 -06:00
parent 6f155c62f8
commit 19ef6933df
11 changed files with 58 additions and 61 deletions
+2 -2
View File
@@ -1,6 +1,6 @@
"""Base package for markdown parsing."""
from basic_memory.markdown.parser import EntityParser
from basic_memory.markdown.knowledge_parser import KnowledgeParser
from basic_memory.markdown.schemas import (
Entity,
EntityContent,
@@ -16,7 +16,7 @@ __all__ = [
"EntityContent",
"EntityFrontmatter",
"EntityMetadata",
"EntityParser",
"KnowledgeParser",
"Observation",
"Relation",
"ParseError",
+3 -5
View File
@@ -80,9 +80,7 @@ class MarkdownParser(ABC, Generic[T]):
# Create final document
return await self.create_document(
frontmatter=parsed_frontmatter,
content=parsed_content,
metadata=parsed_metadata
frontmatter=parsed_frontmatter, content=parsed_content, metadata=parsed_metadata
)
except Exception as e:
@@ -164,11 +162,11 @@ class MarkdownParser(ABC, Generic[T]):
pass
@abstractmethod
async def parse_metadata(self, metadata: Optional[str]) -> Any:
async def parse_metadata(self, metadata_section: Optional[str]) -> Any:
"""Parse metadata section."""
pass
@abstractmethod
async def create_document(self, frontmatter: Any, content: Any, metadata: Any) -> T:
"""Create document from parsed sections."""
pass
pass
@@ -17,7 +17,7 @@ from basic_memory.markdown.schemas import (
)
class EntityParser(MarkdownParser[Entity]):
class KnowledgeParser(MarkdownParser[Entity]):
"""Parser for entity markdown files.
Entity files must have: