class SearchResult:
    search_result: Any       # The actual result content
    dataset_id: UUID | None  # Associated dataset UUID
    dataset_name: str | None # Associated dataset name

class PipelineRunInfo:
    status: str                # Pipeline status
    pipeline_run_id: UUID      # Unique run identifier
    dataset_id: UUID           # Associated dataset
    dataset_name: str          # Dataset name
    payload: Any | None        # Execution payload
    data_ingestion_info: list | None  # Ingestion details

from cognee.modules.pipelines import Task

# Wrap any async generator, generator, coroutine, or function
task = Task(my_function, task_config={"batch_size": 10})

class DataPoint:
    id: UUID                           # Unique identifier
    created_at: int                    # Creation timestamp (ms)
    updated_at: int                    # Update timestamp (ms)
    version: int                       # Version number
    type: str                          # Data type name
    ontology_valid: bool               # Ontology validation status
    topological_rank: int | None       # Topological rank
    belongs_to_set: list[str] | None   # Node set membership
    source_pipeline: str | None        # Source pipeline name
    source_node_set: str | None        # Source node set

class KnowledgeGraph(BaseModel):
    nodes: list[Node]    # Graph nodes
    edges: list[Edge]    # Graph edges
    summary: str         # Graph summary
    description: str     # Graph description

class Node:
    id: str
    name: str
    type: str
    description: str

class Edge:
    source_node_id: str
    target_node_id: str
    relationship_name: str