From a173f4d451833b2d9215ce6150c8a595c0eb5352 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 9 Jul 2025 12:18:57 -0300 Subject: [PATCH 01/13] SIENTIAPDE-1141 Update GITHUB_BRANCH in values.yaml and add debug logging for schedule ID in temporal_manager.py --- orchestrator/activities/temporal_manager.py | 2 ++ values.yaml | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/orchestrator/activities/temporal_manager.py b/orchestrator/activities/temporal_manager.py index 7fd520e..510adab 100644 --- a/orchestrator/activities/temporal_manager.py +++ b/orchestrator/activities/temporal_manager.py @@ -52,6 +52,8 @@ class TemporalManager(BaseActivity): if search_attrs.get("orchestrated", ["false"]) == ["true"]: schedule_id = schedule.id + self.logger.debug("Schedule id: %s", schedule_id) + handle = self.temporal_client.get_schedule_handle(schedule_id) self.schedule_handles[schedule_id] = handle diff --git a/values.yaml b/values.yaml index e346cfd..e96d9bc 100644 --- a/values.yaml +++ b/values.yaml @@ -123,7 +123,7 @@ env: - name: GITHUB_REPO_URL value: "git@github.com:Aignosi/sientia-dataops-orchestrator_temporal.git" - name: GITHUB_BRANCH - value: "SIENTIAPDE-1110-criar-testes-e-2-e" + value: "SIENTIA-1141-criar-testes-de-carga" - name: PYTHON_APP value: "orchestrator.worker.worker" From 9d63647cf1eb65aa253c6e5a2e4dbcdece38b959 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 9 Jul 2025 12:24:27 -0300 Subject: [PATCH 02/13] SIENTIAPDE-1141 Update GITHUB_BRANCH in values.yaml and enhance debug logging in temporal_manager.py --- orchestrator/activities/temporal_manager.py | 6 ++++++ values.yaml | 2 +- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/orchestrator/activities/temporal_manager.py b/orchestrator/activities/temporal_manager.py index 510adab..34c0a21 100644 --- a/orchestrator/activities/temporal_manager.py +++ b/orchestrator/activities/temporal_manager.py @@ -56,10 +56,16 @@ class TemporalManager(BaseActivity): handle = self.temporal_client.get_schedule_handle(schedule_id) + self.logger.debug("Handle acquired") + self.schedule_handles[schedule_id] = handle + self.logger.debug("Describing schedule...") + desc = await handle.describe() + self.logger.debug("Parsing args...") + for arg in desc.schedule.action.args: data = MessageToDict(arg)['data'] data = base64.b64decode(data).decode('utf-8') diff --git a/values.yaml b/values.yaml index e96d9bc..7bb0447 100644 --- a/values.yaml +++ b/values.yaml @@ -123,7 +123,7 @@ env: - name: GITHUB_REPO_URL value: "git@github.com:Aignosi/sientia-dataops-orchestrator_temporal.git" - name: GITHUB_BRANCH - value: "SIENTIA-1141-criar-testes-de-carga" + value: "SIENTIAPDE-1141-criar-testes-de-carga" - name: PYTHON_APP value: "orchestrator.worker.worker" From 545929c61860f38dc30ae9509a8b414ef73b5fda Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 9 Jul 2025 12:33:12 -0300 Subject: [PATCH 03/13] SIENTIAPDE-1141 feat: add asynchronous sleep in temporal_manager to improve schedule processing --- orchestrator/activities/temporal_manager.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/orchestrator/activities/temporal_manager.py b/orchestrator/activities/temporal_manager.py index 34c0a21..ab1cb29 100644 --- a/orchestrator/activities/temporal_manager.py +++ b/orchestrator/activities/temporal_manager.py @@ -1,3 +1,4 @@ +from asyncio import sleep from temporalio import activity, workflow from temporalio.client import ( Client, Schedule, ScheduleActionStartWorkflow, ScheduleIntervalSpec, ScheduleSpec, ScheduleUpdate, ScheduleUpdateInput) @@ -77,6 +78,8 @@ class TemporalManager(BaseActivity): 'data': json.loads(data), } + await sleep(0.075) + self.logger.info("Found %d orchestrated schedules", len(orchestrated_schedules)) From 68c7e2cf3b77eb8e2fc24ffba23a49bafc458ddb Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 9 Jul 2025 12:50:26 -0300 Subject: [PATCH 04/13] SIENTIAPDE-1141 fix: increase sleep duration in temporal_manager to enhance schedule processing efficiency --- orchestrator/activities/temporal_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orchestrator/activities/temporal_manager.py b/orchestrator/activities/temporal_manager.py index ab1cb29..86f0a0b 100644 --- a/orchestrator/activities/temporal_manager.py +++ b/orchestrator/activities/temporal_manager.py @@ -78,7 +78,7 @@ class TemporalManager(BaseActivity): 'data': json.loads(data), } - await sleep(0.075) + await sleep(0.2) self.logger.info("Found %d orchestrated schedules", len(orchestrated_schedules)) From c92017345d167943113090c9dd6f212f0d076d29 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 9 Jul 2025 12:52:48 -0300 Subject: [PATCH 05/13] SIENTIAPDE-1141 fix: increase start_to_close_timeout in orchestrator.py to improve workflow execution duration --- orchestrator/workflows/orchestrator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orchestrator/workflows/orchestrator.py b/orchestrator/workflows/orchestrator.py index b731eeb..e7f90d5 100644 --- a/orchestrator/workflows/orchestrator.py +++ b/orchestrator/workflows/orchestrator.py @@ -23,7 +23,7 @@ class Orchestrator: 'model_id': '-' }, retry_policy=retry_policy, - start_to_close_timeout=timedelta(seconds=60) + start_to_close_timeout=timedelta(seconds=600) ) # pipeline_config_handler = workflow.execute_local_activity_method( From 4df1b0240a047ba6797cf5c0cf03cad399c150ef Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 9 Jul 2025 13:00:05 -0300 Subject: [PATCH 06/13] SIENTIAPDE-1141 refactor: remove commented-out code and improve timeout settings in orchestrator.py and test_activities.py --- orchestrator/workflows/orchestrator.py | 32 +------------------ .../activities/test_activities.py | 21 +----------- 2 files changed, 2 insertions(+), 51 deletions(-) diff --git a/orchestrator/workflows/orchestrator.py b/orchestrator/workflows/orchestrator.py index e7f90d5..7bece5f 100644 --- a/orchestrator/workflows/orchestrator.py +++ b/orchestrator/workflows/orchestrator.py @@ -14,27 +14,6 @@ class Orchestrator: input_data['workflow_name'] = 'orchestrator' - await workflow.execute_local_activity_method( - Activities.prepare_activity, - { - 'workflow_name': input_data['workflow_name'], - 'schedule_name': input_data['schedule_name'], - 'model_name': '-', - 'model_id': '-' - }, - retry_policy=retry_policy, - start_to_close_timeout=timedelta(seconds=600) - ) - - # pipeline_config_handler = workflow.execute_local_activity_method( - # Activities.load_query_from_couchbase, - # { - # 'query': input_data['pipelines_query'] - # }, - # retry_policy=retry_policy, - # start_to_close_timeout=timedelta(seconds=60) - # ) - pipeline_config_handler = workflow.execute_local_activity_method( Activities.aggregate_documents_in_mongodb, { @@ -44,15 +23,6 @@ class Orchestrator: start_to_close_timeout=timedelta(seconds=60) ) - # opc_servers_handler = workflow.execute_local_activity_method( - # Activities.load_query_from_couchbase, - # { - # 'query': input_data['opc_servers_query'] - # }, - # retry_policy=retry_policy, - # start_to_close_timeout=timedelta(seconds=60) - # ) - opc_servers_handler = workflow.execute_local_activity_method( Activities.find_documents_in_mongodb, { @@ -65,7 +35,7 @@ class Orchestrator: orchestrated_schedules_handler = workflow.execute_local_activity_method( Activities.load_schedule, retry_policy=retry_policy, - start_to_close_timeout=timedelta(seconds=60) + start_to_close_timeout=timedelta(seconds=600) ) current_slot_config_handler = workflow.execute_local_activity_method( diff --git a/tests/orchestrator/activities/test_activities.py b/tests/orchestrator/activities/test_activities.py index 4ebbc18..557ec8f 100644 --- a/tests/orchestrator/activities/test_activities.py +++ b/tests/orchestrator/activities/test_activities.py @@ -8,21 +8,13 @@ from orchestrator.activities.slot_manager import SlotManager from orchestrator.activities.formatters import Formatters -@patch('orchestrator.activities.couchbase.Couchbase.__init__') @patch('orchestrator.activities.mongo_db.MongoDB.__init__') @patch('orchestrator.activities.temporal_manager.TemporalManager.__init__') @patch('orchestrator.activities.slot_manager.SlotManager.__init__') @patch('orchestrator.activities.formatters.Formatters.__init__') def test___init__(mock_formatters_init, mock_slot_manager_init, mock_temporal_manager_init, - mock_mongodb_init, - mock_couchbase_init): - - couchbase_config = { - 'connection_string': 'couchbase://localhost', - 'username': 'admin', - 'password': 'password' - } + mock_mongodb_init): mongo_db_config = { 'connection_string': 'mongodb://localhost:27017', @@ -42,7 +34,6 @@ def test___init__(mock_formatters_init, mock_slot_manager_init, activities = Activities( temporal_client=temporal_client, - # couchbase_config=couchbase_config, redis_config=redis_config, mongodb_config=mongo_db_config, logger=logger, @@ -50,7 +41,6 @@ def test___init__(mock_formatters_init, mock_slot_manager_init, ) assert isinstance(activities, Activities) - # assert isinstance(activities, Couchbase) assert isinstance(activities, MongoDB) assert isinstance(activities, TemporalManager) assert isinstance(activities, SlotManager) @@ -66,15 +56,6 @@ def test___init__(mock_formatters_init, mock_slot_manager_init, notification_handler=notification_handler ) - # mock_couchbase_init.assert_called_once_with( - # ANY, - # connection_string=couchbase_config['connection_string'], - # username=couchbase_config['username'], - # password=couchbase_config['password'], - # logger=logger, - # notification_handler=notification_handler - # ) - mock_mongodb_init.assert_called_once_with( ANY, connection_string='mongodb://localhost:27017', From bb5d0f15f9e3aab8f631e7328ac729c374dd8d90 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 9 Jul 2025 16:22:15 -0300 Subject: [PATCH 07/13] SIENTIAPDE-1141 fix: update host configuration and execution counts in test.ipynb; refactor schedule creation parameters in orchestrator --- orchestrator/worker/worker.py | 1 - test.ipynb | 75 +++++++++++++++++++++++++++-------- 2 files changed, 58 insertions(+), 18 deletions(-) diff --git a/orchestrator/worker/worker.py b/orchestrator/worker/worker.py index 383c1be..8aa019f 100644 --- a/orchestrator/worker/worker.py +++ b/orchestrator/worker/worker.py @@ -41,7 +41,6 @@ async def main(): activities = Activities( temporal_client=temporal_client, - # couchbase_config=build_couchbase_config(), redis_config=build_redis_config(), mongodb_config=build_mongodb_config(), logger=logger, diff --git a/test.ipynb b/test.ipynb index ecf11d1..948e129 100644 --- a/test.ipynb +++ b/test.ipynb @@ -120,7 +120,7 @@ "import os\n", "from unittest.mock import MagicMock\n", "\n", - "host = \"localhost:7233\"\n", + "host = \"localhost:44795\"\n", "logger = MagicMock(info=MagicMock(side_effect=print), debug=MagicMock(side_effect=print))\n", "\n", "temporal_client = await client.Client.connect(\n", @@ -136,17 +136,17 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": 4, "id": "bb750ae6", "metadata": {}, "outputs": [ { "data": { "text/plain": [ - "" + "" ] }, - "execution_count": 6, + "execution_count": 4, "metadata": {}, "output_type": "execute_result" } @@ -163,29 +163,70 @@ ")\n", "from temporalio.common import TypedSearchAttributes, SearchAttributeKey, SearchAttributePair\n", "\n", + "# temporal operator search-attribute create --namespace default --name model_id --type Text\n", + "# temporal operator search-attribute create --namespace default --name orchestrated --type Text\n", + "# temporal operator search-attribute create --namespace default --name model_name --type Text\n", "\n", - "customer_id_key = SearchAttributeKey.for_keyword(\"orchestrated\")\n", - "search_attributes = TypedSearchAttributes([\n", - " SearchAttributePair(customer_id_key, \"true\")\n", - "])\n", "await temporal_client.create_schedule(\n", - " \"meu-schedule-id5\",\n", + " \"orchestrator\",\n", " Schedule(\n", " action=ScheduleActionStartWorkflow(\n", - " 'scouter-test2',\n", + " 'orchestrator',\n", " {\n", - " 'args': {\n", - " 'arg1': 'value1'\n", + " \"schedule_name\": \"orchestrator-test\",\n", + " \"pipelines_query\": {\n", + " \"collection\": \"pipelines\",\n", + " \"aggregation\": [\n", + " {\n", + " \"$lookup\": {\n", + " \"from\": \"models\",\n", + " \"localField\": \"model_id\",\n", + " \"foreignField\": \"id\",\n", + " \"as\": \"model_docs\"\n", + " }\n", + " },\n", + " {\n", + " \"$match\": {\n", + " \"models.active\": True\n", + " }\n", + " },\n", + " {\n", + " \"$match\": {\n", + " \"pipelines.active\": True\n", + " }\n", + " },\n", + " {\n", + " \"$addFields\": {\n", + " \"models\": {\n", + " \"$arrayElemAt\": [\n", + " \"$model_docs\",\n", + " 0\n", + " ]\n", + " }\n", + " }\n", + " },\n", + " {\n", + " \"$project\": {\n", + " \"model_docs\": 0\n", + " }\n", + " }\n", + " ]\n", + " },\n", + " \"opc_servers_query\": {\n", + " \"collection\": \"opc-servers\",\n", + " \"filters\": {\n", + "\n", + " }\n", " }\n", " },\n", - " id=\"workflow-id-unico\",\n", - " task_queue=\"nome-da-task-queue\",\n", + " id=\"orchestrator\",\n", + " task_queue=\"orchestrator-queue\",\n", + " execution_timeout=timedelta(minutes=600)\n", " ),\n", " spec=ScheduleSpec(\n", - " intervals=[ScheduleIntervalSpec(every=timedelta(minutes=10))]\n", + " intervals=[ScheduleIntervalSpec(every=timedelta(minutes=60))]\n", " )\n", - " ),\n", - " search_attributes=search_attributes,\n", + " )\n", ")\n" ] }, From 70f9d4e8376f33852d8a407836054e3c3b518014 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Wed, 9 Jul 2025 17:21:35 -0300 Subject: [PATCH 08/13] SIENTIAPDE-1141 fix: update execution counts and refactor query structure in test.ipynb for improved clarity --- test.ipynb | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/test.ipynb b/test.ipynb index 948e129..c8e7a10 100644 --- a/test.ipynb +++ b/test.ipynb @@ -136,17 +136,17 @@ }, { "cell_type": "code", - "execution_count": 4, + "execution_count": 6, "id": "bb750ae6", "metadata": {}, "outputs": [ { "data": { "text/plain": [ - "" + "" ] }, - "execution_count": 4, + "execution_count": 6, "metadata": {}, "output_type": "execute_result" } @@ -187,12 +187,7 @@ " },\n", " {\n", " \"$match\": {\n", - " \"models.active\": True\n", - " }\n", - " },\n", - " {\n", - " \"$match\": {\n", - " \"pipelines.active\": True\n", + " \"active\": True\n", " }\n", " },\n", " {\n", @@ -206,6 +201,11 @@ " }\n", " },\n", " {\n", + " \"$match\": {\n", + " \"models.active\": True\n", + " }\n", + " },\n", + " {\n", " \"$project\": {\n", " \"model_docs\": 0\n", " }\n", From 861cfde9ef2801edf6d75b60e8072722c2a7f684 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 10 Jul 2025 15:42:38 -0300 Subject: [PATCH 09/13] SIENTIAPDE-1141 fix: adjust execution counts and reduce sleep duration in test.ipynb and temporal_manager.py for improved performance --- orchestrator/activities/temporal_manager.py | 2 +- test.ipynb | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/orchestrator/activities/temporal_manager.py b/orchestrator/activities/temporal_manager.py index 86f0a0b..ab1cb29 100644 --- a/orchestrator/activities/temporal_manager.py +++ b/orchestrator/activities/temporal_manager.py @@ -78,7 +78,7 @@ class TemporalManager(BaseActivity): 'data': json.loads(data), } - await sleep(0.2) + await sleep(0.075) self.logger.info("Found %d orchestrated schedules", len(orchestrated_schedules)) diff --git a/test.ipynb b/test.ipynb index c8e7a10..67d426a 100644 --- a/test.ipynb +++ b/test.ipynb @@ -110,7 +110,7 @@ }, { "cell_type": "code", - "execution_count": 1, + "execution_count": 3, "id": "7d01f160", "metadata": {}, "outputs": [], @@ -120,7 +120,7 @@ "import os\n", "from unittest.mock import MagicMock\n", "\n", - "host = \"localhost:44795\"\n", + "host = \"localhost:7233\"\n", "logger = MagicMock(info=MagicMock(side_effect=print), debug=MagicMock(side_effect=print))\n", "\n", "temporal_client = await client.Client.connect(\n", @@ -136,17 +136,17 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": 4, "id": "bb750ae6", "metadata": {}, "outputs": [ { "data": { "text/plain": [ - "" + "" ] }, - "execution_count": 6, + "execution_count": 4, "metadata": {}, "output_type": "execute_result" } From 351fce905a81506890ca7f6795e5cfe06bd55b91 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 10 Jul 2025 16:29:01 -0300 Subject: [PATCH 10/13] SIENTIAPDE-1141 fix: increase sleep duration in temporal_manager.py to improve schedule processing efficiency --- orchestrator/activities/temporal_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orchestrator/activities/temporal_manager.py b/orchestrator/activities/temporal_manager.py index ab1cb29..384ca1c 100644 --- a/orchestrator/activities/temporal_manager.py +++ b/orchestrator/activities/temporal_manager.py @@ -78,7 +78,7 @@ class TemporalManager(BaseActivity): 'data': json.loads(data), } - await sleep(0.075) + await sleep(0.1) self.logger.info("Found %d orchestrated schedules", len(orchestrated_schedules)) From dfd77c17685c4e4f0943dba914a7bb3ae9d5703f Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 10 Jul 2025 16:32:58 -0300 Subject: [PATCH 11/13] SIENTIAPDE-1141 fix: set RPC timeout to 60 seconds in temporal_manager.py for improved schedule description handling --- orchestrator/activities/temporal_manager.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/orchestrator/activities/temporal_manager.py b/orchestrator/activities/temporal_manager.py index 384ca1c..fc4c5a1 100644 --- a/orchestrator/activities/temporal_manager.py +++ b/orchestrator/activities/temporal_manager.py @@ -63,7 +63,9 @@ class TemporalManager(BaseActivity): self.logger.debug("Describing schedule...") - desc = await handle.describe() + desc = await handle.describe( + rpc_timeout=timedelta(seconds=60) + ) self.logger.debug("Parsing args...") From f43e962392a61dd7efcc0763514f9d19ec62dec4 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Fri, 11 Jul 2025 08:08:08 -0300 Subject: [PATCH 12/13] SIENTIAPDE-1141 fix: update execution counts and add schedule deletion logic in test.ipynb for improved error handling --- test.ipynb | 106 ++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 file changed, 105 insertions(+), 1 deletion(-) diff --git a/test.ipynb b/test.ipynb index 67d426a..7a97f4b 100644 --- a/test.ipynb +++ b/test.ipynb @@ -110,7 +110,7 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 14, "id": "7d01f160", "metadata": {}, "outputs": [], @@ -230,6 +230,87 @@ ")\n" ] }, + { + "cell_type": "code", + "execution_count": null, + "id": "b840347d", + "metadata": {}, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "Error deleting schedule scouter-load-test-1: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-2: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-3: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-4: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-5: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-6: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-7: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-8: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-9: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-10: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-11: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-12: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-13: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-14: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-15: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-16: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-17: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-18: workflow execution already completed\n", + "Error deleting schedule scouter-load-test-19: workflow execution already completed\n" + ] + } + ], + "source": [ + "# Delete schedule by ID\n", + "schedule_id = \"scouter-load-test-num\"\n", + "\n", + "for i in range(0, 70):\n", + "\n", + " try:\n", + " handle = temporal_client.get_schedule_handle(\n", + " schedule_id.replace(\"num\", str(i)))\n", + " \n", + " await handle.delete()\n", + " except Exception as e:\n", + " print(f\"Error deleting schedule {schedule_id.replace('num', str(i))}: {e}\")\n", + "\n", + "# Delete schedule by ID\n", + "schedule_id = \"laborious-load-test-num\"\n", + "\n", + "for i in range(0, 70):\n", + "\n", + " try:\n", + " handle = temporal_client.get_schedule_handle(\n", + " schedule_id.replace(\"num\", str(i)))\n", + " \n", + " await handle.delete()\n", + " except Exception as e:\n", + " print(f\"Error deleting schedule {schedule_id.replace('num', str(i))}: {e}\")\n" + ] + }, + { + "cell_type": "code", + "execution_count": 7, + "id": "374b5b0e", + "metadata": {}, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "scouter-load-test-num\n" + ] + } + ], + "source": [ + "schedule_id = \"scouter-load-test-num\"\n", + "schedule_id.replace(\"-\", \"_\")\n", + "\n", + "print(schedule_id)" + ] + }, { "cell_type": "code", "execution_count": 2, @@ -299,6 +380,29 @@ " " ] }, + { + "cell_type": "code", + "execution_count": 17, + "id": "f81b3728", + "metadata": {}, + "outputs": [ + { + "name": "stdout", + "output_type": "stream", + "text": [ + "{'_client': , 'id': 'orchestrator'}\n" + ] + } + ], + "source": [ + "async for schedule in await temporal_client.list_schedules():\n", + " id = schedule.id\n", + "\n", + " handle = temporal_client.get_schedule_handle(id)\n", + "\n", + " print(vars(handle))" + ] + }, { "cell_type": "code", "execution_count": 13, From fae942136b278287ab3bc8472e2326197a4baf67 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Fri, 11 Jul 2025 08:16:28 -0300 Subject: [PATCH 13/13] SIENTIAPDE-1141 fix: remove outdated assertions in test_orchestrator.py to streamline workflow testing --- tests/orchestrator/workflows/test_orchestrator.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/tests/orchestrator/workflows/test_orchestrator.py b/tests/orchestrator/workflows/test_orchestrator.py index 3a131a4..c4fd4c3 100644 --- a/tests/orchestrator/workflows/test_orchestrator.py +++ b/tests/orchestrator/workflows/test_orchestrator.py @@ -20,20 +20,6 @@ async def test_run(workflow_mock, orchestrator): await orchestrator.run(input_data) - workflow_mock.execute_local_activity_method.assert_has_calls([ - call( - Activities.prepare_activity, - { - "workflow_name": "orchestrator", - "schedule_name": "test-schedule-name", - "model_name": "-", - "model_id": "-" - }, - retry_policy=ANY, - start_to_close_timeout=ANY - ) - ]) - workflow_mock.execute_local_activity_method.assert_has_calls([ call( Activities.aggregate_documents_in_mongodb,