vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Do not enforce LLMType in the TaskGeneration schema (#451)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-06-07 16:27:32 -07:00
committed by GitHub
parent d18fc5b59c
commit 037d86939f
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
skyvern/forge/sdk/schemas/task_generations.py
View File

@@ -1,14 +1,9 @@
from datetime import datetime from datetime import datetime
from enum import StrEnum
from typing import Any from typing import Any
from pydantic import BaseModel, ConfigDict from pydantic import BaseModel, ConfigDict
class LLMType(StrEnum):
OPENAI_GPT4O = "OPENAI_GPT4O"
class TaskGenerationBase(BaseModel): class TaskGenerationBase(BaseModel):
model_config = ConfigDict(from_attributes=True) model_config = ConfigDict(from_attributes=True)
@@ -19,7 +14,7 @@ class TaskGenerationBase(BaseModel):
navigation_payload: dict[str, Any] | None = None navigation_payload: dict[str, Any] | None = None
data_extraction_goal: str | None = None data_extraction_goal: str | None = None
extracted_information_schema: dict[str, Any] | None = None extracted_information_schema: dict[str, Any] | None = None
llm: LLMType | None = None llm: str | None = None
llm_prompt: str | None = None llm_prompt: str | None = None
llm_response: str | None = None llm_response: str | None = None