@stigmer/sdk 0.0.36 → 0.0.40

package/src/gen/identityprovider.ts

@@ -1,9 +1,11 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { type DeleteResourceInput, type ResourceRef } from "./types";
 import { create } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
+import { ApiResourceKind } from "@stigmer/protos/ai/stigmer/commons/apiresource/apiresourcekind/api_resource_kind_pb";
 import { ApiResourceIdSchema, ApiResourceReferenceSchema, ApiResourceDeleteInputSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 import { IdentityProviderSchema, type IdentityProvider } from "@stigmer/protos/ai/stigmer/iam/identityprovider/v1/api_pb";
@@ -57,7 +59,7 @@ export class IdentityProviderClient {
 
   async getByReference(ref: ResourceRef): Promise<IdentityProvider> {
     try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, ref));
+      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.identity_provider }));
     } catch (e) { throw wrapError(e); }
   }
 }
@@ -65,7 +67,9 @@ export class IdentityProviderClient {
 /** Input for creating/updating a IdentityProvider. */
 export interface IdentityProviderInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   displayName?: string;
   jwksUri?: string;
   allowedIssuers?: string[];
@@ -81,14 +85,16 @@ function buildIdentityProviderProto(input: IdentityProviderInput): IdentityProvi
     metadata: Object.assign(create(ApiResourceMetadataSchema), {
       name: input.name,
       org: input.org,
+      ...(input.slug && { slug: input.slug }),
+      ...(input.labels && { labels: input.labels }),
     }),
-    spec: Object.assign(create(IdentityProviderSpecSchema), {
+    spec: Object.assign(create(IdentityProviderSpecSchema), stripUndefined({
       displayName: input.displayName,
       jwksUri: input.jwksUri,
       allowedIssuers: input.allowedIssuers,
       expectedAudience: input.expectedAudience,
       rateLimitBudget: input.rateLimitBudget,
       userinfoEndpoint: input.userinfoEndpoint,
-    }),
+    })),
   }) as IdentityProvider;
 }

package/src/gen/mcpserver.ts

@@ -1,16 +1,18 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { type DeleteResourceInput, type ListParams, type ListResult, type EnvSpecInput, type ResourceRef } from "./types";
 import { create } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
+import { EnvironmentSpecSchema, EnvironmentValueSchema } from "@stigmer/protos/ai/stigmer/agentic/environment/v1/spec_pb";
 import { McpServerSchema, type McpServer } from "@stigmer/protos/ai/stigmer/agentic/mcpserver/v1/api_pb";
 import { McpServerCommandController } from "@stigmer/protos/ai/stigmer/agentic/mcpserver/v1/command_pb";
 import { UpdateDiscoveredCapabilitiesInputSchema, type UpdateDiscoveredCapabilitiesInput } from "@stigmer/protos/ai/stigmer/agentic/mcpserver/v1/io_pb";
 import { McpServerQueryController } from "@stigmer/protos/ai/stigmer/agentic/mcpserver/v1/query_pb";
-import { McpServerSpecSchema } from "@stigmer/protos/ai/stigmer/agentic/mcpserver/v1/spec_pb";
+import { McpServerSpecSchema, StdioServerConfigSchema, HttpServerConfigSchema, ToolApprovalPolicySchema } from "@stigmer/protos/ai/stigmer/agentic/mcpserver/v1/spec_pb";
 import { ApiResourceKind } from "@stigmer/protos/ai/stigmer/commons/apiresource/apiresourcekind/api_resource_kind_pb";
-import { ApiResourceIdSchema, ApiResourceReferenceSchema, ApiResourceDeleteInputSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
+import { ApiResourceIdSchema, ApiResourceReferenceSchema, ApiResourceDeleteInputSchema, type UpdateVisibilityInput } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 import { PageInfoSchema } from "@stigmer/protos/ai/stigmer/commons/rpc/pagination_pb";
 import { SearchRequestSchema } from "@stigmer/protos/ai/stigmer/search/v1/io_pb";
@@ -56,6 +58,12 @@ export class McpServerClient {
     } catch (e) { throw wrapError(e); }
   }
 
