pass data to ts workflow (#2049)

.vscode/tasks.json

@@ -10,12 +10,16 @@
       }
     },
     {
-      "label": "Moose lib build",
+      "label": "TS moose lib build",
       "type": "shell",
       "command": "pnpm --filter=@514labs/moose-lib run build",
       "options": {
         "cwd": "${workspaceRoot}"
       }
+    },
+    {
+      "label": "Build all",
+      "dependsOn": ["CLI build", "TS moose lib build"]
     }
   ]
 }

apps/framework-cli/src/framework/python/wrappers/consumption_runner.py

@@ -118,11 +118,11 @@ def __init__(self, temporal_client: TemporalClient):
     # Test workflow executor in rust if this changes significantly
     def execute(self, name: str, input_data: Any) -> Dict[str, Any]:
         try:
-            asyncio.run(self._start_workflow_async(name, input_data))
+            run_id = asyncio.run(self._start_workflow_async(name, input_data))
             print(f"WorkflowClient - started workflow: {name}")
             return {
                 "status": 200,
-                "body": f"Workflow started: {name}"
+                "body": f"Workflow started: {name}. View it in the Temporal dashboard: http://localhost:8080/namespaces/default/workflows/{name}/{run_id}/history"
             }
         except Exception as e:
             print(f"WorkflowClient - error while starting workflow: {e}")
@@ -160,7 +160,7 @@ async def _start_workflow_async(self, name: str, input_data: Any):
             except json.JSONDecodeError as e:
                 raise ValueError(f"Invalid JSON input data: {e}")
 
-        await self.temporal_client.start_workflow(
+        workflow_handle = await self.temporal_client.start_workflow(
             "ScriptWorkflow",
             args=[f"{os.getcwd()}/app/scripts/{name}", input_data],
             id=name,
@@ -169,6 +169,8 @@ async def _start_workflow_async(self, name: str, input_data: Any):
             run_timeout=run_timeout
         )
 
+        return workflow_handle.result_run_id
+
     def load_consolidated_configs(self):
         try:
             file_path = os.path.join(os.getcwd(), ".moose", "workflow_configs.json")

apps/framework-cli/src/framework/typescript/templates.rs

@@ -173,7 +173,9 @@ export interface ParsedActivity {
 
 pub static TS_BASE_SCRIPT_TEMPLATE: &str = r#"import { TaskFunction, TaskDefinition } from "@514labs/moose-lib";
 
-const {{name}}: TaskFunction = async () => {
+// The initial input data and data passed between tasks can be
+// defined in the task function parameter
+const {{name}}: TaskFunction = async (input?: any) => {
     // The body of your script goes here
     console.log("Hello world from {{name}}");
 
@@ -187,7 +189,7 @@ const {{name}}: TaskFunction = async () => {
     };
 };
 
-export default function createTask(input?: object) {
+export default function createTask() {
     return {
         task: {{name}},
         config: {

packages/ts-moose-lib/src/scripts/activity.ts

@@ -16,11 +16,10 @@ export const activities = {
 
       logger.info(`Task received input: ${JSON.stringify(inputData)}`);
 
-      // TODO: Handle initial input data & passing data between steps
       const processedInput = (inputData || {})?.data || {};
       const scriptModule = await require(scriptPath);
       const execResult = await scriptModule.default();
-      const result = await execResult.task();
+      const result = await execResult.task(processedInput);
 
       return result;
     } catch (error) {

packages/ts-moose-lib/src/scripts/task.ts

@@ -1,5 +1,5 @@
 export interface TaskFunction {
-  (): Promise<{ task: string; data: Record<string, any> }>;
+  (input?: any): Promise<{ task: string; data: any }>;
 }
 
 export interface TaskConfig {