Function _create_run_metadata

tests/unit/zen_stores/test_run_statistics.py:122–147

(
    sql_store: SqlZenStore,
    *,
    project_id: UUID,
    run_id: UUID,
    key: str,
    value: object,
    type_: MetadataTypeEnum,
)

def _create_run_metadata(
    sql_store: SqlZenStore,
    *,
    project_id: UUID,
    run_id: UUID,
    key: str,
    value: object,
    type_: MetadataTypeEnum,
) -> None:
    rm = RunMetadataSchema(
        project_id=project_id,
        key=key,
        value=json.dumps(value),
        type=type_.value,
    )
    with Session(sql_store.engine) as session:
        session.add(rm)
        session.flush()
        session.add(
            RunMetadataResourceSchema(
                resource_id=run_id,
                resource_type=MetadataResourceTypes.PIPELINE_RUN.value,
                run_metadata_id=rm.id,
            )
        )
        session.commit()


def _create_pipeline(sql_store: SqlZenStore, project_id: UUID) -> UUID: