Update from https://github.com/cloudimpl/next-coder/commit/836c7f3206e72862b21eeee3e0fd1ae2a492db74

Author: CloudimplBot

polycode/client.go

@@ -198,6 +198,25 @@ type ListFilePageResponse struct {
 	IsTruncated           bool               `json:"isTruncated"`
 }
 
+type SignalEmitRequest struct {
+	TaskId     string `json:"taskId"`
+	SignalName string `json:"signalName"`
+	Output     any    `json:"output"`
+	IsError    bool   `json:"isError"`
+	Error      Error  `json:"error"`
+}
+
+type SignalWaitRequest struct {
+	SignalName string `json:"signalName"`
+}
+
+type SignalWaitResponse struct {
+	IsAsync bool  `json:"isAsync"`
+	Output  any   `json:"output"`
+	IsError bool  `json:"isError"`
+	Error   Error `json:"error"`
+}
+
 type IncrementCounterRequest struct {
 	Group string `json:"group"`
 	Name  string `json:"name"`
@@ -380,6 +399,16 @@ func (sc *ServiceClient) CreateFolder(sessionId string, req CreateFolderRequest)
 	return executeApiWithoutResponse(sc.httpClient, sc.baseURL, sessionId, "v1/context/file/create-folder", req)
 }
 
+func (sc *ServiceClient) EmitSignal(sessionId string, req SignalEmitRequest) error {
+	return executeApiWithoutResponse(sc.httpClient, sc.baseURL, sessionId, "v1/context/signal/emit", req)
+}
+
+func (sc *ServiceClient) WaitForSignal(sessionId string, req SignalWaitRequest) (SignalWaitResponse, error) {
+	res := SignalWaitResponse{}
+	err := executeApiWithResponse(sc.httpClient, sc.baseURL, sessionId, "v1/context/signal/await", req, &res)
+	return res, err
+}
+
 func (sc *ServiceClient) IncrementCounter(sessionId string, req IncrementCounterRequest) (IncrementCounterResponse, error) {
 	var res IncrementCounterResponse
 	err := executeApiWithResponse(sc.httpClient, sc.baseURL, sessionId, "v1/utils/counter/increment", req, &res)

polycode/context.go

@@ -2,6 +2,7 @@ package polycode
 
 import (
 	"context"
+	"fmt"
 	"time"
 )
 
@@ -29,6 +30,9 @@ type WorkflowContext interface {
 	ControllerEx(envId string, controller string) RemoteController
 	UnsafeDb() *UnsafeDataStoreBuilder
 	Memo(getter func() (any, error)) Response
+	SignalAwait(signalName string) Response
+	SignalEmitData(taskId string, signalName string, data any) error
+	SignalEmitError(taskId string, signalName string, err Error) error
 }
 
 type ApiContext interface {
@@ -134,6 +138,51 @@ func (s ContextImpl) Acknowledge() error {
 	return s.serviceClient.Acknowledge(s.sessionId)
 }
 
+func (s ContextImpl) SignalAwait(signalName string) Response {
+	req := SignalWaitRequest{
+		SignalName: signalName,
+	}
+
+	output, err := s.serviceClient.WaitForSignal(s.sessionId, req)
+	if err != nil {
+		fmt.Printf("client: signal await error: %v\n", err)
+		return Response{
+			output:  nil,
+			isError: true,
+			error:   ErrTaskExecError.Wrap(err),
+		}
+	}
+
+	fmt.Printf("client: signal await output: %v\n", output)
+	return Response{
+		output:  output.Output,
+		isError: output.IsError,
+		error:   output.Error,
+	}
+}
+
+func (s ContextImpl) SignalEmitData(taskId string, signalName string, data any) error {
+	req := SignalEmitRequest{
+		TaskId:     taskId,
+		SignalName: signalName,
+		Output:     data,
+		IsError:    false,
+	}
+
+	return s.serviceClient.EmitSignal(s.sessionId, req)
+}
+
+func (s ContextImpl) SignalEmitError(taskId string, signalName string, err Error) error {
+	req := SignalEmitRequest{
+		TaskId:     taskId,
+		SignalName: signalName,
+		IsError:    true,
+		Error:      err,
+	}
+
+	return s.serviceClient.EmitSignal(s.sessionId, req)
+}
+
 func (s ContextImpl) Counter(group string, name string, ttl int64) Counter {
 	return Counter{
 		client:    s.serviceClient,

polycode/model.go

@@ -117,6 +117,7 @@ type ContextMeta struct {
 	TaskGroup    string `json:"taskGroup"`
 	TaskName     string `json:"taskName"`
 	TaskId       string `json:"taskId"`
+	ParentId     string `json:"parentId"`
 	TraceId      string `json:"traceId"`
 	InputId      string `json:"inputId"`
 }