@takeshape/schema 11.99.0 → 11.99.2

package/dist/agents.d.ts

@@ -1,4 +1,4 @@
-import type { AgentJSON, AgentMap, GuardJSON, ProjectSchemaJSON, QueryJSON } from './project-schema/index.ts';
+import type { AgentJSON, AgentMap, GuardJSON, GuardMap, ProjectSchemaJSON, QueryJSON } from './project-schema/index.ts';
 export declare const START_AGENT_EXECUTION = "START";
 /**
  * Special transition destination ID for agent transitions that end the agent execution.
@@ -27,9 +27,12 @@ export declare const AGENTS_SCHEMA_PATH: readonly ["ai-experimental", "agents"];
  * Check if a guard is enabled.
  */
 export declare const isGuardEnabled: (guard: GuardJSON | undefined) => boolean;
+export declare function getGuardMap(schema: ProjectSchemaJSON): GuardMap | undefined;
 export declare function getGuards(schema: ProjectSchemaJSON): (GuardJSON & {
     id: string;
 })[];
+export declare function setGuardMap<S extends Partial<ProjectSchemaJSON>>(schema: S, guards: GuardMap): S;
+export declare function setGuard<S extends Partial<ProjectSchemaJSON>>(schema: S, id: string, guard: GuardJSON): S;
 /**
  * Get a guard configuration from the project schema.
  */
