Skip to content

graflo.architecture.graph_types.index_config

Index and weight configuration models.

Classes

ABCFields

Bases: ConfigBaseModel

Base model for entities that have fields.

Attributes:

Name Type Description
name str | None

Optional name of the entity
fields list[str]

List of field names
Source code in graflo/architecture/graph_types/index_config.py 
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class ABCFields(ConfigBaseModel):
    """Base model for entities that have fields.

    Attributes:
        name: Optional name of the entity
        fields: List of field names
    """

    name: str | None = Field(
        default=None,
        description="Optional name of the entity (e.g. vertex name for composite field prefix).",
    )
    fields: list[str] = Field(
        default_factory=list,
        description="List of field names for this entity.",
    )
    keep_vertex_name: bool = Field(
        default=True,
        description="If True, composite field names use entity@field format; otherwise use field only.",
    )

    def cfield(self, x: str) -> str:
        """Creates a composite field name by combining the entity name with a field name.

        Args:
            x: Field name to combine with entity name

        Returns:
            Composite field name in format "entity@field"
        """
        return f"{self.name}@{x}" if self.keep_vertex_name else x

Attributes

fields = Field(default_factory=list, description='List of field names for this entity.') class-attribute instance-attribute
keep_vertex_name = Field(default=True, description='If True, composite field names use entity@field format; otherwise use field only.') class-attribute instance-attribute
name = Field(default=None, description='Optional name of the entity (e.g. vertex name for composite field prefix).') class-attribute instance-attribute

Methods:

cfield(x)

Creates a composite field name by combining the entity name with a field name.

Parameters:

Name Type Description Default
x str

Field name to combine with entity name

 required

Returns:

Type Description
str

Composite field name in format "entity@field"
Source code in graflo/architecture/graph_types/index_config.py 
33
34
35
36
37
38
39
40
41
42
def cfield(self, x: str) -> str:
    """Creates a composite field name by combining the entity name with a field name.

    Args:
        x: Field name to combine with entity name

    Returns:
        Composite field name in format "entity@field"
    """
    return f"{self.name}@{x}" if self.keep_vertex_name else x

Index

Bases: ConfigBaseModel

Configuration for database indexes.

Attributes:

Name Type Description
name str | None

Optional name of the index
fields list[str]

List of fields to index
unique bool

Whether the index enforces uniqueness
type IndexType

Type of index to create
deduplicate bool

Whether to deduplicate index entries
sparse bool

Whether to create a sparse index
exclude_edge_endpoints bool

Whether to exclude edge endpoints from index
Source code in graflo/architecture/graph_types/index_config.py 
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
class Index(ConfigBaseModel):
    """Configuration for database indexes.

    Attributes:
        name: Optional name of the index
        fields: List of fields to index
        unique: Whether the index enforces uniqueness
        type: Type of index to create
        deduplicate: Whether to deduplicate index entries
        sparse: Whether to create a sparse index
        exclude_edge_endpoints: Whether to exclude edge endpoints from index
    """

    name: str | None = Field(
        default=None,
        description="Optional index name. For edges, can reference a vertex name for composite fields.",
    )
    fields: list[str] = Field(
        default_factory=list,
        description="List of field names included in this index.",
    )
    unique: bool = Field(
        default=True,
        description="If True, index enforces uniqueness on the field combination.",
    )
    type: IndexType = Field(
        default=IndexType.PERSISTENT,
        description="Index type (PERSISTENT, HASH, SKIPLIST, FULLTEXT).",
    )
    deduplicate: bool = Field(
        default=True,
        description="Whether to deduplicate index entries (e.g. ArangoDB).",
    )
    sparse: bool = Field(
        default=False,
        description="If True, create a sparse index (exclude null/missing values).",
    )
    exclude_edge_endpoints: bool = Field(
        default=False,
        description="If True, do not add _from/_to to edge index (e.g. ArangoDB).",
    )

    def __iter__(self):
        """Iterate over the indexed fields."""
        return iter(self.fields)

    def db_form(self, db_type: DBType) -> dict:
        """Convert index configuration to database-specific format.

        Args:
            db_type: Type of database (ARANGO or NEO4J)

        Returns:
            Dictionary of index configuration in database-specific format

        Raises:
            ValueError: If db_type is not supported
        """
        r = dict(self.to_dict())
        if db_type == DBType.ARANGO:
            r.pop("name", None)
            r.pop("exclude_edge_endpoints", None)
        return r

Attributes

deduplicate = Field(default=True, description='Whether to deduplicate index entries (e.g. ArangoDB).') class-attribute instance-attribute
exclude_edge_endpoints = Field(default=False, description='If True, do not add _from/_to to edge index (e.g. ArangoDB).') class-attribute instance-attribute
fields = Field(default_factory=list, description='List of field names included in this index.') class-attribute instance-attribute
name = Field(default=None, description='Optional index name. For edges, can reference a vertex name for composite fields.') class-attribute instance-attribute
sparse = Field(default=False, description='If True, create a sparse index (exclude null/missing values).') class-attribute instance-attribute
type = Field(default=IndexType.PERSISTENT, description='Index type (PERSISTENT, HASH, SKIPLIST, FULLTEXT).') class-attribute instance-attribute
unique = Field(default=True, description='If True, index enforces uniqueness on the field combination.') class-attribute instance-attribute

Methods:

__iter__()

Iterate over the indexed fields.

Source code in graflo/architecture/graph_types/index_config.py 
105
106
107
def __iter__(self):
    """Iterate over the indexed fields."""
    return iter(self.fields)
db_form(db_type)

Convert index configuration to database-specific format.

Parameters:

Name Type Description Default
db_type DBType

Type of database (ARANGO or NEO4J)

 required

Returns:

Type Description
dict

Dictionary of index configuration in database-specific format

Raises:

Type Description
ValueError

If db_type is not supported
Source code in graflo/architecture/graph_types/index_config.py 
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
def db_form(self, db_type: DBType) -> dict:
    """Convert index configuration to database-specific format.

    Args:
        db_type: Type of database (ARANGO or NEO4J)

    Returns:
        Dictionary of index configuration in database-specific format

    Raises:
        ValueError: If db_type is not supported
    """
    r = dict(self.to_dict())
    if db_type == DBType.ARANGO:
        r.pop("name", None)
        r.pop("exclude_edge_endpoints", None)
    return r

Weight

Bases: ABCFields

Defines weight configuration for edges.

Attributes:

Name Type Description
map dict

Dictionary mapping field values to weights
filter dict

Dictionary of filter conditions for weights
Source code in graflo/architecture/graph_types/index_config.py 
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
class Weight(ABCFields):
    """Defines weight configuration for edges.

    Attributes:
        map: Dictionary mapping field values to weights
        filter: Dictionary of filter conditions for weights
    """

    map: dict = Field(
        default_factory=dict,
        description="Mapping of field values to weight values for vertex-based edge attributes.",
    )
    filter: dict = Field(
        default_factory=dict,
        description="Filter conditions applied when resolving vertex-based weights.",
    )

Attributes

filter = Field(default_factory=dict, description='Filter conditions applied when resolving vertex-based weights.') class-attribute instance-attribute
map = Field(default_factory=dict, description='Mapping of field values to weight values for vertex-based edge attributes.') class-attribute instance-attribute