Spaces:
Runtime error
Runtime error
File size: 722 Bytes
35b22df |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
"""Struct store schema."""
from dataclasses import dataclass, field
from typing import Any, Dict
from dataclasses_json import DataClassJsonMixin
from gpt_index.data_structs.data_structs import IndexStruct
@dataclass
class StructDatapoint(DataClassJsonMixin):
"""Struct outputs."""
# map from field name to StructValue
fields: Dict[str, Any]
@dataclass
class BaseStructTable(IndexStruct):
"""Struct outputs."""
@dataclass
class SQLStructTable(BaseStructTable):
"""SQL struct outputs."""
context_dict: Dict[str, str] = field(default_factory=dict)
@classmethod
def get_type(cls) -> str:
"""Get type."""
# TODO: consolidate with IndexStructType
return "sql"
|