@stigmer/sdk 0.0.36 → 0.0.40

package/src/gen/agent.ts

@@ -1,16 +1,18 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { 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 { AgentSchema, type Agent } from "@stigmer/protos/ai/stigmer/agentic/agent/v1/api_pb";
 import { AgentCommandController } from "@stigmer/protos/ai/stigmer/agentic/agent/v1/command_pb";
-import { AgentIdSchema } from "@stigmer/protos/ai/stigmer/agentic/agent/v1/io_pb";
+import { AgentIdSchema, GetDefaultAgentRequestSchema, type GetDefaultAgentRequest } from "@stigmer/protos/ai/stigmer/agentic/agent/v1/io_pb";
 import { AgentQueryController } from "@stigmer/protos/ai/stigmer/agentic/agent/v1/query_pb";
-import { AgentSpecSchema } from "@stigmer/protos/ai/stigmer/agentic/agent/v1/spec_pb";
+import { AgentSpecSchema, ToolApprovalOverrideSchema, McpServerUsageSchema, McpAccessSchema, SubAgentSchema } from "@stigmer/protos/ai/stigmer/agentic/agent/v1/spec_pb";
+import { EnvironmentSpecSchema, EnvironmentValueSchema } from "@stigmer/protos/ai/stigmer/agentic/environment/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";
@@ -46,6 +48,12 @@ export class AgentClient {
     } catch (e) { throw wrapError(e); }
   }
 
+  async updateVisibility(input: UpdateVisibilityInput): Promise<Agent> {
+    try {
+      return await this.command.updateVisibility(input);
+    } catch (e) { throw wrapError(e); }
+  }
+
   async delete(id: string): Promise<Agent> {
     try {
       return await this.command.delete(create(AgentIdSchema, { value: id }));
@@ -60,7 +68,13 @@ export class AgentClient {
 
   async getByReference(ref: ResourceRef): Promise<Agent> {
     try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, ref));
+      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.agent }));
+    } catch (e) { throw wrapError(e); }
+  }
+
+  async getDefault(input: GetDefaultAgentRequest): Promise<Agent> {
+    try {
+      return await this.query.getDefault(input);
     } catch (e) { throw wrapError(e); }
   }
 
@@ -85,7 +99,9 @@ export class AgentClient {
 /** Input for creating/updating a Agent. */
 export interface AgentInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   description?: string;
   iconUrl?: string;
   instructions?: string;
@@ -125,22 +141,69 @@ export interface McpAccessInput {
   enabledTools?: string[];
 }
 
+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 buildMcpAccessProto(input: McpAccessInput) {
+  return Object.assign(create(McpAccessSchema), stripUndefined({
+    mcpServer: input.mcpServer,
+    enabledTools: input.enabledTools,
+  }));
+}
+
+function buildSubAgentProto(input: SubAgentInput) {
+  const msg = create(SubAgentSchema);
+  if (input.name !== undefined) msg.name = input.name;
+  if (input.description !== undefined) msg.description = input.description;
+  if (input.instructions !== undefined) msg.instructions = input.instructions;
+  if (input.mcpAccess) msg.mcpAccess = input.mcpAccess.map(buildMcpAccessProto);
+  if (input.skillRefs) msg.skillRefs = input.skillRefs.map(r => create(ApiResourceReferenceSchema, r));
+  if (input.modelOverride !== undefined) msg.modelOverride = input.modelOverride;
+  return msg;
+}
+
 function buildAgentProto(input: AgentInput): Agent {
+  const mcpServerUsages = input.mcpServerUsages?.map(buildMcpServerUsageProto);
+  const skillRefs = input.skillRefs?.map(r => create(ApiResourceReferenceSchema, r));
+  const subAgents = input.subAgents?.map(buildSubAgentProto);
+  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(AgentSchema), {
     apiVersion: "agentic.stigmer.ai/v1",
     kind: "Agent",
     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(AgentSpecSchema), {
+    spec: Object.assign(create(AgentSpecSchema), stripUndefined({
       description: input.description,
       iconUrl: input.iconUrl,
       instructions: input.instructions,
-      mcpServerUsages: input.mcpServerUsages,
-      skillRefs: input.skillRefs,
-      subAgents: input.subAgents,
-      envSpec: input.envSpec,
-    }),
+      mcpServerUsages,
+      skillRefs,
+      subAgents,
+      envSpec,
+    })),
   }) as Agent;
 }