@@ -38,7 +41,12 @@ export declare function getGuardConfig(schema: ProjectSchemaJSON, guardId: strin
  * Get a guard configuration from the project schema, but only if it is enabled.
  */
 export declare function getEnabledGuardConfig(schema: ProjectSchemaJSON, guardId: string): GuardJSON | undefined;
-export declare function getAgentMap(projectSchema: ProjectSchemaJSON): AgentMap | undefined;
+export declare function getAgentMap(projectSchema: Pick<ProjectSchemaJSON, 'ai-experimental'>): AgentMap | undefined;
+export declare function getAgents(projectSchema: Pick<ProjectSchemaJSON, 'ai-experimental'>): (AgentJSON & {
+    name: string;
+})[] | undefined;
+export declare function setAgentMap<S extends Partial<ProjectSchemaJSON>>(projectSchema: S, agents: AgentMap): S;
+export declare function setAgent<S extends Partial<ProjectSchemaJSON>>(projectSchema: S, name: string, agent: AgentJSON): S;
 export declare function getAgent(projectSchema: ProjectSchemaJSON, name: string): AgentJSON | undefined;
 export declare function ensureAgent(projectSchema: ProjectSchemaJSON, name: string): AgentJSON;
 export declare function isAgentQuery(query: QueryJSON, agentName: string): boolean;

package/dist/agents.js

@@ -1,3 +1,4 @@
+import set from 'lodash/fp/set.js';
 import get from 'lodash/get.js';
 import uniq from 'lodash/uniq.js';
 import upperFirst from 'lodash/upperFirst.js';
@@ -144,10 +145,19 @@ export const AGENTS_SCHEMA_PATH = ['ai-experimental', 'agents'];
 export const isGuardEnabled = (guard) => {
     return Boolean(guard && guard.enabled !== false);
 };
+export function getGuardMap(schema) {
+    return schema['ai-experimental']?.guards;
+}
 export function getGuards(schema) {
-    const guards = schema['ai-experimental']?.guards;
+    const guards = getGuardMap(schema);
     return guards ? Object.entries(guards).map(([id, guard]) => ({ id, ...guard })) : [];
 }
+export function setGuardMap(schema, guards) {
+    return set(GUARDS_SCHEMA_PATH, guards, schema);
+}
+export function setGuard(schema, id, guard) {
+    return set([...GUARDS_SCHEMA_PATH, id], guard, schema);
+}
 /**
  * Get a guard configuration from the project schema.
  */
@@ -166,6 +176,16 @@ export function getEnabledGuardConfig(schema, guardId) {
 export function getAgentMap(projectSchema) {
     return projectSchema['ai-experimental']?.agents;
 }
+export function getAgents(projectSchema) {
+    const agentMap = getAgentMap(projectSchema);
+    return agentMap && Object.entries(agentMap).map(([name, agent]) => ({ name, ...agent }));
+}
+export function setAgentMap(projectSchema, agents) {
+    return set(AGENTS_SCHEMA_PATH, agents, projectSchema);
+}
+export function setAgent(projectSchema, name, agent) {
+    return set([...AGENTS_SCHEMA_PATH, name], agent, projectSchema);
+}
 export function getAgent(projectSchema, name) {
     return getAgentMap(projectSchema)?.[name];
 }

package/dist/index.d.ts

@@ -11,6 +11,7 @@ export * from './get-is-leaf.ts';
 export * from './interfaces.ts';
 export * from './migration/index.ts';
 export type { ProjectSchemaUpdate } from './migration/types.ts';
+export { AGENTS_UPDATE_KEY, GUARDS_UPDATE_KEY, MCP_TOOLS_UPDATE_KEY } from './migration/types.ts';
 export * from './mocks.ts';
 export * from './models/runtime-schema.ts';
 export * from './models/user-schema.ts';

package/dist/index.js

@@ -9,6 +9,7 @@ export * from "./flatten-templates.js";
 export * from "./get-is-leaf.js";
 export * from "./interfaces.js";
 export * from "./migration/index.js";
+export { AGENTS_UPDATE_KEY, GUARDS_UPDATE_KEY, MCP_TOOLS_UPDATE_KEY } from "./migration/types.js";
 export * from "./mocks.js";
 export * from "./models/runtime-schema.js";
 export * from "./models/user-schema.js";

package/dist/migration/types.d.ts

@@ -5,13 +5,16 @@ export type ProjectSchemaMigrationContext = {
     encryptFn: EncryptFn;
     decryptFn: SafeDecryptFn;
 };
+export declare const AGENTS_UPDATE_KEY = "ai-experimental.agents";
+export declare const GUARDS_UPDATE_KEY = "ai-experimental.guards";
+export declare const MCP_TOOLS_UPDATE_KEY = "ai-experimental.mcp.tools";
 export type ProjectSchemaUpdate = {
     queries?: Record<string, QueryJSON | null>;
     mutations?: Record<string, QueryJSON | null>;
     shapes?: Record<string, ShapeJSON | null>;
-    'ai-experimental.agents'?: Record<string, AgentJSON | null>;
-    'ai-experimental.guards'?: Record<string, GuardJSON | null>;
-    'ai-experimental.mcp.tools'?: Record<string, ToolJSON | null>;
+    [AGENTS_UPDATE_KEY]?: Record<string, AgentJSON | null>;
+    [GUARDS_UPDATE_KEY]?: Record<string, GuardJSON | null>;
+    [MCP_TOOLS_UPDATE_KEY]?: Record<string, ToolJSON | null>;
     forms?: Record<string, FormConfig | null>;
     workflows?: Record<string, Workflow | null>;
     services?: Record<string, ServiceConfigJSON | null>;

package/dist/migration/types.js

@@ -1 +1,3 @@
-export {};
+export const AGENTS_UPDATE_KEY = 'ai-experimental.agents';
+export const GUARDS_UPDATE_KEY = 'ai-experimental.guards';
+export const MCP_TOOLS_UPDATE_KEY = 'ai-experimental.mcp.tools';

package/dist/refs.js

@@ -526,7 +526,7 @@ export function getAllPropertyRefs(projectSchema) {
             return false;
         }
         const parent = get(projectSchema, path.slice(0, -1));
-        return isRecord(parent) && parent.type === 'generate';
+        return isRecord(parent) && (parent.type === 'generate' || parent.type === 'chat');
     };
     const pushRef = (refStr) => {
         const parsed = parsePropertyRef(refStr);

package/dist/schema-util.js

@@ -6,6 +6,7 @@ import isString from 'lodash/isString.js';
 import keyBy from 'lodash/keyBy.js';
 import mapValues from 'lodash/mapValues.js';
 import uniq from 'lodash/uniq.js';
+import { AGENTS_SCHEMA_PATH, getAgentMap } from './agents.js';
 import { workflowsEnabled } from "./api-version.js";
 import { builtInForms, builtInShapes } from "./builtin-schema.js";
 import { SERVICE_OBJECT_PATTERN_NAME } from "./constants.js";
@@ -711,10 +712,10 @@ export function getAllRefsInShapes(projectSchema, predicate) {
 }
 export function getRefsInAgents(context, predicate) {
     const refs = [];
-    const agents = context['ai-experimental']?.agents;
+    const agents = getAgentMap(context);
     if (agents) {
         for (const [agentKey, agent] of Object.entries(agents)) {
-            const agentPath = ['services', 'ai-experimental', 'agents', agentKey];
+            const agentPath = [...AGENTS_SCHEMA_PATH, agentKey];
             agent.api.inputs.forEach((input, index) => {
                 const ref = {
                     ...parseRef(context, input.args),

package/dist/util/mcp.d.ts

@@ -4,5 +4,5 @@ export declare function getMcpToolMap(projectSchema: ProjectSchemaJSON): ToolMap
 export declare function getMcpTools(projectSchema: ProjectSchemaJSON): ToolJSON[] | undefined;
 export declare function getMcpToolNames(projectSchema: ProjectSchemaJSON): string[];
 export declare function getMcpTool(projectSchema: ProjectSchemaJSON, toolName: string): ToolJSON | undefined;
-export declare function setMcpToolMap<S extends ProjectSchemaJSON>(projectSchema: S, tools: ToolMap): S;
-export declare function setMcpTool<S extends ProjectSchemaJSON>(projectSchema: S, tool: ToolJSON): S;
+export declare function setMcpToolMap<S extends Partial<ProjectSchemaJSON>>(projectSchema: S, tools: ToolMap): S;
+export declare function setMcpTool<S extends Partial<ProjectSchemaJSON>>(projectSchema: S, tool: ToolJSON): S;

package/dist/util/patch-schema.d.ts

@@ -1,4 +1,4 @@
-import type { ProjectSchemaUpdate } from '../migration/types.ts';
+import { type ProjectSchemaUpdate } from '../migration/types.ts';
 import type { ProjectSchemaJSON } from '../project-schema/index.ts';
 /**
  * Apply a schema update to a schema. This can operate on a schema with frozen properties,

package/dist/util/patch-schema.js

@@ -2,6 +2,7 @@ import set from 'lodash/fp/set.js';
 import get from 'lodash/get.js';
 import omit from 'lodash/omit.js';
 import unset from 'lodash/unset.js';
+import { AGENTS_UPDATE_KEY, GUARDS_UPDATE_KEY, MCP_TOOLS_UPDATE_KEY } from "../migration/types.js";
 /**
  * Array of lodash.get paths to keys that are merged when applying a schema update.
  */
@@ -12,9 +13,9 @@ const schemaUpdateMergedKeys = [
     'workflows',
     'forms',
     'services',
-    'ai-experimental.agents',
-    'ai-experimental.guards',
-    'ai-experimental.mcp.tools'
+    AGENTS_UPDATE_KEY,
+    GUARDS_UPDATE_KEY,
+    MCP_TOOLS_UPDATE_KEY
 ];
 function shallowMerge(original, update) {
     const result = { ...original };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@takeshape/schema",
-  "version": "11.99.0",
+  "version": "11.99.2",
   "description": "TakeShape Schema",
   "homepage": "https://www.takeshape.io",
   "repository": {
@@ -58,9 +58,9 @@
     "p-reduce": "^2.1.0",
     "semver": "^7.3.2",
     "tiny-invariant": "^1.2.0",
-    "@takeshape/errors": "11.99.0",
-    "@takeshape/json-schema": "11.99.0",
-    "@takeshape/util": "11.99.0"
+    "@takeshape/errors": "11.99.2",
+    "@takeshape/json-schema": "11.99.2",
+    "@takeshape/util": "11.99.2"
   },
   "devDependencies": {
     "@takeshape/json-schema-to-typescript": "^11.0.0",
@@ -76,7 +76,7 @@
     "glob": "^7.1.6",
     "json-schema-to-ts": "^3.1.1",
     "shortid": "^2.2.15",
-    "@takeshape/infra": "11.99.0"
+    "@takeshape/infra": "11.99.2"
   },
   "engines": {
     "node": ">=20"