@fluidframework/tree-agent 2.70.0 → 2.71.0

package/CHANGELOG.md

@@ -1,5 +1,9 @@
 # @fluidframework/tree-agent
 
+## 2.71.0
+
+Dependency updates only.
+
 ## 2.70.0
 
 ### Minor Changes

package/dist/typeGeneration.js

@@ -136,15 +136,15 @@ function getOrCreateType(definitionMap, definition, objectCache, bindableSchemas
                     .describe(simpleNodeSchema.metadata?.description ?? ""));
             }
             case internal_3.NodeKind.Map: {
-                const zodType = zod_1.z.map(zod_1.z.string(), getTypeForAllowedTypes(definitionMap, simpleNodeSchema.allowedTypesIdentifiers, objectCache, bindableSchemas));
+                const zodType = zod_1.z.map(zod_1.z.string(), getTypeForAllowedTypes(definitionMap, new Set(simpleNodeSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas));
                 return (type = addBindingIntersectionIfNeeded("map", zodType, definition, simpleNodeSchema, bindableSchemas));
             }
             case internal_3.NodeKind.Record: {
-                const zodType = zod_1.z.record(getTypeForAllowedTypes(definitionMap, simpleNodeSchema.allowedTypesIdentifiers, objectCache, bindableSchemas));
+                const zodType = zod_1.z.record(getTypeForAllowedTypes(definitionMap, new Set(simpleNodeSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas));
                 return (type = addBindingIntersectionIfNeeded("record", zodType, definition, simpleNodeSchema, bindableSchemas));
             }
             case internal_3.NodeKind.Array: {
-                const zodType = zod_1.z.array(getTypeForAllowedTypes(definitionMap, simpleNodeSchema.allowedTypesIdentifiers, objectCache, bindableSchemas));
+                const zodType = zod_1.z.array(getTypeForAllowedTypes(definitionMap, new Set(simpleNodeSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas));
                 return (type = addBindingIntersectionIfNeeded("array", zodType, definition, simpleNodeSchema, bindableSchemas));
             }
             case internal_3.NodeKind.Leaf: {
@@ -183,7 +183,7 @@ function getOrCreateTypeForField(definitionMap, fieldSchema, objectCache, bindab
             throw new internal_2.UsageError(`The ${utils_js_1.llmDefault.description} property is only permitted on optional fields.`);
         }
     }
-    const field = getTypeForAllowedTypes(definitionMap, fieldSchema.allowedTypesIdentifiers, objectCache, bindableSchemas).describe(getDefault === undefined
+    const field = getTypeForAllowedTypes(definitionMap, new Set(fieldSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas).describe(getDefault === undefined
         ? (fieldSchema.metadata?.description ?? "")
         : "Do not populate this field. It will be automatically supplied by the system after insertion.");
     switch (fieldSchema.kind) {

package/dist/typeGeneration.js.map

@@ -1 +1 @@
-{"version":3,"file":"typeGeneration.js","sourceRoot":"","sources":["../src/typeGeneration.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,kEAA8E;AAC9E,uEAAsE;AACtE,4DAUuC;AAQvC,6BAAyC;AAGzC,yDAAwE;AACxE,yCAQoB;AAEpB;;;;;;;;;GASG;AAEH;;GAEG;AACH,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAGlC,CAAC;AAEJ;;GAEG;AACH,SAAgB,0BAA0B,CACzC,UAA+B,EAC/B,MAAwB;IAIxB,OAAO,IAAA,sBAAW,EAAC,iBAAiB,EAAE,MAAM,EAAE,GAAG,EAAE;QAClD,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC7C,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAkB,CAAC;QACpD,IAAA,0BAAe,EAAC,UAAU,EAAE;YAC3B,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;gBACX,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAClB,IACC,CAAC,YAAY,2BAAgB;oBAC7B,CAAC,YAAY,wBAAa;oBAC1B,CAAC,YAAY,0BAAe;oBAC5B,CAAC,YAAY,2BAAgB,EAC5B,CAAC;oBACF,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBACzB,MAAM,cAAc,GAAG,IAAA,oCAAiB,EAAC,CAAC,CAAC,CAAC;oBAC5C,KAAK,MAAM,CAAC,IAAI,cAAc,CAAC,eAAe,EAAE,CAAC;wBAChD,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;wBAClB,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBAC1B,CAAC;gBACF,CAAC;YACF,CAAC;SACD,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,CAAC,GAAG,iBAAiB,CAAC,MAAM,EAAE,CAAqB,CAAC;QACxE,MAAM,eAAe,GAAG,IAAI,GAAG,CAC9B,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CACpE,CAAC;QACF,OAAO,iBAAiB,CACvB,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,0BAAe,EAAC,CAAC,CAAC,CAAC,EACvD,IAAI,GAAG,EAAE,EACT,eAAe,CACf,CAAC;IACH,CAAC,CAAC,CAAC;AACJ,CAAC;AArCD,gEAqCC;AAED;;;;;;;GAOG;AACH,SAAS,iBAAiB,CACzB,OAAmC,EACnC,WAAoD,EACpD,eAA4C;IAI5C,MAAM,WAAW,GAA+B,EAAE,CAAC;IACnD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC9B,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;YAC9C,oIAAoI;YACpI,WAAW,CAAC,IAAI,CAAC,GAAG,eAAe,CAClC,MAAM,CAAC,WAAW,EAClB,IAAI,EACJ,WAAW,EACX,eAAe,CACf,CAAC;QACH,CAAC;IACF,CAAC;IAED,OAAO;QACN,WAAW;KACX,CAAC;AACH,CAAC;AAED,SAAS,0BAA0B,CAAC,cAA8B;IAIjE,MAAM,WAAW,GAA2B,EAAE,CAAC;IAC/C,MAAM,OAAO,GAAG,IAAA,oCAAiB,EAAC,cAAc,CAAC,CAAC;IAClD,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,OAAC,CAAC,UAAU,CAAC,kCAAe,CAAC,CAAC;QAClD,0GAA0G;QACzG,WAAmB,CAAC,MAAM,GAAG,MAAM,CAAC;QACrC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,CAAC;AAC3E,CAAC;AAED,SAAS,eAAe,CACvB,UAAkB,EAClB,eAA4C;IAE5C,MAAM,cAAc,GAAG,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAA,eAAI,EAAC,oBAAoB,CAAC,CAAC;IACrF,OAAO,0BAA0B,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC;AAC3D,CAAC;AAED,SAAS,8BAA8B,CACtC,UAAsC,EACtC,YAAwB,EACxB,UAAkB,EAClB,gBAAkC,EAClC,eAA4C;IAE5C,IAAI,OAAO,GAAG,YAAY,CAAC;IAC3B,IAAI,WAAW,GAAG,gBAAgB,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC;IAC/D,MAAM,YAAY,GAAG,eAAe,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;IAClE,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAiC,EAAE,CAAC;QACjD,KAAK,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,YAAY,EAAE,CAAC;YAChD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;gBACjC,MAAM,IAAI,qBAAU,CACnB,UAAU,IAAI,oDAAoD,UAAU,EAAE,CAC9E,CAAC;YACH,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC;QAC7B,CAAC;QACD,OAAO,GAAG,OAAC,CAAC,YAAY,CAAC,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QACrD,MAAM,UAAU,GAAG,cAAc,UAAU,kDAAkD,CAAC;QAC9F,WAAW,GAAG,WAAW,KAAK,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,WAAW,MAAM,UAAU,EAAE,CAAC;IAClF,CAAC;IACD,OAAO,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,eAAe,CACvB,aAAoD,EACpD,UAAkB,EAClB,WAAoD,EACpD,eAA4C;IAE5C,MAAM,gBAAgB,GAAG,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAA,eAAI,EAAC,uBAAuB,CAAC,CAAC;IACxF,OAAO,IAAA,sBAAW,EAAC,WAAW,EAAE,gBAAgB,EAAE,GAAG,EAAE;QACtD,8HAA8H;QAC9H,IAAI,IAA4B,CAAC;QACjC,WAAW,CAAC,GAAG,CACd,gBAAgB,EAChB,OAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,IAAI,IAAA,eAAI,EAAC,qCAAqC,CAAC,CAAC,CACjE,CAAC;QACF,QAAQ,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAC/B,KAAK,mBAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;gBACtB,mEAAmE;gBACnE,MAAM,UAAU,GAAiC,MAAM,CAAC,WAAW,CAClE,CAAC,GAAG,gBAAgB,CAAC,MAAM,CAAC;qBAC1B,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;oBACrB,OAAO;wBACN,GAAG;wBACH,uBAAuB,CAAC,aAAa,EAAE,KAAK,EAAE,WAAW,EAAE,eAAe,CAAC;qBAC3E,CAAC;gBACH,CAAC,CAAC;qBACD,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC,CAC5C,CAAC;gBAEF,8GAA8G;gBAC9G,KAAK,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,eAAe,CAAC,UAAU,EAAE,eAAe,CAAC,EAAE,CAAC;oBAChF,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;wBACpC,MAAM,IAAI,qBAAU,CACnB,UAAU,IAAI,oDAAoD,UAAU,EAAE,CAC9E,CAAC;oBACH,CAAC;oBACD,UAAU,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC;gBAChC,CAAC;gBAED,OAAO,CAAC,IAAI,GAAG,OAAC;qBACd,MAAM,CAAC,UAAU,CAAC;qBAClB,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3D,CAAC;YACD,KAAK,mBAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;gBACnB,MAAM,OAAO,GAAG,OAAC,CAAC,GAAG,CACpB,OAAC,CAAC,MAAM,EAAE,EACV,sBAAsB,CACrB,aAAa,EACb,gBAAgB,CAAC,uBAAuB,EACxC,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,KAAK,EACL,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,mBAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;gBACtB,MAAM,OAAO,GAAG,OAAC,CAAC,MAAM,CACvB,sBAAsB,CACrB,aAAa,EACb,gBAAgB,CAAC,uBAAuB,EACxC,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,QAAQ,EACR,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,mBAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;gBACrB,MAAM,OAAO,GAAG,OAAC,CAAC,KAAK,CACtB,sBAAsB,CACrB,aAAa,EACb,gBAAgB,CAAC,uBAAuB,EACxC,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,OAAO,EACP,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,mBAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpB,QAAQ,gBAAgB,CAAC,QAAQ,EAAE,CAAC;oBACnC,KAAK,sBAAW,CAAC,OAAO,CAAC,CAAC,CAAC;wBAC1B,OAAO,CAAC,IAAI,GAAG,OAAC,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC7B,CAAC;oBACD,KAAK,sBAAW,CAAC,MAAM,CAAC,CAAC,CAAC;wBACzB,OAAO,CAAC,IAAI,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC5B,CAAC;oBACD,KAAK,sBAAW,CAAC,MAAM,CAAC,CAAC,CAAC;wBACzB,OAAO,CAAC,IAAI,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC5B,CAAC;oBACD,KAAK,sBAAW,CAAC,IAAI,CAAC,CAAC,CAAC;wBACvB,OAAO,CAAC,IAAI,GAAG,OAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC1B,CAAC;oBACD,OAAO,CAAC,CAAC,CAAC;wBACT,MAAM,IAAI,KAAK,CAAC,yBAAyB,mBAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;oBAClF,CAAC;gBACF,CAAC;YACF,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACT,OAAO,IAAA,0BAAe,EAAC,gBAAgB,EAAE,mBAAmB,CAAC,CAAC;YAC/D,CAAC;QACF,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAC/B,aAAoD,EACpD,WAA8B,EAC9B,WAAoD,EACpD,eAA4C;IAE5C,MAAM,cAAc,GAAG,WAAW,CAAC,QAAQ,CAAC,MAEhC,CAAC;IACb,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC,qBAAU,CAAC,CAAC;IAChD,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC9B,IAAI,OAAO,UAAU,KAAK,UAAU,EAAE,CAAC;YACtC,MAAM,IAAI,qBAAU,CACnB,qBAAqB,qBAAU,CAAC,WAAW,uCAAuC,OAAO,UAAU,EAAE,CACrG,CAAC;QACH,CAAC;QAED,IAAI,WAAW,CAAC,IAAI,KAAK,oBAAS,CAAC,QAAQ,EAAE,CAAC;YAC7C,MAAM,IAAI,qBAAU,CACnB,OAAO,qBAAU,CAAC,WAAW,iDAAiD,CAC9E,CAAC;QACH,CAAC;IACF,CAAC;IAED,MAAM,KAAK,GAAG,sBAAsB,CACnC,aAAa,EACb,WAAW,CAAC,uBAAuB,EACnC,WAAW,EACX,eAAe,CACf,CAAC,QAAQ,CACT,UAAU,KAAK,SAAS;QACvB,CAAC,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC;QAC3C,CAAC,CAAC,8FAA8F,CACjG,CAAC;IAEF,QAAQ,WAAW,CAAC,IAAI,EAAE,CAAC;QAC1B,KAAK,oBAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzB,OAAO,KAAK,CAAC;QACd,CAAC;QACD,KAAK,oBAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzB,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;QACzB,CAAC;QACD,KAAK,oBAAS,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3B,OAAO,KAAK;iBACV,QAAQ,EAAE;iBACV,QAAQ,CACR,uGAAuG,CACvG,CAAC;QACJ,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,0BAA0B,mBAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1E,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,sBAAsB,CAC9B,aAAoD,EACpD,YAAiC,EACjC,WAAoD,EACpD,eAA4C;IAE5C,MAAM,MAAM,GAAG,IAAA,0BAAe,EAAC,YAAY,CAAC,CAAC;IAC7C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG;YACb,GAAG,IAAA,sBAAW,EAAC,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE;gBACrC,OAAO,eAAe,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,EAAE,eAAe,CAAC,CAAC;YAC3E,CAAC,CAAC;SACF,CAAC;QACF,IAAA,iBAAM,EAAC,IAAA,wBAAa,EAAC,KAAK,CAAC,EAAE,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACtE,OAAO,OAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;SAAM,CAAC;QACP,OAAO,eAAe,CAAC,aAAa,EAAE,MAAM,EAAE,WAAW,EAAE,eAAe,CAAC,CAAC;IAC7E,CAAC;AACF,CAAC","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\nimport { assert, unreachableCase } from \"@fluidframework/core-utils/internal\";\nimport { UsageError } from \"@fluidframework/telemetry-utils/internal\";\nimport {\n\tArrayNodeSchema,\n\tFieldKind,\n\tgetSimpleSchema,\n\tMapNodeSchema,\n\tNodeKind,\n\tObjectNodeSchema,\n\tRecordNodeSchema,\n\tValueSchema,\n\twalkFieldSchema,\n} from \"@fluidframework/tree/internal\";\nimport type {\n\tImplicitFieldSchema,\n\tSimpleFieldSchema,\n\tSimpleNodeSchema,\n\tSimpleTreeSchema,\n\tTreeNodeSchema,\n} from \"@fluidframework/tree/internal\";\nimport { z, type ZodTypeAny } from \"zod\";\n\nimport type { BindableSchema } from \"./methodBinding.js\";\nimport { FunctionWrapper, getExposedMethods } from \"./methodBinding.js\";\nimport {\n\tfail,\n\tgetOrCreate,\n\thasAtLeastTwo,\n\tllmDefault,\n\tmapIterable,\n\ttryGetSingleton,\n\ttype MapGetSet,\n} from \"./utils.js\";\n\n/**\n *\n * TODO: Add a prompt suggestion API!\n *\n * TODO: Handle rate limit errors.\n *\n * TODO: Pass descriptions from schema metadata to the generated TS types that we put in the prompt\n *\n * TODO make the Ids be \"Vector-2\" instead of \"Vector2\" (or else it gets weird when you have a type called \"Vector2\")\n */\n\n/**\n * Cache used to prevent repeatedly generating the same Zod validation objects for the same {@link SimpleTreeSchema} as generate propts for repeated calls to an LLM\n */\nconst promptSchemaCache = new WeakMap<\n\tSimpleTreeSchema,\n\tReturnType<typeof generateEditTypes>\n>();\n\n/**\n * TODO\n */\nexport function generateEditTypesForPrompt(\n\trootSchema: ImplicitFieldSchema,\n\tschema: SimpleTreeSchema,\n): {\n\tdomainTypes: Record<string, ZodTypeAny>;\n} {\n\treturn getOrCreate(promptSchemaCache, schema, () => {\n\t\tconst allSchemas = new Set<TreeNodeSchema>();\n\t\tconst objectTypeSchemas = new Set<TreeNodeSchema>();\n\t\twalkFieldSchema(rootSchema, {\n\t\t\tnode: (n) => {\n\t\t\t\tallSchemas.add(n);\n\t\t\t\tif (\n\t\t\t\t\tn instanceof ObjectNodeSchema ||\n\t\t\t\t\tn instanceof MapNodeSchema ||\n\t\t\t\t\tn instanceof ArrayNodeSchema ||\n\t\t\t\t\tn instanceof RecordNodeSchema\n\t\t\t\t) {\n\t\t\t\t\tobjectTypeSchemas.add(n);\n\t\t\t\t\tconst exposedMethods = getExposedMethods(n);\n\t\t\t\t\tfor (const t of exposedMethods.referencedTypes) {\n\t\t\t\t\t\tallSchemas.add(t);\n\t\t\t\t\t\tobjectTypeSchemas.add(t);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t});\n\t\tconst nodeSchemas = [...objectTypeSchemas.values()] as BindableSchema[];\n\t\tconst bindableSchemas = new Map<string, BindableSchema>(\n\t\t\tnodeSchemas.map((nodeSchema) => [nodeSchema.identifier, nodeSchema]),\n\t\t);\n\t\treturn generateEditTypes(\n\t\t\t[...allSchemas.values()].map((s) => getSimpleSchema(s)),\n\t\t\tnew Map(),\n\t\t\tbindableSchemas,\n\t\t);\n\t});\n}\n\n/**\n * Generates a set of ZOD validation objects for the various types of data that can be put into the provided {@link SimpleTreeSchema}\n * and then uses those sets to generate an all-encompassing ZOD object for each type of {@link TreeEdit} that can validate any of the types of data that can be put into the tree.\n *\n * @returns a Record of schema names to Zod validation objects, and the name of the root schema used to encompass all of the other schemas.\n *\n * @remarks The return type of this function is designed to work with Typechat's createZodJsonValidator as well as be used as the JSON schema for OpenAi's structured output response format.\n */\nfunction generateEditTypes(\n\tschemas: Iterable<SimpleTreeSchema>,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): {\n\tdomainTypes: Record<string, ZodTypeAny>;\n} {\n\tconst domainTypes: Record<string, ZodTypeAny> = {};\n\tfor (const schema of schemas) {\n\t\tfor (const name of schema.definitions.keys()) {\n\t\t\t// If this does overwrite anything in domainTypes, it is guaranteed to be overwritten with an identical value due to the getOrCreate\n\t\t\tdomainTypes[name] = getOrCreateType(\n\t\t\t\tschema.definitions,\n\t\t\t\tname,\n\t\t\t\tobjectCache,\n\t\t\t\tbindableSchemas,\n\t\t\t);\n\t\t}\n\t}\n\n\treturn {\n\t\tdomainTypes,\n\t};\n}\n\nfunction getBoundMethodsForBindable(bindableSchema: BindableSchema): {\n\treferencedTypes: Set<TreeNodeSchema>;\n\tmethods: [string, ZodTypeAny][];\n} {\n\tconst methodTypes: [string, ZodTypeAny][] = [];\n\tconst methods = getExposedMethods(bindableSchema);\n\tfor (const [name, method] of Object.entries(methods.methods)) {\n\t\tconst zodFunction = z.instanceof(FunctionWrapper);\n\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any\n\t\t(zodFunction as any).method = method;\n\t\tmethodTypes.push([name, zodFunction]);\n\t}\n\treturn { methods: methodTypes, referencedTypes: methods.referencedTypes };\n}\n\nfunction getBoundMethods(\n\tdefinition: string,\n\tbindableSchemas: Map<string, BindableSchema>,\n): [string, ZodTypeAny][] {\n\tconst bindableSchema = bindableSchemas.get(definition) ?? fail(\"Unknown definition\");\n\treturn getBoundMethodsForBindable(bindableSchema).methods;\n}\n\nfunction addBindingIntersectionIfNeeded(\n\ttypeString: \"array\" | \"map\" | \"record\",\n\tzodTypeBound: ZodTypeAny,\n\tdefinition: string,\n\tsimpleNodeSchema: SimpleNodeSchema,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tlet zodType = zodTypeBound;\n\tlet description = simpleNodeSchema.metadata?.description ?? \"\";\n\tconst boundMethods = getBoundMethods(definition, bindableSchemas);\n\tif (boundMethods.length > 0) {\n\t\tconst methods: Record<string, z.ZodTypeAny> = {};\n\t\tfor (const [name, zodFunction] of boundMethods) {\n\t\t\tif (methods[name] !== undefined) {\n\t\t\t\tthrow new UsageError(\n\t\t\t\t\t`Method ${name} conflicts with field of the same name in schema ${definition}`,\n\t\t\t\t);\n\t\t\t}\n\t\t\tmethods[name] = zodFunction;\n\t\t}\n\t\tzodType = z.intersection(zodType, z.object(methods));\n\t\tconst methodNote = `Note: this ${typeString} has custom user-defined methods directly on it.`;\n\t\tdescription = description === \"\" ? methodNote : `${description} - ${methodNote}`;\n\t}\n\treturn zodType.describe(description);\n}\n\nfunction getOrCreateType(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tdefinition: string,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst simpleNodeSchema = definitionMap.get(definition) ?? fail(\"Unexpected definition\");\n\treturn getOrCreate(objectCache, simpleNodeSchema, () => {\n\t\t// Handle recursive types: temporarily create a zod \"lazy\" type that can be referenced by a recursive call to getOrCreateType.\n\t\tlet type: ZodTypeAny | undefined;\n\t\tobjectCache.set(\n\t\t\tsimpleNodeSchema,\n\t\t\tz.lazy(() => type ?? fail(\"Recursive type used before creation\")),\n\t\t);\n\t\tswitch (simpleNodeSchema.kind) {\n\t\t\tcase NodeKind.Object: {\n\t\t\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment\n\t\t\t\tconst properties: Record<string, z.ZodTypeAny> = Object.fromEntries(\n\t\t\t\t\t[...simpleNodeSchema.fields]\n\t\t\t\t\t\t.map(([key, field]) => {\n\t\t\t\t\t\t\treturn [\n\t\t\t\t\t\t\t\tkey,\n\t\t\t\t\t\t\t\tgetOrCreateTypeForField(definitionMap, field, objectCache, bindableSchemas),\n\t\t\t\t\t\t\t];\n\t\t\t\t\t\t})\n\t\t\t\t\t\t.filter(([, value]) => value !== undefined),\n\t\t\t\t);\n\n\t\t\t\t// Unlike arrays/maps/records, object nodes include methods directly on them rather than using an intersection\n\t\t\t\tfor (const [name, zodFunction] of getBoundMethods(definition, bindableSchemas)) {\n\t\t\t\t\tif (properties[name] !== undefined) {\n\t\t\t\t\t\tthrow new UsageError(\n\t\t\t\t\t\t\t`Method ${name} conflicts with field of the same name in schema ${definition}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t\tproperties[name] = zodFunction;\n\t\t\t\t}\n\n\t\t\t\treturn (type = z\n\t\t\t\t\t.object(properties)\n\t\t\t\t\t.describe(simpleNodeSchema.metadata?.description ?? \"\"));\n\t\t\t}\n\t\t\tcase NodeKind.Map: {\n\t\t\t\tconst zodType = z.map(\n\t\t\t\t\tz.string(),\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tsimpleNodeSchema.allowedTypesIdentifiers,\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"map\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Record: {\n\t\t\t\tconst zodType = z.record(\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tsimpleNodeSchema.allowedTypesIdentifiers,\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"record\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Array: {\n\t\t\t\tconst zodType = z.array(\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tsimpleNodeSchema.allowedTypesIdentifiers,\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"array\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Leaf: {\n\t\t\t\tswitch (simpleNodeSchema.leafKind) {\n\t\t\t\t\tcase ValueSchema.Boolean: {\n\t\t\t\t\t\treturn (type = z.boolean());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.Number: {\n\t\t\t\t\t\treturn (type = z.number());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.String: {\n\t\t\t\t\t\treturn (type = z.string());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.Null: {\n\t\t\t\t\t\treturn (type = z.null());\n\t\t\t\t\t}\n\t\t\t\t\tdefault: {\n\t\t\t\t\t\tthrow new Error(`Unsupported leaf kind ${NodeKind[simpleNodeSchema.leafKind]}.`);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\tdefault: {\n\t\t\t\treturn unreachableCase(simpleNodeSchema, \"Unknown node kind\");\n\t\t\t}\n\t\t}\n\t});\n}\n\nfunction getOrCreateTypeForField(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tfieldSchema: SimpleFieldSchema,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst customMetadata = fieldSchema.metadata.custom as\n\t\t| Record<string | symbol, unknown>\n\t\t| undefined;\n\tconst getDefault = customMetadata?.[llmDefault];\n\tif (getDefault !== undefined) {\n\t\tif (typeof getDefault !== \"function\") {\n\t\t\tthrow new UsageError(\n\t\t\t\t`Expected value of ${llmDefault.description} property to be a function, but got ${typeof getDefault}`,\n\t\t\t);\n\t\t}\n\n\t\tif (fieldSchema.kind !== FieldKind.Optional) {\n\t\t\tthrow new UsageError(\n\t\t\t\t`The ${llmDefault.description} property is only permitted on optional fields.`,\n\t\t\t);\n\t\t}\n\t}\n\n\tconst field = getTypeForAllowedTypes(\n\t\tdefinitionMap,\n\t\tfieldSchema.allowedTypesIdentifiers,\n\t\tobjectCache,\n\t\tbindableSchemas,\n\t).describe(\n\t\tgetDefault === undefined\n\t\t\t? (fieldSchema.metadata?.description ?? \"\")\n\t\t\t: \"Do not populate this field. It will be automatically supplied by the system after insertion.\",\n\t);\n\n\tswitch (fieldSchema.kind) {\n\t\tcase FieldKind.Required: {\n\t\t\treturn field;\n\t\t}\n\t\tcase FieldKind.Optional: {\n\t\t\treturn field.optional();\n\t\t}\n\t\tcase FieldKind.Identifier: {\n\t\t\treturn field\n\t\t\t\t.optional()\n\t\t\t\t.describe(\n\t\t\t\t\t\"This is an ID automatically generated by the system. Do not supply it when constructing a new object.\",\n\t\t\t\t);\n\t\t}\n\t\tdefault: {\n\t\t\tthrow new Error(`Unsupported field kind ${NodeKind[fieldSchema.kind]}.`);\n\t\t}\n\t}\n}\n\nfunction getTypeForAllowedTypes(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tallowedTypes: ReadonlySet<string>,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst single = tryGetSingleton(allowedTypes);\n\tif (single === undefined) {\n\t\tconst types = [\n\t\t\t...mapIterable(allowedTypes, (name) => {\n\t\t\t\treturn getOrCreateType(definitionMap, name, objectCache, bindableSchemas);\n\t\t\t}),\n\t\t];\n\t\tassert(hasAtLeastTwo(types), 0xa7e /* Expected at least two types */);\n\t\treturn z.union(types);\n\t} else {\n\t\treturn getOrCreateType(definitionMap, single, objectCache, bindableSchemas);\n\t}\n}\n"]}
+{"version":3,"file":"typeGeneration.js","sourceRoot":"","sources":["../src/typeGeneration.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,kEAA8E;AAC9E,uEAAsE;AACtE,4DAUuC;AAQvC,6BAAyC;AAGzC,yDAAwE;AACxE,yCAQoB;AAEpB;;;;;;;;;GASG;AAEH;;GAEG;AACH,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAGlC,CAAC;AAEJ;;GAEG;AACH,SAAgB,0BAA0B,CACzC,UAA+B,EAC/B,MAAwB;IAIxB,OAAO,IAAA,sBAAW,EAAC,iBAAiB,EAAE,MAAM,EAAE,GAAG,EAAE;QAClD,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC7C,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAkB,CAAC;QACpD,IAAA,0BAAe,EAAC,UAAU,EAAE;YAC3B,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;gBACX,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAClB,IACC,CAAC,YAAY,2BAAgB;oBAC7B,CAAC,YAAY,wBAAa;oBAC1B,CAAC,YAAY,0BAAe;oBAC5B,CAAC,YAAY,2BAAgB,EAC5B,CAAC;oBACF,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBACzB,MAAM,cAAc,GAAG,IAAA,oCAAiB,EAAC,CAAC,CAAC,CAAC;oBAC5C,KAAK,MAAM,CAAC,IAAI,cAAc,CAAC,eAAe,EAAE,CAAC;wBAChD,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;wBAClB,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBAC1B,CAAC;gBACF,CAAC;YACF,CAAC;SACD,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,CAAC,GAAG,iBAAiB,CAAC,MAAM,EAAE,CAAqB,CAAC;QACxE,MAAM,eAAe,GAAG,IAAI,GAAG,CAC9B,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CACpE,CAAC;QACF,OAAO,iBAAiB,CACvB,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,0BAAe,EAAC,CAAC,CAAC,CAAC,EACvD,IAAI,GAAG,EAAE,EACT,eAAe,CACf,CAAC;IACH,CAAC,CAAC,CAAC;AACJ,CAAC;AArCD,gEAqCC;AAED;;;;;;;GAOG;AACH,SAAS,iBAAiB,CACzB,OAAmC,EACnC,WAAoD,EACpD,eAA4C;IAI5C,MAAM,WAAW,GAA+B,EAAE,CAAC;IACnD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC9B,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;YAC9C,oIAAoI;YACpI,WAAW,CAAC,IAAI,CAAC,GAAG,eAAe,CAClC,MAAM,CAAC,WAAW,EAClB,IAAI,EACJ,WAAW,EACX,eAAe,CACf,CAAC;QACH,CAAC;IACF,CAAC;IAED,OAAO;QACN,WAAW;KACX,CAAC;AACH,CAAC;AAED,SAAS,0BAA0B,CAAC,cAA8B;IAIjE,MAAM,WAAW,GAA2B,EAAE,CAAC;IAC/C,MAAM,OAAO,GAAG,IAAA,oCAAiB,EAAC,cAAc,CAAC,CAAC;IAClD,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,OAAC,CAAC,UAAU,CAAC,kCAAe,CAAC,CAAC;QAClD,0GAA0G;QACzG,WAAmB,CAAC,MAAM,GAAG,MAAM,CAAC;QACrC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,CAAC;AAC3E,CAAC;AAED,SAAS,eAAe,CACvB,UAAkB,EAClB,eAA4C;IAE5C,MAAM,cAAc,GAAG,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAA,eAAI,EAAC,oBAAoB,CAAC,CAAC;IACrF,OAAO,0BAA0B,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC;AAC3D,CAAC;AAED,SAAS,8BAA8B,CACtC,UAAsC,EACtC,YAAwB,EACxB,UAAkB,EAClB,gBAAkC,EAClC,eAA4C;IAE5C,IAAI,OAAO,GAAG,YAAY,CAAC;IAC3B,IAAI,WAAW,GAAG,gBAAgB,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC;IAC/D,MAAM,YAAY,GAAG,eAAe,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;IAClE,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAiC,EAAE,CAAC;QACjD,KAAK,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,YAAY,EAAE,CAAC;YAChD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;gBACjC,MAAM,IAAI,qBAAU,CACnB,UAAU,IAAI,oDAAoD,UAAU,EAAE,CAC9E,CAAC;YACH,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC;QAC7B,CAAC;QACD,OAAO,GAAG,OAAC,CAAC,YAAY,CAAC,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QACrD,MAAM,UAAU,GAAG,cAAc,UAAU,kDAAkD,CAAC;QAC9F,WAAW,GAAG,WAAW,KAAK,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,WAAW,MAAM,UAAU,EAAE,CAAC;IAClF,CAAC;IACD,OAAO,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,eAAe,CACvB,aAAoD,EACpD,UAAkB,EAClB,WAAoD,EACpD,eAA4C;IAE5C,MAAM,gBAAgB,GAAG,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAA,eAAI,EAAC,uBAAuB,CAAC,CAAC;IACxF,OAAO,IAAA,sBAAW,EAAC,WAAW,EAAE,gBAAgB,EAAE,GAAG,EAAE;QACtD,8HAA8H;QAC9H,IAAI,IAA4B,CAAC;QACjC,WAAW,CAAC,GAAG,CACd,gBAAgB,EAChB,OAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,IAAI,IAAA,eAAI,EAAC,qCAAqC,CAAC,CAAC,CACjE,CAAC;QACF,QAAQ,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAC/B,KAAK,mBAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;gBACtB,mEAAmE;gBACnE,MAAM,UAAU,GAAiC,MAAM,CAAC,WAAW,CAClE,CAAC,GAAG,gBAAgB,CAAC,MAAM,CAAC;qBAC1B,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;oBACrB,OAAO;wBACN,GAAG;wBACH,uBAAuB,CAAC,aAAa,EAAE,KAAK,EAAE,WAAW,EAAE,eAAe,CAAC;qBAC3E,CAAC;gBACH,CAAC,CAAC;qBACD,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC,CAC5C,CAAC;gBAEF,8GAA8G;gBAC9G,KAAK,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,eAAe,CAAC,UAAU,EAAE,eAAe,CAAC,EAAE,CAAC;oBAChF,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;wBACpC,MAAM,IAAI,qBAAU,CACnB,UAAU,IAAI,oDAAoD,UAAU,EAAE,CAC9E,CAAC;oBACH,CAAC;oBACD,UAAU,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC;gBAChC,CAAC;gBAED,OAAO,CAAC,IAAI,GAAG,OAAC;qBACd,MAAM,CAAC,UAAU,CAAC;qBAClB,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3D,CAAC;YACD,KAAK,mBAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;gBACnB,MAAM,OAAO,GAAG,OAAC,CAAC,GAAG,CACpB,OAAC,CAAC,MAAM,EAAE,EACV,sBAAsB,CACrB,aAAa,EACb,IAAI,GAAG,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,EACnD,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,KAAK,EACL,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,mBAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;gBACtB,MAAM,OAAO,GAAG,OAAC,CAAC,MAAM,CACvB,sBAAsB,CACrB,aAAa,EACb,IAAI,GAAG,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,EACnD,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,QAAQ,EACR,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,mBAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;gBACrB,MAAM,OAAO,GAAG,OAAC,CAAC,KAAK,CACtB,sBAAsB,CACrB,aAAa,EACb,IAAI,GAAG,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,EACnD,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,OAAO,EACP,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,mBAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpB,QAAQ,gBAAgB,CAAC,QAAQ,EAAE,CAAC;oBACnC,KAAK,sBAAW,CAAC,OAAO,CAAC,CAAC,CAAC;wBAC1B,OAAO,CAAC,IAAI,GAAG,OAAC,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC7B,CAAC;oBACD,KAAK,sBAAW,CAAC,MAAM,CAAC,CAAC,CAAC;wBACzB,OAAO,CAAC,IAAI,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC5B,CAAC;oBACD,KAAK,sBAAW,CAAC,MAAM,CAAC,CAAC,CAAC;wBACzB,OAAO,CAAC,IAAI,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC5B,CAAC;oBACD,KAAK,sBAAW,CAAC,IAAI,CAAC,CAAC,CAAC;wBACvB,OAAO,CAAC,IAAI,GAAG,OAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC1B,CAAC;oBACD,OAAO,CAAC,CAAC,CAAC;wBACT,MAAM,IAAI,KAAK,CAAC,yBAAyB,mBAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;oBAClF,CAAC;gBACF,CAAC;YACF,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACT,OAAO,IAAA,0BAAe,EAAC,gBAAgB,EAAE,mBAAmB,CAAC,CAAC;YAC/D,CAAC;QACF,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAC/B,aAAoD,EACpD,WAA8B,EAC9B,WAAoD,EACpD,eAA4C;IAE5C,MAAM,cAAc,GAAG,WAAW,CAAC,QAAQ,CAAC,MAEhC,CAAC;IACb,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC,qBAAU,CAAC,CAAC;IAChD,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC9B,IAAI,OAAO,UAAU,KAAK,UAAU,EAAE,CAAC;YACtC,MAAM,IAAI,qBAAU,CACnB,qBAAqB,qBAAU,CAAC,WAAW,uCAAuC,OAAO,UAAU,EAAE,CACrG,CAAC;QACH,CAAC;QAED,IAAI,WAAW,CAAC,IAAI,KAAK,oBAAS,CAAC,QAAQ,EAAE,CAAC;YAC7C,MAAM,IAAI,qBAAU,CACnB,OAAO,qBAAU,CAAC,WAAW,iDAAiD,CAC9E,CAAC;QACH,CAAC;IACF,CAAC;IAED,MAAM,KAAK,GAAG,sBAAsB,CACnC,aAAa,EACb,IAAI,GAAG,CAAC,WAAW,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,EAC9C,WAAW,EACX,eAAe,CACf,CAAC,QAAQ,CACT,UAAU,KAAK,SAAS;QACvB,CAAC,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC;QAC3C,CAAC,CAAC,8FAA8F,CACjG,CAAC;IAEF,QAAQ,WAAW,CAAC,IAAI,EAAE,CAAC;QAC1B,KAAK,oBAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzB,OAAO,KAAK,CAAC;QACd,CAAC;QACD,KAAK,oBAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzB,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;QACzB,CAAC;QACD,KAAK,oBAAS,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3B,OAAO,KAAK;iBACV,QAAQ,EAAE;iBACV,QAAQ,CACR,uGAAuG,CACvG,CAAC;QACJ,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,0BAA0B,mBAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1E,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,sBAAsB,CAC9B,aAAoD,EACpD,YAAiC,EACjC,WAAoD,EACpD,eAA4C;IAE5C,MAAM,MAAM,GAAG,IAAA,0BAAe,EAAC,YAAY,CAAC,CAAC;IAC7C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG;YACb,GAAG,IAAA,sBAAW,EAAC,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE;gBACrC,OAAO,eAAe,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,EAAE,eAAe,CAAC,CAAC;YAC3E,CAAC,CAAC;SACF,CAAC;QACF,IAAA,iBAAM,EAAC,IAAA,wBAAa,EAAC,KAAK,CAAC,EAAE,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACtE,OAAO,OAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;SAAM,CAAC;QACP,OAAO,eAAe,CAAC,aAAa,EAAE,MAAM,EAAE,WAAW,EAAE,eAAe,CAAC,CAAC;IAC7E,CAAC;AACF,CAAC","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\nimport { assert, unreachableCase } from \"@fluidframework/core-utils/internal\";\nimport { UsageError } from \"@fluidframework/telemetry-utils/internal\";\nimport {\n\tArrayNodeSchema,\n\tFieldKind,\n\tgetSimpleSchema,\n\tMapNodeSchema,\n\tNodeKind,\n\tObjectNodeSchema,\n\tRecordNodeSchema,\n\tValueSchema,\n\twalkFieldSchema,\n} from \"@fluidframework/tree/internal\";\nimport type {\n\tImplicitFieldSchema,\n\tSimpleFieldSchema,\n\tSimpleNodeSchema,\n\tSimpleTreeSchema,\n\tTreeNodeSchema,\n} from \"@fluidframework/tree/internal\";\nimport { z, type ZodTypeAny } from \"zod\";\n\nimport type { BindableSchema } from \"./methodBinding.js\";\nimport { FunctionWrapper, getExposedMethods } from \"./methodBinding.js\";\nimport {\n\tfail,\n\tgetOrCreate,\n\thasAtLeastTwo,\n\tllmDefault,\n\tmapIterable,\n\ttryGetSingleton,\n\ttype MapGetSet,\n} from \"./utils.js\";\n\n/**\n *\n * TODO: Add a prompt suggestion API!\n *\n * TODO: Handle rate limit errors.\n *\n * TODO: Pass descriptions from schema metadata to the generated TS types that we put in the prompt\n *\n * TODO make the Ids be \"Vector-2\" instead of \"Vector2\" (or else it gets weird when you have a type called \"Vector2\")\n */\n\n/**\n * Cache used to prevent repeatedly generating the same Zod validation objects for the same {@link SimpleTreeSchema} as generate propts for repeated calls to an LLM\n */\nconst promptSchemaCache = new WeakMap<\n\tSimpleTreeSchema,\n\tReturnType<typeof generateEditTypes>\n>();\n\n/**\n * TODO\n */\nexport function generateEditTypesForPrompt(\n\trootSchema: ImplicitFieldSchema,\n\tschema: SimpleTreeSchema,\n): {\n\tdomainTypes: Record<string, ZodTypeAny>;\n} {\n\treturn getOrCreate(promptSchemaCache, schema, () => {\n\t\tconst allSchemas = new Set<TreeNodeSchema>();\n\t\tconst objectTypeSchemas = new Set<TreeNodeSchema>();\n\t\twalkFieldSchema(rootSchema, {\n\t\t\tnode: (n) => {\n\t\t\t\tallSchemas.add(n);\n\t\t\t\tif (\n\t\t\t\t\tn instanceof ObjectNodeSchema ||\n\t\t\t\t\tn instanceof MapNodeSchema ||\n\t\t\t\t\tn instanceof ArrayNodeSchema ||\n\t\t\t\t\tn instanceof RecordNodeSchema\n\t\t\t\t) {\n\t\t\t\t\tobjectTypeSchemas.add(n);\n\t\t\t\t\tconst exposedMethods = getExposedMethods(n);\n\t\t\t\t\tfor (const t of exposedMethods.referencedTypes) {\n\t\t\t\t\t\tallSchemas.add(t);\n\t\t\t\t\t\tobjectTypeSchemas.add(t);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t});\n\t\tconst nodeSchemas = [...objectTypeSchemas.values()] as BindableSchema[];\n\t\tconst bindableSchemas = new Map<string, BindableSchema>(\n\t\t\tnodeSchemas.map((nodeSchema) => [nodeSchema.identifier, nodeSchema]),\n\t\t);\n\t\treturn generateEditTypes(\n\t\t\t[...allSchemas.values()].map((s) => getSimpleSchema(s)),\n\t\t\tnew Map(),\n\t\t\tbindableSchemas,\n\t\t);\n\t});\n}\n\n/**\n * Generates a set of ZOD validation objects for the various types of data that can be put into the provided {@link SimpleTreeSchema}\n * and then uses those sets to generate an all-encompassing ZOD object for each type of {@link TreeEdit} that can validate any of the types of data that can be put into the tree.\n *\n * @returns a Record of schema names to Zod validation objects, and the name of the root schema used to encompass all of the other schemas.\n *\n * @remarks The return type of this function is designed to work with Typechat's createZodJsonValidator as well as be used as the JSON schema for OpenAi's structured output response format.\n */\nfunction generateEditTypes(\n\tschemas: Iterable<SimpleTreeSchema>,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): {\n\tdomainTypes: Record<string, ZodTypeAny>;\n} {\n\tconst domainTypes: Record<string, ZodTypeAny> = {};\n\tfor (const schema of schemas) {\n\t\tfor (const name of schema.definitions.keys()) {\n\t\t\t// If this does overwrite anything in domainTypes, it is guaranteed to be overwritten with an identical value due to the getOrCreate\n\t\t\tdomainTypes[name] = getOrCreateType(\n\t\t\t\tschema.definitions,\n\t\t\t\tname,\n\t\t\t\tobjectCache,\n\t\t\t\tbindableSchemas,\n\t\t\t);\n\t\t}\n\t}\n\n\treturn {\n\t\tdomainTypes,\n\t};\n}\n\nfunction getBoundMethodsForBindable(bindableSchema: BindableSchema): {\n\treferencedTypes: Set<TreeNodeSchema>;\n\tmethods: [string, ZodTypeAny][];\n} {\n\tconst methodTypes: [string, ZodTypeAny][] = [];\n\tconst methods = getExposedMethods(bindableSchema);\n\tfor (const [name, method] of Object.entries(methods.methods)) {\n\t\tconst zodFunction = z.instanceof(FunctionWrapper);\n\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any\n\t\t(zodFunction as any).method = method;\n\t\tmethodTypes.push([name, zodFunction]);\n\t}\n\treturn { methods: methodTypes, referencedTypes: methods.referencedTypes };\n}\n\nfunction getBoundMethods(\n\tdefinition: string,\n\tbindableSchemas: Map<string, BindableSchema>,\n): [string, ZodTypeAny][] {\n\tconst bindableSchema = bindableSchemas.get(definition) ?? fail(\"Unknown definition\");\n\treturn getBoundMethodsForBindable(bindableSchema).methods;\n}\n\nfunction addBindingIntersectionIfNeeded(\n\ttypeString: \"array\" | \"map\" | \"record\",\n\tzodTypeBound: ZodTypeAny,\n\tdefinition: string,\n\tsimpleNodeSchema: SimpleNodeSchema,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tlet zodType = zodTypeBound;\n\tlet description = simpleNodeSchema.metadata?.description ?? \"\";\n\tconst boundMethods = getBoundMethods(definition, bindableSchemas);\n\tif (boundMethods.length > 0) {\n\t\tconst methods: Record<string, z.ZodTypeAny> = {};\n\t\tfor (const [name, zodFunction] of boundMethods) {\n\t\t\tif (methods[name] !== undefined) {\n\t\t\t\tthrow new UsageError(\n\t\t\t\t\t`Method ${name} conflicts with field of the same name in schema ${definition}`,\n\t\t\t\t);\n\t\t\t}\n\t\t\tmethods[name] = zodFunction;\n\t\t}\n\t\tzodType = z.intersection(zodType, z.object(methods));\n\t\tconst methodNote = `Note: this ${typeString} has custom user-defined methods directly on it.`;\n\t\tdescription = description === \"\" ? methodNote : `${description} - ${methodNote}`;\n\t}\n\treturn zodType.describe(description);\n}\n\nfunction getOrCreateType(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tdefinition: string,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst simpleNodeSchema = definitionMap.get(definition) ?? fail(\"Unexpected definition\");\n\treturn getOrCreate(objectCache, simpleNodeSchema, () => {\n\t\t// Handle recursive types: temporarily create a zod \"lazy\" type that can be referenced by a recursive call to getOrCreateType.\n\t\tlet type: ZodTypeAny | undefined;\n\t\tobjectCache.set(\n\t\t\tsimpleNodeSchema,\n\t\t\tz.lazy(() => type ?? fail(\"Recursive type used before creation\")),\n\t\t);\n\t\tswitch (simpleNodeSchema.kind) {\n\t\t\tcase NodeKind.Object: {\n\t\t\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment\n\t\t\t\tconst properties: Record<string, z.ZodTypeAny> = Object.fromEntries(\n\t\t\t\t\t[...simpleNodeSchema.fields]\n\t\t\t\t\t\t.map(([key, field]) => {\n\t\t\t\t\t\t\treturn [\n\t\t\t\t\t\t\t\tkey,\n\t\t\t\t\t\t\t\tgetOrCreateTypeForField(definitionMap, field, objectCache, bindableSchemas),\n\t\t\t\t\t\t\t];\n\t\t\t\t\t\t})\n\t\t\t\t\t\t.filter(([, value]) => value !== undefined),\n\t\t\t\t);\n\n\t\t\t\t// Unlike arrays/maps/records, object nodes include methods directly on them rather than using an intersection\n\t\t\t\tfor (const [name, zodFunction] of getBoundMethods(definition, bindableSchemas)) {\n\t\t\t\t\tif (properties[name] !== undefined) {\n\t\t\t\t\t\tthrow new UsageError(\n\t\t\t\t\t\t\t`Method ${name} conflicts with field of the same name in schema ${definition}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t\tproperties[name] = zodFunction;\n\t\t\t\t}\n\n\t\t\t\treturn (type = z\n\t\t\t\t\t.object(properties)\n\t\t\t\t\t.describe(simpleNodeSchema.metadata?.description ?? \"\"));\n\t\t\t}\n\t\t\tcase NodeKind.Map: {\n\t\t\t\tconst zodType = z.map(\n\t\t\t\t\tz.string(),\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tnew Set(simpleNodeSchema.simpleAllowedTypes.keys()),\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"map\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Record: {\n\t\t\t\tconst zodType = z.record(\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tnew Set(simpleNodeSchema.simpleAllowedTypes.keys()),\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"record\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Array: {\n\t\t\t\tconst zodType = z.array(\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tnew Set(simpleNodeSchema.simpleAllowedTypes.keys()),\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"array\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Leaf: {\n\t\t\t\tswitch (simpleNodeSchema.leafKind) {\n\t\t\t\t\tcase ValueSchema.Boolean: {\n\t\t\t\t\t\treturn (type = z.boolean());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.Number: {\n\t\t\t\t\t\treturn (type = z.number());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.String: {\n\t\t\t\t\t\treturn (type = z.string());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.Null: {\n\t\t\t\t\t\treturn (type = z.null());\n\t\t\t\t\t}\n\t\t\t\t\tdefault: {\n\t\t\t\t\t\tthrow new Error(`Unsupported leaf kind ${NodeKind[simpleNodeSchema.leafKind]}.`);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\tdefault: {\n\t\t\t\treturn unreachableCase(simpleNodeSchema, \"Unknown node kind\");\n\t\t\t}\n\t\t}\n\t});\n}\n\nfunction getOrCreateTypeForField(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tfieldSchema: SimpleFieldSchema,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst customMetadata = fieldSchema.metadata.custom as\n\t\t| Record<string | symbol, unknown>\n\t\t| undefined;\n\tconst getDefault = customMetadata?.[llmDefault];\n\tif (getDefault !== undefined) {\n\t\tif (typeof getDefault !== \"function\") {\n\t\t\tthrow new UsageError(\n\t\t\t\t`Expected value of ${llmDefault.description} property to be a function, but got ${typeof getDefault}`,\n\t\t\t);\n\t\t}\n\n\t\tif (fieldSchema.kind !== FieldKind.Optional) {\n\t\t\tthrow new UsageError(\n\t\t\t\t`The ${llmDefault.description} property is only permitted on optional fields.`,\n\t\t\t);\n\t\t}\n\t}\n\n\tconst field = getTypeForAllowedTypes(\n\t\tdefinitionMap,\n\t\tnew Set(fieldSchema.simpleAllowedTypes.keys()),\n\t\tobjectCache,\n\t\tbindableSchemas,\n\t).describe(\n\t\tgetDefault === undefined\n\t\t\t? (fieldSchema.metadata?.description ?? \"\")\n\t\t\t: \"Do not populate this field. It will be automatically supplied by the system after insertion.\",\n\t);\n\n\tswitch (fieldSchema.kind) {\n\t\tcase FieldKind.Required: {\n\t\t\treturn field;\n\t\t}\n\t\tcase FieldKind.Optional: {\n\t\t\treturn field.optional();\n\t\t}\n\t\tcase FieldKind.Identifier: {\n\t\t\treturn field\n\t\t\t\t.optional()\n\t\t\t\t.describe(\n\t\t\t\t\t\"This is an ID automatically generated by the system. Do not supply it when constructing a new object.\",\n\t\t\t\t);\n\t\t}\n\t\tdefault: {\n\t\t\tthrow new Error(`Unsupported field kind ${NodeKind[fieldSchema.kind]}.`);\n\t\t}\n\t}\n}\n\nfunction getTypeForAllowedTypes(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tallowedTypes: ReadonlySet<string>,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst single = tryGetSingleton(allowedTypes);\n\tif (single === undefined) {\n\t\tconst types = [\n\t\t\t...mapIterable(allowedTypes, (name) => {\n\t\t\t\treturn getOrCreateType(definitionMap, name, objectCache, bindableSchemas);\n\t\t\t}),\n\t\t];\n\t\tassert(hasAtLeastTwo(types), 0xa7e /* Expected at least two types */);\n\t\treturn z.union(types);\n\t} else {\n\t\treturn getOrCreateType(definitionMap, single, objectCache, bindableSchemas);\n\t}\n}\n"]}

package/lib/typeGeneration.js

@@ -132,15 +132,15 @@ function getOrCreateType(definitionMap, definition, objectCache, bindableSchemas
                     .describe(simpleNodeSchema.metadata?.description ?? ""));
             }
             case NodeKind.Map: {
-                const zodType = z.map(z.string(), getTypeForAllowedTypes(definitionMap, simpleNodeSchema.allowedTypesIdentifiers, objectCache, bindableSchemas));
+                const zodType = z.map(z.string(), getTypeForAllowedTypes(definitionMap, new Set(simpleNodeSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas));
                 return (type = addBindingIntersectionIfNeeded("map", zodType, definition, simpleNodeSchema, bindableSchemas));
             }
             case NodeKind.Record: {
-                const zodType = z.record(getTypeForAllowedTypes(definitionMap, simpleNodeSchema.allowedTypesIdentifiers, objectCache, bindableSchemas));
+                const zodType = z.record(getTypeForAllowedTypes(definitionMap, new Set(simpleNodeSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas));
                 return (type = addBindingIntersectionIfNeeded("record", zodType, definition, simpleNodeSchema, bindableSchemas));
             }
             case NodeKind.Array: {
-                const zodType = z.array(getTypeForAllowedTypes(definitionMap, simpleNodeSchema.allowedTypesIdentifiers, objectCache, bindableSchemas));
+                const zodType = z.array(getTypeForAllowedTypes(definitionMap, new Set(simpleNodeSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas));
                 return (type = addBindingIntersectionIfNeeded("array", zodType, definition, simpleNodeSchema, bindableSchemas));
             }
             case NodeKind.Leaf: {
@@ -179,7 +179,7 @@ function getOrCreateTypeForField(definitionMap, fieldSchema, objectCache, bindab
             throw new UsageError(`The ${llmDefault.description} property is only permitted on optional fields.`);
         }
     }
-    const field = getTypeForAllowedTypes(definitionMap, fieldSchema.allowedTypesIdentifiers, objectCache, bindableSchemas).describe(getDefault === undefined
+    const field = getTypeForAllowedTypes(definitionMap, new Set(fieldSchema.simpleAllowedTypes.keys()), objectCache, bindableSchemas).describe(getDefault === undefined
         ? (fieldSchema.metadata?.description ?? "")
         : "Do not populate this field. It will be automatically supplied by the system after insertion.");
     switch (fieldSchema.kind) {

package/lib/typeGeneration.js.map

@@ -1 +1 @@
-{"version":3,"file":"typeGeneration.js","sourceRoot":"","sources":["../src/typeGeneration.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAC;AAC9E,OAAO,EAAE,UAAU,EAAE,MAAM,0CAA0C,CAAC;AACtE,OAAO,EACN,eAAe,EACf,SAAS,EACT,eAAe,EACf,aAAa,EACb,QAAQ,EACR,gBAAgB,EAChB,gBAAgB,EAChB,WAAW,EACX,eAAe,GACf,MAAM,+BAA+B,CAAC;AAQvC,OAAO,EAAE,CAAC,EAAmB,MAAM,KAAK,CAAC;AAGzC,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACxE,OAAO,EACN,IAAI,EACJ,WAAW,EACX,aAAa,EACb,UAAU,EACV,WAAW,EACX,eAAe,GAEf,MAAM,YAAY,CAAC;AAEpB;;;;;;;;;GASG;AAEH;;GAEG;AACH,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAGlC,CAAC;AAEJ;;GAEG;AACH,MAAM,UAAU,0BAA0B,CACzC,UAA+B,EAC/B,MAAwB;IAIxB,OAAO,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,GAAG,EAAE;QAClD,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC7C,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAkB,CAAC;QACpD,eAAe,CAAC,UAAU,EAAE;YAC3B,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;gBACX,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAClB,IACC,CAAC,YAAY,gBAAgB;oBAC7B,CAAC,YAAY,aAAa;oBAC1B,CAAC,YAAY,eAAe;oBAC5B,CAAC,YAAY,gBAAgB,EAC5B,CAAC;oBACF,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBACzB,MAAM,cAAc,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;oBAC5C,KAAK,MAAM,CAAC,IAAI,cAAc,CAAC,eAAe,EAAE,CAAC;wBAChD,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;wBAClB,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBAC1B,CAAC;gBACF,CAAC;YACF,CAAC;SACD,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,CAAC,GAAG,iBAAiB,CAAC,MAAM,EAAE,CAAqB,CAAC;QACxE,MAAM,eAAe,GAAG,IAAI,GAAG,CAC9B,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CACpE,CAAC;QACF,OAAO,iBAAiB,CACvB,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,EACvD,IAAI,GAAG,EAAE,EACT,eAAe,CACf,CAAC;IACH,CAAC,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,iBAAiB,CACzB,OAAmC,EACnC,WAAoD,EACpD,eAA4C;IAI5C,MAAM,WAAW,GAA+B,EAAE,CAAC;IACnD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC9B,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;YAC9C,oIAAoI;YACpI,WAAW,CAAC,IAAI,CAAC,GAAG,eAAe,CAClC,MAAM,CAAC,WAAW,EAClB,IAAI,EACJ,WAAW,EACX,eAAe,CACf,CAAC;QACH,CAAC;IACF,CAAC;IAED,OAAO;QACN,WAAW;KACX,CAAC;AACH,CAAC;AAED,SAAS,0BAA0B,CAAC,cAA8B;IAIjE,MAAM,WAAW,GAA2B,EAAE,CAAC;IAC/C,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAC;IAClD,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;QAClD,0GAA0G;QACzG,WAAmB,CAAC,MAAM,GAAG,MAAM,CAAC;QACrC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,CAAC;AAC3E,CAAC;AAED,SAAS,eAAe,CACvB,UAAkB,EAClB,eAA4C;IAE5C,MAAM,cAAc,GAAG,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACrF,OAAO,0BAA0B,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC;AAC3D,CAAC;AAED,SAAS,8BAA8B,CACtC,UAAsC,EACtC,YAAwB,EACxB,UAAkB,EAClB,gBAAkC,EAClC,eAA4C;IAE5C,IAAI,OAAO,GAAG,YAAY,CAAC;IAC3B,IAAI,WAAW,GAAG,gBAAgB,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC;IAC/D,MAAM,YAAY,GAAG,eAAe,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;IAClE,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAiC,EAAE,CAAC;QACjD,KAAK,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,YAAY,EAAE,CAAC;YAChD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;gBACjC,MAAM,IAAI,UAAU,CACnB,UAAU,IAAI,oDAAoD,UAAU,EAAE,CAC9E,CAAC;YACH,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC;QAC7B,CAAC;QACD,OAAO,GAAG,CAAC,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QACrD,MAAM,UAAU,GAAG,cAAc,UAAU,kDAAkD,CAAC;QAC9F,WAAW,GAAG,WAAW,KAAK,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,WAAW,MAAM,UAAU,EAAE,CAAC;IAClF,CAAC;IACD,OAAO,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,eAAe,CACvB,aAAoD,EACpD,UAAkB,EAClB,WAAoD,EACpD,eAA4C;IAE5C,MAAM,gBAAgB,GAAG,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACxF,OAAO,WAAW,CAAC,WAAW,EAAE,gBAAgB,EAAE,GAAG,EAAE;QACtD,8HAA8H;QAC9H,IAAI,IAA4B,CAAC;QACjC,WAAW,CAAC,GAAG,CACd,gBAAgB,EAChB,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC,qCAAqC,CAAC,CAAC,CACjE,CAAC;QACF,QAAQ,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAC/B,KAAK,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;gBACtB,mEAAmE;gBACnE,MAAM,UAAU,GAAiC,MAAM,CAAC,WAAW,CAClE,CAAC,GAAG,gBAAgB,CAAC,MAAM,CAAC;qBAC1B,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;oBACrB,OAAO;wBACN,GAAG;wBACH,uBAAuB,CAAC,aAAa,EAAE,KAAK,EAAE,WAAW,EAAE,eAAe,CAAC;qBAC3E,CAAC;gBACH,CAAC,CAAC;qBACD,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC,CAC5C,CAAC;gBAEF,8GAA8G;gBAC9G,KAAK,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,eAAe,CAAC,UAAU,EAAE,eAAe,CAAC,EAAE,CAAC;oBAChF,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;wBACpC,MAAM,IAAI,UAAU,CACnB,UAAU,IAAI,oDAAoD,UAAU,EAAE,CAC9E,CAAC;oBACH,CAAC;oBACD,UAAU,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC;gBAChC,CAAC;gBAED,OAAO,CAAC,IAAI,GAAG,CAAC;qBACd,MAAM,CAAC,UAAU,CAAC;qBAClB,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3D,CAAC;YACD,KAAK,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;gBACnB,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,CACpB,CAAC,CAAC,MAAM,EAAE,EACV,sBAAsB,CACrB,aAAa,EACb,gBAAgB,CAAC,uBAAuB,EACxC,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,KAAK,EACL,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;gBACtB,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CACvB,sBAAsB,CACrB,aAAa,EACb,gBAAgB,CAAC,uBAAuB,EACxC,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,QAAQ,EACR,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;gBACrB,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CACtB,sBAAsB,CACrB,aAAa,EACb,gBAAgB,CAAC,uBAAuB,EACxC,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,OAAO,EACP,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpB,QAAQ,gBAAgB,CAAC,QAAQ,EAAE,CAAC;oBACnC,KAAK,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;wBAC1B,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC7B,CAAC;oBACD,KAAK,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;wBACzB,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC5B,CAAC;oBACD,KAAK,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;wBACzB,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC5B,CAAC;oBACD,KAAK,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;wBACvB,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC1B,CAAC;oBACD,OAAO,CAAC,CAAC,CAAC;wBACT,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;oBAClF,CAAC;gBACF,CAAC;YACF,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACT,OAAO,eAAe,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,CAAC;YAC/D,CAAC;QACF,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAC/B,aAAoD,EACpD,WAA8B,EAC9B,WAAoD,EACpD,eAA4C;IAE5C,MAAM,cAAc,GAAG,WAAW,CAAC,QAAQ,CAAC,MAEhC,CAAC;IACb,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC,UAAU,CAAC,CAAC;IAChD,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC9B,IAAI,OAAO,UAAU,KAAK,UAAU,EAAE,CAAC;YACtC,MAAM,IAAI,UAAU,CACnB,qBAAqB,UAAU,CAAC,WAAW,uCAAuC,OAAO,UAAU,EAAE,CACrG,CAAC;QACH,CAAC;QAED,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC,QAAQ,EAAE,CAAC;YAC7C,MAAM,IAAI,UAAU,CACnB,OAAO,UAAU,CAAC,WAAW,iDAAiD,CAC9E,CAAC;QACH,CAAC;IACF,CAAC;IAED,MAAM,KAAK,GAAG,sBAAsB,CACnC,aAAa,EACb,WAAW,CAAC,uBAAuB,EACnC,WAAW,EACX,eAAe,CACf,CAAC,QAAQ,CACT,UAAU,KAAK,SAAS;QACvB,CAAC,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC;QAC3C,CAAC,CAAC,8FAA8F,CACjG,CAAC;IAEF,QAAQ,WAAW,CAAC,IAAI,EAAE,CAAC;QAC1B,KAAK,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzB,OAAO,KAAK,CAAC;QACd,CAAC;QACD,KAAK,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzB,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;QACzB,CAAC;QACD,KAAK,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3B,OAAO,KAAK;iBACV,QAAQ,EAAE;iBACV,QAAQ,CACR,uGAAuG,CACvG,CAAC;QACJ,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1E,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,sBAAsB,CAC9B,aAAoD,EACpD,YAAiC,EACjC,WAAoD,EACpD,eAA4C;IAE5C,MAAM,MAAM,GAAG,eAAe,CAAC,YAAY,CAAC,CAAC;IAC7C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG;YACb,GAAG,WAAW,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE;gBACrC,OAAO,eAAe,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,EAAE,eAAe,CAAC,CAAC;YAC3E,CAAC,CAAC;SACF,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACtE,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;SAAM,CAAC;QACP,OAAO,eAAe,CAAC,aAAa,EAAE,MAAM,EAAE,WAAW,EAAE,eAAe,CAAC,CAAC;IAC7E,CAAC;AACF,CAAC","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\nimport { assert, unreachableCase } from \"@fluidframework/core-utils/internal\";\nimport { UsageError } from \"@fluidframework/telemetry-utils/internal\";\nimport {\n\tArrayNodeSchema,\n\tFieldKind,\n\tgetSimpleSchema,\n\tMapNodeSchema,\n\tNodeKind,\n\tObjectNodeSchema,\n\tRecordNodeSchema,\n\tValueSchema,\n\twalkFieldSchema,\n} from \"@fluidframework/tree/internal\";\nimport type {\n\tImplicitFieldSchema,\n\tSimpleFieldSchema,\n\tSimpleNodeSchema,\n\tSimpleTreeSchema,\n\tTreeNodeSchema,\n} from \"@fluidframework/tree/internal\";\nimport { z, type ZodTypeAny } from \"zod\";\n\nimport type { BindableSchema } from \"./methodBinding.js\";\nimport { FunctionWrapper, getExposedMethods } from \"./methodBinding.js\";\nimport {\n\tfail,\n\tgetOrCreate,\n\thasAtLeastTwo,\n\tllmDefault,\n\tmapIterable,\n\ttryGetSingleton,\n\ttype MapGetSet,\n} from \"./utils.js\";\n\n/**\n *\n * TODO: Add a prompt suggestion API!\n *\n * TODO: Handle rate limit errors.\n *\n * TODO: Pass descriptions from schema metadata to the generated TS types that we put in the prompt\n *\n * TODO make the Ids be \"Vector-2\" instead of \"Vector2\" (or else it gets weird when you have a type called \"Vector2\")\n */\n\n/**\n * Cache used to prevent repeatedly generating the same Zod validation objects for the same {@link SimpleTreeSchema} as generate propts for repeated calls to an LLM\n */\nconst promptSchemaCache = new WeakMap<\n\tSimpleTreeSchema,\n\tReturnType<typeof generateEditTypes>\n>();\n\n/**\n * TODO\n */\nexport function generateEditTypesForPrompt(\n\trootSchema: ImplicitFieldSchema,\n\tschema: SimpleTreeSchema,\n): {\n\tdomainTypes: Record<string, ZodTypeAny>;\n} {\n\treturn getOrCreate(promptSchemaCache, schema, () => {\n\t\tconst allSchemas = new Set<TreeNodeSchema>();\n\t\tconst objectTypeSchemas = new Set<TreeNodeSchema>();\n\t\twalkFieldSchema(rootSchema, {\n\t\t\tnode: (n) => {\n\t\t\t\tallSchemas.add(n);\n\t\t\t\tif (\n\t\t\t\t\tn instanceof ObjectNodeSchema ||\n\t\t\t\t\tn instanceof MapNodeSchema ||\n\t\t\t\t\tn instanceof ArrayNodeSchema ||\n\t\t\t\t\tn instanceof RecordNodeSchema\n\t\t\t\t) {\n\t\t\t\t\tobjectTypeSchemas.add(n);\n\t\t\t\t\tconst exposedMethods = getExposedMethods(n);\n\t\t\t\t\tfor (const t of exposedMethods.referencedTypes) {\n\t\t\t\t\t\tallSchemas.add(t);\n\t\t\t\t\t\tobjectTypeSchemas.add(t);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t});\n\t\tconst nodeSchemas = [...objectTypeSchemas.values()] as BindableSchema[];\n\t\tconst bindableSchemas = new Map<string, BindableSchema>(\n\t\t\tnodeSchemas.map((nodeSchema) => [nodeSchema.identifier, nodeSchema]),\n\t\t);\n\t\treturn generateEditTypes(\n\t\t\t[...allSchemas.values()].map((s) => getSimpleSchema(s)),\n\t\t\tnew Map(),\n\t\t\tbindableSchemas,\n\t\t);\n\t});\n}\n\n/**\n * Generates a set of ZOD validation objects for the various types of data that can be put into the provided {@link SimpleTreeSchema}\n * and then uses those sets to generate an all-encompassing ZOD object for each type of {@link TreeEdit} that can validate any of the types of data that can be put into the tree.\n *\n * @returns a Record of schema names to Zod validation objects, and the name of the root schema used to encompass all of the other schemas.\n *\n * @remarks The return type of this function is designed to work with Typechat's createZodJsonValidator as well as be used as the JSON schema for OpenAi's structured output response format.\n */\nfunction generateEditTypes(\n\tschemas: Iterable<SimpleTreeSchema>,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): {\n\tdomainTypes: Record<string, ZodTypeAny>;\n} {\n\tconst domainTypes: Record<string, ZodTypeAny> = {};\n\tfor (const schema of schemas) {\n\t\tfor (const name of schema.definitions.keys()) {\n\t\t\t// If this does overwrite anything in domainTypes, it is guaranteed to be overwritten with an identical value due to the getOrCreate\n\t\t\tdomainTypes[name] = getOrCreateType(\n\t\t\t\tschema.definitions,\n\t\t\t\tname,\n\t\t\t\tobjectCache,\n\t\t\t\tbindableSchemas,\n\t\t\t);\n\t\t}\n\t}\n\n\treturn {\n\t\tdomainTypes,\n\t};\n}\n\nfunction getBoundMethodsForBindable(bindableSchema: BindableSchema): {\n\treferencedTypes: Set<TreeNodeSchema>;\n\tmethods: [string, ZodTypeAny][];\n} {\n\tconst methodTypes: [string, ZodTypeAny][] = [];\n\tconst methods = getExposedMethods(bindableSchema);\n\tfor (const [name, method] of Object.entries(methods.methods)) {\n\t\tconst zodFunction = z.instanceof(FunctionWrapper);\n\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any\n\t\t(zodFunction as any).method = method;\n\t\tmethodTypes.push([name, zodFunction]);\n\t}\n\treturn { methods: methodTypes, referencedTypes: methods.referencedTypes };\n}\n\nfunction getBoundMethods(\n\tdefinition: string,\n\tbindableSchemas: Map<string, BindableSchema>,\n): [string, ZodTypeAny][] {\n\tconst bindableSchema = bindableSchemas.get(definition) ?? fail(\"Unknown definition\");\n\treturn getBoundMethodsForBindable(bindableSchema).methods;\n}\n\nfunction addBindingIntersectionIfNeeded(\n\ttypeString: \"array\" | \"map\" | \"record\",\n\tzodTypeBound: ZodTypeAny,\n\tdefinition: string,\n\tsimpleNodeSchema: SimpleNodeSchema,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tlet zodType = zodTypeBound;\n\tlet description = simpleNodeSchema.metadata?.description ?? \"\";\n\tconst boundMethods = getBoundMethods(definition, bindableSchemas);\n\tif (boundMethods.length > 0) {\n\t\tconst methods: Record<string, z.ZodTypeAny> = {};\n\t\tfor (const [name, zodFunction] of boundMethods) {\n\t\t\tif (methods[name] !== undefined) {\n\t\t\t\tthrow new UsageError(\n\t\t\t\t\t`Method ${name} conflicts with field of the same name in schema ${definition}`,\n\t\t\t\t);\n\t\t\t}\n\t\t\tmethods[name] = zodFunction;\n\t\t}\n\t\tzodType = z.intersection(zodType, z.object(methods));\n\t\tconst methodNote = `Note: this ${typeString} has custom user-defined methods directly on it.`;\n\t\tdescription = description === \"\" ? methodNote : `${description} - ${methodNote}`;\n\t}\n\treturn zodType.describe(description);\n}\n\nfunction getOrCreateType(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tdefinition: string,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst simpleNodeSchema = definitionMap.get(definition) ?? fail(\"Unexpected definition\");\n\treturn getOrCreate(objectCache, simpleNodeSchema, () => {\n\t\t// Handle recursive types: temporarily create a zod \"lazy\" type that can be referenced by a recursive call to getOrCreateType.\n\t\tlet type: ZodTypeAny | undefined;\n\t\tobjectCache.set(\n\t\t\tsimpleNodeSchema,\n\t\t\tz.lazy(() => type ?? fail(\"Recursive type used before creation\")),\n\t\t);\n\t\tswitch (simpleNodeSchema.kind) {\n\t\t\tcase NodeKind.Object: {\n\t\t\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment\n\t\t\t\tconst properties: Record<string, z.ZodTypeAny> = Object.fromEntries(\n\t\t\t\t\t[...simpleNodeSchema.fields]\n\t\t\t\t\t\t.map(([key, field]) => {\n\t\t\t\t\t\t\treturn [\n\t\t\t\t\t\t\t\tkey,\n\t\t\t\t\t\t\t\tgetOrCreateTypeForField(definitionMap, field, objectCache, bindableSchemas),\n\t\t\t\t\t\t\t];\n\t\t\t\t\t\t})\n\t\t\t\t\t\t.filter(([, value]) => value !== undefined),\n\t\t\t\t);\n\n\t\t\t\t// Unlike arrays/maps/records, object nodes include methods directly on them rather than using an intersection\n\t\t\t\tfor (const [name, zodFunction] of getBoundMethods(definition, bindableSchemas)) {\n\t\t\t\t\tif (properties[name] !== undefined) {\n\t\t\t\t\t\tthrow new UsageError(\n\t\t\t\t\t\t\t`Method ${name} conflicts with field of the same name in schema ${definition}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t\tproperties[name] = zodFunction;\n\t\t\t\t}\n\n\t\t\t\treturn (type = z\n\t\t\t\t\t.object(properties)\n\t\t\t\t\t.describe(simpleNodeSchema.metadata?.description ?? \"\"));\n\t\t\t}\n\t\t\tcase NodeKind.Map: {\n\t\t\t\tconst zodType = z.map(\n\t\t\t\t\tz.string(),\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tsimpleNodeSchema.allowedTypesIdentifiers,\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"map\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Record: {\n\t\t\t\tconst zodType = z.record(\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tsimpleNodeSchema.allowedTypesIdentifiers,\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"record\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Array: {\n\t\t\t\tconst zodType = z.array(\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tsimpleNodeSchema.allowedTypesIdentifiers,\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"array\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Leaf: {\n\t\t\t\tswitch (simpleNodeSchema.leafKind) {\n\t\t\t\t\tcase ValueSchema.Boolean: {\n\t\t\t\t\t\treturn (type = z.boolean());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.Number: {\n\t\t\t\t\t\treturn (type = z.number());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.String: {\n\t\t\t\t\t\treturn (type = z.string());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.Null: {\n\t\t\t\t\t\treturn (type = z.null());\n\t\t\t\t\t}\n\t\t\t\t\tdefault: {\n\t\t\t\t\t\tthrow new Error(`Unsupported leaf kind ${NodeKind[simpleNodeSchema.leafKind]}.`);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\tdefault: {\n\t\t\t\treturn unreachableCase(simpleNodeSchema, \"Unknown node kind\");\n\t\t\t}\n\t\t}\n\t});\n}\n\nfunction getOrCreateTypeForField(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tfieldSchema: SimpleFieldSchema,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst customMetadata = fieldSchema.metadata.custom as\n\t\t| Record<string | symbol, unknown>\n\t\t| undefined;\n\tconst getDefault = customMetadata?.[llmDefault];\n\tif (getDefault !== undefined) {\n\t\tif (typeof getDefault !== \"function\") {\n\t\t\tthrow new UsageError(\n\t\t\t\t`Expected value of ${llmDefault.description} property to be a function, but got ${typeof getDefault}`,\n\t\t\t);\n\t\t}\n\n\t\tif (fieldSchema.kind !== FieldKind.Optional) {\n\t\t\tthrow new UsageError(\n\t\t\t\t`The ${llmDefault.description} property is only permitted on optional fields.`,\n\t\t\t);\n\t\t}\n\t}\n\n\tconst field = getTypeForAllowedTypes(\n\t\tdefinitionMap,\n\t\tfieldSchema.allowedTypesIdentifiers,\n\t\tobjectCache,\n\t\tbindableSchemas,\n\t).describe(\n\t\tgetDefault === undefined\n\t\t\t? (fieldSchema.metadata?.description ?? \"\")\n\t\t\t: \"Do not populate this field. It will be automatically supplied by the system after insertion.\",\n\t);\n\n\tswitch (fieldSchema.kind) {\n\t\tcase FieldKind.Required: {\n\t\t\treturn field;\n\t\t}\n\t\tcase FieldKind.Optional: {\n\t\t\treturn field.optional();\n\t\t}\n\t\tcase FieldKind.Identifier: {\n\t\t\treturn field\n\t\t\t\t.optional()\n\t\t\t\t.describe(\n\t\t\t\t\t\"This is an ID automatically generated by the system. Do not supply it when constructing a new object.\",\n\t\t\t\t);\n\t\t}\n\t\tdefault: {\n\t\t\tthrow new Error(`Unsupported field kind ${NodeKind[fieldSchema.kind]}.`);\n\t\t}\n\t}\n}\n\nfunction getTypeForAllowedTypes(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tallowedTypes: ReadonlySet<string>,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst single = tryGetSingleton(allowedTypes);\n\tif (single === undefined) {\n\t\tconst types = [\n\t\t\t...mapIterable(allowedTypes, (name) => {\n\t\t\t\treturn getOrCreateType(definitionMap, name, objectCache, bindableSchemas);\n\t\t\t}),\n\t\t];\n\t\tassert(hasAtLeastTwo(types), 0xa7e /* Expected at least two types */);\n\t\treturn z.union(types);\n\t} else {\n\t\treturn getOrCreateType(definitionMap, single, objectCache, bindableSchemas);\n\t}\n}\n"]}
+{"version":3,"file":"typeGeneration.js","sourceRoot":"","sources":["../src/typeGeneration.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAC;AAC9E,OAAO,EAAE,UAAU,EAAE,MAAM,0CAA0C,CAAC;AACtE,OAAO,EACN,eAAe,EACf,SAAS,EACT,eAAe,EACf,aAAa,EACb,QAAQ,EACR,gBAAgB,EAChB,gBAAgB,EAChB,WAAW,EACX,eAAe,GACf,MAAM,+BAA+B,CAAC;AAQvC,OAAO,EAAE,CAAC,EAAmB,MAAM,KAAK,CAAC;AAGzC,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACxE,OAAO,EACN,IAAI,EACJ,WAAW,EACX,aAAa,EACb,UAAU,EACV,WAAW,EACX,eAAe,GAEf,MAAM,YAAY,CAAC;AAEpB;;;;;;;;;GASG;AAEH;;GAEG;AACH,MAAM,iBAAiB,GAAG,IAAI,OAAO,EAGlC,CAAC;AAEJ;;GAEG;AACH,MAAM,UAAU,0BAA0B,CACzC,UAA+B,EAC/B,MAAwB;IAIxB,OAAO,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,GAAG,EAAE;QAClD,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC7C,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAkB,CAAC;QACpD,eAAe,CAAC,UAAU,EAAE;YAC3B,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;gBACX,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAClB,IACC,CAAC,YAAY,gBAAgB;oBAC7B,CAAC,YAAY,aAAa;oBAC1B,CAAC,YAAY,eAAe;oBAC5B,CAAC,YAAY,gBAAgB,EAC5B,CAAC;oBACF,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBACzB,MAAM,cAAc,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;oBAC5C,KAAK,MAAM,CAAC,IAAI,cAAc,CAAC,eAAe,EAAE,CAAC;wBAChD,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;wBAClB,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBAC1B,CAAC;gBACF,CAAC;YACF,CAAC;SACD,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,CAAC,GAAG,iBAAiB,CAAC,MAAM,EAAE,CAAqB,CAAC;QACxE,MAAM,eAAe,GAAG,IAAI,GAAG,CAC9B,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CACpE,CAAC;QACF,OAAO,iBAAiB,CACvB,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,EACvD,IAAI,GAAG,EAAE,EACT,eAAe,CACf,CAAC;IACH,CAAC,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,iBAAiB,CACzB,OAAmC,EACnC,WAAoD,EACpD,eAA4C;IAI5C,MAAM,WAAW,GAA+B,EAAE,CAAC;IACnD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC9B,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;YAC9C,oIAAoI;YACpI,WAAW,CAAC,IAAI,CAAC,GAAG,eAAe,CAClC,MAAM,CAAC,WAAW,EAClB,IAAI,EACJ,WAAW,EACX,eAAe,CACf,CAAC;QACH,CAAC;IACF,CAAC;IAED,OAAO;QACN,WAAW;KACX,CAAC;AACH,CAAC;AAED,SAAS,0BAA0B,CAAC,cAA8B;IAIjE,MAAM,WAAW,GAA2B,EAAE,CAAC;IAC/C,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAC;IAClD,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;QAClD,0GAA0G;QACzG,WAAmB,CAAC,MAAM,GAAG,MAAM,CAAC;QACrC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,CAAC;AAC3E,CAAC;AAED,SAAS,eAAe,CACvB,UAAkB,EAClB,eAA4C;IAE5C,MAAM,cAAc,GAAG,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACrF,OAAO,0BAA0B,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC;AAC3D,CAAC;AAED,SAAS,8BAA8B,CACtC,UAAsC,EACtC,YAAwB,EACxB,UAAkB,EAClB,gBAAkC,EAClC,eAA4C;IAE5C,IAAI,OAAO,GAAG,YAAY,CAAC;IAC3B,IAAI,WAAW,GAAG,gBAAgB,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC;IAC/D,MAAM,YAAY,GAAG,eAAe,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;IAClE,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAiC,EAAE,CAAC;QACjD,KAAK,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,YAAY,EAAE,CAAC;YAChD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;gBACjC,MAAM,IAAI,UAAU,CACnB,UAAU,IAAI,oDAAoD,UAAU,EAAE,CAC9E,CAAC;YACH,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC;QAC7B,CAAC;QACD,OAAO,GAAG,CAAC,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QACrD,MAAM,UAAU,GAAG,cAAc,UAAU,kDAAkD,CAAC;QAC9F,WAAW,GAAG,WAAW,KAAK,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,WAAW,MAAM,UAAU,EAAE,CAAC;IAClF,CAAC;IACD,OAAO,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,eAAe,CACvB,aAAoD,EACpD,UAAkB,EAClB,WAAoD,EACpD,eAA4C;IAE5C,MAAM,gBAAgB,GAAG,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACxF,OAAO,WAAW,CAAC,WAAW,EAAE,gBAAgB,EAAE,GAAG,EAAE;QACtD,8HAA8H;QAC9H,IAAI,IAA4B,CAAC;QACjC,WAAW,CAAC,GAAG,CACd,gBAAgB,EAChB,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC,qCAAqC,CAAC,CAAC,CACjE,CAAC;QACF,QAAQ,gBAAgB,CAAC,IAAI,EAAE,CAAC;YAC/B,KAAK,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;gBACtB,mEAAmE;gBACnE,MAAM,UAAU,GAAiC,MAAM,CAAC,WAAW,CAClE,CAAC,GAAG,gBAAgB,CAAC,MAAM,CAAC;qBAC1B,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;oBACrB,OAAO;wBACN,GAAG;wBACH,uBAAuB,CAAC,aAAa,EAAE,KAAK,EAAE,WAAW,EAAE,eAAe,CAAC;qBAC3E,CAAC;gBACH,CAAC,CAAC;qBACD,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC,CAC5C,CAAC;gBAEF,8GAA8G;gBAC9G,KAAK,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,eAAe,CAAC,UAAU,EAAE,eAAe,CAAC,EAAE,CAAC;oBAChF,IAAI,UAAU,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;wBACpC,MAAM,IAAI,UAAU,CACnB,UAAU,IAAI,oDAAoD,UAAU,EAAE,CAC9E,CAAC;oBACH,CAAC;oBACD,UAAU,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC;gBAChC,CAAC;gBAED,OAAO,CAAC,IAAI,GAAG,CAAC;qBACd,MAAM,CAAC,UAAU,CAAC;qBAClB,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3D,CAAC;YACD,KAAK,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;gBACnB,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,CACpB,CAAC,CAAC,MAAM,EAAE,EACV,sBAAsB,CACrB,aAAa,EACb,IAAI,GAAG,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,EACnD,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,KAAK,EACL,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;gBACtB,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CACvB,sBAAsB,CACrB,aAAa,EACb,IAAI,GAAG,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,EACnD,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,QAAQ,EACR,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;gBACrB,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CACtB,sBAAsB,CACrB,aAAa,EACb,IAAI,GAAG,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,EACnD,WAAW,EACX,eAAe,CACf,CACD,CAAC;gBACF,OAAO,CAAC,IAAI,GAAG,8BAA8B,CAC5C,OAAO,EACP,OAAO,EACP,UAAU,EACV,gBAAgB,EAChB,eAAe,CACf,CAAC,CAAC;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpB,QAAQ,gBAAgB,CAAC,QAAQ,EAAE,CAAC;oBACnC,KAAK,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;wBAC1B,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC7B,CAAC;oBACD,KAAK,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;wBACzB,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC5B,CAAC;oBACD,KAAK,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;wBACzB,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC5B,CAAC;oBACD,KAAK,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;wBACvB,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC1B,CAAC;oBACD,OAAO,CAAC,CAAC,CAAC;wBACT,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,CAAC,gBAAgB,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;oBAClF,CAAC;gBACF,CAAC;YACF,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACT,OAAO,eAAe,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,CAAC;YAC/D,CAAC;QACF,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAC/B,aAAoD,EACpD,WAA8B,EAC9B,WAAoD,EACpD,eAA4C;IAE5C,MAAM,cAAc,GAAG,WAAW,CAAC,QAAQ,CAAC,MAEhC,CAAC;IACb,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC,UAAU,CAAC,CAAC;IAChD,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC9B,IAAI,OAAO,UAAU,KAAK,UAAU,EAAE,CAAC;YACtC,MAAM,IAAI,UAAU,CACnB,qBAAqB,UAAU,CAAC,WAAW,uCAAuC,OAAO,UAAU,EAAE,CACrG,CAAC;QACH,CAAC;QAED,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,CAAC,QAAQ,EAAE,CAAC;YAC7C,MAAM,IAAI,UAAU,CACnB,OAAO,UAAU,CAAC,WAAW,iDAAiD,CAC9E,CAAC;QACH,CAAC;IACF,CAAC;IAED,MAAM,KAAK,GAAG,sBAAsB,CACnC,aAAa,EACb,IAAI,GAAG,CAAC,WAAW,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,EAC9C,WAAW,EACX,eAAe,CACf,CAAC,QAAQ,CACT,UAAU,KAAK,SAAS;QACvB,CAAC,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,WAAW,IAAI,EAAE,CAAC;QAC3C,CAAC,CAAC,8FAA8F,CACjG,CAAC;IAEF,QAAQ,WAAW,CAAC,IAAI,EAAE,CAAC;QAC1B,KAAK,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzB,OAAO,KAAK,CAAC;QACd,CAAC;QACD,KAAK,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzB,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;QACzB,CAAC;QACD,KAAK,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3B,OAAO,KAAK;iBACV,QAAQ,EAAE;iBACV,QAAQ,CACR,uGAAuG,CACvG,CAAC;QACJ,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1E,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,sBAAsB,CAC9B,aAAoD,EACpD,YAAiC,EACjC,WAAoD,EACpD,eAA4C;IAE5C,MAAM,MAAM,GAAG,eAAe,CAAC,YAAY,CAAC,CAAC;IAC7C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG;YACb,GAAG,WAAW,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE;gBACrC,OAAO,eAAe,CAAC,aAAa,EAAE,IAAI,EAAE,WAAW,EAAE,eAAe,CAAC,CAAC;YAC3E,CAAC,CAAC;SACF,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACtE,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;SAAM,CAAC;QACP,OAAO,eAAe,CAAC,aAAa,EAAE,MAAM,EAAE,WAAW,EAAE,eAAe,CAAC,CAAC;IAC7E,CAAC;AACF,CAAC","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\nimport { assert, unreachableCase } from \"@fluidframework/core-utils/internal\";\nimport { UsageError } from \"@fluidframework/telemetry-utils/internal\";\nimport {\n\tArrayNodeSchema,\n\tFieldKind,\n\tgetSimpleSchema,\n\tMapNodeSchema,\n\tNodeKind,\n\tObjectNodeSchema,\n\tRecordNodeSchema,\n\tValueSchema,\n\twalkFieldSchema,\n} from \"@fluidframework/tree/internal\";\nimport type {\n\tImplicitFieldSchema,\n\tSimpleFieldSchema,\n\tSimpleNodeSchema,\n\tSimpleTreeSchema,\n\tTreeNodeSchema,\n} from \"@fluidframework/tree/internal\";\nimport { z, type ZodTypeAny } from \"zod\";\n\nimport type { BindableSchema } from \"./methodBinding.js\";\nimport { FunctionWrapper, getExposedMethods } from \"./methodBinding.js\";\nimport {\n\tfail,\n\tgetOrCreate,\n\thasAtLeastTwo,\n\tllmDefault,\n\tmapIterable,\n\ttryGetSingleton,\n\ttype MapGetSet,\n} from \"./utils.js\";\n\n/**\n *\n * TODO: Add a prompt suggestion API!\n *\n * TODO: Handle rate limit errors.\n *\n * TODO: Pass descriptions from schema metadata to the generated TS types that we put in the prompt\n *\n * TODO make the Ids be \"Vector-2\" instead of \"Vector2\" (or else it gets weird when you have a type called \"Vector2\")\n */\n\n/**\n * Cache used to prevent repeatedly generating the same Zod validation objects for the same {@link SimpleTreeSchema} as generate propts for repeated calls to an LLM\n */\nconst promptSchemaCache = new WeakMap<\n\tSimpleTreeSchema,\n\tReturnType<typeof generateEditTypes>\n>();\n\n/**\n * TODO\n */\nexport function generateEditTypesForPrompt(\n\trootSchema: ImplicitFieldSchema,\n\tschema: SimpleTreeSchema,\n): {\n\tdomainTypes: Record<string, ZodTypeAny>;\n} {\n\treturn getOrCreate(promptSchemaCache, schema, () => {\n\t\tconst allSchemas = new Set<TreeNodeSchema>();\n\t\tconst objectTypeSchemas = new Set<TreeNodeSchema>();\n\t\twalkFieldSchema(rootSchema, {\n\t\t\tnode: (n) => {\n\t\t\t\tallSchemas.add(n);\n\t\t\t\tif (\n\t\t\t\t\tn instanceof ObjectNodeSchema ||\n\t\t\t\t\tn instanceof MapNodeSchema ||\n\t\t\t\t\tn instanceof ArrayNodeSchema ||\n\t\t\t\t\tn instanceof RecordNodeSchema\n\t\t\t\t) {\n\t\t\t\t\tobjectTypeSchemas.add(n);\n\t\t\t\t\tconst exposedMethods = getExposedMethods(n);\n\t\t\t\t\tfor (const t of exposedMethods.referencedTypes) {\n\t\t\t\t\t\tallSchemas.add(t);\n\t\t\t\t\t\tobjectTypeSchemas.add(t);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t},\n\t\t});\n\t\tconst nodeSchemas = [...objectTypeSchemas.values()] as BindableSchema[];\n\t\tconst bindableSchemas = new Map<string, BindableSchema>(\n\t\t\tnodeSchemas.map((nodeSchema) => [nodeSchema.identifier, nodeSchema]),\n\t\t);\n\t\treturn generateEditTypes(\n\t\t\t[...allSchemas.values()].map((s) => getSimpleSchema(s)),\n\t\t\tnew Map(),\n\t\t\tbindableSchemas,\n\t\t);\n\t});\n}\n\n/**\n * Generates a set of ZOD validation objects for the various types of data that can be put into the provided {@link SimpleTreeSchema}\n * and then uses those sets to generate an all-encompassing ZOD object for each type of {@link TreeEdit} that can validate any of the types of data that can be put into the tree.\n *\n * @returns a Record of schema names to Zod validation objects, and the name of the root schema used to encompass all of the other schemas.\n *\n * @remarks The return type of this function is designed to work with Typechat's createZodJsonValidator as well as be used as the JSON schema for OpenAi's structured output response format.\n */\nfunction generateEditTypes(\n\tschemas: Iterable<SimpleTreeSchema>,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): {\n\tdomainTypes: Record<string, ZodTypeAny>;\n} {\n\tconst domainTypes: Record<string, ZodTypeAny> = {};\n\tfor (const schema of schemas) {\n\t\tfor (const name of schema.definitions.keys()) {\n\t\t\t// If this does overwrite anything in domainTypes, it is guaranteed to be overwritten with an identical value due to the getOrCreate\n\t\t\tdomainTypes[name] = getOrCreateType(\n\t\t\t\tschema.definitions,\n\t\t\t\tname,\n\t\t\t\tobjectCache,\n\t\t\t\tbindableSchemas,\n\t\t\t);\n\t\t}\n\t}\n\n\treturn {\n\t\tdomainTypes,\n\t};\n}\n\nfunction getBoundMethodsForBindable(bindableSchema: BindableSchema): {\n\treferencedTypes: Set<TreeNodeSchema>;\n\tmethods: [string, ZodTypeAny][];\n} {\n\tconst methodTypes: [string, ZodTypeAny][] = [];\n\tconst methods = getExposedMethods(bindableSchema);\n\tfor (const [name, method] of Object.entries(methods.methods)) {\n\t\tconst zodFunction = z.instanceof(FunctionWrapper);\n\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access, @typescript-eslint/no-explicit-any\n\t\t(zodFunction as any).method = method;\n\t\tmethodTypes.push([name, zodFunction]);\n\t}\n\treturn { methods: methodTypes, referencedTypes: methods.referencedTypes };\n}\n\nfunction getBoundMethods(\n\tdefinition: string,\n\tbindableSchemas: Map<string, BindableSchema>,\n): [string, ZodTypeAny][] {\n\tconst bindableSchema = bindableSchemas.get(definition) ?? fail(\"Unknown definition\");\n\treturn getBoundMethodsForBindable(bindableSchema).methods;\n}\n\nfunction addBindingIntersectionIfNeeded(\n\ttypeString: \"array\" | \"map\" | \"record\",\n\tzodTypeBound: ZodTypeAny,\n\tdefinition: string,\n\tsimpleNodeSchema: SimpleNodeSchema,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tlet zodType = zodTypeBound;\n\tlet description = simpleNodeSchema.metadata?.description ?? \"\";\n\tconst boundMethods = getBoundMethods(definition, bindableSchemas);\n\tif (boundMethods.length > 0) {\n\t\tconst methods: Record<string, z.ZodTypeAny> = {};\n\t\tfor (const [name, zodFunction] of boundMethods) {\n\t\t\tif (methods[name] !== undefined) {\n\t\t\t\tthrow new UsageError(\n\t\t\t\t\t`Method ${name} conflicts with field of the same name in schema ${definition}`,\n\t\t\t\t);\n\t\t\t}\n\t\t\tmethods[name] = zodFunction;\n\t\t}\n\t\tzodType = z.intersection(zodType, z.object(methods));\n\t\tconst methodNote = `Note: this ${typeString} has custom user-defined methods directly on it.`;\n\t\tdescription = description === \"\" ? methodNote : `${description} - ${methodNote}`;\n\t}\n\treturn zodType.describe(description);\n}\n\nfunction getOrCreateType(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tdefinition: string,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst simpleNodeSchema = definitionMap.get(definition) ?? fail(\"Unexpected definition\");\n\treturn getOrCreate(objectCache, simpleNodeSchema, () => {\n\t\t// Handle recursive types: temporarily create a zod \"lazy\" type that can be referenced by a recursive call to getOrCreateType.\n\t\tlet type: ZodTypeAny | undefined;\n\t\tobjectCache.set(\n\t\t\tsimpleNodeSchema,\n\t\t\tz.lazy(() => type ?? fail(\"Recursive type used before creation\")),\n\t\t);\n\t\tswitch (simpleNodeSchema.kind) {\n\t\t\tcase NodeKind.Object: {\n\t\t\t\t// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment\n\t\t\t\tconst properties: Record<string, z.ZodTypeAny> = Object.fromEntries(\n\t\t\t\t\t[...simpleNodeSchema.fields]\n\t\t\t\t\t\t.map(([key, field]) => {\n\t\t\t\t\t\t\treturn [\n\t\t\t\t\t\t\t\tkey,\n\t\t\t\t\t\t\t\tgetOrCreateTypeForField(definitionMap, field, objectCache, bindableSchemas),\n\t\t\t\t\t\t\t];\n\t\t\t\t\t\t})\n\t\t\t\t\t\t.filter(([, value]) => value !== undefined),\n\t\t\t\t);\n\n\t\t\t\t// Unlike arrays/maps/records, object nodes include methods directly on them rather than using an intersection\n\t\t\t\tfor (const [name, zodFunction] of getBoundMethods(definition, bindableSchemas)) {\n\t\t\t\t\tif (properties[name] !== undefined) {\n\t\t\t\t\t\tthrow new UsageError(\n\t\t\t\t\t\t\t`Method ${name} conflicts with field of the same name in schema ${definition}`,\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t\tproperties[name] = zodFunction;\n\t\t\t\t}\n\n\t\t\t\treturn (type = z\n\t\t\t\t\t.object(properties)\n\t\t\t\t\t.describe(simpleNodeSchema.metadata?.description ?? \"\"));\n\t\t\t}\n\t\t\tcase NodeKind.Map: {\n\t\t\t\tconst zodType = z.map(\n\t\t\t\t\tz.string(),\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tnew Set(simpleNodeSchema.simpleAllowedTypes.keys()),\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"map\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Record: {\n\t\t\t\tconst zodType = z.record(\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tnew Set(simpleNodeSchema.simpleAllowedTypes.keys()),\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"record\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Array: {\n\t\t\t\tconst zodType = z.array(\n\t\t\t\t\tgetTypeForAllowedTypes(\n\t\t\t\t\t\tdefinitionMap,\n\t\t\t\t\t\tnew Set(simpleNodeSchema.simpleAllowedTypes.keys()),\n\t\t\t\t\t\tobjectCache,\n\t\t\t\t\t\tbindableSchemas,\n\t\t\t\t\t),\n\t\t\t\t);\n\t\t\t\treturn (type = addBindingIntersectionIfNeeded(\n\t\t\t\t\t\"array\",\n\t\t\t\t\tzodType,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tsimpleNodeSchema,\n\t\t\t\t\tbindableSchemas,\n\t\t\t\t));\n\t\t\t}\n\t\t\tcase NodeKind.Leaf: {\n\t\t\t\tswitch (simpleNodeSchema.leafKind) {\n\t\t\t\t\tcase ValueSchema.Boolean: {\n\t\t\t\t\t\treturn (type = z.boolean());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.Number: {\n\t\t\t\t\t\treturn (type = z.number());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.String: {\n\t\t\t\t\t\treturn (type = z.string());\n\t\t\t\t\t}\n\t\t\t\t\tcase ValueSchema.Null: {\n\t\t\t\t\t\treturn (type = z.null());\n\t\t\t\t\t}\n\t\t\t\t\tdefault: {\n\t\t\t\t\t\tthrow new Error(`Unsupported leaf kind ${NodeKind[simpleNodeSchema.leafKind]}.`);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\tdefault: {\n\t\t\t\treturn unreachableCase(simpleNodeSchema, \"Unknown node kind\");\n\t\t\t}\n\t\t}\n\t});\n}\n\nfunction getOrCreateTypeForField(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tfieldSchema: SimpleFieldSchema,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst customMetadata = fieldSchema.metadata.custom as\n\t\t| Record<string | symbol, unknown>\n\t\t| undefined;\n\tconst getDefault = customMetadata?.[llmDefault];\n\tif (getDefault !== undefined) {\n\t\tif (typeof getDefault !== \"function\") {\n\t\t\tthrow new UsageError(\n\t\t\t\t`Expected value of ${llmDefault.description} property to be a function, but got ${typeof getDefault}`,\n\t\t\t);\n\t\t}\n\n\t\tif (fieldSchema.kind !== FieldKind.Optional) {\n\t\t\tthrow new UsageError(\n\t\t\t\t`The ${llmDefault.description} property is only permitted on optional fields.`,\n\t\t\t);\n\t\t}\n\t}\n\n\tconst field = getTypeForAllowedTypes(\n\t\tdefinitionMap,\n\t\tnew Set(fieldSchema.simpleAllowedTypes.keys()),\n\t\tobjectCache,\n\t\tbindableSchemas,\n\t).describe(\n\t\tgetDefault === undefined\n\t\t\t? (fieldSchema.metadata?.description ?? \"\")\n\t\t\t: \"Do not populate this field. It will be automatically supplied by the system after insertion.\",\n\t);\n\n\tswitch (fieldSchema.kind) {\n\t\tcase FieldKind.Required: {\n\t\t\treturn field;\n\t\t}\n\t\tcase FieldKind.Optional: {\n\t\t\treturn field.optional();\n\t\t}\n\t\tcase FieldKind.Identifier: {\n\t\t\treturn field\n\t\t\t\t.optional()\n\t\t\t\t.describe(\n\t\t\t\t\t\"This is an ID automatically generated by the system. Do not supply it when constructing a new object.\",\n\t\t\t\t);\n\t\t}\n\t\tdefault: {\n\t\t\tthrow new Error(`Unsupported field kind ${NodeKind[fieldSchema.kind]}.`);\n\t\t}\n\t}\n}\n\nfunction getTypeForAllowedTypes(\n\tdefinitionMap: ReadonlyMap<string, SimpleNodeSchema>,\n\tallowedTypes: ReadonlySet<string>,\n\tobjectCache: MapGetSet<SimpleNodeSchema, ZodTypeAny>,\n\tbindableSchemas: Map<string, BindableSchema>,\n): ZodTypeAny {\n\tconst single = tryGetSingleton(allowedTypes);\n\tif (single === undefined) {\n\t\tconst types = [\n\t\t\t...mapIterable(allowedTypes, (name) => {\n\t\t\t\treturn getOrCreateType(definitionMap, name, objectCache, bindableSchemas);\n\t\t\t}),\n\t\t];\n\t\tassert(hasAtLeastTwo(types), 0xa7e /* Expected at least two types */);\n\t\treturn z.union(types);\n\t} else {\n\t\treturn getOrCreateType(definitionMap, single, objectCache, bindableSchemas);\n\t}\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fluidframework/tree-agent",
-  "version": "2.70.0",
+  "version": "2.71.0",
   "description": "Experimental package to simplify integrating AI into Fluid-based applications",
   "homepage": "https://fluidframework.com",
   "repository": {
@@ -70,24 +70,24 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.39.0",
-    "@fluidframework/core-utils": "~2.70.0",
-    "@fluidframework/runtime-utils": "~2.70.0",
-    "@fluidframework/telemetry-utils": "~2.70.0",
-    "@fluidframework/tree": "~2.70.0",
+    "@fluidframework/core-utils": "~2.71.0",
+    "@fluidframework/runtime-utils": "~2.71.0",
+    "@fluidframework/telemetry-utils": "~2.71.0",
+    "@fluidframework/tree": "~2.71.0",
     "uuid": "^11.1.0",
     "zod": "^3.25.32"
   },
   "devDependencies": {
     "@arethetypeswrong/cli": "^0.17.1",
     "@biomejs/biome": "~1.9.3",
-    "@fluid-internal/mocha-test-setup": "~2.70.0",
+    "@fluid-internal/mocha-test-setup": "~2.71.0",
     "@fluid-tools/build-cli": "^0.58.3",
     "@fluidframework/build-common": "^2.0.3",
     "@fluidframework/build-tools": "^0.58.3",
-    "@fluidframework/eslint-config-fluid": "^6.1.0",
-    "@fluidframework/id-compressor": "~2.70.0",
-    "@fluidframework/runtime-utils": "~2.70.0",
-    "@fluidframework/test-runtime-utils": "~2.70.0",
+    "@fluidframework/eslint-config-fluid": "^7.0.0",
+    "@fluidframework/id-compressor": "~2.71.0",
+    "@fluidframework/runtime-utils": "~2.71.0",
+    "@fluidframework/test-runtime-utils": "~2.71.0",
     "@langchain/anthropic": "^0.3.24",
     "@langchain/core": "^0.3.78",
     "@langchain/google-genai": "^0.2.16",
@@ -149,8 +149,8 @@
     "check:format": "npm run check:biome",
     "ci:build:docs": "api-extractor run",
     "clean": "rimraf --glob dist lib {alpha,beta,internal,legacy}.d.ts \"**/*.tsbuildinfo\" \"**/*.build.log\" _api-extractor-temp nyc",
-    "eslint": "eslint --format stylish src",
-    "eslint:fix": "eslint --format stylish src --fix --fix-type problem,suggestion,layout",
+    "eslint": "eslint --quiet --format stylish src",
+    "eslint:fix": "eslint --quiet --format stylish src --fix --fix-type problem,suggestion,layout",
     "format": "npm run format:biome",
     "format:biome": "biome check . --write",
     "lint": "fluid-build . --task lint",

package/src/typeGeneration.ts

@@ -228,7 +228,7 @@ function getOrCreateType(
 					z.string(),
 					getTypeForAllowedTypes(
 						definitionMap,
-						simpleNodeSchema.allowedTypesIdentifiers,
+						new Set(simpleNodeSchema.simpleAllowedTypes.keys()),
 						objectCache,
 						bindableSchemas,
 					),
@@ -245,7 +245,7 @@ function getOrCreateType(
 				const zodType = z.record(
 					getTypeForAllowedTypes(
 						definitionMap,
-						simpleNodeSchema.allowedTypesIdentifiers,
+						new Set(simpleNodeSchema.simpleAllowedTypes.keys()),
 						objectCache,
 						bindableSchemas,
 					),
@@ -262,7 +262,7 @@ function getOrCreateType(
 				const zodType = z.array(
 					getTypeForAllowedTypes(
 						definitionMap,
-						simpleNodeSchema.allowedTypesIdentifiers,
+						new Set(simpleNodeSchema.simpleAllowedTypes.keys()),
 						objectCache,
 						bindableSchemas,
 					),
@@ -327,7 +327,7 @@ function getOrCreateTypeForField(
 
 	const field = getTypeForAllowedTypes(
 		definitionMap,
-		fieldSchema.allowedTypesIdentifiers,
+		new Set(fieldSchema.simpleAllowedTypes.keys()),
 		objectCache,
 		bindableSchemas,
 	).describe(