package/src/gen/agentexecution.ts

@@ -1,14 +1,16 @@
 // 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 { AgentExecutionSchema, type AgentExecution } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/api_pb";
 import { AgentExecutionCommandController } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/command_pb";
-import { AgentExecutionIdSchema, AgentExecutionUpdateStatusInputSchema, SubmitApprovalInputSchema, CancelAgentExecutionInputSchema, TerminateAgentExecutionInputSchema, RecoverAgentExecutionInputSchema, PauseAgentExecutionInputSchema, ResumeAgentExecutionInputSchema, UploadAttachmentRequestSchema, UploadAttachmentResponseSchema, ListAgentExecutionsRequestSchema, AgentExecutionListSchema, ListAgentExecutionsBySessionRequestSchema, GetArtifactDownloadUrlRequestSchema, GetArtifactDownloadUrlResponseSchema, GetSessionUsageReportInputSchema, GetSessionUsageReportOutputSchema, GetAgentUsageReportInputSchema, GetAgentUsageReportOutputSchema, GetOrgUsageReportInputSchema, GetOrgUsageReportOutputSchema, type AgentExecutionUpdateStatusInput, type SubmitApprovalInput, type CancelAgentExecutionInput, type TerminateAgentExecutionInput, type RecoverAgentExecutionInput, type PauseAgentExecutionInput, type ResumeAgentExecutionInput, type UploadAttachmentRequest, type UploadAttachmentResponse, type ListAgentExecutionsRequest, type AgentExecutionList, type ListAgentExecutionsBySessionRequest, type GetArtifactDownloadUrlRequest, type GetArtifactDownloadUrlResponse, type GetSessionUsageReportInput, type GetSessionUsageReportOutput, type GetAgentUsageReportInput, type GetAgentUsageReportOutput, type GetOrgUsageReportInput, type GetOrgUsageReportOutput } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/io_pb";
+import { AgentExecutionIdSchema, AgentExecutionUpdateStatusInputSchema, SubmitApprovalInputSchema, CancelAgentExecutionInputSchema, TerminateAgentExecutionInputSchema, RecoverAgentExecutionInputSchema, PauseAgentExecutionInputSchema, ResumeAgentExecutionInputSchema, UploadAttachmentRequestSchema, UploadAttachmentResponseSchema, ListAgentExecutionsRequestSchema, AgentExecutionListSchema, ListAgentExecutionsBySessionRequestSchema, GetArtifactDownloadUrlRequestSchema, GetArtifactDownloadUrlResponseSchema, GetArtifactContentRequestSchema, GetArtifactContentResponseSchema, GetSessionUsageReportInputSchema, GetSessionUsageReportOutputSchema, GetAgentUsageReportInputSchema, GetAgentUsageReportOutputSchema, GetOrgUsageReportInputSchema, GetOrgUsageReportOutputSchema, type AgentExecutionUpdateStatusInput, type SubmitApprovalInput, type CancelAgentExecutionInput, type TerminateAgentExecutionInput, type RecoverAgentExecutionInput, type PauseAgentExecutionInput, type ResumeAgentExecutionInput, type UploadAttachmentRequest, type UploadAttachmentResponse, type ListAgentExecutionsRequest, type AgentExecutionList, type ListAgentExecutionsBySessionRequest, type GetArtifactDownloadUrlRequest, type GetArtifactDownloadUrlResponse, type GetArtifactContentRequest, type GetArtifactContentResponse, type GetSessionUsageReportInput, type GetSessionUsageReportOutput, type GetAgentUsageReportInput, type GetAgentUsageReportOutput, type GetOrgUsageReportInput, type GetOrgUsageReportOutput } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/io_pb";
 import { AgentExecutionQueryController } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/query_pb";
-import { AgentExecutionSpecSchema } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/spec_pb";
+import { AgentExecutionSpecSchema, ContextManagementConfigSchema, ExecutionConfigSchema, AttachmentSchema } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/spec_pb";
+import { ExecutionValueSchema } from "@stigmer/protos/ai/stigmer/agentic/executioncontext/v1/spec_pb";
 import { ApiResourceIdSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 
@@ -120,6 +122,12 @@ export class AgentExecutionClient {
     } catch (e) { throw wrapError(e); }
   }
 
