Merge pull request #3 from sebastientinel/master

Update create_task_log.py
StackStorm-Exchange · Sep 4, 2019 · 7ab2efe · 7ab2efe
2 parents 9d92039 + a3e8e38
commit 7ab2efe
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/actions/create_task_log.py b/actions/create_task_log.py
@@ -1,4 +1,5 @@
 from thehive4py.api import TheHiveApi
+from thehive4py.models import CaseTaskLog
 from st2common.runners.base_action import Action
 
 __all__ = [
@@ -9,6 +10,7 @@
 class CreateTaskLogAction(Action):
     def run(self, task_id, log):
         api = TheHiveApi(self.config['thehive_url'], self.config['thehive_api_key'])
-        response = api.create_task_log(task_id, log)
+        case_task_log = CaseTaskLog(message=log)
+        response = api.create_task_log(task_id, case_task_log)
 
         return response.json()