| """Serializable records produced by the CodeRAG indexers.""" |
|
|
| from dataclasses import asdict, dataclass, field |
|
|
|
|
| @dataclass |
| class FunctionRecord: |
| function_id: str |
| name: str |
| qualified_name: str |
| symbol_type: str |
| file_path: str |
| start_line: int |
| end_line: int |
| signature: str |
| docstring: str | None |
| raw_code: str |
| own_sketch: str |
| own_summary: str |
| direct_callees: list[str] = field(default_factory=list) |
| unresolved_calls: list[str] = field(default_factory=list) |
| callee_context: str = "" |
| propagated_text: str = "" |
| code_domain_scores: dict[str, float] = field(default_factory=dict) |
| top_code_domains: list[str] = field(default_factory=list) |
| structural_signals: dict = field(default_factory=dict) |
| data_flow_signals: dict = field(default_factory=dict) |
| semantic_signals: dict = field(default_factory=dict) |
|
|
| def to_dict(self) -> dict: |
| return asdict(self) |
|
|
|
|
| @dataclass |
| class FileIndex: |
| schema_version: str |
| file_path: str |
| function_count: int |
| call_edges: list[dict[str, str]] |
| indexing_order: list[str] |
| functions: list[FunctionRecord] |
|
|
| def to_dict(self) -> dict: |
| result = asdict(self) |
| result["functions"] = [record.to_dict() for record in self.functions] |
| return result |
|
|
|
|
| @dataclass |
| class RepositoryIndex: |
| schema_version: str |
| root_path: str |
| file_count: int |
| function_count: int |
| call_edges: list[dict[str, str]] |
| indexing_order: list[str] |
| functions: list[FunctionRecord] |
| files: list[str] |
|
|
| def to_dict(self) -> dict: |
| result = asdict(self) |
| result["functions"] = [record.to_dict() for record in self.functions] |
| return result |
|
|
