Add suggested title to task generation (#719)

Co-authored-by: Muhammed Salih Altun <muhammedsalihaltun@gmail.com>

skyvern/forge/sdk/db/client.py

@@ -1391,6 +1391,7 @@ class AgentDB:
         navigation_payload: dict[str, Any] | None = None,
         data_extraction_goal: str | None = None,
         extracted_information_schema: dict[str, Any] | None = None,
+        suggested_title: str | None = None,
         llm: str | None = None,
         llm_prompt: str | None = None,
         llm_response: str | None = None,
@@ -1407,6 +1408,7 @@ class AgentDB:
                 llm=llm,
                 llm_prompt=llm_prompt,
                 llm_response=llm_response,
+                suggested_title=suggested_title,
             )
             session.add(new_task_generation)
             await session.commit()

skyvern/forge/sdk/db/models.py

@@ -380,6 +380,7 @@ class TaskGenerationModel(Base):
     navigation_payload = Column(JSON)
     data_extraction_goal = Column(String)
     extracted_information_schema = Column(JSON)
+    suggested_title = Column(String)  # task title suggested by the language model
 
     llm = Column(String)  # language model to use
     llm_prompt = Column(String)  # The prompt sent to the language model

skyvern/forge/sdk/routes/agent_protocol.py

@@ -779,6 +779,7 @@ async def generate_task(
             navigation_payload=parsed_task_generation_obj.navigation_payload,
             data_extraction_goal=parsed_task_generation_obj.data_extraction_goal,
             extracted_information_schema=parsed_task_generation_obj.extracted_information_schema,
+            suggested_title=parsed_task_generation_obj.suggested_title,
             llm=SettingsManager.get_settings().LLM_KEY,
             llm_prompt=llm_prompt,
             llm_response=str(llm_response),

skyvern/forge/sdk/schemas/task_generations.py

@@ -17,6 +17,7 @@ class TaskGenerationBase(BaseModel):
     llm: str | None = None
     llm_prompt: str | None = None
     llm_response: str | None = None
+    suggested_title: str | None = None
 
 
 class TaskGenerationCreate(TaskGenerationBase):