Skip to content

Commit 09f48a9

Browse files
add change 6
1 parent 662e80e commit 09f48a9

File tree

1 file changed

+19
-1
lines changed

1 file changed

+19
-1
lines changed

src/backend/common/models/messages_kernel.py

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ class Plan(BaseDataModel):
120120
"""Represents a plan containing multiple steps."""
121121

122122
data_type: Literal[DataType.plan] = Field(DataType.plan, Literal=True)
123-
plan_id: str
123+
plan_id: str = Field(default_factory=lambda: str(uuid.uuid4()))
124124
user_id: str
125125
initial_goal: str
126126
overall_status: PlanStatus = PlanStatus.in_progress
@@ -276,3 +276,21 @@ class AgentMessageData(BaseDataModel):
276276
raw_data: str
277277
steps: List[Any] = Field(default_factory=list)
278278
next_steps: List[Any] = Field(default_factory=list)
279+
280+
281+
class ActionRequest(BaseDataModel):
282+
"""Message sent to an agent to perform an action."""
283+
284+
step_id: str
285+
plan_id: str
286+
action: str
287+
agent: AgentType
288+
289+
290+
class HumanFeedback(BaseDataModel):
291+
"""Message containing human feedback on a step."""
292+
293+
step_id: str
294+
plan_id: str
295+
approved: bool
296+
human_feedback: Optional[str] = None

0 commit comments

Comments
 (0)