From e279b4afa2971d6884f7543cbef67c898111e520 Mon Sep 17 00:00:00 2001
From: Davidson Gomes <davidson.gomes@brius.com.br>
Date: Thu, 8 May 2025 13:17:34 -0300
Subject: [PATCH] feat(schemas): add folder_id field to AgentBase and Agent
 schemas

---
 src/schemas/schemas.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/schemas/schemas.py b/src/schemas/schemas.py
index 11a8b86f..95d6aafc 100644
--- a/src/schemas/schemas.py
+++ b/src/schemas/schemas.py
@@ -78,6 +78,9 @@ class AgentBase(BaseModel):
     agent_card_url: Optional[str] = Field(
         None, description="Agent card URL (required for a2a type)"
     )
+    folder_id: Optional[UUID4] = Field(
+        None, description="ID of the folder this agent belongs to"
+    )
     config: Any = Field(None, description="Agent configuration based on type")
 
     @validator("name")
@@ -182,6 +185,7 @@ class Agent(AgentBase):
     created_at: datetime
     updated_at: Optional[datetime] = None
     agent_card_url: Optional[str] = None
+    folder_id: Optional[UUID4] = None
 
     class Config:
         from_attributes = True