Closed maheshsattala closed 2 months ago
This pull request introduces significant enhancements across multiple files in the Kairon project. Key modifications include the addition of a task_type
parameter in various method signatures, enabling better categorization of tasks. The CloudUtility
class has been updated to improve logging and error handling during lambda function invocations. Additionally, a new TASK_TYPE
enumeration has been introduced to provide structured task categorization. The changes also extend to test cases, ensuring that task execution details are logged and verified correctly.
Files | Change Summary |
---|---|
kairon/async_callback/processor.py |
Added task_type parameter to run_pyscript function, utilizing TASK_TYPE.CALLBACK.value in CloudUtility.trigger_lambda . |
kairon/events/executors/base.py |
Modified execute_task to accept **kwargs , added log_task method for detailed logging of task execution. |
kairon/events/executors/dramatiq.py |
Updated execute_task method to include **kwargs , enhanced logging and error handling during task execution. |
kairon/events/executors/lamda.py |
Modified execute_task to accept **kwargs , extracted task_type from kwargs for use in CloudUtility.trigger_lambda . |
kairon/events/executors/standalone.py |
Updated execute_task method to accept **kwargs , improved error handling and logging based on execution outcome. |
kairon/events/scheduler/base.py |
Modified add_job and update_job methods to include task_type parameter. |
kairon/events/scheduler/kscheduler.py |
Updated add_job and update_job methods to include task_type , allowing dynamic job context adjustment. |
kairon/events/utility.py |
Added task_type parameter to add_job and update_job methods, standardizing task type for scheduled events. |
kairon/shared/actions/utils.py |
Added task_type argument to perform_web_search and run_pyscript functions for lambda invocations. |
kairon/shared/cloud/utils.py |
Enhanced trigger_lambda method with task_type and from_executor parameters, added log_task method for logging execution details. |
kairon/shared/data/constant.py |
Introduced TASK_TYPE enumeration with constants for ACTION , EVENT , and CALLBACK . |
kairon/shared/events/data_objects.py |
Added ExecutorLogs class for logging execution events with various fields. |
tests/unit_test/callback_test.py |
Modified test_run_pyscript to include task_type in the mock call for trigger_lambda . |
tests/unit_test/cloud_utils_test.py |
Added new tests for CloudUtility class, focusing on lambda invocation scenarios and logging. |
tests/unit_test/events/executors_test.py |
Updated tests for DramatiqExecutor and StandaloneExecutor to include task_type in execute_task calls, added logging assertions. |
tests/unit_test/events/scheduler_test.py |
Modified tests for KScheduler to replace hardcoded event types with TASK_TYPE.EVENT.value , enhancing maintainability. |
run_pyscript
function for better task categorization.🐰 In the meadow where bunnies hop,
New tasks are added, we can't stop!
With types defined, our work's a breeze,
Logging details brings us ease.
So let us cheer for changes bright,
In Kairon's code, all feels just right! 🌼✨
Thank you for using CodeRabbit. We offer it for free to the OSS community and would appreciate your support in helping us grow. If you find it useful, would you consider giving us a shout-out on your favorite social media?
Added code for executor logs and fixed respective test cases and added test cases for the same.
Summary by CodeRabbit
New Features
TASK_TYPE
enumeration for better task categorization.log_task
method.execute_task
methods across various executors to accept flexible parameters and improve error handling.Bug Fixes
Tests
TASK_TYPE
for better maintainability and clarity.