+  async getArtifactContent(input: GetArtifactContentRequest): Promise<GetArtifactContentResponse> {
+    try {
+      return await this.query.getArtifactContent(input);
+    } catch (e) { throw wrapError(e); }
+  }
+
   async getSessionUsageReport(input: GetSessionUsageReportInput): Promise<GetSessionUsageReportOutput> {
     try {
       return await this.query.getSessionUsageReport(input);
@@ -142,7 +150,9 @@ export class AgentExecutionClient {
 /** Input for creating/updating a AgentExecution. */
 export interface AgentExecutionInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   sessionId?: string;
   agentId?: string;
   message?: string;
@@ -181,25 +191,63 @@ export interface AttachmentInput {
   localPath?: string;
 }
 
+function buildContextManagementConfigProto(input: ContextManagementConfigInput) {
+  return Object.assign(create(ContextManagementConfigSchema), stripUndefined({
+    disableSummarization: input.disableSummarization,
+    customTriggerThreshold: input.customTriggerThreshold,
+    customTargetTokens: input.customTargetTokens,
+  }));
+}
+
+function buildExecutionConfigProto(input: ExecutionConfigInput) {
+  const msg = create(ExecutionConfigSchema);
+  if (input.modelName !== undefined) msg.modelName = input.modelName;
+  if (input.contextManagement) msg.contextManagement = buildContextManagementConfigProto(input.contextManagement);
+  if (input.maxToolRounds !== undefined) msg.maxToolRounds = input.maxToolRounds;
+  if (input.maxToolResultChars !== undefined) msg.maxToolResultChars = input.maxToolResultChars;
+  if (input.maxCostUsd !== undefined) msg.maxCostUsd = input.maxCostUsd;
+  return msg;
+}
+
+function buildAttachmentProto(input: AttachmentInput) {
+  return Object.assign(create(AttachmentSchema), stripUndefined({
+    filename: input.filename,
+    storageKey: input.storageKey,
+    mountPath: input.mountPath,
+    contentType: input.contentType,
+    extract: input.extract,
+    localPath: input.localPath,
+  }));
+}
+
 function buildAgentExecutionProto(input: AgentExecutionInput): AgentExecution {
+  const executionConfig = input.executionConfig ? buildExecutionConfigProto(input.executionConfig) : undefined;
+  let runtimeEnv;
+  if (input.runtimeEnv) {
+    runtimeEnv = Object.fromEntries(Object.entries(input.runtimeEnv).map(([k, v]) =>
+      [k, create(ExecutionValueSchema, { value: v.value, isSecret: v.isSecret })]));
+  }
+  const attachments = input.attachments?.map(buildAttachmentProto);
   return Object.assign(create(AgentExecutionSchema), {
     apiVersion: "agentic.stigmer.ai/v1",
     kind: "AgentExecution",
     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(AgentExecutionSpecSchema), {
+    spec: Object.assign(create(AgentExecutionSpecSchema), stripUndefined({
       sessionId: input.sessionId,
       agentId: input.agentId,
       message: input.message,
-      executionConfig: input.executionConfig,
-      runtimeEnv: input.runtimeEnv,
+      executionConfig,
+      runtimeEnv,
       callbackToken: input.callbackToken,
       autoApproveAll: input.autoApproveAll,
       parentWorkflowId: input.parentWorkflowId,
-      attachments: input.attachments,
+      attachments,
       workspaceFileRefs: input.workspaceFileRefs,
-    }),
+    })),
   }) as AgentExecution;
 }

package/src/gen/agentinstance.ts

@@ -1,14 +1,16 @@
 // 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 { AgentInstanceSchema, type AgentInstance } from "@stigmer/protos/ai/stigmer/agentic/agentinstance/v1/api_pb";
 import { AgentInstanceCommandController } from "@stigmer/protos/ai/stigmer/agentic/agentinstance/v1/command_pb";
-import { AgentInstanceIdSchema, GetAgentInstancesByAgentRequestSchema, AgentInstanceListSchema, type GetAgentInstancesByAgentRequest, type AgentInstanceList } from "@stigmer/protos/ai/stigmer/agentic/agentinstance/v1/io_pb";
+import { AgentInstanceIdSchema, GetAgentInstancesByAgentRequestSchema, AgentInstanceListSchema, ListAgentInstancesRequestSchema, type GetAgentInstancesByAgentRequest, type AgentInstanceList, type ListAgentInstancesRequest } from "@stigmer/protos/ai/stigmer/agentic/agentinstance/v1/io_pb";
 import { AgentInstanceQueryController } from "@stigmer/protos/ai/stigmer/agentic/agentinstance/v1/query_pb";
 import { AgentInstanceSpecSchema } from "@stigmer/protos/ai/stigmer/agentic/agentinstance/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";
 
@@ -60,7 +62,13 @@ export class AgentInstanceClient {
 
   async getByReference(ref: ResourceRef): Promise<AgentInstance> {
     try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, ref));
+      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.agent_instance }));
+    } catch (e) { throw wrapError(e); }
+  }
+
+  async list(input: ListAgentInstancesRequest): Promise<AgentInstanceList> {
+    try {
+      return await this.query.list(input);
     } catch (e) { throw wrapError(e); }
   }
 }
