Skip to content

Commit 450f237

Browse files
committed
feat: add description field to TaskOutput in validation feedback tests
1 parent 66e53fc commit 450f237

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/praisonai/tests/unit/test_validation_feedback.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,12 +60,14 @@ def test_validation_feedback_captured_on_invalid_decision(self):
6060
# Simulate task execution results
6161
collect_task.result = TaskOutput(
6262
raw="Collected only 3 items",
63+
description="Collected data items",
6364
agent="agent1"
6465
)
6566
collect_task.status = "completed"
6667

6768
validate_task.result = TaskOutput(
6869
raw="Not enough results, need at least 10",
70+
description="Validation result",
6971
agent="agent2"
7072
)
7173
validate_task.status = "completed"
@@ -220,12 +222,14 @@ def test_validation_feedback_with_context_tasks(self):
220222
# Simulate task execution results
221223
data_task.result = TaskOutput(
222224
raw="Generated data with errors",
225+
description="Generated data",
223226
agent="agent1"
224227
)
225228
data_task.status = "completed"
226229

227230
validate_task.result = TaskOutput(
228231
raw="Data has errors, please fix",
232+
description="Data validation result",
229233
agent="agent2"
230234
)
231235
validate_task.status = "completed"

0 commit comments

Comments
 (0)