+  async updateVisibility(input: UpdateVisibilityInput): Promise<McpServer> {
+    try {
+      return await this.command.updateVisibility(input);
+    } catch (e) { throw wrapError(e); }
+  }
+
   async updateDiscoveredCapabilities(input: UpdateDiscoveredCapabilitiesInput): Promise<McpServer> {
     try {
       return await this.command.updateDiscoveredCapabilities(input);
@@ -70,7 +78,7 @@ export class McpServerClient {
 
   async getByReference(ref: ResourceRef): Promise<McpServer> {
     try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, ref));
+      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.mcp_server }));
     } catch (e) { throw wrapError(e); }
   }
 
@@ -95,7 +103,9 @@ export class McpServerClient {
 /** Input for creating/updating a McpServer. */
 export interface McpServerInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   description?: string;
   iconUrl?: string;
   stdio?: StdioServerConfigInput;
@@ -126,22 +136,61 @@ export interface ToolApprovalPolicyInput {
   message?: string;
 }
 
+function buildStdioServerConfigProto(input: StdioServerConfigInput) {
+  return Object.assign(create(StdioServerConfigSchema), stripUndefined({
+    command: input.command,
+    args: input.args,
+    workingDir: input.workingDir,
+  }));
+}
+
+function buildHttpServerConfigProto(input: HttpServerConfigInput) {
+  return Object.assign(create(HttpServerConfigSchema), stripUndefined({
+    url: input.url,
+    headers: input.headers,
+    queryParams: input.queryParams,
+    timeoutSeconds: input.timeoutSeconds,
+  }));
+}
+
+function buildToolApprovalPolicyProto(input: ToolApprovalPolicyInput) {
+  return Object.assign(create(ToolApprovalPolicySchema), stripUndefined({
+    toolName: input.toolName,
+    message: input.message,
+  }));
+}
+
 function buildMcpServerProto(input: McpServerInput): McpServer {
+  let envSpec;
+  if (input.envSpec) {
+    const es = create(EnvironmentSpecSchema);
+    for (const [k, v] of Object.entries(input.envSpec.variables)) {
+      es.data[k] = create(EnvironmentValueSchema, { value: v.value, isSecret: v.isSecret, description: v.description });
+    }
+    envSpec = es;
+  }
+  const defaultToolApprovals = input.defaultToolApprovals?.map(buildToolApprovalPolicyProto);
+  const spec = Object.assign(create(McpServerSpecSchema), stripUndefined({
+    description: input.description,
+    iconUrl: input.iconUrl,
+    defaultEnabledTools: input.defaultEnabledTools,
+    envSpec,
+    defaultToolApprovals,
+  }));
+  if (input.stdio) {
+    spec.serverType = { case: "stdio", value: buildStdioServerConfigProto(input.stdio) };
+  } else if (input.http) {
+    spec.serverType = { case: "http", value: buildHttpServerConfigProto(input.http) };
+  }
   return Object.assign(create(McpServerSchema), {
     apiVersion: "agentic.stigmer.ai/v1",
     kind: "McpServer",
     metadata: Object.assign(create(ApiResourceMetadataSchema), {
       name: input.name,
       org: input.org,
+      ...(input.slug && { slug: input.slug }),
+      ...(input.labels && { labels: input.labels }),
     }),
-    spec: Object.assign(create(McpServerSpecSchema), {
-      description: input.description,
-      iconUrl: input.iconUrl,
-      stdio: input.stdio,
-      http: input.http,
-      defaultEnabledTools: input.defaultEnabledTools,
-      envSpec: input.envSpec,
-      defaultToolApprovals: input.defaultToolApprovals,
-    }),
+    spec,
   }) as McpServer;
 }

package/src/gen/organization.ts

@@ -1,11 +1,12 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { type ResourceRef } from "./types";
 import { create } from "@bufbuild/protobuf";
 import { EmptySchema } from "@bufbuild/protobuf/wkt";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
-import { type FindApiResourcesRequest } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
+import { ApiResourceReferenceSchema, type FindApiResourcesRequest } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 import { OrganizationSchema, type Organization } from "@stigmer/protos/ai/stigmer/tenancy/organization/v1/api_pb";
 import { OrganizationCommandController } from "@stigmer/protos/ai/stigmer/tenancy/organization/v1/command_pb";
@@ -76,28 +77,35 @@ export class OrganizationClient {
 /** Input for creating/updating a Organization. */
 export interface OrganizationInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   description?: string;
   logoUrl?: string;
   managementMode?: ManagementMode;
   identityProviderRef?: ResourceRef;
   externalOrgId?: string;
+  isPersonal?: boolean;
 }
 
 function buildOrganizationProto(input: OrganizationInput): Organization {
+  const identityProviderRef = input.identityProviderRef ? create(ApiResourceReferenceSchema, input.identityProviderRef) : undefined;
   return Object.assign(create(OrganizationSchema), {
     apiVersion: "tenancy.stigmer.ai/v1",
     kind: "Organization",
     metadata: Object.assign(create(ApiResourceMetadataSchema), {
       name: input.name,
       org: input.org,
+      ...(input.slug && { slug: input.slug }),
+      ...(input.labels && { labels: input.labels }),
     }),
-    spec: Object.assign(create(OrganizationSpecSchema), {
+    spec: Object.assign(create(OrganizationSpecSchema), stripUndefined({
       description: input.description,
       logoUrl: input.logoUrl,
       managementMode: input.managementMode,
-      identityProviderRef: input.identityProviderRef,
+      identityProviderRef,
       externalOrgId: input.externalOrgId,
-    }),
+      isPersonal: input.isPersonal,
+    })),
   }) as Organization;
 }

package/src/gen/project.ts

@@ -1,9 +1,11 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { type ResourceRef } from "./types";
 import { create } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
+import { ApiResourceKind } from "@stigmer/protos/ai/stigmer/commons/apiresource/apiresourcekind/api_resource_kind_pb";
 import { ApiResourceReferenceSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 import { ProjectSchema, type Project } from "@stigmer/protos/ai/stigmer/tenancy/project/v1/api_pb";
@@ -54,7 +56,7 @@ export class ProjectClient {
 
   async getByReference(ref: ResourceRef): Promise<Project> {
     try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, ref));
+      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.project }));
     } catch (e) { throw wrapError(e); }
   }
 }
