@takeshape/schema 11.80.2 → 11.82.2

package/dist/builtin-schema.js

@@ -1,3 +1,4 @@
+import { embeddingSearchArgsSchema } from "./resolvers/ai/embedding-search.js";
 export function isAsset(content) {
     return content._contentTypeId === 'ASSET' || content._shapeId === 'ASSET';
 }
@@ -423,6 +424,12 @@ export const builtInShapes = {
                 currentValue: { type: 'object' }
             }
         }
+    },
+    TSEmbeddingSearchArgs: {
+        id: 'TSEmbeddingSearchArgs',
+        name: 'TSEmbeddingSearchArgs',
+        title: 'Embedding Search Args',
+        schema: embeddingSearchArgsSchema
     }
 };
 export const builtInForms = {

package/dist/index.d.ts

@@ -26,6 +26,7 @@ export * from './service-dependencies.ts';
 export * from './services/services.ts';
 export * from './services/types.ts';
 export * from './services/util.ts';
+export * from './shape-paths.ts';
 export * from './taxonomies.ts';
 export * from './template-shapes/index.ts';
 export * from './types/transforms.ts';

package/dist/index.js

@@ -24,6 +24,7 @@ export * from "./service-dependencies.js";
 export * from "./services/services.js";
 export * from "./services/types.js";
 export * from "./services/util.js";
+export * from "./shape-paths.js";
 export * from "./taxonomies.js";
 export * from "./template-shapes/index.js";
 export * from "./types/transforms.js";

package/dist/layers/refs.d.ts

@@ -77,7 +77,7 @@ export declare function refItemToShape(shapes: ShapeMapV4_0_0, refItem: RefItem)
 /**
  * Helper fn to omit `ref` props from the target schema, and then extend it with the source schema.
  */
-export declare const omitRefAndExtend: (targetSchema: PropertySchemaV4_0_0, sourceSchema: PropertySchemaV4_0_0) => import("lodash").Omit<PropertySchemaV4_0_0, "@ref" | "$ref"> & PropertySchemaV4_0_0;
+export declare const omitRefAndExtend: (targetSchema: PropertySchemaV4_0_0, sourceSchema: PropertySchemaV4_0_0) => import("lodash").Omit<PropertySchemaV4_0_0, "$ref" | "@ref"> & PropertySchemaV4_0_0;
 /**
  * If there is a $ref, this will insert a content schema from the top level
  * in place of the reference.

package/dist/project-schema/latest.d.ts

@@ -290,7 +290,6 @@ export type ShapeSchema = ShapeSchemaAllOf | ShapeSchemaExtends | ShapeSchemaOne
 export type AgentAPI = AgentAPIChat | AgentAPIGenerate;
 export type InteractionTimeoutMilliseconds = number;
 export type InteractionTimeoutMilliseconds1 = number;
-export type MemoryLocations = AgentSessionMemoryLocation[];
 export type Variables = AgentVariable[];
 export type AgentExecution = AgentExecutionGraphQL | AgentExecutionGenerate | AgentExecutionChat;
 export type AgentAIStateInput = AgentAIStateInputArgument | AgentAIStateInputTemplate;
@@ -1772,11 +1771,7 @@ export interface AgentMemory {
     sessionMemory: AgentSessionMemory;
 }
 export interface AgentSessionMemory {
-    memoryLocations: MemoryLocations;
-}
-export interface AgentSessionMemoryLocation {
-    key: string;
-    shape: string;
+    shape?: string;
 }
 /**
  * Configuration for the start of an agent's execution
@@ -1920,17 +1915,17 @@ export interface AgentChatHistoryClear {
 }
 export interface AgentStateSessionMemoryLocationContent {
     type: 'content';
-    memoryKey: string;
+    memoryPath?: string;
     path?: string;
 }
 export interface AgentStateSessionMemoryLocationArtifact {
     type: 'artifactValue';
-    memoryKey: string;
+    memoryPath?: string;
     artifactPath?: string;
 }
 export interface AgentStateSessionMemoryLocationStaticValue {
     type: 'staticValue';
-    memoryKey: string;
+    memoryPath?: string;
     value?: string;
 }
 export interface AgentGuard {

package/dist/project-schema/v3.48.0.d.ts

@@ -265,7 +265,6 @@ export type ShapeSchemaV3_48_0 = ShapeSchemaAllOfV3_48_0 | ShapeSchemaExtendsV3_
 export type AgentAPIV3_48_0 = AgentAPIChatV3_48_0 | AgentAPIGenerateV3_48_0;
 export type InteractionTimeoutMillisecondsV3_48_0 = number;
 export type InteractionTimeoutMillisecondsV3_48_01 = number;
-export type MemoryLocationsV3_48_0 = AgentSessionMemoryLocationV3_48_0[];
 export type VariablesV3_48_0 = AgentVariableV3_48_0[];
 export type AgentExecutionV3_48_0 = AgentExecutionGraphQLV3_48_0 | AgentExecutionGenerateV3_48_0 | AgentExecutionChatV3_48_0;
 export type AgentAIStateInputV3_48_0 = AgentAIStateInputArgumentV3_48_0 | AgentAIStateInputTemplateV3_48_0;
@@ -1620,11 +1619,7 @@ export interface AgentMemoryV3_48_0 {
     sessionMemory: AgentSessionMemoryV3_48_0;
 }
 export interface AgentSessionMemoryV3_48_0 {
-    memoryLocations: MemoryLocationsV3_48_0;
-}
-export interface AgentSessionMemoryLocationV3_48_0 {
-    key: string;
-    shape: string;
+    shape?: string;
 }
 /**
  * Configuration for the start of an agent's execution
@@ -1768,17 +1763,17 @@ export interface AgentChatHistoryClearV3_48_0 {
 }
 export interface AgentStateSessionMemoryLocationContentV3_48_0 {
     type: 'content';
-    memoryKey: string;
+    memoryPath?: string;
     path?: string;
 }
 export interface AgentStateSessionMemoryLocationArtifactV3_48_0 {
     type: 'artifactValue';
-    memoryKey: string;
+    memoryPath?: string;
     artifactPath?: string;
 }
 export interface AgentStateSessionMemoryLocationStaticValueV3_48_0 {
     type: 'staticValue';
-    memoryKey: string;
+    memoryPath?: string;
     value?: string;
 }
 export interface AgentGuardV3_48_0 {

package/dist/project-schema/v3.49.0.d.ts

@@ -265,7 +265,6 @@ export type ShapeSchemaV3_49_0 = ShapeSchemaAllOfV3_49_0 | ShapeSchemaExtendsV3_
 export type AgentAPIV3_49_0 = AgentAPIChatV3_49_0 | AgentAPIGenerateV3_49_0;
 export type InteractionTimeoutMillisecondsV3_49_0 = number;
 export type InteractionTimeoutMillisecondsV3_49_01 = number;
-export type MemoryLocationsV3_49_0 = AgentSessionMemoryLocationV3_49_0[];
 export type VariablesV3_49_0 = AgentVariableV3_49_0[];
 export type AgentExecutionV3_49_0 = AgentExecutionGraphQLV3_49_0 | AgentExecutionGenerateV3_49_0 | AgentExecutionChatV3_49_0;
 export type AgentAIStateInputV3_49_0 = AgentAIStateInputArgumentV3_49_0 | AgentAIStateInputTemplateV3_49_0;
@@ -1620,11 +1619,7 @@ export interface AgentMemoryV3_49_0 {
     sessionMemory: AgentSessionMemoryV3_49_0;
 }
 export interface AgentSessionMemoryV3_49_0 {
-    memoryLocations: MemoryLocationsV3_49_0;
-}
-export interface AgentSessionMemoryLocationV3_49_0 {
-    key: string;
-    shape: string;
+    shape?: string;
 }
 /**
  * Configuration for the start of an agent's execution
@@ -1768,17 +1763,17 @@ export interface AgentChatHistoryClearV3_49_0 {
 }
 export interface AgentStateSessionMemoryLocationContentV3_49_0 {
     type: 'content';
-    memoryKey: string;
+    memoryPath?: string;
     path?: string;
 }
 export interface AgentStateSessionMemoryLocationArtifactV3_49_0 {
     type: 'artifactValue';
-    memoryKey: string;
+    memoryPath?: string;
     artifactPath?: string;
 }
 export interface AgentStateSessionMemoryLocationStaticValueV3_49_0 {
     type: 'staticValue';
-    memoryKey: string;
+    memoryPath?: string;
     value?: string;
 }
 export interface AgentGuardV3_49_0 {

package/dist/project-schema/v3.50.0.d.ts

@@ -277,7 +277,6 @@ export type ShapeSchemaV3_50_0 = ShapeSchemaAllOfV3_50_0 | ShapeSchemaExtendsV3_
 export type AgentAPIV3_50_0 = AgentAPIChatV3_50_0 | AgentAPIGenerateV3_50_0;
 export type InteractionTimeoutMillisecondsV3_50_0 = number;
 export type InteractionTimeoutMillisecondsV3_50_01 = number;
-export type MemoryLocationsV3_50_0 = AgentSessionMemoryLocationV3_50_0[];
 export type VariablesV3_50_0 = AgentVariableV3_50_0[];
 export type AgentExecutionV3_50_0 = AgentExecutionGraphQLV3_50_0 | AgentExecutionGenerateV3_50_0 | AgentExecutionChatV3_50_0;
 export type AgentAIStateInputV3_50_0 = AgentAIStateInputArgumentV3_50_0 | AgentAIStateInputTemplateV3_50_0;
@@ -1672,11 +1671,7 @@ export interface AgentMemoryV3_50_0 {
     sessionMemory: AgentSessionMemoryV3_50_0;
 }
 export interface AgentSessionMemoryV3_50_0 {
-    memoryLocations: MemoryLocationsV3_50_0;
-}
-export interface AgentSessionMemoryLocationV3_50_0 {
-    key: string;
-    shape: string;
+    shape?: string;
 }
 /**
  * Configuration for the start of an agent's execution
@@ -1820,17 +1815,17 @@ export interface AgentChatHistoryClearV3_50_0 {
 }
 export interface AgentStateSessionMemoryLocationContentV3_50_0 {
     type: 'content';
-    memoryKey: string;
+    memoryPath?: string;
     path?: string;
 }
 export interface AgentStateSessionMemoryLocationArtifactV3_50_0 {
     type: 'artifactValue';
-    memoryKey: string;
+    memoryPath?: string;
     artifactPath?: string;
 }
 export interface AgentStateSessionMemoryLocationStaticValueV3_50_0 {
     type: 'staticValue';
-    memoryKey: string;
+    memoryPath?: string;
     value?: string;
 }
 export interface AgentGuardV3_50_0 {

package/dist/project-schema/v3.51.0.d.ts

@@ -277,7 +277,6 @@ export type ShapeSchemaV3_51_0 = ShapeSchemaAllOfV3_51_0 | ShapeSchemaExtendsV3_
 export type AgentAPIV3_51_0 = AgentAPIChatV3_51_0 | AgentAPIGenerateV3_51_0;
 export type InteractionTimeoutMillisecondsV3_51_0 = number;
 export type InteractionTimeoutMillisecondsV3_51_01 = number;
-export type MemoryLocationsV3_51_0 = AgentSessionMemoryLocationV3_51_0[];
 export type VariablesV3_51_0 = AgentVariableV3_51_0[];
 export type AgentExecutionV3_51_0 = AgentExecutionGraphQLV3_51_0 | AgentExecutionGenerateV3_51_0 | AgentExecutionChatV3_51_0;
 export type AgentAIStateInputV3_51_0 = AgentAIStateInputArgumentV3_51_0 | AgentAIStateInputTemplateV3_51_0;
@@ -1672,11 +1671,7 @@ export interface AgentMemoryV3_51_0 {
     sessionMemory: AgentSessionMemoryV3_51_0;
 }
 export interface AgentSessionMemoryV3_51_0 {
-    memoryLocations: MemoryLocationsV3_51_0;
-}
-export interface AgentSessionMemoryLocationV3_51_0 {
-    key: string;
-    shape: string;
+    shape?: string;
 }
 /**
  * Configuration for the start of an agent's execution
@@ -1820,17 +1815,17 @@ export interface AgentChatHistoryClearV3_51_0 {
 }
 export interface AgentStateSessionMemoryLocationContentV3_51_0 {
     type: 'content';
-    memoryKey: string;
+    memoryPath?: string;
     path?: string;
 }
 export interface AgentStateSessionMemoryLocationArtifactV3_51_0 {
     type: 'artifactValue';
-    memoryKey: string;
+    memoryPath?: string;
     artifactPath?: string;
 }
 export interface AgentStateSessionMemoryLocationStaticValueV3_51_0 {
     type: 'staticValue';
-    memoryKey: string;
+    memoryPath?: string;
     value?: string;
 }
 export interface AgentGuardV3_51_0 {

package/dist/project-schema/v3.52.0.d.ts

@@ -277,7 +277,6 @@ export type ShapeSchemaV3_52_0 = ShapeSchemaAllOfV3_52_0 | ShapeSchemaExtendsV3_
 export type AgentAPIV3_52_0 = AgentAPIChatV3_52_0 | AgentAPIGenerateV3_52_0;
 export type InteractionTimeoutMillisecondsV3_52_0 = number;
 export type InteractionTimeoutMillisecondsV3_52_01 = number;
-export type MemoryLocationsV3_52_0 = AgentSessionMemoryLocationV3_52_0[];
 export type VariablesV3_52_0 = AgentVariableV3_52_0[];
 export type AgentExecutionV3_52_0 = AgentExecutionGraphQLV3_52_0 | AgentExecutionGenerateV3_52_0 | AgentExecutionChatV3_52_0;
 export type AgentAIStateInputV3_52_0 = AgentAIStateInputArgumentV3_52_0 | AgentAIStateInputTemplateV3_52_0;
@@ -1672,11 +1671,7 @@ export interface AgentMemoryV3_52_0 {
     sessionMemory: AgentSessionMemoryV3_52_0;
 }
 export interface AgentSessionMemoryV3_52_0 {
-    memoryLocations: MemoryLocationsV3_52_0;
-}
-export interface AgentSessionMemoryLocationV3_52_0 {
-    key: string;
-    shape: string;
+    shape?: string;
 }
 /**
  * Configuration for the start of an agent's execution
@@ -1820,17 +1815,17 @@ export interface AgentChatHistoryClearV3_52_0 {
 }
 export interface AgentStateSessionMemoryLocationContentV3_52_0 {
     type: 'content';
-    memoryKey: string;
+    memoryPath?: string;
     path?: string;
 }
 export interface AgentStateSessionMemoryLocationArtifactV3_52_0 {
     type: 'artifactValue';
-    memoryKey: string;
+    memoryPath?: string;
     artifactPath?: string;
 }
 export interface AgentStateSessionMemoryLocationStaticValueV3_52_0 {
     type: 'staticValue';
-    memoryKey: string;
+    memoryPath?: string;
     value?: string;
 }
 export interface AgentGuardV3_52_0 {

package/dist/project-schema/v3.53.0.d.ts

@@ -286,7 +286,6 @@ export type ShapeSchemaV3_53_0 = ShapeSchemaAllOfV3_53_0 | ShapeSchemaExtendsV3_
 export type AgentAPIV3_53_0 = AgentAPIChatV3_53_0 | AgentAPIGenerateV3_53_0;
 export type InteractionTimeoutMillisecondsV3_53_0 = number;
 export type InteractionTimeoutMillisecondsV3_53_01 = number;
-export type MemoryLocationsV3_53_0 = AgentSessionMemoryLocationV3_53_0[];
 export type VariablesV3_53_0 = AgentVariableV3_53_0[];
 export type AgentExecutionV3_53_0 = AgentExecutionGraphQLV3_53_0 | AgentExecutionGenerateV3_53_0 | AgentExecutionChatV3_53_0;
 export type AgentAIStateInputV3_53_0 = AgentAIStateInputArgumentV3_53_0 | AgentAIStateInputTemplateV3_53_0;
@@ -1753,11 +1752,7 @@ export interface AgentMemoryV3_53_0 {
     sessionMemory: AgentSessionMemoryV3_53_0;
 }
 export interface AgentSessionMemoryV3_53_0 {
-    memoryLocations: MemoryLocationsV3_53_0;
-}
-export interface AgentSessionMemoryLocationV3_53_0 {
-    key: string;
-    shape: string;
+    shape?: string;
 }
 /**
  * Configuration for the start of an agent's execution
@@ -1901,17 +1896,17 @@ export interface AgentChatHistoryClearV3_53_0 {
 }
 export interface AgentStateSessionMemoryLocationContentV3_53_0 {
     type: 'content';
-    memoryKey: string;
+    memoryPath?: string;
     path?: string;
 }
 export interface AgentStateSessionMemoryLocationArtifactV3_53_0 {
     type: 'artifactValue';
-    memoryKey: string;
+    memoryPath?: string;
     artifactPath?: string;
 }
 export interface AgentStateSessionMemoryLocationStaticValueV3_53_0 {
     type: 'staticValue';
-    memoryKey: string;
+    memoryPath?: string;
     value?: string;
 }
 export interface AgentGuardV3_53_0 {

package/dist/project-schema/v3.54.0.d.ts

@@ -290,7 +290,6 @@ export type ShapeSchemaV3_54_0 = ShapeSchemaAllOfV3_54_0 | ShapeSchemaExtendsV3_
 export type AgentAPIV3_54_0 = AgentAPIChatV3_54_0 | AgentAPIGenerateV3_54_0;
 export type InteractionTimeoutMillisecondsV3_54_0 = number;
 export type InteractionTimeoutMillisecondsV3_54_01 = number;
-export type MemoryLocationsV3_54_0 = AgentSessionMemoryLocationV3_54_0[];
 export type VariablesV3_54_0 = AgentVariableV3_54_0[];
 export type AgentExecutionV3_54_0 = AgentExecutionGraphQLV3_54_0 | AgentExecutionGenerateV3_54_0 | AgentExecutionChatV3_54_0;
 export type AgentAIStateInputV3_54_0 = AgentAIStateInputArgumentV3_54_0 | AgentAIStateInputTemplateV3_54_0;
@@ -1772,11 +1771,7 @@ export interface AgentMemoryV3_54_0 {
     sessionMemory: AgentSessionMemoryV3_54_0;
 }
 export interface AgentSessionMemoryV3_54_0 {
-    memoryLocations: MemoryLocationsV3_54_0;
-}
-export interface AgentSessionMemoryLocationV3_54_0 {
-    key: string;
-    shape: string;
+    shape?: string;
 }
 /**
  * Configuration for the start of an agent's execution
@@ -1920,17 +1915,17 @@ export interface AgentChatHistoryClearV3_54_0 {
 }
 export interface AgentStateSessionMemoryLocationContentV3_54_0 {
     type: 'content';
-    memoryKey: string;
+    memoryPath?: string;
     path?: string;
 }
 export interface AgentStateSessionMemoryLocationArtifactV3_54_0 {
     type: 'artifactValue';
-    memoryKey: string;
+    memoryPath?: string;
     artifactPath?: string;
 }
 export interface AgentStateSessionMemoryLocationStaticValueV3_54_0 {
     type: 'staticValue';
-    memoryKey: string;
+    memoryPath?: string;
     value?: string;
 }
 export interface AgentGuardV3_54_0 {

package/dist/refs.d.ts

@@ -167,7 +167,7 @@ export declare function anyRefToShapeSchemaPath(context: ServicesContext, refSch
 /**
  * Helper fn to omit `ref` props from the target schema, and then extend it with the source schema.
  */
-export declare const omitRefAndExtend: (targetSchema: PropertySchema, sourceSchema: PropertySchema) => import("lodash").Omit<PropertySchema, "@ref" | "$ref"> & PropertySchema;
+export declare const omitRefAndExtend: (targetSchema: PropertySchema, sourceSchema: PropertySchema) => import("lodash").Omit<PropertySchema, "$ref" | "@ref"> & PropertySchema;
 /**
  * If there is a $ref, this will insert a property schema from the top level
  * in place of the reference.

package/dist/resolvers/ai/{rag-query.d.ts → embedding-search.d.ts}

@@ -1,5 +1,5 @@
 import type { FromSchema } from 'json-schema-to-ts';
-export declare const ragQueryArgsSchema: {
+export declare const embeddingSearchArgsSchema: {
     readonly type: "object";
     readonly properties: {
         readonly text: {
@@ -33,17 +33,22 @@ export declare const ragQueryArgsSchema: {
                 readonly additionalProperties: false;
             };
         };
+        readonly threshold: {
+            readonly type: "number";
+            readonly description: "The minimum score a document must have to be returned.";
+        };
     };
     readonly required: ["text"];
     readonly additionalProperties: false;
 };
-export type RagQueryArgs = FromSchema<typeof ragQueryArgsSchema>;
-export declare const validateRagQueryArgs: import("ajv").ValidateFunction<{
+export type EmbeddingSearchArgs = FromSchema<typeof embeddingSearchArgsSchema>;
+export declare const validateEmbeddingSearchArgs: import("ajv").ValidateFunction<{
     sort?: {
         field: string;
         order: "asc" | "desc";
     }[] | undefined;
     size?: number | undefined;
     from?: number | undefined;
+    threshold?: number | undefined;
     text: string;
 }>;

package/dist/resolvers/ai/{rag-query.js → embedding-search.js}

@@ -1,5 +1,5 @@
 import { createTypedValidator } from '@takeshape/json-schema';
-export const ragQueryArgsSchema = {
+export const embeddingSearchArgsSchema = {
     type: 'object',
     properties: {
         text: {
@@ -32,9 +32,15 @@ export const ragQueryArgsSchema = {
                 required: ['field', 'order'],
                 additionalProperties: false
             }
+        },
+        threshold: {
+            type: 'number',
+            description: 'The minimum score a document must have to be returned.'
         }
     },
     required: ['text'],
     additionalProperties: false
 };
-export const validateRagQueryArgs = createTypedValidator(ragQueryArgsSchema, { useDefaults: true });
+export const validateEmbeddingSearchArgs = createTypedValidator(embeddingSearchArgsSchema, {
+    useDefaults: true
+});

package/dist/schema-util.d.ts

@@ -46,6 +46,72 @@ export declare function getArgsReference(projectSchema: ServicesContext, argsSch
 export declare function getArgsReferenceWithPath(context: ServicesContext, argsSchema: SchemaWithArgs, schemaPath?: SchemaPath): Maybe<RefItemWithPath>;
 export declare function getArgsShapeSchema(context: ServicesShapesContext, argsSchema: SchemaWithArgs): Maybe<ShapeSchema>;
 export declare function shapeToPropertySchema(projectSchema: ProjectSchemaJSON, shape: ShapeJSON): PropertySchema;
+/**
+ * Currently this is not 100% valid JSON schema, but its output can be used with createSchemaValidator
+ */
+export declare function shapeToJsonSchema(projectSchema: ProjectSchemaJSON, shape: ShapeJSON): {
+    definitions: Record<string, PropertySchema>;
+    $ref?: string;
+    title?: string;
+    description?: string;
+    default?: any;
+    readOnly?: boolean;
+    multipleOf?: number;
+    maximum?: number;
+    exclusiveMaximum?: number;
+    minimum?: number;
+    exclusiveMinimum?: number;
+    maxLength?: import("./project-schema/latest.ts").NonNegativeInteger;
+    minLength?: import("./project-schema/latest.ts").NonNegativeIntegerDefault0;
+    pattern?: string;
+    additionalItems?: PropertySchema;
+    items?: PropertySchema;
+    maxItems?: import("./project-schema/latest.ts").NonNegativeInteger;
+    minItems?: import("./project-schema/latest.ts").NonNegativeIntegerDefault0;
+    uniqueItems?: boolean;
+    contains?: PropertySchema;
+    maxProperties?: import("./project-schema/latest.ts").NonNegativeInteger;
+    minProperties?: import("./project-schema/latest.ts").NonNegativeIntegerDefault0;
+    required?: import("./project-schema/latest.ts").StringArray;
+    additionalProperties?: PropertySchema | false;
+    properties?: {
+        [k: string]: PropertySchema;
+    };
+    const?: any;
+    enum?: any[];
+    type?: import("./project-schema/latest.ts").SimpleTypes | import("./project-schema/latest.ts").SimpleTypes[];
+    format?: string;
+    contentMediaType?: string;
+    contentEncoding?: string;
+    allOf?: import("./project-schema/latest.ts").SchemaArray;
+    discriminator?: {
+        propertyName?: string;
+        [k: string]: any;
+    };
+    oneOf?: import("./project-schema/latest.ts").SchemaArray;
+    extends?: import("./project-schema/latest.ts").ObjectOrRefArray;
+    '@backreference'?: import("./project-schema/latest.ts").BackReference;
+    '@input'?: PropertySchema;
+    '@output'?: PropertySchema;
+    '@syncLocaleStructure'?: boolean;
+    '@sensitive'?: boolean;
+    '@draftjs'?: boolean;
+    '@l10n'?: boolean;
+    '@key'?: string;
+    '@workflow'?: string;
+    '@mapping'?: FieldMapping;
+    '@tag'?: string;
+    '@tags'?: string[];
+    '@deprecationReason'?: string;
+    '@args'?: import("./project-schema/latest.ts").Args;
+    '@resolver'?: import("./project-schema/latest.ts").Resolver;
+    '@dependencies'?: import("./project-schema/latest.ts").Dependencies;
+    '@ref'?: import("./project-schema/latest.ts").Ref;
+    '@derivedFrom'?: string;
+    '@indexed'?: import("./project-schema/latest.ts").Indexed;
+    '@graphql'?: import("./project-schema/latest.ts").GraphQLConfig;
+    $id: string;
+};
 export declare function projectSchemaToJSONSchema(projectSchema: ProjectSchemaJSON): {
     definitions: Record<string, PropertySchema>;
 };

package/dist/schema-util.js

@@ -471,7 +471,7 @@ export function getArgsShapeSchema(context, argsSchema) {
         }
     }
 }
