mirror/temporal-ai-agent
1
0
Fork 0
mirror of https://github.com/temporal-community/temporal-ai-agent.git synced 2026-03-15 14:08:08 +01:00
Code Issues Packages Projects Releases Wiki Activity

no parent workflow needed, all in main wf loop

Browse Source
This commit is contained in:
Steve Androulakis
2025-01-01 17:31:40 -08:00
parent 01271efcb4
commit f6fde7a7cd
2 changed files with 1 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/run_worker.py
View File

@@ -7,7 +7,6 @@ from temporalio.worker import Worker
from activities.tool_activities import ToolActivities, dynamic_tool_activity from activities.tool_activities import ToolActivities, dynamic_tool_activity
from workflows.tool_workflow import ToolWorkflow from workflows.tool_workflow import ToolWorkflow
from workflows.parent_workflow import ParentWorkflow
async def main(): async def main():
@@ -20,7 +19,7 @@ async def main():
worker = Worker( worker = Worker(
client, client,
task_queue="ollama-task-queue", task_queue="ollama-task-queue",
workflows=[ToolWorkflow, ParentWorkflow], workflows=[ToolWorkflow],
activities=[ activities=[
activities.prompt_llm, activities.prompt_llm,
activities.parse_tool_data, activities.parse_tool_data,

14
workflows/parent_workflow.py
View File

@@ -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