@@ -62,24 +64,29 @@ export class ProjectClient {
 /** Input for creating/updating a Project. */
 export interface ProjectInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   entryPoint?: string;
   description?: string;
   members?: ResourceRef[];
 }
 
 function buildProjectProto(input: ProjectInput): Project {
+  const members = input.members?.map(r => create(ApiResourceReferenceSchema, r));
   return Object.assign(create(ProjectSchema), {
     apiVersion: "tenancy.stigmer.ai/v1",
     kind: "Project",
     metadata: Object.assign(create(ApiResourceMetadataSchema), {
       name: input.name,
       org: input.org,
+      ...(input.slug && { slug: input.slug }),
+      ...(input.labels && { labels: input.labels }),
     }),
-    spec: Object.assign(create(ProjectSpecSchema), {
+    spec: Object.assign(create(ProjectSpecSchema), stripUndefined({
       entryPoint: input.entryPoint,
       description: input.description,
-      members: input.members,
-    }),
+      members,
+    })),
   }) as Project;
 }

package/src/gen/proto-utils.ts

@@ -0,0 +1,15 @@
+// Code generated by stigmer-codegen. DO NOT EDIT.
+
+/**
+ * Remove keys whose values are `undefined` so that `Object.assign`
+ * does not overwrite protobuf default values (empty maps, empty arrays).
+ */
+export function stripUndefined<T extends Record<string, unknown>>(obj: T): Partial<T> {
+  const result: Record<string, unknown> = {};
+  for (const key of Object.keys(obj)) {
+    if (obj[key] !== undefined) {
+      result[key] = obj[key];
+    }
+  }
+  return result as Partial<T>;
+}

package/src/gen/session.ts

@@ -1,13 +1,18 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
+import { type ResourceRef } from "./types";
 import { create } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