@@ -68,24 +76,29 @@ export class AgentInstanceClient {
 /** Input for creating/updating a AgentInstance. */
 export interface AgentInstanceInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   agentId?: string;
   description?: string;
   environmentRefs?: ResourceRef[];
 }
 
 function buildAgentInstanceProto(input: AgentInstanceInput): AgentInstance {
+  const environmentRefs = input.environmentRefs?.map(r => create(ApiResourceReferenceSchema, r));
   return Object.assign(create(AgentInstanceSchema), {
     apiVersion: "agentic.stigmer.ai/v1",
     kind: "AgentInstance",
     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(AgentInstanceSpecSchema), {
+    spec: Object.assign(create(AgentInstanceSpecSchema), stripUndefined({
       agentId: input.agentId,
       description: input.description,
-      environmentRefs: input.environmentRefs,
-    }),
+      environmentRefs,
+    })),
   }) as AgentInstance;
 }

package/src/gen/apikey.ts

@@ -1,6 +1,7 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { create } from "@bufbuild/protobuf";
 import { EmptySchema } from "@bufbuild/protobuf/wkt";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
@@ -61,7 +62,9 @@ export class ApiKeyClient {
 /** Input for creating/updating a ApiKey. */
 export interface ApiKeyInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   keyHash?: string;
   fingerprint?: string;
   expiresAt?: Date | string;
@@ -75,12 +78,14 @@ function buildApiKeyProto(input: ApiKeyInput): ApiKey {
     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(ApiKeySpecSchema), {
+    spec: Object.assign(create(ApiKeySpecSchema), stripUndefined({
       keyHash: input.keyHash,
       fingerprint: input.fingerprint,
       expiresAt: input.expiresAt,
       neverExpires: input.neverExpires,
-    }),
+    })),
   }) as ApiKey;
 }

package/src/gen/environment.ts

@@ -1,13 +1,16 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { type DeleteResourceInput, type ResourceRef, type EnvVarInput } from "./types";
 import { create } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
 import { EnvironmentSchema, type Environment } from "@stigmer/protos/ai/stigmer/agentic/environment/v1/api_pb";
 import { EnvironmentCommandController } from "@stigmer/protos/ai/stigmer/agentic/environment/v1/command_pb";
+import { UpdateEnvironmentVariablesRequestSchema, RemoveEnvironmentVariablesRequestSchema, EnvironmentSecretValueInputSchema, ListEnvironmentsRequestSchema, EnvironmentListSchema, type UpdateEnvironmentVariablesRequest, type RemoveEnvironmentVariablesRequest, type EnvironmentSecretValueInput, type ListEnvironmentsRequest, type EnvironmentList } from "@stigmer/protos/ai/stigmer/agentic/environment/v1/io_pb";
 import { EnvironmentQueryController } from "@stigmer/protos/ai/stigmer/agentic/environment/v1/query_pb";
-import { EnvironmentSpecSchema } from "@stigmer/protos/ai/stigmer/agentic/environment/v1/spec_pb";
+import { EnvironmentValueSchema, EnvironmentSpecSchema, type EnvironmentValue } from "@stigmer/protos/ai/stigmer/agentic/environment/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 { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 
@@ -49,6 +52,18 @@ export class EnvironmentClient {
     } catch (e) { throw wrapError(e); }
   }
 
