open-notebook/open_notebook/domain/transformation.py
2024-11-18 22:01:11 -03:00

21 lines
540 B
Python

from typing import ClassVar, Optional
from pydantic import Field
from open_notebook.domain.base import ObjectModel, RecordModel
class Transformation(ObjectModel):
table_name: ClassVar[str] = "transformation"
name: str
title: str
description: str
prompt: str
apply_default: bool
class DefaultPrompts(RecordModel):
record_id: ClassVar[str] = "open_notebook:default_prompts"
transformation_instructions: Optional[str] = Field(
None, description="Instructions for executing a transformation"
)