From 037d86939f51e8b37a44177868da107ebd07c235 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Fri, 7 Jun 2024 16:27:32 -0700 Subject: [PATCH] Do not enforce LLMType in the TaskGeneration schema (#451) --- skyvern/forge/sdk/schemas/task_generations.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/skyvern/forge/sdk/schemas/task_generations.py b/skyvern/forge/sdk/schemas/task_generations.py index 7e1ff4a8e..141023f13 100644 --- a/skyvern/forge/sdk/schemas/task_generations.py +++ b/skyvern/forge/sdk/schemas/task_generations.py @@ -1,14 +1,9 @@ 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) @@ -19,7 +14,7 @@ class TaskGenerationBase(BaseModel): 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: str | None = None llm_prompt: str | None = None llm_response: str | None = None