@algolia/wizard 0.20.0 → 0.21.0

package/dist/main.js

@@ -3263,10 +3263,18 @@ var RECORD_MODEL = "claude-haiku-4-5";
 var MAX_RECORDS = 100;
 var BATCH_SIZE = 10;
 var MAX_BATCH_ATTEMPTS = 3;
-var anthropic = createAnthropic({
-  apiKey: process.env.PROVIDER_API_KEY ?? ""
-});
-function generateRecordTool(ctx) {
+function defaultCreateModel() {
+  const token = getAuthToken();
+  if (!token) {
+    throw new Error("Not authenticated: no user token available");
+  }
+  return createAnthropic({
+    apiKey: token,
+    baseURL: PROXY_BASE_URL,
+    fetch: proxyFetch
+  });
+}
+function generateRecordTool(ctx, createModel = defaultCreateModel) {
   return tool10({
     description: "Generate realistic sample records for an entity and write them to a JSON file in the worktree. Provide the entity name and its attributes; this tool asks a model to invent varied, realistic values, each with a unique objectID, and returns the file path to read them from at runtime. Do not invent the record values or objectIDs yourself, and do not inline the returned records into the script \u2014 call this tool and read the file it writes.",
     inputSchema: z17.object({
@@ -3278,6 +3286,7 @@ function generateRecordTool(ctx) {
     execute: async ({ entityName, attributes, count, hint }) => {
       logger.info({ entityName, count }, "called generateRecord tool");
       try {
+        const anthropic = createModel();
         const value = z17.union([z17.string(), z17.number(), z17.boolean(), z17.null()]);
         const recordSchema = z17.object(
           Object.fromEntries(attributes.map((attr) => [attr, value]))
@@ -3296,10 +3305,6 @@ function generateRecordTool(ctx) {
                 prompt: [
                   `Generate ${batchCount} realistic, varied sample records for the "${entityName}" entity.`,
                   hint ? `Context: ${hint}` : "",
-                  // Batches run concurrently with identical prompts, so each call
-                  // would otherwise drift to the same high-probability values and
-                  // collide across batches. This per-batch seed pushes each call
-                  // into a different region of the output space.
                   `Variety seed: ${nanoid2()}. Use it to diversify values.`
                 ].filter(Boolean).join("\n")
               });
@@ -3332,8 +3337,15 @@ function generateRecordTool(ctx) {
           return `Refused: ${resolved2.target} is outside the repo root (${ctx.root}).`;
         }
         await mkdir5(dirname6(resolved2.target), { recursive: true });
-        await writeFile6(resolved2.target, JSON.stringify(records, null, 2), "utf8");
-        logger.info({ entityName, count: records.length, relPath }, "generateRecord wrote records to disk");
+        await writeFile6(
+          resolved2.target,
+          JSON.stringify(records, null, 2),
+          "utf8"
+        );
+        logger.info(
+          { entityName, count: records.length, relPath },
+          "generateRecord wrote records to disk"
+        );
         return {
           filePath: relPath,
           count: records.length,
@@ -3427,7 +3439,7 @@ async function runAgent(req) {
   if (!token) {
     throw new Error("Not authenticated: no user token available");
   }
-  const anthropic2 = createAnthropic2({
+  const anthropic = createAnthropic2({
     apiKey: token,
     baseURL: PROXY_BASE_URL,
     fetch: proxyFetch
@@ -3443,7 +3455,7 @@ async function runAgent(req) {
     "Call notifyUser whenever you start a new phase of work or your focus shifts (e.g. moving from reading code to writing files) \u2014 a short, plain-language, high-level update. Do not call it for every tool use."
   ];
   const agent = new ToolLoopAgent({
-    model: anthropic2(MODEL_BY_SIZE[req.modelSize ?? "medium"]),
+    model: anthropic(MODEL_BY_SIZE[req.modelSize ?? "medium"]),
     // Cache tools + system on the last system block. Tools render before
     // system, so one breakpoint here caches both, reused on every loop turn
     // after the first.
@@ -3630,7 +3642,7 @@ async function runAnalysis(mode, extraInstructions = []) {
 // package.json
 var package_default = {
   name: "@algolia/wizard",
-  version: "0.20.0",
+  version: "0.21.0",
   description: "Magically implement Algolia functionality in your codebase",
   type: "module",
   engines: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@algolia/wizard",
-  "version": "0.20.0",
+  "version": "0.21.0",
   "description": "Magically implement Algolia functionality in your codebase",
   "type": "module",
   "engines": {