use cached prompt generation (#768)
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
from pydantic import BaseModel, ConfigDict, Field
|
||||
|
||||
|
||||
class TaskGenerationBase(BaseModel):
|
||||
@@ -9,6 +9,7 @@ class TaskGenerationBase(BaseModel):
|
||||
|
||||
organization_id: str | None = None
|
||||
user_prompt: str | None = None
|
||||
user_prompt_hash: str | None = None
|
||||
url: str | None = None
|
||||
navigation_goal: str | None = None
|
||||
navigation_payload: dict[str, Any] | None = None
|
||||
@@ -20,19 +21,16 @@ class TaskGenerationBase(BaseModel):
|
||||
suggested_title: 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
|
||||
user_prompt_hash: str
|
||||
|
||||
created_at: datetime
|
||||
modified_at: datetime
|
||||
|
||||
|
||||
class GenerateTaskRequest(BaseModel):
|
||||
prompt: str
|
||||
# prompt needs to be at least 1 character long
|
||||
prompt: str = Field(..., min_length=1)
|
||||
|
||||
Reference in New Issue
Block a user