zangief 0.1.0 → 0.1.1

package/dist/index.js

@@ -60866,7 +60866,11 @@ AVAILABLE GADGETS`);
 `);
         for (const gadget of gadgets) {
           const gadgetName = gadget.name ?? gadget.constructor.name;
-          const instruction = gadget.getInstruction(this.argPrefix);
+          const instruction = gadget.getInstruction({
+            argPrefix: this.argPrefix,
+            startPrefix: this.startPrefix,
+            endPrefix: this.endPrefix
+          });
           const schemaMarker = `
 
 Input Schema (BLOCK):`;
@@ -64792,13 +64796,13 @@ var init_openai = __esm2({
       async generateSpeech(options) {
         const client = this.client;
         const spec = getOpenAISpeechModelSpec(options.model);
-        const format = options.responseFormat ?? spec?.defaultFormat ?? "mp3";
+        const format2 = options.responseFormat ?? spec?.defaultFormat ?? "mp3";
         const voice = options.voice ?? spec?.defaultVoice ?? "alloy";
         const response = await client.audio.speech.create({
           model: options.model,
           input: options.input,
           voice,
-          response_format: format,
+          response_format: format2,
           speed: options.speed ?? 1
         });
         const audioBuffer = await response.arrayBuffer();
@@ -64810,7 +64814,7 @@ var init_openai = __esm2({
             characterCount: options.input.length
           },
           cost,
-          format
+          format: format2
         };
       }
       buildApiRequest(options, descriptor, spec, messages) {
@@ -66320,7 +66324,8 @@ var init_executor = __esm2({
           } else {
             validatedParameters = schemaAwareParameters;
           }
-          const timeoutMs = gadget.timeoutMs ?? this.defaultGadgetTimeoutMs;
+          const subagentTimeout = this.subagentConfig?.[call.gadgetName]?.timeoutMs;
+          const timeoutMs = subagentTimeout ?? gadget.timeoutMs ?? this.defaultGadgetTimeoutMs;
           const abortController = new AbortController;
           let callbackCost = 0;
           const reportCost = (amount) => {
@@ -66347,7 +66352,8 @@ var init_executor = __esm2({
             nodeId: gadgetNodeId,
             depth: gadgetDepth,
             hostExports: getHostExportsInternal(),
-            logger: this.logger
+            logger: this.logger,
+            requestHumanInput: this.requestHumanInput
           };
           let rawResult;
           if (timeoutMs && timeoutMs > 0) {
@@ -66543,6 +66549,9 @@ var init_stream_processor = __esm2({
       failedInvocations = /* @__PURE__ */ new Set;
       inFlightExecutions = /* @__PURE__ */ new Map;
       completedResultsQueue = [];
+      subagentConfig;
+      activeCountByGadget = /* @__PURE__ */ new Map;
+      concurrencyQueue = /* @__PURE__ */ new Map;
       priorCompletedInvocations;
       priorFailedInvocations;
       constructor(options) {
@@ -66555,6 +66564,7 @@ var init_stream_processor = __esm2({
         this.baseDepth = options.baseDepth ?? 0;
         this.priorCompletedInvocations = options.priorCompletedInvocations ?? /* @__PURE__ */ new Set;
         this.priorFailedInvocations = options.priorFailedInvocations ?? /* @__PURE__ */ new Set;
+        this.subagentConfig = options.subagentConfig;
         this.parser = new GadgetCallParser({
           startPrefix: options.gadgetStartPrefix,
           endPrefix: options.gadgetEndPrefix,
@@ -66779,9 +66789,53 @@ var init_stream_processor = __esm2({
           }
           return;
         }
-        const executionPromise = this.executeGadgetAndCollect(call);
+        const limit2 = this.getConcurrencyLimit(call.gadgetName);
+        const activeCount = this.activeCountByGadget.get(call.gadgetName) ?? 0;
+        if (limit2 > 0 && activeCount >= limit2) {
+          this.logger.debug("Gadget queued due to concurrency limit", {
+            gadgetName: call.gadgetName,
+            invocationId: call.invocationId,
+            activeCount,
+            limit: limit2
+          });
+          const queue = this.concurrencyQueue.get(call.gadgetName) ?? [];
+          queue.push(call);
+          this.concurrencyQueue.set(call.gadgetName, queue);
+          return;
+        }
+        this.startGadgetWithConcurrencyTracking(call);
+      }
+      getConcurrencyLimit(gadgetName) {
+        const config2 = this.subagentConfig?.[gadgetName];
+        return config2?.maxConcurrent ?? 0;
+      }
+      startGadgetWithConcurrencyTracking(call) {
+        const gadgetName = call.gadgetName;
+        const currentCount = this.activeCountByGadget.get(gadgetName) ?? 0;
+        this.activeCountByGadget.set(gadgetName, currentCount + 1);
+        const executionPromise = this.executeGadgetAndCollect(call).finally(() => {
+          const newCount = (this.activeCountByGadget.get(gadgetName) ?? 1) - 1;
+          this.activeCountByGadget.set(gadgetName, newCount);
+          this.processQueuedGadget(gadgetName);
+        });
         this.inFlightExecutions.set(call.invocationId, executionPromise);
       }
+      processQueuedGadget(gadgetName) {
+        const queue = this.concurrencyQueue.get(gadgetName);
+        if (!queue || queue.length === 0)
+          return;
+        const limit2 = this.getConcurrencyLimit(gadgetName);
+        const activeCount = this.activeCountByGadget.get(gadgetName) ?? 0;
+        if (limit2 === 0 || activeCount < limit2) {
+          const nextCall = queue.shift();
+          this.logger.debug("Processing queued gadget", {
+            gadgetName,
+            invocationId: nextCall.invocationId,
+            remainingInQueue: queue.length
+          });
+          this.startGadgetWithConcurrencyTracking(nextCall);
+        }
+      }
       async* executeGadgetGenerator(call) {
         if (call.parseError) {
           this.logger.warn("Gadget has parse error", {
@@ -66946,27 +67000,49 @@ var init_stream_processor = __esm2({
         }
       }
       async* waitForInFlightExecutions() {
-        if (this.inFlightExecutions.size === 0) {
+        if (this.inFlightExecutions.size === 0 && !this.hasQueuedGadgets()) {
           return;
         }
         this.logger.debug("Waiting for in-flight gadget executions", {
           count: this.inFlightExecutions.size,
-          invocationIds: Array.from(this.inFlightExecutions.keys())
+          invocationIds: Array.from(this.inFlightExecutions.keys()),
+          queuedCount: this.getQueuedGadgetCount()
         });
-        const allDone = Promise.all(this.inFlightExecutions.values()).then(() => "done");
         const POLL_INTERVAL_MS = 100;
-        while (true) {
+        while (this.inFlightExecutions.size > 0 || this.hasQueuedGadgets()) {
+          const allDone = this.inFlightExecutions.size > 0 ? Promise.all(this.inFlightExecutions.values()).then(() => "done") : Promise.resolve("done");
           const result = await Promise.race([
             allDone,
             new Promise((resolve) => setTimeout(() => resolve("poll"), POLL_INTERVAL_MS))
           ]);
           yield* this.drainCompletedResults();
-          if (result === "done") {
+          if (result === "done" && this.getTotalActiveGadgetCount() === 0 && !this.hasQueuedGadgets()) {
             break;
           }
         }
         this.inFlightExecutions.clear();
       }
+      hasQueuedGadgets() {
+        for (const queue of this.concurrencyQueue.values()) {
+          if (queue.length > 0)
+            return true;
+        }
+        return false;
+      }
+      getQueuedGadgetCount() {
+        let count = 0;
+        for (const queue of this.concurrencyQueue.values()) {
+          count += queue.length;
+        }
+        return count;
+      }
+      getTotalActiveGadgetCount() {
+        let total = 0;
+        for (const count of this.activeCountByGadget.values()) {
+          total += count;
+        }
+        return total;
+      }
       async handleFailedDependency(call, failedDep) {
         const events = [];
         const depResult = this.completedResults.get(failedDep);
@@ -68010,6 +68086,9 @@ var init_builder = __esm2({
         if (ctx.signal && !this.signal) {
           this.signal = ctx.signal;
         }
+        if (ctx.logger && !this.logger) {
+          this.logger = ctx.logger;
+        }
         return this;
       }
       withTrailingMessage(message) {
@@ -68237,6 +68316,30 @@ init_anthropic();
 init_discovery();
 init_gemini();
 init_openai();
+function formatBytes(bytes2, decimals = 1) {
+  if (bytes2 === 0)
+    return "0 B";
+  const k = 1024;
+  const sizes = ["B", "KB", "MB", "GB", "TB", "PB"];
+  const i2 = Math.floor(Math.log(bytes2) / Math.log(k));
+  const size = bytes2 / Math.pow(k, i2);
+  const formatted = size % 1 === 0 ? size.toString() : size.toFixed(decimals);
+  return `${formatted} ${sizes[i2]}`;
+}
+function formatDate(isoDate, options = {
+  year: "numeric",
+  month: "short",
+  day: "numeric",
+  hour: "2-digit",
+  minute: "2-digit"
+}) {
+  try {
+    const date5 = new Date(isoDate);
+    return date5.toLocaleString(undefined, options);
+  } catch {
+    return isoDate;
+  }
+}
 
 // src/client.ts
 var SOURCEGRAPH_API_URL = "https://sourcegraph.com/.api/graphql";
@@ -68426,27 +68529,6 @@ function formatFileContent(repo, path5, content, byteSize) {
   return lines.join(`
 `);
 }
-function formatBytes(bytes2) {
-  if (bytes2 < 1024)
-    return `${bytes2} bytes`;
-  if (bytes2 < 1024 * 1024)
-    return `${(bytes2 / 1024).toFixed(1)} KB`;
-  return `${(bytes2 / (1024 * 1024)).toFixed(1)} MB`;
-}
-function formatDate(isoDate) {
-  try {
-    const date5 = new Date(isoDate);
-    return date5.toLocaleDateString("en-US", {
-      year: "numeric",
-      month: "short",
-      day: "numeric",
-      hour: "2-digit",
-      minute: "2-digit"
-    });
-  } catch {
-    return isoDate;
-  }
-}
 
 // src/gadgets/search.ts
 var SEARCH_QUERY = `

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zangief",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "type": "module",
   "description": "llmist gadgets for exploring GitHub repos via Sourcegraph API",
   "main": "dist/index.js",
@@ -15,7 +15,7 @@
     "dist"
   ],
   "scripts": {
-    "build": "bun build src/index.ts --outdir dist --target node",
+    "build": "bun build src/index.ts --outdir dist --target node --external:tiktoken",
     "dev": "bun run src/index.ts",
     "demo": "bun run examples/demo.ts",
     "typecheck": "tsc --noEmit"