dataflow.models

Models for the Haive Registry System.

This module defines the core models used by the registry system to represent different types of entities, configurations, dependencies, etc.

Classes

ConfigType	Types of configurations.
Configuration	Configuration for a registry item.
Dependency	Dependency between registry items.
DependencyType	Types of dependencies between entities.
EntityType	Types of entities that can be registered.
EnvironmentVar	Environment variable requirement for a registry item.
GraphDefinition	Graph definition for a registry item.
ImportLogItem	Log entry for import operations.
ImportStatus	Import operation status.
RegistryItem	Base model for registry items.

Module Contents

class dataflow.models.ConfigType

Bases: str, enum.Enum

Types of configurations.

ENGINE = 'engine'

GRAPH = 'graph'

INPUT_SCHEMA = 'input_schema'

NODE = 'node'

OUTPUT_SCHEMA = 'output_schema'

PROMPT = 'prompt'

STATE_SCHEMA = 'state_schema'

class dataflow.models.Configuration(/, **data: Any)

Bases: pydantic.BaseModel

Configuration for a registry item.

config_data: dict[str, Any]

config_type: ConfigType

created_at: datetime.datetime | None = None

id: str = None

registry_id: str

updated_at: datetime.datetime | None = None

class dataflow.models.Dependency(/, **data: Any)

Bases: pydantic.BaseModel

Dependency between registry items.

created_at: datetime.datetime | None = None

dependency_type: DependencyType

dependent_id: str

id: str = None

registry_id: str

class dataflow.models.DependencyType

Bases: str, enum.Enum

Types of dependencies between entities.

EXTENDS = 'extends'

REQUIRES = 'requires'

USES = 'uses'

class dataflow.models.EntityType

Bases: str, enum.Enum

Types of entities that can be registered.

AGENT = 'agent'

ENGINE = 'engine'

GAME = 'game'

LLM_MODEL = 'llm_model'

LLM_PROVIDER = 'llm_provider'

TOOL = 'tool'

TOOLKIT = 'toolkit'

class dataflow.models.EnvironmentVar(/, **data: Any)

Bases: pydantic.BaseModel

Environment variable requirement for a registry item.

created_at: datetime.datetime | None = None

default_value: str | None = None

env_name: str

id: str = None

is_required: bool = False

registry_id: str

class dataflow.models.GraphDefinition(/, **data: Any)

Bases: pydantic.BaseModel

Graph definition for a registry item.

created_at: datetime.datetime | None = None

edges: list[dict[str, Any]] = None

id: str = None

nodes: list[dict[str, Any]] = None

registry_id: str

updated_at: datetime.datetime | None = None

class dataflow.models.ImportLogItem(/, **data: Any)

Bases: pydantic.BaseModel

Log entry for import operations.

created_at: datetime.datetime | None = None

entity_name: str

entity_type: str

id: str = None

import_session: str

message: str | None = None

status: ImportStatus

traceback: str | None = None

class dataflow.models.ImportStatus

Bases: str, enum.Enum

Import operation status.

FAILURE = 'failure'

SUCCESS = 'success'

class dataflow.models.RegistryItem(/, **data: Any)

Bases: pydantic.BaseModel

Base model for registry items.

class_name: str | None = None

created_at: datetime.datetime | None = None

description: str | None = None

id: str = None

metadata: dict[str, Any] = None

module_path: str | None = None

name: str

type: EntityType

updated_at: datetime.datetime | None = None