348610e73c
* fix: handle rate limit error coming from llm model * fix: fixes lost edges and nodes in get_graph_from_model * fix: fixes database pruning issue in pgvector (#261) * fix: cognee_demo notebook pipeline is not saving summaries --------- Co-authored-by: hajdul88 <52442977+hajdul88@users.noreply.github.com>
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from typing import List, Optional
|
|
from cognee.infrastructure.engine import DataPoint
|
|
|
|
class Repository(DataPoint):
|
|
__tablename__ = "Repository"
|
|
path: str
|
|
type: Optional[str] = "Repository"
|
|
|
|
class CodeFile(DataPoint):
|
|
__tablename__ = "codefile"
|
|
extracted_id: str # actually file path
|
|
type: Optional[str] = "CodeFile"
|
|
source_code: Optional[str] = None
|
|
part_of: Optional[Repository] = None
|
|
depends_on: Optional[List["CodeFile"]] = None
|
|
depends_directly_on: Optional[List["CodeFile"]] = None
|
|
contains: Optional[List["CodePart"]] = None
|
|
|
|
_metadata: dict = {
|
|
"index_fields": ["source_code"]
|
|
}
|
|
|
|
class CodePart(DataPoint):
|
|
__tablename__ = "codepart"
|
|
# part_of: Optional[CodeFile]
|
|
source_code: str
|
|
type: Optional[str] = "CodePart"
|
|
|
|
_metadata: dict = {
|
|
"index_fields": ["source_code"]
|
|
}
|
|
|
|
class CodeRelationship(DataPoint):
|
|
source_id: str
|
|
target_id: str
|
|
type: str # between files
|
|
relation: str # depends on or depends directly
|
|
|
|
CodeFile.model_rebuild()
|
|
CodePart.model_rebuild()
|