+import { ToolApprovalOverrideSchema, McpServerUsageSchema } from "@stigmer/protos/ai/stigmer/agentic/agent/v1/spec_pb";
 import { SessionSchema, type Session } from "@stigmer/protos/ai/stigmer/agentic/session/v1/api_pb";
 import { SessionCommandController } from "@stigmer/protos/ai/stigmer/agentic/session/v1/command_pb";
 import { SessionIdSchema, ListSessionsRequestSchema, SessionListSchema, ListSessionsByAgentRequestSchema, type ListSessionsRequest, type SessionList, type ListSessionsByAgentRequest } from "@stigmer/protos/ai/stigmer/agentic/session/v1/io_pb";
 import { SessionQueryController } from "@stigmer/protos/ai/stigmer/agentic/session/v1/query_pb";
 import { SessionSpecSchema } from "@stigmer/protos/ai/stigmer/agentic/session/v1/spec_pb";
+import { GitRepoSourceSchema, LocalPathSourceSchema, WorkspaceSourceSchema, WorkspaceEntrySchema } from "@stigmer/protos/ai/stigmer/agentic/session/v1/workspace_pb";
+import { ApiResourceReferenceSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 
 /** Provides operations on session resources. */
@@ -66,13 +71,17 @@ export class SessionClient {
 /** Input for creating/updating a Session. */
 export interface SessionInput {
   name: string;
+  slug?: string;
   org: string;
-  agentInstanceId: string;
+  labels?: Record<string, string>;
+  agentInstanceId?: string;
   subject?: string;
   threadId?: string;
   sandboxId?: string;
   metadata?: Record<string, string>;
   workspaceEntries?: WorkspaceEntryInput[];
+  mcpServerUsages?: McpServerUsageInput[];
+  skillRefs?: ResourceRef[];
 }
 
 /** SDK input type for WorkspaceEntry. */
@@ -100,21 +109,90 @@ export interface LocalPathSourceInput {
   path?: string;
 }
 
+/** SDK input type for McpServerUsage. */
+export interface McpServerUsageInput {
+  mcpServerRef: ResourceRef;
+  enabledTools?: string[];
+  toolApprovalOverrides?: ToolApprovalOverrideInput[];
+}
+
+/** SDK input type for ToolApprovalOverride. */
+export interface ToolApprovalOverrideInput {
+  toolName?: string;
+  requiresApproval?: boolean;
+  message?: string;
+}
+
+function buildGitRepoSourceProto(input: GitRepoSourceInput) {
+  return Object.assign(create(GitRepoSourceSchema), stripUndefined({
+    url: input.url,
+    branch: input.branch,
+    commit: input.commit,
+    depth: input.depth,
+  }));
+}
+
+function buildLocalPathSourceProto(input: LocalPathSourceInput) {
+  return Object.assign(create(LocalPathSourceSchema), stripUndefined({
+    path: input.path,
+  }));
+}
+
+function buildWorkspaceSourceProto(input: WorkspaceSourceInput) {
+  const msg = create(WorkspaceSourceSchema);
+  if (input.gitRepo) {
+    msg.source = { case: "gitRepo", value: buildGitRepoSourceProto(input.gitRepo) };
+  } else if (input.localPath) {
+    msg.source = { case: "localPath", value: buildLocalPathSourceProto(input.localPath) };
+  }
+  return msg;
+}
+
+function buildWorkspaceEntryProto(input: WorkspaceEntryInput) {
+  const msg = create(WorkspaceEntrySchema);
+  if (input.name !== undefined) msg.name = input.name;
+  if (input.source) msg.source = buildWorkspaceSourceProto(input.source);
+  return msg;
+}
+
+function buildToolApprovalOverrideProto(input: ToolApprovalOverrideInput) {
+  return Object.assign(create(ToolApprovalOverrideSchema), stripUndefined({
+    toolName: input.toolName,
+    requiresApproval: input.requiresApproval,
+    message: input.message,
+  }));
+}
+
+function buildMcpServerUsageProto(input: McpServerUsageInput) {
+  const msg = create(McpServerUsageSchema);
+  if (input.mcpServerRef) msg.mcpServerRef = create(ApiResourceReferenceSchema, input.mcpServerRef);
+  if (input.enabledTools) msg.enabledTools = input.enabledTools;
+  if (input.toolApprovalOverrides) msg.toolApprovalOverrides = input.toolApprovalOverrides.map(buildToolApprovalOverrideProto);
+  return msg;
+}
+
 function buildSessionProto(input: SessionInput): Session {
+  const workspaceEntries = input.workspaceEntries?.map(buildWorkspaceEntryProto);
+  const mcpServerUsages = input.mcpServerUsages?.map(buildMcpServerUsageProto);
+  const skillRefs = input.skillRefs?.map(r => create(ApiResourceReferenceSchema, r));
   return Object.assign(create(SessionSchema), {
     apiVersion: "agentic.stigmer.ai/v1",
     kind: "Session",
     metadata: Object.assign(create(ApiResourceMetadataSchema), {
       name: input.name,
       org: input.org,
+      ...(input.slug && { slug: input.slug }),
+      ...(input.labels && { labels: input.labels }),
     }),
-    spec: Object.assign(create(SessionSpecSchema), {
+    spec: Object.assign(create(SessionSpecSchema), stripUndefined({
       agentInstanceId: input.agentInstanceId,
       subject: input.subject,
       threadId: input.threadId,
       sandboxId: input.sandboxId,
       metadata: input.metadata,
-      workspaceEntries: input.workspaceEntries,
-    }),
+      workspaceEntries,
+      mcpServerUsages,
+      skillRefs,
+    })),
   }) as Session;
 }

package/src/gen/skill.ts

@@ -1,16 +1,17 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { type ListParams, type ListResult, type ResourceRef } from "./types";
 import { create } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
 import { SkillSchema, type Skill } from "@stigmer/protos/ai/stigmer/agentic/skill/v1/api_pb";
 import { SkillCommandController } from "@stigmer/protos/ai/stigmer/agentic/skill/v1/command_pb";
-import { SkillIdSchema, PushSkillRequestSchema, GetArtifactRequestSchema, GetArtifactResponseSchema, type PushSkillRequest, type GetArtifactRequest, type GetArtifactResponse } from "@stigmer/protos/ai/stigmer/agentic/skill/v1/io_pb";
+import { SkillIdSchema, PushSkillRequestSchema, PushSkillFromExecutionArtifactRequestSchema, GetArtifactRequestSchema, GetArtifactResponseSchema, type PushSkillRequest, type PushSkillFromExecutionArtifactRequest, type GetArtifactRequest, type GetArtifactResponse } from "@stigmer/protos/ai/stigmer/agentic/skill/v1/io_pb";
 import { SkillQueryController } from "@stigmer/protos/ai/stigmer/agentic/skill/v1/query_pb";
 import { SkillSpecSchema } from "@stigmer/protos/ai/stigmer/agentic/skill/v1/spec_pb";
 import { ApiResourceKind } from "@stigmer/protos/ai/stigmer/commons/apiresource/apiresourcekind/api_resource_kind_pb";
-import { ApiResourceReferenceSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
+import { ApiResourceReferenceSchema, type UpdateVisibilityInput } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 import { PageInfoSchema } from "@stigmer/protos/ai/stigmer/commons/rpc/pagination_pb";
 import { SearchRequestSchema } from "@stigmer/protos/ai/stigmer/search/v1/io_pb";
@@ -34,6 +35,18 @@ export class SkillClient {
     } catch (e) { throw wrapError(e); }
   }
 
+  async pushFromExecutionArtifact(input: PushSkillFromExecutionArtifactRequest): Promise<Skill> {
+    try {
+      return await this.command.pushFromExecutionArtifact(input);
+    } catch (e) { throw wrapError(e); }
+  }
+
+  async updateVisibility(input: UpdateVisibilityInput): Promise<Skill> {
+    try {
+      return await this.command.updateVisibility(input);
+    } catch (e) { throw wrapError(e); }
+  }
+
   async delete(id: string): Promise<Skill> {
     try {
       return await this.command.delete(create(SkillIdSchema, { value: id }));
@@ -48,7 +61,7 @@ export class SkillClient {
 
   async getByReference(ref: ResourceRef): Promise<Skill> {
     try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, ref));
+      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.skill }));
     } catch (e) { throw wrapError(e); }
   }
 