+  async updateVariables(input: UpdateEnvironmentVariablesRequest): Promise<Environment> {
+    try {
+      return await this.command.updateVariables(input);
+    } catch (e) { throw wrapError(e); }
+  }
+
+  async removeVariables(input: RemoveEnvironmentVariablesRequest): Promise<Environment> {
+    try {
+      return await this.command.removeVariables(input);
+    } catch (e) { throw wrapError(e); }
+  }
+
   async get(id: string): Promise<Environment> {
     try {
       return await this.query.get(create(ApiResourceIdSchema, { value: id }));
@@ -57,7 +72,19 @@ export class EnvironmentClient {
 
   async getByReference(ref: ResourceRef): Promise<Environment> {
     try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, ref));
+      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.environment }));
+    } catch (e) { throw wrapError(e); }
+  }
+
+  async getSecretValue(input: EnvironmentSecretValueInput): Promise<EnvironmentValue> {
+    try {
+      return await this.query.getSecretValue(input);
+    } catch (e) { throw wrapError(e); }
+  }
+
+  async list(input: ListEnvironmentsRequest): Promise<EnvironmentList> {
+    try {
+      return await this.query.list(input);
     } catch (e) { throw wrapError(e); }
   }
 }
@@ -65,22 +92,31 @@ export class EnvironmentClient {
 /** Input for creating/updating a Environment. */
 export interface EnvironmentInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   description?: string;
   data?: Record<string, EnvVarInput>;
 }
 
 function buildEnvironmentProto(input: EnvironmentInput): Environment {
+  let data;
+  if (input.data) {
+    data = Object.fromEntries(Object.entries(input.data).map(([k, v]) =>
+      [k, create(EnvironmentValueSchema, { value: v.value, isSecret: v.isSecret, description: v.description })]));
+  }
   return Object.assign(create(EnvironmentSchema), {
     apiVersion: "agentic.stigmer.ai/v1",
     kind: "Environment",
     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(EnvironmentSpecSchema), {
+    spec: Object.assign(create(EnvironmentSpecSchema), stripUndefined({
       description: input.description,
-      data: input.data,
-    }),
+      data,
+    })),
   }) as Environment;
 }

package/src/gen/executioncontext.ts

@@ -1,6 +1,7 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { type DeleteResourceInput, type ResourceRef, type EnvVarInput } from "./types";
 import { create } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
@@ -8,7 +9,8 @@ import { ExecutionContextSchema, type ExecutionContext } from "@stigmer/protos/a
 import { ExecutionContextCommandController } from "@stigmer/protos/ai/stigmer/agentic/executioncontext/v1/command_pb";
 import { ExecutionContextIdSchema, ExecutionContextExecutionIdInputSchema, type ExecutionContextExecutionIdInput } from "@stigmer/protos/ai/stigmer/agentic/executioncontext/v1/io_pb";
 import { ExecutionContextQueryController } from "@stigmer/protos/ai/stigmer/agentic/executioncontext/v1/query_pb";
-import { ExecutionContextSpecSchema } from "@stigmer/protos/ai/stigmer/agentic/executioncontext/v1/spec_pb";
+import { ExecutionContextSpecSchema, ExecutionValueSchema } from "@stigmer/protos/ai/stigmer/agentic/executioncontext/v1/spec_pb";
+import { ApiResourceKind } from "@stigmer/protos/ai/stigmer/commons/apiresource/apiresourcekind/api_resource_kind_pb";
 import { ApiResourceReferenceSchema, ApiResourceDeleteInputSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 
@@ -52,7 +54,7 @@ export class ExecutionContextClient {
 
   async getByReference(ref: ResourceRef): Promise<ExecutionContext> {
     try {
-      return await this.query.getByReference(create(ApiResourceReferenceSchema, ref));
+      return await this.query.getByReference(create(ApiResourceReferenceSchema, { ...ref, kind: ApiResourceKind.execution_context }));
     } catch (e) { throw wrapError(e); }
   }
 
@@ -66,22 +68,31 @@ export class ExecutionContextClient {
 /** Input for creating/updating a ExecutionContext. */
 export interface ExecutionContextInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   executionId?: string;
   data?: Record<string, EnvVarInput>;
 }
 
 function buildExecutionContextProto(input: ExecutionContextInput): ExecutionContext {
+  let data;
+  if (input.data) {
+    data = Object.fromEntries(Object.entries(input.data).map(([k, v]) =>
+      [k, create(ExecutionValueSchema, { value: v.value, isSecret: v.isSecret })]));
+  }
   return Object.assign(create(ExecutionContextSchema), {
     apiVersion: "agentic.stigmer.ai/v1",
     kind: "ExecutionContext",
     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(ExecutionContextSpecSchema), {
+    spec: Object.assign(create(ExecutionContextSpecSchema), stripUndefined({
       executionId: input.executionId,
-      data: input.data,
-    }),
+      data,
+    })),
   }) as ExecutionContext;
 }

package/src/gen/iampolicy.ts

@@ -1,6 +1,7 @@
 // Code generated by stigmer-codegen. DO NOT EDIT.
 
 import { wrapError } from "./errors";
+import { stripUndefined } from "./proto-utils";
 import { create } from "@bufbuild/protobuf";
 import { createClient, type Client, type Transport } from "@connectrpc/connect";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
@@ -32,12 +33,6 @@ export class IamPolicyClient {
     } catch (e) { throw wrapError(e); }
   }
 
