22from datetime import datetime , timezone
33from enum import Enum
44from typing import Any , Dict , List , Literal , Optional
5- from pydantic import BaseModel , Field
65
6+ from semantic_kernel .kernel_pydantic import Field , KernelBaseModel
77
88
99class DataType (str , Enum ):
@@ -75,7 +75,7 @@ class MessageRole(str, Enum):
7575 function = "function"
7676
7777
78- class BaseDataModel (BaseModel ):
78+ class BaseDataModel (KernelBaseModel ):
7979 """Base data model with common fields."""
8080
8181 id : str = Field (default_factory = lambda : str (uuid .uuid4 ()))
@@ -155,7 +155,7 @@ class TeamSelectionRequest(BaseDataModel):
155155 team_id : str
156156
157157
158- class TeamAgent (BaseModel ):
158+ class TeamAgent (KernelBaseModel ):
159159 """Represents an agent within a team."""
160160
161161 input_key : str
@@ -173,7 +173,7 @@ class TeamAgent(BaseModel):
173173 coding_tools : bool = False
174174
175175
176- class StartingTask (BaseModel ):
176+ class StartingTask (KernelBaseModel ):
177177 """Represents a starting task for a team."""
178178
179179 id : str
@@ -245,15 +245,15 @@ def update_step_counts(self):
245245
246246
247247# Message classes for communication between agents
248- class InputTask (BaseModel ):
248+ class InputTask (KernelBaseModel ):
249249 """Message representing the initial input task from the user."""
250250
251251 session_id : str
252252 description : str # Initial goal
253253 # team_id: str
254254
255255
256- class UserLanguage (BaseModel ):
256+ class UserLanguage (KernelBaseModel ):
257257 language : str
258258
259259
@@ -275,4 +275,4 @@ class AgentMessageData(BaseDataModel):
275275 content : str
276276 raw_data : str
277277 steps : List [Any ] = Field (default_factory = list )
278- next_steps : List [Any ] = Field (default_factory = list )
278+ next_steps : List [Any ] = Field (default_factory = list )
0 commit comments