From f6fde7a7cdd586b1cdc88a325d39ccebf563b2a6 Mon Sep 17 00:00:00 2001 From: Steve Androulakis Date: Wed, 1 Jan 2025 17:31:40 -0800 Subject: [PATCH] no parent workflow needed, all in main wf loop --- scripts/run_worker.py | 3 +-- workflows/parent_workflow.py | 14 -------------- 2 files changed, 1 insertion(+), 16 deletions(-) delete mode 100644 workflows/parent_workflow.py diff --git a/scripts/run_worker.py b/scripts/run_worker.py index b8732a6..cdd11b8 100644 --- a/scripts/run_worker.py +++ b/scripts/run_worker.py @@ -7,7 +7,6 @@ from temporalio.worker import Worker from activities.tool_activities import ToolActivities, dynamic_tool_activity from workflows.tool_workflow import ToolWorkflow -from workflows.parent_workflow import ParentWorkflow async def main(): @@ -20,7 +19,7 @@ async def main(): worker = Worker( client, task_queue="ollama-task-queue", - workflows=[ToolWorkflow, ParentWorkflow], + workflows=[ToolWorkflow], activities=[ activities.prompt_llm, activities.parse_tool_data, diff --git a/workflows/parent_workflow.py b/workflows/parent_workflow.py deleted file mode 100644 index fdddcf3..0000000 --- a/workflows/parent_workflow.py +++ /dev/null @@ -1,14 +0,0 @@ -from temporalio import workflow -from .tool_workflow import ToolWorkflow, CombinedInput, ToolWorkflowParams - - -@workflow.defn -class ParentWorkflow: - @workflow.run - async def run(self, some_input: dict) -> dict: - combined_input = CombinedInput( - tool_params=ToolWorkflowParams(None, None), tools_data=some_input - ) - child = workflow.start_child_workflow(ToolWorkflow.run, combined_input) - result = await child - return result