-  async createPlatformLink(input: IamPolicySpec): Promise<IamPolicy> {
-    try {
-      return await this.command.createPlatformLink(input);
-    } catch (e) { throw wrapError(e); }
-  }
-
   async bootstrapPolicy(input: IamPolicySpec): Promise<IamPolicy> {
     try {
       return await this.command.bootstrapPolicy(input);
@@ -78,7 +73,9 @@ export class IamPolicyClient {
 /** Input for creating/updating a IamPolicy. */
 export interface IamPolicyInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   principal: ApiResourceRefInput;
   resource: ApiResourceRefInput;
   relation: string;
@@ -91,18 +88,30 @@ export interface ApiResourceRefInput {
   relation?: string;
 }
 
+function buildApiResourceRefProto(input: ApiResourceRefInput) {
+  return Object.assign(create(ApiResourceRefSchema), stripUndefined({
+    kind: input.kind,
+    id: input.id,
+    relation: input.relation,
+  }));
+}
+
 function buildIamPolicyProto(input: IamPolicyInput): IamPolicy {
+  const principal = input.principal ? buildApiResourceRefProto(input.principal) : undefined;
+  const resource = input.resource ? buildApiResourceRefProto(input.resource) : undefined;
   return Object.assign(create(IamPolicySchema), {
     apiVersion: "iam.stigmer.ai/v1",
     kind: "IamPolicy",
     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(IamPolicySpecSchema), {
-      principal: input.principal,
-      resource: input.resource,
+    spec: Object.assign(create(IamPolicySpecSchema), stripUndefined({
+      principal,
+      resource,
       relation: input.relation,
-    }),
+    })),
   }) as IamPolicy;
 }

package/src/gen/identityaccount.ts

@@ -1,10 +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 { ApiResourceReferenceSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/io_pb";
 import { ApiResourceMetadataSchema } from "@stigmer/protos/ai/stigmer/commons/apiresource/metadata_pb";
 import { type ApiResourceAuditActor } from "@stigmer/protos/ai/stigmer/commons/apiresource/status_pb";
 import { IdentityAccountSchema, type IdentityAccount } from "@stigmer/protos/ai/stigmer/iam/identityaccount/v1/api_pb";
@@ -82,7 +84,9 @@ export class IdentityAccountClient {
 /** Input for creating/updating a IdentityAccount. */
 export interface IdentityAccountInput {
   name: string;
+  slug?: string;
   org: string;
+  labels?: Record<string, string>;
   idpId: string;
   email?: string;
   firstName?: string;
@@ -94,14 +98,17 @@ export interface IdentityAccountInput {
 }
 
 function buildIdentityAccountProto(input: IdentityAccountInput): IdentityAccount {
+  const identityProviderRef = input.identityProviderRef ? create(ApiResourceReferenceSchema, input.identityProviderRef) : undefined;
   return Object.assign(create(IdentityAccountSchema), {
     apiVersion: "iam.stigmer.ai/v1",
     kind: "IdentityAccount",
     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(IdentityAccountSpecSchema), {
+    spec: Object.assign(create(IdentityAccountSpecSchema), stripUndefined({
       idpId: input.idpId,
       email: input.email,
       firstName: input.firstName,
@@ -109,7 +116,7 @@ function buildIdentityAccountProto(input: IdentityAccountInput): IdentityAccount
       pictureUrl: input.pictureUrl,
       isMachineAccount: input.isMachineAccount,
       provisioningMode: input.provisioningMode,
-      identityProviderRef: input.identityProviderRef,
-    }),
+      identityProviderRef,
+    })),
   }) as IdentityAccount;
 }