From 8d3df0592f548699b48bdac31fc74b9fec019356 Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Sun, 11 Feb 2024 18:45:48 -0700 Subject: [PATCH] added pydantic base model for Agent and Skill --- swarms/structs/schemas.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/swarms/structs/schemas.py b/swarms/structs/schemas.py index f7f5441e..901fb452 100644 --- a/swarms/structs/schemas.py +++ b/swarms/structs/schemas.py @@ -155,3 +155,12 @@ class Step(StepRequestBody): False, description="Whether this is the last step in the task.", ) + +class Skill(BaseModel): + name: str + level: int + +class Agent(BaseModel): + name: str + age: int + skills: List[Skill] \ No newline at end of file