@@ -79,7 +92,9 @@ export class SkillClient {
 /** Input for creating/updating a Skill. */
 export interface SkillInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   skillMd?: string;
   tag?: string;
   description?: string;
@@ -92,11 +107,13 @@ function buildSkillProto(input: SkillInput): Skill {
     metadata: Object.assign(create(ApiResourceMetadataSchema), {
       name: input.name,
       org: input.org,
+      ...(input.slug && { slug: input.slug }),
+      ...(input.labels && { labels: input.labels }),
     }),
-    spec: Object.assign(create(SkillSpecSchema), {
+    spec: Object.assign(create(SkillSpecSchema), stripUndefined({
       skillMd: input.skillMd,
       tag: input.tag,
       description: input.description,
-    }),
+    })),
   }) as Skill;
 }

package/src/gen/workflow.ts

@@ -1,15 +1,18 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { type EnvSpecInput, type ResourceRef } from "./types";
 import { create, type JsonObject } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
+import { EnvironmentSpecSchema, EnvironmentValueSchema } from "@stigmer/protos/ai/stigmer/agentic/environment/v1/spec_pb";
 import { WorkflowSchema, type Workflow } from "@stigmer/protos/ai/stigmer/agentic/workflow/v1/api_pb";
 import { WorkflowCommandController } from "@stigmer/protos/ai/stigmer/agentic/workflow/v1/command_pb";
 import { WorkflowTaskKind } from "@stigmer/protos/ai/stigmer/agentic/workflow/v1/enum_pb";
 import { WorkflowIdSchema } from "@stigmer/protos/ai/stigmer/agentic/workflow/v1/io_pb";
 import { WorkflowQueryController } from "@stigmer/protos/ai/stigmer/agentic/workflow/v1/query_pb";
