kiarina-agi-data-builder

What is this?

kiarina-agi-data-builder provides builders for AI agent messages, events, histories, tool info, file info, and file segments.

Dependencies

Required Dependencies

Optional Dependencies

The all Extra installs every optional dependency listed above.

Installation

pip install "kiarina-agi-data-builder[all]"

Features

Capability-aware PDF fallback

PDF analysis bundles select content from chat model capabilities.

Set analysis_dpi on the PDF file specification to control the resolution of fallback page images.

from kiarina.agi.file_info_builder_impl.pdf import create_pdf_file_info_builder
from kiarina.agi.run_context import RunContext
from kiarina.utils.file.asyncio import read_file


async def build_pdf_info():
    file_blob = await read_file("/path/to/document.pdf")
    assert file_blob is not None
    builder = create_pdf_file_info_builder(analysis_enabled=True)
    return await builder.build(
        {
            "uri_or_file_path": file_blob.file_path,
            "analysis_dpi": 144,
        },
        file_blob,
        run_context=RunContext(),
    )

Capability-aware video fallback

Video analysis bundles select content from chat model capabilities.

Set analysis_fps on the video file specification to control the frame rate of the prepared video and fallback images.

from kiarina.agi.file_info_builder_impl.video import (
    create_video_file_info_builder,
)
from kiarina.agi.run_context import RunContext
from kiarina.utils.file.asyncio import read_file


async def build_video_info():
    file_blob = await read_file("/path/to/video.mp4")
    assert file_blob is not None
    builder = create_video_file_info_builder(
        analysis_enabled=True,
        audio_consumers=["stt", "ambient"],
    )
    return await builder.build(
        {
            "uri_or_file_path": file_blob.file_path,
            "analysis_fps": 1.0,
        },
        file_blob,
        run_context=RunContext(),
    )

API Reference

kiarina.agi.file_info_builder_impl.pdf

from kiarina.agi.file_info_builder_impl.pdf import (
    PDFFileInfoBuilder,
    PDFFileInfoBuilderSettings,
    create_pdf_file_info_builder,
    settings_manager,
)

create_pdf_file_info_builder

def create_pdf_file_info_builder(**kwargs: Any) -> PDFFileInfoBuilder: ...

Creates a PDF file info builder from managed settings and optional overrides.

PDFFileInfoBuilder

class PDFFileInfoBuilder(BaseFileInfoBuilder):
    def __init__(self, settings: PDFFileInfoBuilderSettings) -> None: ...

    async def build(
        self,
        file_info_spec: FileInfoSpec,
        file_blob: FileBlob,
        *,
        run_context: RunContext,
    ) -> BuildResult: ...

PDFFileInfoBuilderSettings

class PDFFileInfoBuilderSettings(BaseSettings):
    analysis_enabled: bool = False

settings_manager is the SettingsManager[PDFFileInfoBuilderSettings] instance used by the factory.

kiarina.agi.file_info_builder_impl.video

from kiarina.agi.file_info_builder_impl.video import (
    VideoFileInfoBuilder,
    VideoFileInfoBuilderSettings,
    create_video_file_info_builder,
    settings_manager,
)

create_video_file_info_builder

def create_video_file_info_builder(**kwargs: Any) -> VideoFileInfoBuilder: ...

Creates a video file info builder from managed settings and optional overrides.

VideoFileInfoBuilder

class VideoFileInfoBuilder(BaseFileInfoBuilder):
    def __init__(self, settings: VideoFileInfoBuilderSettings) -> None: ...

    async def build(
        self,
        file_info_spec: FileInfoSpec,
        file_blob: FileBlob,
        *,
        run_context: RunContext,
    ) -> BuildResult: ...

VideoFileInfoBuilderSettings

class VideoFileInfoBuilderSettings(BaseSettings):
    analysis_enabled: bool = False
    audio_source: AudioSourceSpecifier = "file?sample_rate=16000&start_timestamp=0.0"
    audio_consumers: list[AudioConsumerSpecifier] = [
        "stt?diarization_enabled=true",
        "ambient?window_seconds=10.0&top_k=3",
    ]
    audio_event_bundlers: list[AudioEventBundlerSpecifier] = ["stt", "ambient"]

settings_manager is the SettingsManager[VideoFileInfoBuilderSettings] instance used by the factory.