Skip to content

Commit 591b7e2

Browse files
fix: add return_dict=True to fix TypeError in prompt_chaining tests
The workflow.start() method returns a string by default (for backward compatibility), but the test files were expecting a dictionary. Fixed by adding return_dict=True parameter to get the expected dictionary format with task_results and task_status keys. Fixes #841 Co-authored-by: Mervin Praison <[email protected]>
1 parent 5d8816e commit 591b7e2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/python/general/prompt_chaining.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def get_time_check():
7272
)
7373

7474
# Run the workflow
75-
results = workflow.start()
75+
results = workflow.start(return_dict=True)
7676

7777
# Print results
7878
print("\nWorkflow Results:")

src/praisonai-agents/tests/prompt_chaining.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def get_time_check():
6969
)
7070

7171
# Run the workflow
72-
results = workflow.start()
72+
results = workflow.start(return_dict=True)
7373

7474
# Print results
7575
print("\nWorkflow Results:")

0 commit comments

Comments
 (0)