-import { WorkflowSpecSchema } from "@stigmer/protos/ai/stigmer/agentic/workflow/v1/spec_pb";
+import { WorkflowSpecSchema, WorkflowDocumentSchema, ExportSchema, FlowControlSchema, WorkflowTaskSchema } from "@stigmer/protos/ai/stigmer/agentic/workflow/v1/spec_pb";
+import { ApiResourceKind } from "@stigmer/protos/ai/stigmer/commons/apiresource/apiresourcekind/api_resource_kind_pb";
 import { ApiResourceReferenceSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 
@@ -55,7 +58,7 @@ export class WorkflowClient {
 
   async getByReference(ref: ResourceRef): Promise<Workflow> {
     try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, ref));
+      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.workflow }));
     } catch (e) { throw wrapError(e); }
   }
 }
@@ -63,7 +66,9 @@ export class WorkflowClient {
 /** Input for creating/updating a Workflow. */
 export interface WorkflowInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   description?: string;
   document: WorkflowDocumentInput;
   tasks?: WorkflowTaskInput[];
@@ -98,19 +103,63 @@ export interface FlowControlInput {
   then?: string;
 }
 
+function buildWorkflowDocumentProto(input: WorkflowDocumentInput) {
+  return Object.assign(create(WorkflowDocumentSchema), stripUndefined({
+    dsl: input.dsl,
+    namespace: input.namespace,
+    name: input.name,
+    version: input.version,
+    description: input.description,
+  }));
+}
+
+function buildExportProto(input: ExportInput) {
+  return Object.assign(create(ExportSchema), stripUndefined({
+    as: input.as,
+  }));
+}
+
+function buildFlowControlProto(input: FlowControlInput) {
+  return Object.assign(create(FlowControlSchema), stripUndefined({
+    then: input.then,
+  }));
+}
+
+function buildWorkflowTaskProto(input: WorkflowTaskInput) {
+  const msg = create(WorkflowTaskSchema);
+  if (input.name !== undefined) msg.name = input.name;
+  if (input.kind !== undefined) msg.kind = input.kind;
+  if (input.taskConfig !== undefined) msg.taskConfig = input.taskConfig;
+  if (input.export) msg.export = buildExportProto(input.export);
+  if (input.flow) msg.flow = buildFlowControlProto(input.flow);
+  return msg;
+}
+
 function buildWorkflowProto(input: WorkflowInput): Workflow {
+  const document = input.document ? buildWorkflowDocumentProto(input.document) : undefined;
+  const tasks = input.tasks?.map(buildWorkflowTaskProto);
+  let envSpec;
+  if (input.envSpec) {
+    const es = create(EnvironmentSpecSchema);
+    for (const [k, v] of Object.entries(input.envSpec.variables)) {
+      es.data[k] = create(EnvironmentValueSchema, { value: v.value, isSecret: v.isSecret, description: v.description });
+    }
+    envSpec = es;
+  }
   return Object.assign(create(WorkflowSchema), {
     apiVersion: "agentic.stigmer.ai/v1",
     kind: "Workflow",
     metadata: Object.assign(create(ApiResourceMetadataSchema), {
       name: input.name,
       org: input.org,
+      ...(input.slug && { slug: input.slug }),
+      ...(input.labels && { labels: input.labels }),
     }),
-    spec: Object.assign(create(WorkflowSpecSchema), {
+    spec: Object.assign(create(WorkflowSpecSchema), stripUndefined({
       description: input.description,
-      document: input.document,
-      tasks: input.tasks,
-      envSpec: input.envSpec,
-    }),
+      document,
+      tasks,
+      envSpec,
+    })),
   }) as Workflow;
 }

package/src/gen/workflowexecution.ts

@@ -1,9 +1,11 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { type EnvVarInput } from "./types";
 import { create } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
+import { ExecutionValueSchema } from "@stigmer/protos/ai/stigmer/agentic/executioncontext/v1/spec_pb";
 import { WorkflowExecutionSchema, type WorkflowExecution } from "@stigmer/protos/ai/stigmer/agentic/workflowexecution/v1/api_pb";
 import { WorkflowExecutionCommandController } from "@stigmer/protos/ai/stigmer/agentic/workflowexecution/v1/command_pb";
 import { WorkflowExecutionIdSchema, WorkflowExecutionUpdateStatusInputSchema, SubmitWorkflowApprovalInputSchema, SendSignalInputSchema, CancelWorkflowExecutionInputSchema, TerminateWorkflowExecutionInputSchema, RecoverWorkflowExecutionInputSchema, PauseWorkflowExecutionInputSchema, ResumeWorkflowExecutionInputSchema, ListWorkflowExecutionsRequestSchema, WorkflowExecutionListSchema, ListWorkflowExecutionsByWorkflowRequestSchema, SubscribeWorkflowExecutionRequestSchema, type WorkflowExecutionUpdateStatusInput, type SubmitWorkflowApprovalInput, type SendSignalInput, type CancelWorkflowExecutionInput, type TerminateWorkflowExecutionInput, type RecoverWorkflowExecutionInput, type PauseWorkflowExecutionInput, type ResumeWorkflowExecutionInput, type ListWorkflowExecutionsRequest, type WorkflowExecutionList, type ListWorkflowExecutionsByWorkflowRequest, type SubscribeWorkflowExecutionRequest } from "@stigmer/protos/ai/stigmer/agentic/workflowexecution/v1/io_pb";
@@ -118,7 +120,9 @@ export class WorkflowExecutionClient {
 /** Input for creating/updating a WorkflowExecution. */
 export interface WorkflowExecutionInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   workflowInstanceId?: string;
   workflowId?: string;
   triggerMessage?: string;
@@ -128,20 +132,27 @@ export interface WorkflowExecutionInput {
 }
 
 function buildWorkflowExecutionProto(input: WorkflowExecutionInput): WorkflowExecution {
+  let runtimeEnv;
+  if (input.runtimeEnv) {
+    runtimeEnv = Object.fromEntries(Object.entries(input.runtimeEnv).map(([k, v]) =>
+      [k, create(ExecutionValueSchema, { value: v.value, isSecret: v.isSecret })]));
+  }
   return Object.assign(create(WorkflowExecutionSchema), {
     apiVersion: "agentic.stigmer.ai/v1",
     kind: "WorkflowExecution",
     metadata: Object.assign(create(ApiResourceMetadataSchema), {
       name: input.name,
       org: input.org,
+      ...(input.slug && { slug: input.slug }),
+      ...(input.labels && { labels: input.labels }),
     }),
-    spec: Object.assign(create(WorkflowExecutionSpecSchema), {
+    spec: Object.assign(create(WorkflowExecutionSpecSchema), stripUndefined({
       workflowInstanceId: input.workflowInstanceId,
       workflowId: input.workflowId,
       triggerMessage: input.triggerMessage,
       triggerMetadata: input.triggerMetadata,
-      runtimeEnv: input.runtimeEnv,
+      runtimeEnv,
       callbackToken: input.callbackToken,
-    }),
+    })),
   }) as WorkflowExecution;
 }