Schemas

class lumigator_schemas.completions.CompletionRequest(*, text: str)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.completions.CompletionResponse(*, text: str)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.datasets.DatasetDownloadResponse(*, id: UUID, download_urls: list[str])

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.datasets.DatasetFormat(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

class lumigator_schemas.datasets.DatasetResponse(*, id: UUID, filename: str, format: DatasetFormat, size: int, ground_truth: bool, run_id: UUID | None, generated: bool, generated_by: str | None, created_at: datetime)

model_config: ClassVar[ConfigDict] = {'from_attributes': True}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.Job(*, type: str | None = None, submission_id: str | None = None, driver_info: str | None = None, status: ~lumigator_schemas.jobs.JobStatus, entrypoint: str | None = None, message: str | None = None, error_type: str | None = None, start_time: ~datetime.datetime | None = None, end_time: ~datetime.datetime | None = None, metadata: dict = <factory>, runtime_env: dict = <factory>, driver_agent_http_address: str | None = None, driver_node_id: str | None = None, driver_exit_code: int | None = None, id: ~uuid.UUID, name: str, description: str, created_at: ~datetime.datetime, experiment_id: ~uuid.UUID | None = None, updated_at: ~datetime.datetime | None = None)

Job represents the composition of JobResponse and JobSubmissionResponse.

JobSubmissionResponse was formerly returned from some /health/jobs related endpoints, while JobResponse was used by /jobs related endpoints.

The only conflicting field in the two schemas is ‘status’ which is consistent in what it intends to represent, but uses different types (JobStatus/str).

The Job type has both id and submission_id which will contain the same data.

NOTE: Job is intended to reduce breaking changes experienced by the UI and other consumers. Tt was not conceived as a type that will be around for long, as the API needs to be refactored to better support experiments.

model_config: ClassVar[ConfigDict] = {'from_attributes': True}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobAnnotateCreate(*, name: str, description: str = '', dataset: UUID, max_samples: int = -1, task: str | None = 'summarization')

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobConfig(*, job_id: UUID, job_type: JobType, command: str, args: dict[str, Any] | None = None)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobEvalCreate(*, name: str, description: str = '', model: str, dataset: UUID, max_samples: int = -1, model_url: str | None = None, system_prompt: str | None = None, config_template: str | None = None, skip_inference: bool = False)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobEvalLiteCreate(*, name: str, description: str = '', model: str, dataset: UUID, max_samples: int = -1, config_template: str | None = None)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobEvent(*, job_id: UUID, job_type: JobType, status: JobStatus, detail: str | None = None)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobInferenceCreate(*, name: str, description: str = '', model: str, dataset: UUID, max_samples: int = -1, task: str | None = 'summarization', accelerator: str | None = 'auto', revision: str | None = 'main', use_fast: bool = True, trust_remote_code: bool = False, torch_dtype: str = 'auto', model_url: str | None = None, system_prompt: str | None = None, output_field: str | None = 'predictions', max_tokens: int = 1024, frequency_penalty: float = 0.0, temperature: float = 1.0, top_p: float = 1.0, config_template: str | None = None)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobLogsResponse(*, logs: str | None = None)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobResponse(*, id: UUID, name: str, description: str, status: JobStatus, created_at: datetime, experiment_id: UUID | None = None, updated_at: datetime | None = None)

model_config: ClassVar[ConfigDict] = {'from_attributes': True}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobResultDownloadResponse(*, id: UUID, download_url: str)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobResultResponse(*, id: UUID, job_id: UUID)

model_config: ClassVar[ConfigDict] = {'from_attributes': True}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobStatus(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

class lumigator_schemas.jobs.JobSubmissionResponse(*, type: str | None = None, submission_id: str | None = None, driver_info: str | None = None, status: str | None = None, entrypoint: str | None = None, message: str | None = None, error_type: str | None = None, start_time: ~datetime.datetime | None = None, end_time: ~datetime.datetime | None = None, metadata: dict = <factory>, runtime_env: dict = <factory>, driver_agent_http_address: str | None = None, driver_node_id: str | None = None, driver_exit_code: int | None = None)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.jobs.JobType(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

class lumigator_schemas.extras.DeploymentType(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

class lumigator_schemas.extras.HealthResponse(*, status: str, deployment_type: DeploymentType)

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class lumigator_schemas.extras.ListingResponse(*, total: int, items: list[ItemType])

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].