@juspay/neurolink 11.11.2 → 11.11.4

package/dist/cli/commands/proxy.d.ts

@@ -58,6 +58,23 @@ export declare function isRollingHandoffCapable(state: ProxySupervisorState | nu
  *   confirm a mismatch" and falls through to the args-only result.
  */
 export declare function processLooksLikeProxySupervisor(pid: number, expectedStartTimeIso?: string): Promise<boolean>;
+declare function getOpenCodeConfigDir(): string;
+declare function getOpenCodeConfigPath(): string;
+declare function setOpenCodeProxySettings(baseUrl: string, proxyKey?: string): Promise<boolean>;
+declare function clearOpenCodeProxySettings(expectedBaseUrl?: string): Promise<boolean>;
+/**
+ * Test-only export (CLAUDE.md rule 15 determinism exception). The OpenCode
+ * client writers resolve paths from the environment and are only reachable
+ * from `proxy start` / `proxy setup`, neither of which can be driven against a
+ * throwaway HOME without starting a real server. Consumed by
+ * test/continuous-test-suite-proxy.ts.
+ */
+export declare const __openCodeTestHooks: {
+    getOpenCodeConfigDir: typeof getOpenCodeConfigDir;
+    getOpenCodeConfigPath: typeof getOpenCodeConfigPath;
+    setOpenCodeProxySettings: typeof setOpenCodeProxySettings;
+    clearOpenCodeProxySettings: typeof clearOpenCodeProxySettings;
+};
 export declare function probeProxyHealth(host: string, port: number, timeoutMs: number): Promise<ProxyHealthProbe>;
 export declare function mapClaudeErrorTypeToStatus(errorType?: string): number;
 export declare function createProxyStartApp(params: {
@@ -84,3 +101,4 @@ export declare const proxyGuardCommand: CommandModule<object, ProxyGuardArgs>;
 export declare const proxySetupCommand: CommandModule;
 export declare const proxyInstallCommand: CommandModule;
 export declare const proxyUninstallCommand: CommandModule;
+export {};

package/dist/cli/commands/proxy.js

@@ -513,13 +513,17 @@ async function clearClaudeProxySettings(expectedBaseUrl) {
 // OPENCODE AUTO-CONFIGURATION
 // =============================================================================
 function getOpenCodeConfigDir() {
-    if (process.platform === "darwin") {
-        return join(homedir(), "Library", "Application Support", "opencode");
-    }
-    // Linux/other: XDG_CONFIG_HOME or ~/.config
+    // OpenCode resolves this with the unmodified `xdg-basedir` package —
+    // `XDG_CONFIG_HOME || ~/.config` — on every platform, macOS included. There
+    // is deliberately no darwin branch here: `~/Library/Application Support/
+    // opencode` is not a path OpenCode reads. (The similar-looking literal in
+    // OpenCode's binary is `systemManagedConfigDir()`, an MDM policy directory
+    // at the filesystem root with no $HOME prefix.)
     return join(process.env.XDG_CONFIG_HOME || join(homedir(), ".config"), "opencode");
 }
-const OPENCODE_CONFIG_PATH = join(getOpenCodeConfigDir(), "opencode.json");
+function getOpenCodeConfigPath() {
+    return join(getOpenCodeConfigDir(), "opencode.json");
+}
 /**
  * Key under which we persist the snapshot of the user's pre-existing
  * `provider.neurolink` config inside `opencode.json` itself. Persisting (rather
@@ -536,12 +540,14 @@ async function setOpenCodeProxySettings(baseUrl, proxyKey) {
         fs.accessSync(configDir);
     }
     catch {
-        // OpenCode not installed — config directory does not exist, skip silently
-        return;
+        // OpenCode not installed — config directory does not exist. Report the
+        // skip so the caller does not print a success message for work that did
+        // not happen.
+        return false;
     }
     let config;
     try {
-        config = JSON.parse(fs.readFileSync(OPENCODE_CONFIG_PATH, "utf8"));
+        config = JSON.parse(fs.readFileSync(getOpenCodeConfigPath(), "utf8"));
     }
     catch {
         // file missing/invalid — create fresh config object
@@ -571,13 +577,14 @@ async function setOpenCodeProxySettings(baseUrl, proxyKey) {
         },
     };
     config.provider = provider;
-    fs.writeFileSync(OPENCODE_CONFIG_PATH, JSON.stringify(config, null, 2));
+    fs.writeFileSync(getOpenCodeConfigPath(), JSON.stringify(config, null, 2));
+    return true;
 }
 async function clearOpenCodeProxySettings(expectedBaseUrl) {
     const fs = await import("fs");
     let config;
     try {
-        config = JSON.parse(fs.readFileSync(OPENCODE_CONFIG_PATH, "utf8"));
+        config = JSON.parse(fs.readFileSync(getOpenCodeConfigPath(), "utf8"));
     }
     catch {
         return false;
@@ -622,9 +629,22 @@ async function clearOpenCodeProxySettings(expectedBaseUrl) {
         return false;
     }
     config.provider = provider;
-    fs.writeFileSync(OPENCODE_CONFIG_PATH, JSON.stringify(config, null, 2));
+    fs.writeFileSync(getOpenCodeConfigPath(), JSON.stringify(config, null, 2));
     return hadNeurolink;
 }
+/**
+ * Test-only export (CLAUDE.md rule 15 determinism exception). The OpenCode
+ * client writers resolve paths from the environment and are only reachable
+ * from `proxy start` / `proxy setup`, neither of which can be driven against a
+ * throwaway HOME without starting a real server. Consumed by
+ * test/continuous-test-suite-proxy.ts.
+ */
+export const __openCodeTestHooks = {
+    getOpenCodeConfigDir,
+    getOpenCodeConfigPath,
+    setOpenCodeProxySettings,
+    clearOpenCodeProxySettings,
+};
 // =============================================================================
 // CODEX (ChatGPT) AUTO-CONFIGURATION
 // =============================================================================
@@ -2709,9 +2729,10 @@ async function startProxyRuntime(params) {
                 (error instanceof Error ? error.message : String(error)));
         }
         try {
-            await setOpenCodeProxySettings(`${url}/v1`);
-            logger.always(chalk.green("  ✓ Auto-configured OpenCode settings"));
-            logger.always(chalk.dim("    Restart OpenCode to connect through proxy"));
+            if (await setOpenCodeProxySettings(`${url}/v1`)) {
+                logger.always(chalk.green("  ✓ Auto-configured OpenCode settings"));
+                logger.always(chalk.dim("    Restart OpenCode to connect through proxy"));
+            }
         }
         catch (error) {
             logger.debug("[proxy] Failed to auto-configure OpenCode: " +
@@ -4155,8 +4176,9 @@ export const proxySetupCommand = {
                 console.info(chalk.yellow(`  Set manually: ANTHROPIC_BASE_URL=${url}`));
             }
             try {
-                await setOpenCodeProxySettings(`${url}/v1`);
-                console.info(chalk.green("  ✓ OpenCode configured"));
+                if (await setOpenCodeProxySettings(`${url}/v1`)) {
+                    console.info(chalk.green("  ✓ OpenCode configured"));
+                }
             }
             catch (e) {
                 console.info(chalk.yellow(`  ⚠ Could not auto-configure OpenCode: ${e instanceof Error ? e.message : String(e)}`));

package/dist/lib/providers/googleAiStudio/client.js

@@ -6,6 +6,9 @@ import { ATTR, tracers, withClientSpan, withClientStreamSpan, withSpan, } from "
 import { AuthenticationError, InvalidModelError, NetworkError, ProviderError, RateLimitError, } from "../../types/index.js";
 import { ERROR_CODES, NeuroLinkError } from "../../utils/errorHandling.js";
 import { logger } from "../../utils/logger.js";
+import { createGeminiLoopAdapter } from "../../core/geminiLoopAdapter.js";
+import { runAgenticLoop } from "../../core/loopEngine.js";
+import { DEFAULT_TOOL_MAX_RETRIES } from "../../core/constants.js";
 import { isToolsSchemaExclusionInForce } from "../../core/modules/structuredOutputPolicy.js";
 import { GEMINI_ELISION_NOTE, planGeminiLoopReclaim, previewGeminiToolResponseText, } from "../../context/geminiLoopGuard.js";
 import { getAvailableInputTokens, getContextWindowSize, } from "../../constants/contextWindows.js";
@@ -14,7 +17,7 @@ import { withTimeout } from "../../utils/async/index.js";
 import { estimateTokens } from "../../utils/tokenEstimation.js";
 import { transformToolExecutions } from "../../utils/transformationUtils.js";
 import { resolveToolExecutionRecords } from "../../core/toolExecutionRecorder.js";
-import { buildGeminiResponseSchema, buildNativeConfig, collectStreamChunks, collectStreamChunksIncremental, computeMaxSteps, createContextGuard, buildUserPartsWithMultimodal, executeNativeToolCalls, extractTextFromParts, extractThoughtSignature, handleMaxStepsTermination, prependConversationMessages, pushModelResponseToHistory, refreshNativeToolDeclarations, DedupExecuteMap, } from "../googleNativeGemini3/index.js";
+import { buildGeminiResponseSchema, buildNativeConfig, collectStreamChunks, computeMaxSteps, createContextGuard, buildUserPartsWithMultimodal, executeNativeToolCalls, extractTextFromParts, extractThoughtSignature, handleMaxStepsTermination, prependConversationMessages, pushModelResponseToHistory, refreshNativeToolDeclarations, DedupExecuteMap, } from "../googleNativeGemini3/index.js";
 import { createStreamChannel } from "../../core/streamChannel.js";
 import { toNativeToolDeclarations } from "../../core/nativeToolFormat.js";
 import { createProxyFetch } from "../../proxy/proxyFetch.js";
@@ -649,8 +652,6 @@ export class GoogleAIStudioProvider extends BaseProvider {
                 });
                 // Convert tools
                 let toolsConfig;
-                let executeMap = new DedupExecuteMap();
-                let originalNameMap = new Map();
                 let declarationsResult;
                 if (options.tools &&
                     Object.keys(options.tools).length > 0 &&
@@ -658,8 +659,6 @@ export class GoogleAIStudioProvider extends BaseProvider {
                     const result = toNativeToolDeclarations(options.tools, "functionDeclarations");
                     declarationsResult = result;
                     toolsConfig = result.toolsConfig;
-                    executeMap = result.executeMap;
-                    originalNameMap = result.originalNameMap;
                     logger.debug("[GoogleAIStudio] Converted tools for native SDK", {
                         toolCount: toolsConfig[0].functionDeclarations.length,
                         toolNames: toolsConfig[0].functionDeclarations.map((t) => t.name),
@@ -720,8 +719,6 @@ export class GoogleAIStudioProvider extends BaseProvider {
                     let totalCacheReadTokens = 0;
                     let totalReasoningTokens = 0;
                     let step = 0;
-                    let completedWithFinalAnswer = false;
-                    const failedTools = new Map();
                     // Cheap trigger for the in-turn reclaim, mirroring the Vertex twin.
                     // Planning serializes the WHOLE accumulated history to estimate it,
                     // so running it unconditionally charges that once per step for the
@@ -731,98 +728,88 @@ export class GoogleAIStudioProvider extends BaseProvider {
                     const contextGuard = createContextGuard(getContextWindowSize("googleAiStudio", modelName));
                     try {
                         // Agentic loop for tool calling
-                        while (step < maxSteps) {
-                            // In-turn context guard: this loop appends a model turn plus a
-                            // tool turn every step with nothing bounding growth. No-op
-                            // while the request still fits, so a loop that fits never pays
-                            // a cache invalidation. Step 0 still plans unconditionally —
-                            // the guard has no usage to go on yet, and the incoming history
-                            // can already be oversized before the first call.
-                            if (step === 0 || contextGuard.shouldStop()) {
-                                if (reclaimAiStudioContext(currentContents, modelName, contextGuard.projectedNextPromptTokens)) {
-                                    contextGuard.resetAfterReclaim();
+                        // The turn itself now runs on the shared engine: the step cap,
+                        // tool dispatch, the failure breaker and usage accumulation are
+                        // engine-owned. Everything below is the provider half — building
+                        // one request, and the per-step side effects the old loop
+                        // performed inline.
+                        const baseAdapter = createGeminiLoopAdapter({
+                            providerLabel: "GoogleAIStudio",
+                            maxSteps,
+                            // Same threshold the hand-rolled dispatcher used, which is
+                            // what makes an always-failing tool dispatch exactly twice.
+                            toolFailureBreaker: { maxRetries: DEFAULT_TOOL_MAX_RETRIES },
+                            liveTools: options.tools ?? {},
+                            ...(declarationsResult
+                                ? { declarations: declarationsResult }
+                                : {}),
+                            buildRequest: (contents) => ({
+                                model: modelName,
+                                contents,
+                                config,
+                                ...(composedSignal
+                                    ? { httpOptions: { signal: composedSignal } }
+                                    : {}),
+                            }),
+                            sendStep: async (request) => client.models.generateContentStream(request),
+                            noteUsage: (inputTokens, outputTokens) => {
+                                contextGuard.noteUsage(inputTokens, outputTokens);
+                            },
+                            // Pure: the engine assigns what this returns. The old loop
+                            // reclaimed in place because it owned `currentContents`.
+                            planReclaim: (contents, stepIndex) => {
+                                if (stepIndex !== 0 && !contextGuard.shouldStop()) {
+                                    return undefined;
                                 }
-                            }
-                            if (composedSignal?.aborted) {
-                                throw composedSignal.reason instanceof Error
-                                    ? composedSignal.reason
-                                    : new Error("Request aborted");
-                            }
-                            step++;
-                            // Mid-turn discovery sync: advertise tools hydrated into the
-                            // live record by search_tools during the previous step.
-                            if (declarationsResult) {
-                                refreshNativeToolDeclarations(options.tools, declarationsResult);
-                            }
-                            logger.debug(`[GoogleAIStudio] Native SDK step ${step}/${maxSteps}`);
-                            try {
-                                const rawStream = await client.models.generateContentStream({
-                                    model: modelName,
-                                    contents: currentContents,
-                                    config,
-                                    ...(composedSignal
-                                        ? { httpOptions: { signal: composedSignal } }
-                                        : {}),
-                                });
-                                // For every step, use incremental collection so text parts
-                                // are pushed to the channel as they arrive.  For intermediate
-                                // steps (those that produce function calls) we still need the
-                                // complete rawResponseParts for pushModelResponseToHistory,
-                                // which collectStreamChunksIncremental provides at stream end.
-                                const chunkResult = await collectStreamChunksIncremental(rawStream, channel);
-                                totalInputTokens += chunkResult.inputTokens;
-                                totalOutputTokens += chunkResult.outputTokens;
-                                totalCacheReadTokens += chunkResult.cacheReadTokens ?? 0;
-                                totalReasoningTokens += chunkResult.reasoningTokens ?? 0;
-                                // `inputTokens` is this step's promptTokenCount — the FULL
-                                // prompt size for the request just made, which is what the
-                                // guard projects the next request from.
-                                contextGuard.noteUsage(chunkResult.inputTokens, chunkResult.outputTokens);
-                                const stepText = extractTextFromParts(chunkResult.rawResponseParts);
-                                // If no function calls, this was the final step — channel
-                                // already received all text parts incrementally.
-                                if (chunkResult.stepFunctionCalls.length === 0) {
-                                    completedWithFinalAnswer = true;
-                                    break;
+                                const working = [...contents];
+                                if (!reclaimAiStudioContext(working, modelName, contextGuard.projectedNextPromptTokens)) {
+                                    return undefined;
                                 }
-                                lastStepText = stepText;
-                                // Record tool call events on the span
-                                for (const fc of chunkResult.stepFunctionCalls) {
+                                contextGuard.resetAfterReclaim();
+                                return working;
+                            },
+                        });
+                        // Wrapped because these fire once PER STEP in the loop this
+                        // replaces, and buildToolResultMessages is the only hook that
+                        // runs per step with exactly that step's results. Reading them
+                        // off the turn's final result would batch every step into one
+                        // late write and lose the per-step thought signature.
+                        const adapter = {
+                            ...baseAdapter,
+                            buildToolResultMessages: (contents, stepResult, toolResults) => {
+                                step++;
+                                for (const call of stepResult.toolCalls) {
                                     span.addEvent("gen_ai.tool_call", {
-                                        "tool.name": fc.name,
+                                        "tool.name": call.name,
                                         "tool.step": step,
                                     });
                                 }
-                                logger.debug(`[GoogleAIStudio] Executing ${chunkResult.stepFunctionCalls.length} function calls`);
-                                // Add model response with ALL parts (including thoughtSignature) to history
-                                pushModelResponseToHistory(currentContents, chunkResult.rawResponseParts, chunkResult.stepFunctionCalls);
-                                const toolCallsBefore = allToolCalls.length;
-                                const toolExecsBefore = toolExecutions.length;
-                                const functionResponses = await executeNativeToolCalls("[GoogleAIStudio]", chunkResult.stepFunctionCalls, executeMap, failedTools, allToolCalls, {
-                                    abortSignal: composedSignal,
-                                    originalNameMap,
-                                    toolExecutions,
-                                    liveTools: options.tools,
-                                    declarations: declarationsResult,
-                                });
-                                // Persist this step's tool calls/results into conversation
-                                // memory. Without this, tool_call / tool_result rows never
-                                // reach Redis and the chat-history UI loses every tool
-                                // invocation.
-                                const stepToolCalls = allToolCalls.slice(toolCallsBefore);
-                                const stepToolExecs = toolExecutions.slice(toolExecsBefore);
-                                if (stepToolCalls.length > 0 || stepToolExecs.length > 0) {
-                                    const stepThoughtSig = extractThoughtSignature(chunkResult.rawResponseParts);
-                                    withTimeout(this.handleToolExecutionStorage(stepToolCalls.map((tc, i) => ({
-                                        toolName: tc.toolName,
-                                        args: tc.args,
+                                lastStepText = stepResult.text || lastStepText;
+                                for (const call of stepResult.toolCalls) {
+                                    allToolCalls.push({
+                                        toolName: call.name,
+                                        args: call.args,
+                                    });
+                                }
+                                for (const result of toolResults) {
+                                    toolExecutions.push({
+                                        name: result.name,
+                                        input: result.args,
+                                        output: result.output,
+                                    });
+                                }
+                                if (toolResults.length > 0) {
+                                    const stepThoughtSig = extractThoughtSignature(stepResult.raw.rawResponseParts);
+                                    withTimeout(this.handleToolExecutionStorage(stepResult.toolCalls.map((call, i) => ({
+                                        toolName: call.name,
+                                        args: call.args,
                                         ...(i === 0 && stepThoughtSig
                                             ? { thoughtSignature: stepThoughtSig }
                                             : {}),
                                         stepIndex: step,
-                                    })), stepToolExecs.map((te) => ({
-                                        toolName: te.name,
-                                        output: te.output,
+                                    })), toolResults.map((result) => ({
+                                        toolName: result.name,
+                                        output: result.output,
                                         stepIndex: step,
                                     })), options, new Date()), TOOL_STORAGE_TIMEOUT_MS, "tool storage write timed out").catch((error) => {
                                         logger.warn("[GoogleAIStudio] Failed to store native tool executions", {
@@ -832,28 +819,57 @@ export class GoogleAIStudioProvider extends BaseProvider {
                                         });
                                     });
                                 }
-                                // Add function responses to history — the @google/genai SDK
-                                // only accepts "user" and "model" as valid roles in contents.
-                                // Function/tool responses must use role: "user" (matching the
-                                // SDK's own automaticFunctionCalling implementation).
-                                currentContents.push({
-                                    role: "user",
-                                    parts: functionResponses,
-                                });
+                                const next = baseAdapter.buildToolResultMessages(contents, stepResult, toolResults);
                                 // Project this step's growth: the appended tool results ride
                                 // the next prompt, which the provider has not reported on yet.
                                 try {
-                                    contextGuard.noteAppendedChars(JSON.stringify(functionResponses).length);
+                                    const appended = next[next.length - 1];
+                                    contextGuard.noteAppendedChars(JSON.stringify(appended?.parts ?? []).length);
                                 }
                                 catch {
                                     /* estimation is best-effort — never break the loop */
                                 }
+                                return next;
+                            },
+                        };
+                        const engineTools = {};
+                        for (const [name, tool] of Object.entries(options.tools ?? {})) {
+                            const execute = tool?.execute;
+                            if (!execute) {
+                                continue;
                             }
-                            catch (error) {
-                                logger.error("[GoogleAIStudio] Native SDK error", error);
-                                throw this.handleProviderError(error);
+                            engineTools[name] = {
+                                execute: async (args, opts) => execute(args, opts),
+                            };
+                        }
+                        const { stream: engineStream, resultPromise } = runAgenticLoop(adapter, currentContents, {
+                            tools: engineTools,
+                            ...(composedSignal ? { abortSignal: composedSignal } : {}),
+                        });
+                        const pump = (async () => {
+                            for await (const chunk of engineStream) {
+                                channel.push(chunk);
                             }
+                        })();
+                        let engineResult;
+                        try {
+                            engineResult = await resultPromise;
+                        }
+                        catch (error) {
+                            await pump.catch(() => { });
+                            logger.error("[GoogleAIStudio] Native SDK error", error);
+                            throw this.handleProviderError(error);
                         }
+                        await pump;
+                        totalInputTokens += engineResult.usage.inputTokens;
+                        totalOutputTokens += engineResult.usage.outputTokens;
+                        totalCacheReadTokens += engineResult.usage.cacheReadTokens ?? 0;
+                        totalReasoningTokens += engineResult.usage.reasoningTokens ?? 0;
+                        // The turn produced a final answer when the model stopped
+                        // calling tools of its own accord, rather than being cut off at
+                        // the cap.
+                        const completedWithFinalAnswer = engineResult.toolCalls.length === 0 ||
+                            engineResult.finishReason !== "tool-calls";
                         // Handle max-steps termination: if the model was still calling
                         // tools when we hit the limit, push a synthetic final message.
                         const hitStepLimitWithoutFinalAnswer = step >= maxSteps && !completedWithFinalAnswer;