43 lines
997 B
Python
43 lines
997 B
Python
from datetime import datetime
|
|
from enum import StrEnum
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict
|
|
|
|
|
|
class LLMType(StrEnum):
|
|
OPENAI_GPT4O = "OPENAI_GPT4O"
|
|
|
|
|
|
class TaskGenerationBase(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
organization_id: str | None = None
|
|
user_prompt: str | None = None
|
|
url: str | None = None
|
|
navigation_goal: str | None = None
|
|
navigation_payload: dict[str, Any] | None = None
|
|
data_extraction_goal: str | None = None
|
|
extracted_information_schema: dict[str, Any] | None = None
|
|
llm: LLMType | None = None
|
|
llm_prompt: str | None = None
|
|
llm_response: str | None = None
|
|
|
|
|
|
class TaskGenerationCreate(TaskGenerationBase):
|
|
organization_id: str
|
|
user_prompt: str
|
|
|
|
|
|
class TaskGeneration(TaskGenerationBase):
|
|
task_generation_id: str
|
|
organization_id: str
|
|
user_prompt: str
|
|
|
|
created_at: datetime
|
|
modified_at: datetime
|
|
|
|
|
|
class GenerateTaskRequest(BaseModel):
|
|
prompt: str
|