Update from https://github.com/cloudimpl/next-coder/commit/03736fc9ab76c7d909783a71caa484fa94ac3f00

Author: CloudimplBot

polycode/client.go

@@ -30,14 +30,15 @@ type StartAppRequest struct {
 }
 
 type ExecServiceRequest struct {
-	EnvId         string      `json:"envId"`
-	Service       string      `json:"service"`
-	TenantId      string      `json:"tenantId"`
-	PartitionKey  string      `json:"partitionKey"`
-	Method        string      `json:"method"`
-	Options       TaskOptions `json:"options"`
-	FireAndForget bool        `json:"fireAndForget"`
-	Input         any         `json:"input"`
+	EnvId         string            `json:"envId"`
+	Service       string            `json:"service"`
+	TenantId      string            `json:"tenantId"`
+	PartitionKey  string            `json:"partitionKey"`
+	Method        string            `json:"method"`
+	Options       TaskOptions       `json:"options"`
+	FireAndForget bool              `json:"fireAndForget"`
+	Headers       map[string]string `json:"headers"`
+	Input         any               `json:"input"`
 }
 
 type ExecAppRequest struct {

polycode/context.go

@@ -28,7 +28,9 @@ type ServiceContext interface {
 type WorkflowContext interface {
 	BaseContext
 	Service(service string) *RemoteServiceBuilder
+	Agent(agent string) *RemoteAgentBuilder
 	ServiceEx(envId string, service string) *RemoteServiceBuilder
+	AgentEx(envId string, agent string) *RemoteAgentBuilder
 	App(appName string) RemoteApp
 	AppEx(envId string, appName string) RemoteApp
 	Controller(controller string) RemoteController
@@ -107,12 +109,24 @@ func (s ContextImpl) Service(service string) *RemoteServiceBuilder {
 	}
 }
 
+func (s ContextImpl) Agent(agent string) *RemoteAgentBuilder {
+	return &RemoteAgentBuilder{
+		ctx: s.ctx, sessionId: s.sessionId, agent: agent, serviceClient: s.serviceClient,
+	}
+}
+
 func (s ContextImpl) ServiceEx(envId string, service string) *RemoteServiceBuilder {
 	return &RemoteServiceBuilder{
 		ctx: s.ctx, sessionId: s.sessionId, envId: envId, service: service, serviceClient: s.serviceClient,
 	}
 }
 
+func (s ContextImpl) AgentEx(envId string, agent string) *RemoteAgentBuilder {
+	return &RemoteAgentBuilder{
+		ctx: s.ctx, sessionId: s.sessionId, envId: envId, agent: agent, serviceClient: s.serviceClient,
+	}
+}
+
 func (s ContextImpl) App(appName string) RemoteApp {
 	return RemoteApp{
 		ctx: s.ctx, sessionId: s.sessionId, appName: appName, serviceClient: s.serviceClient,

polycode/model.go

@@ -4,6 +4,8 @@ import (
 	"time"
 )
 
+const AgentNameHeader = "X-Sidecar-Agent-Name"
+
 type BackoffStrategy struct {
 	InitialInterval time.Duration `json:"initialInterval"`
 	MaxInterval     time.Duration `json:"maxInterval"`
@@ -136,3 +138,19 @@ type CallerContextMeta struct {
 	TaskName     string `json:"taskName"`
 	TaskId       string `json:"taskId"`
 }
+
+type AgentInput struct {
+	SessionKey string            `json:"sessionKey"`
+	LLMInput   LLMInput          `json:"llmInput"`
+	Labels     map[string]string `json:"labels"`
+}
+
+type LLMInput struct {
+	Text string `json:"text"`
+	//Files []File `json:"files"`
+}
+
+//type File struct {
+//	Data []byte `json:"data"`
+//	Path string `json:"path"`
+//}

polycode/service.go

@@ -132,6 +132,72 @@ func (r RemoteService) Send(options TaskOptions, method string, input any) error
 	}
 }
 
+type RemoteAgentBuilder struct {
+	ctx           context.Context
+	sessionId     string
+	envId         string
+	agent         string
+	serviceClient *ServiceClient
+	tenantId      string
+}
+
+func (r *RemoteAgentBuilder) WithTenantId(tenantId string) *RemoteAgentBuilder {
+	r.tenantId = tenantId
+	return r
+}
+
+func (r *RemoteAgentBuilder) Get() RemoteAgent {
+	return RemoteAgent{
+		ctx:           r.ctx,
+		sessionId:     r.sessionId,
+		envId:         r.envId,
+		agent:         r.agent,
+		serviceClient: r.serviceClient,
+		tenantId:      r.tenantId,
+	}
+}
+
+type RemoteAgent struct {
+	ctx           context.Context
+	sessionId     string
+	envId         string
+	agent         string
+	serviceClient *ServiceClient
+	tenantId      string
+}
+
+func (r RemoteAgent) Call(options TaskOptions, input AgentInput) Response {
+	req := ExecServiceRequest{
+		EnvId:        r.envId,
+		Service:      "agent_service",
+		TenantId:     r.tenantId,
+		PartitionKey: r.agent + ":" + input.SessionKey,
+		Method:       "CallAgent",
+		Options:      options,
+		Headers: map[string]string{
+			AgentNameHeader: r.agent,
+		},
+		Input: input,
+	}
+
+	output, err := r.serviceClient.ExecService(r.sessionId, req)
+	if err != nil {
+		fmt.Printf("client: exec task error: %v\n", err)
+		return Response{
+			output:  nil,
+			isError: true,
+			error:   ErrTaskExecError.Wrap(err),
+		}
+	}
+
+	fmt.Printf("client: exec task output: %v\n", output)
+	return Response{
+		output:  output.Output,
+		isError: output.IsError,
+		error:   output.Error,
+	}
+}
+
 type RemoteApp struct {
 	ctx           context.Context
 	sessionId     string