Move to src dir usage (#99)

This commit is contained in:
Greyson LaLonde
2024-01-10 09:39:36 -05:00
committed by GitHub
parent a944cfc8d0
commit 155368be3b
22 changed files with 17 additions and 6 deletions

View File

@@ -0,0 +1,17 @@
from typing import Optional
from pydantic import BaseModel, Field, model_validator
class TaskOutput(BaseModel):
"""Class that represents the result of a task."""
description: str = Field(description="Description of the task")
summary: Optional[str] = Field(description="Summary of the task", default=None)
result: str = Field(description="Result of the task")
@model_validator(mode="after")
def set_summary(self):
excerpt = " ".join(self.description.split(" ")[:10])
self.summary = f"{excerpt}..."
return self