@xsai/generate-object 0.3.3 → 0.3.5

package/dist/index.d.ts

@@ -5,6 +5,8 @@ interface GenerateObjectOptions<T extends Schema> extends GenerateTextOptions {
     schema: T;
     schemaDescription?: string;
     schemaName?: string;
+    /** @default true */
+    strict?: boolean;
 }
 type GenerateObjectResult<O> = GenerateTextResult & {
     object: O;

package/dist/index.js

@@ -13,8 +13,9 @@ const wrap = (items) => ({
 });
 
 async function generateObject(options) {
-  const { schema: schemaValidator } = options;
-  let schema = await toJsonSchema(schemaValidator).then((schema2) => strictJsonSchema(schema2));
+  let schema = await toJsonSchema(options.schema);
+  if (options.strict !== false)
+    schema = strictJsonSchema(schema);
   if (options.output === "array")
     schema = wrap(schema);
   return generateText({
@@ -24,7 +25,7 @@ async function generateObject(options) {
         description: options.schemaDescription,
         name: options.schemaName ?? "json_schema",
         schema,
-        strict: true
+        strict: options.strict ?? true
       },
       type: "json_schema"
     },
@@ -32,15 +33,17 @@ async function generateObject(options) {
     // Remove schema from options
     schemaDescription: void 0,
     // Remove schemaDescription from options
-    schemaName: void 0
+    schemaName: void 0,
     // Remove schemaName from options
+    strict: void 0
+    // Remove strict from options
   }).then(async ({ finishReason, messages, steps, text, toolCalls, toolResults, usage }) => {
     const json = JSON.parse(text);
     if (options.output === "array") {
       return {
         finishReason,
         messages,
-        object: await Promise.all(json.elements.map(async (element) => validate(schemaValidator, element))),
+        object: await Promise.all(json.elements.map(async (element) => validate(options.schema, element))),
         steps,
         text,
         toolCalls,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@xsai/generate-object",
   "type": "module",
-  "version": "0.3.3",
+  "version": "0.3.5",
   "description": "extra-small AI SDK.",
   "author": "Moeru AI",
   "license": "MIT",
@@ -29,8 +29,8 @@
     "dist"
   ],
   "dependencies": {
-    "@xsai/generate-text": "~0.3.3",
-    "xsschema": "~0.3.3"
+    "@xsai/generate-text": "~0.3.5",
+    "xsschema": "~0.3.5"
   },
   "devDependencies": {
     "@valibot/to-json-schema": "^1.0.0",