@@ -12,8 +12,11 @@ class DataType(str, Enum):
1212 session = "session"
1313 plan = "plan"
1414 step = "step"
15- message = "message"
16- team = "team"
15+ message = "agent_message"
16+ team = "team_config"
17+ user_current_team = "user_current_team"
18+ m_plan = "m_plan"
19+ m_plan_step = "m_plan_step"
1720
1821
1922class AgentType (str , Enum ):
@@ -98,18 +101,20 @@ class Session(BaseDataModel):
98101 current_status : str
99102 message_to_user : Optional [str ] = None
100103
104+
101105class UserCurrentTeam (BaseDataModel ):
102106 """Represents the current team of a user."""
103107
104108 data_type : Literal ["user_current_team" ] = Field ("user_current_team" , Literal = True )
105109 user_id : str
106110 team_id : str
107111
112+
108113class Plan (BaseDataModel ):
109114 """Represents a plan containing multiple steps."""
110115
111116 data_type : Literal ["plan" ] = Field ("plan" , Literal = True )
112- plan_id : str
117+ plan_id : str
113118 session_id : str
114119 user_id : str
115120 initial_goal : str
@@ -119,7 +124,7 @@ class Plan(BaseDataModel):
119124 team_id : Optional [str ] = None
120125 human_clarification_request : Optional [str ] = None
121126 human_clarification_response : Optional [str ] = None
122-
127+
123128
124129class Step (BaseDataModel ):
125130 """Represents an individual step (task) within a plan."""
@@ -137,24 +142,10 @@ class Step(BaseDataModel):
137142 updated_action : Optional [str ] = None
138143
139144
140- class ThreadIdAgent (BaseDataModel ):
141- """Represents an individual thread_id."""
142-
143- data_type : Literal ["thread" ] = Field ("thread" , Literal = True )
144- session_id : str # Partition key
145- user_id : str
146- thread_id : str
147-
148-
149- class AzureIdAgent (BaseDataModel ):
150- """Represents an individual thread_id."""
145+ class TeamSelectionRequest (KernelBaseModel ):
146+ """Request model for team selection."""
151147
152- data_type : Literal ["agent" ] = Field ("agent" , Literal = True )
153- session_id : str # Partition key
154- user_id : str
155- action : str
156- agent : AgentType
157- agent_id : str
148+ team_id : str
158149
159150
160151class TeamAgent (KernelBaseModel ):
@@ -185,9 +176,6 @@ class StartingTask(KernelBaseModel):
185176 creator : str
186177 logo : str
187178
188- class TeamSelectionRequest (KernelBaseModel ):
189- """Request model for team selection."""
190- team_id : str
191179
192180class TeamConfiguration (BaseDataModel ):
193181 """Represents a team configuration stored in the database."""
@@ -260,80 +248,3 @@ class InputTask(KernelBaseModel):
260248
261249class UserLanguage (KernelBaseModel ):
262250 language : str
263-
264-
265- class GeneratePlanRequest (KernelBaseModel ):
266- """Message representing a request to generate a plan from an existing plan ID."""
267-
268- plan_id : str
269-
270-
271- class ApprovalRequest (KernelBaseModel ):
272- """Message sent to HumanAgent to request approval for a step."""
273-
274- step_id : str
275- plan_id : str
276- session_id : str
277- user_id : str
278- action : str
279- agent : AgentType
280-
281-
282- class HumanFeedback (KernelBaseModel ):
283- """Message containing human feedback on a step."""
284-
285- step_id : Optional [str ] = None
286- plan_id : str
287- session_id : str
288- approved : bool
289- human_feedback : Optional [str ] = None
290- updated_action : Optional [str ] = None
291-
292-
293- class HumanClarification (KernelBaseModel ):
294- """Message containing human clarification on a plan."""
295-
296- plan_id : str
297- session_id : str
298- human_clarification : str
299-
300-
301- class ActionRequest (KernelBaseModel ):
302- """Message sent to an agent to perform an action."""
303-
304- step_id : str
305- plan_id : str
306- session_id : str
307- action : str
308- agent : AgentType
309-
310-
311- class ActionResponse (KernelBaseModel ):
312- """Message containing the response from an agent after performing an action."""
313-
314- step_id : str
315- plan_id : str
316- session_id : str
317- result : str
318- status : StepStatus # Should be 'completed' or 'failed'
319-
320-
321- class PlanStateUpdate (KernelBaseModel ):
322- """Optional message for updating the plan state."""
323-
324- plan_id : str
325- session_id : str
326- overall_status : PlanStatus
327-
328-
329- # Define the expected structure of the LLM response
330- class PlannerResponseStep (KernelBaseModel ):
331- action : str
332- agent : AgentType
333-
334-
335- class PlannerResponsePlan (KernelBaseModel ):
336- initial_goal : str
337- steps : List [PlannerResponseStep ]
338- summary_plan_and_steps : str
339- human_clarification_request : Optional [str ] = None
0 commit comments