-const createShapeToPropertySchema = (projectSchema, definitions, definitionNames, { useShapeIdDiscriminator }) => {
+const createShapeToPropertySchema = (projectSchema, definitions, definitionNames, { useShapeIdDiscriminator, additionalProperties }) => {
     const rewriteRefs = (shape) => {
         const handleSchema = (propSchema) => {
             // Unwrap array schema
@@ -508,11 +508,16 @@ const createShapeToPropertySchema = (projectSchema, definitions, definitionNames
                     if (!definitionNames.has(shapeName)) {
                         definitionNames.add(shapeName);
                         const rewritten = rewriteRefs(shape);
-                        if (useShapeIdDiscriminator && isObjectSchema(rewritten)) {
-                            rewritten.properties._shapeId = {
-                                type: 'string',
-                                enum: [shape.id]
-                            };
+                        if (isObjectSchema(rewritten)) {
+                            if (useShapeIdDiscriminator) {
+                                rewritten.properties._shapeId = {
+                                    type: 'string',
+                                    enum: [shape.id]
+                                };
+                            }
+                            if (additionalProperties !== undefined) {
+                                rewritten.additionalProperties = additionalProperties;
+                            }
                         }
                         definitions[shapeName] = rewritten;
                     }
@@ -534,6 +539,9 @@ const createShapeToPropertySchema = (projectSchema, definitions, definitionNames
         for (const schema of Object.values(shapeSchemaCopy.properties)) {
             handleSchema(schema);
         }
+        if (additionalProperties !== undefined) {
+            shapeSchemaCopy.additionalProperties = additionalProperties;
+        }
         return shapeSchemaCopy;
     };
     return rewriteRefs;
@@ -549,6 +557,22 @@ export function shapeToPropertySchema(projectSchema, shape) {
         definitions
     };
 }
+/**
+ * Currently this is not 100% valid JSON schema, but its output can be used with createSchemaValidator
+ */
+export function shapeToJsonSchema(projectSchema, shape) {
+    const definitions = {};
+    const definitionNames = new Set();
+    const convert = createShapeToPropertySchema(projectSchema, definitions, definitionNames, {
+        useShapeIdDiscriminator: false,
+        additionalProperties: false
+    });
+    return {
+        $id: shape.id,
+        ...convert(shape),
+        definitions
+    };
+}
 export function projectSchemaToJSONSchema(projectSchema) {
     const definitions = {};
     const definitionNames = new Set();

package/dist/schemas/project-schema/experimental.json

@@ -106,29 +106,11 @@
       "title": "Agent Session Memory",
       "type": "object",
       "properties": {
-        "memoryLocations": {
-          "title": "Memory Locations",
-          "type": "array",
-          "items": {
-            "$ref": "#/definitions/agentSessionMemoryLocation"
-          }
-        }
-      },
-      "required": ["memoryLocations"],
-      "additionalProperties": false
-    },
-    "agentSessionMemoryLocation": {
-      "title": "Agent Session Memory Location",
-      "type": "object",
-      "properties": {
-        "key": {
-          "type": "string"
-        },
         "shape": {
           "type": "string"
         }
       },
-      "required": ["key", "shape"],
+      "required": [],
       "additionalProperties": false
     },
     "agentVariable": {
@@ -309,14 +291,14 @@
         "type": {
           "enum": ["staticValue"]
         },
-        "memoryKey": {
+        "memoryPath": {
           "type": "string"
         },
         "value": {
           "type": "string"
         }
       },
-      "required": ["type", "memoryKey"],
+      "required": ["type"],
       "additionalProperties": false
     },
     "agentStateSessionMemoryLocationArtifact": {
@@ -326,14 +308,14 @@
         "type": {
           "enum": ["artifactValue"]
         },
-        "memoryKey": {
+        "memoryPath": {
           "type": "string"
         },
         "artifactPath": {
           "type": "string"
         }
       },
-      "required": ["type", "memoryKey"],
+      "required": ["type"],
       "additionalProperties": false
     },
     "agentStateSessionMemoryLocationContent": {
@@ -343,14 +325,14 @@
         "type": {
           "enum": ["content"]
         },
-        "memoryKey": {
+        "memoryPath": {
           "type": "string"
         },
         "path": {
           "type": "string"
         }
       },
-      "required": ["type", "memoryKey"],
+      "required": ["type"],
       "additionalProperties": false
     },
     "agentAiStateInputArg": {

package/dist/shape-paths.d.ts

@@ -0,0 +1,7 @@
+import type { ProjectSchemaJSON, ShapeJSON } from './project-schema/latest.ts';
+export interface GetShapePathsArgs {
+    includeRoot?: boolean;
+    prefix?: string;
+    depthLimit?: number;
+}
+export declare function getShapePaths(projectSchema: ProjectSchemaJSON, shape: ShapeJSON, args?: GetShapePathsArgs): string[];

package/dist/shape-paths.js

@@ -0,0 +1,28 @@
+import { dereferenceSchema } from "./refs.js";
+import { isArraySchema, isObjectSchema } from "./types/utils.js";
+export function getShapePaths(projectSchema, shape, args = {}) {
+    const { includeRoot = false, prefix = '', depthLimit = 2 } = args;
+    const options = [];
+    const visit = (propertySchema, path, depth) => {
+        if (depth > depthLimit) {
+            return;
+        }
+        if (depth !== 0 || includeRoot) {
+            options.push(path.join('.'));
+        }
+        const schema = dereferenceSchema(projectSchema, propertySchema);
+        if (isArraySchema(schema)) {
+            if (isObjectSchema(schema.items)) {
+                visit(schema.items, path, depth + 1);
+            }
+        }
+        else if (isObjectSchema(schema)) {
+            for (const [name, propSchema] of Object.entries(schema.properties)) {
+                const newPath = [...path, propSchema.type === 'array' ? `${name}[0]` : name];
+                visit(propSchema, newPath, depth + 1);
+            }
+        }
+    };
+    visit(shape.schema, prefix ? [prefix] : [], 0);
+    return options;
+}

package/examples/latest/agent-schema.json

@@ -204,9 +204,7 @@
           ]
         },
         "memory": {
-          "sessionMemory": {
-            "memoryLocations": []
-          }
+          "sessionMemory": {}
         },
         "states": {
           "bbb": {

package/examples/source/agent-schema.json

@@ -186,9 +186,7 @@
           ]
         },
         "memory": {
-          "sessionMemory": {
-            "memoryLocations": []
-          }
+          "sessionMemory": {}
         },
         "states": {
           "bbb": {

package/package.json

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