refactor transformation, add graph and admin

This commit is contained in:
LUIS NOVO 2024-11-18 22:01:11 -03:00
parent e3fa445fcc
commit 4a5d47d934
26 changed files with 326 additions and 384 deletions

View file

@ -1,19 +1,21 @@
from typing import ClassVar, List, Optional
from typing import ClassVar, Optional
import yaml
from pydantic import Field
from open_notebook.domain.base import RecordModel
from open_notebook.domain.base import ObjectModel, RecordModel
class Transformation:
@classmethod
def get_all(cls):
with open("transformations.yaml", "r") as file:
transformations = yaml.safe_load(file)
return transformations
class Transformation(ObjectModel):
table_name: ClassVar[str] = "transformation"
name: str
title: str
description: str
prompt: str
apply_default: bool
class DefaultTransformations(RecordModel):
record_id: ClassVar[str] = "open_notebook:default_transformations"
source_insights: Optional[List[str]] = Field(default_factory=list)
class DefaultPrompts(RecordModel):
record_id: ClassVar[str] = "open_notebook:default_prompts"
transformation_instructions: Optional[str] = Field(
None, description="Instructions for executing a transformation"
)