@juspay/neurolink 9.65.1 → 9.65.2

package/dist/providers/googleNativeGemini3.js

@@ -11,11 +11,11 @@
 import { randomUUID } from "node:crypto";
 import { existsSync, readFileSync } from "node:fs";
 import { extname } from "node:path";
-import { jsonSchema as aiJsonSchema, tool as createAISDKTool, } from "ai";
 import { DEFAULT_MAX_STEPS, DEFAULT_TOOL_MAX_RETRIES, } from "../core/constants.js";
 import { logger } from "../utils/logger.js";
 import { convertZodToJsonSchema, ensureNestedSchemaTypes, inlineJsonSchema, isZodSchema, normalizeJsonSchemaObject, } from "../utils/schemaConversion.js";
 import { createNativeThinkingConfig } from "../utils/thinkingConfig.js";
+import { jsonSchema as aiJsonSchema, tool as createAISDKTool, } from "../utils/tool.js";
 // ── Functions ──
 /**
  * Google's `function_declarations[].name` validator regex.

package/dist/providers/googleVertex.d.ts

@@ -1,8 +1,8 @@
 import type { ZodType } from "zod";
-import { type Schema, type LanguageModel } from "ai";
 import { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { EnhancedGenerateResult, TextGenerationOptions, StreamOptions, StreamResult } from "../types/index.js";
+import type { Schema, LanguageModel } from "../types/index.js";
 /**
  * Resolve the effective Vertex region for a given model.
  *

package/dist/providers/groq.d.ts

@@ -1,7 +1,7 @@
-import { type LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Groq Provider
  *

package/dist/providers/groq.js

@@ -1,5 +1,4 @@
 import { createOpenAI } from "@ai-sdk/openai";
-import { stepCountIs, streamText } from "ai";
 import { GroqModels } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
@@ -14,6 +13,8 @@ import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../
 import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 const GROQ_DEFAULT_BASE_URL = "https://api.groq.com/openai/v1";
 const getGroqApiKey = () => validateApiKey(createGroqConfig());
 const getDefaultGroqModel = () => getProviderModel("GROQ_MODEL", GroqModels.LLAMA_3_3_70B_VERSATILE);

package/dist/providers/huggingFace.d.ts

@@ -1,8 +1,8 @@
-import { type LanguageModel, type Schema } from "ai";
 import type { ZodType } from "zod";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult } from "../types/index.js";
+import type { LanguageModel, Schema } from "../types/index.js";
 /**
  * HuggingFace Provider - BaseProvider Implementation
  * Using AI SDK with HuggingFace's OpenAI-compatible endpoint

package/dist/providers/huggingFace.js

@@ -1,5 +1,4 @@
 import { createOpenAI } from "@ai-sdk/openai";
-import { NoOutputGeneratedError, stepCountIs, streamText, } from "ai";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
 import { createProxyFetch } from "../proxy/proxyFetch.js";
@@ -10,6 +9,9 @@ import { buildNoOutputSentinel, detectPostStreamNoOutput, stampNoOutputSpan, } f
 import { createHuggingFaceConfig, getProviderModel, validateApiKey, } from "../utils/providerConfig.js";
 import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
+import { NoOutputGeneratedError } from "../utils/generationErrors.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 // Configuration helpers - now using consolidated utility
 const getHuggingFaceApiKey = () => {
     return validateApiKey(createHuggingFaceConfig());

package/dist/providers/ideogram.d.ts

@@ -1,7 +1,7 @@
-import type { LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { EnhancedGenerateResult, NeurolinkCredentials, StreamOptions, StreamResult, TextGenerationOptions, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Ideogram Provider — direct image generation with strong typography.
  *

package/dist/providers/jina.d.ts

@@ -1,7 +1,7 @@
-import type { LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Jina AI Provider — embeddings + reranking.
  *

package/dist/providers/litellm.d.ts

@@ -1,8 +1,8 @@
-import { type LanguageModel, type Schema } from "ai";
 import type { ZodType } from "zod";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { StreamOptions, StreamResult } from "../types/index.js";
+import type { LanguageModel, Schema } from "../types/index.js";
 /**
  * LiteLLM Provider - BaseProvider Implementation
  * Provides access to 100+ models via LiteLLM proxy server

package/dist/providers/litellm.js

@@ -1,6 +1,5 @@
 import { createOpenAI } from "@ai-sdk/openai";
 import { SpanKind, SpanStatusCode, trace } from "@opentelemetry/api";
-import { NoOutputGeneratedError, Output, stepCountIs, streamText, } from "ai";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
 import { streamAnalyticsCollector } from "../core/streamAnalytics.js";
@@ -12,9 +11,12 @@ import { logger } from "../utils/logger.js";
 import { buildNoOutputSentinel, detectPostStreamNoOutput, stampNoOutputSpan, } from "../utils/noOutputSentinel.js";
 import { calculateCost } from "../utils/pricing.js";
 import { getProviderModel } from "../utils/providerConfig.js";
-import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
+import { composeAbortSignals, createTimeoutController, TimeoutError, withTimeout, } from "../utils/timeout.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { getModelId } from "./providerTypeUtils.js";
+import { NoOutputGeneratedError } from "../utils/generationErrors.js";
+import { Output, stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 const streamTracer = trace.getTracer("neurolink.provider.litellm");
 // Configuration helpers
 const getLiteLLMConfig = () => {
@@ -425,7 +427,7 @@ export class LiteLLMProvider extends BaseProvider {
      * Uses the LiteLLM proxy with OpenAI-compatible embedding API
      */
     async embed(text, modelName) {
-        const { embed: aiEmbed } = await import("ai");
+        const { embed: aiEmbed } = await import("../utils/generation.js");
         const { createOpenAI } = await import("@ai-sdk/openai");
         const config = getLiteLLMConfig();
         const embeddingModelName = modelName ||
@@ -437,7 +439,10 @@ export class LiteLLMProvider extends BaseProvider {
             fetch: createProxyFetch(),
         });
         const embeddingModel = customOpenAI.textEmbeddingModel(embeddingModelName);
-        const result = await aiEmbed({ model: embeddingModel, value: text });
+        // Wrap in withTimeout so stalled upstream embedding requests abort instead
+        // of hanging forever. 30s matches the default for embedding endpoints
+        // across the OpenAI-compatible cluster.
+        const result = await withTimeout(aiEmbed({ model: embeddingModel, value: text }), 30_000, "litellm", "generate");
         return result.embedding;
     }
     /**
@@ -445,7 +450,7 @@ export class LiteLLMProvider extends BaseProvider {
      * Uses the LiteLLM proxy with OpenAI-compatible embedding API
      */
     async embedMany(texts, modelName) {
-        const { embedMany: aiEmbedMany } = await import("ai");
+        const { embedMany: aiEmbedMany } = await import("../utils/generation.js");
         const { createOpenAI } = await import("@ai-sdk/openai");
         const config = getLiteLLMConfig();
         const embeddingModelName = modelName ||
@@ -457,7 +462,8 @@ export class LiteLLMProvider extends BaseProvider {
             fetch: createProxyFetch(),
         });
         const embeddingModel = customOpenAI.textEmbeddingModel(embeddingModelName);
-        const result = await aiEmbedMany({ model: embeddingModel, values: texts });
+        // Wrap in withTimeout so a single slow batch doesn't hang indefinitely.
+        const result = await withTimeout(aiEmbedMany({ model: embeddingModel, values: texts }), 30_000, "litellm", "generate");
         return result.embeddings;
     }
     /**

package/dist/providers/llamaCpp.d.ts

@@ -1,7 +1,7 @@
-import { type LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * llama.cpp Provider
  * Wraps a llama-server process (https://github.com/ggerganov/llama.cpp) that

package/dist/providers/llamaCpp.js

@@ -1,5 +1,4 @@
 import { createOpenAI } from "@ai-sdk/openai";
-import { stepCountIs, streamText } from "ai";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
 import { streamAnalyticsCollector } from "../core/streamAnalytics.js";
@@ -13,6 +12,8 @@ import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../
 import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 const LLAMACPP_DEFAULT_BASE_URL = "http://localhost:8080/v1";
 const LLAMACPP_PLACEHOLDER_KEY = "llamacpp";
 const FALLBACK_MODEL = "loaded-model";

package/dist/providers/lmStudio.d.ts

@@ -1,7 +1,7 @@
-import { type LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * LM Studio Provider
  * Wraps the LM Studio local server (https://lmstudio.ai/) which exposes an

package/dist/providers/lmStudio.js

@@ -1,5 +1,4 @@
 import { createOpenAI } from "@ai-sdk/openai";
-import { stepCountIs, streamText } from "ai";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
 import { streamAnalyticsCollector } from "../core/streamAnalytics.js";
@@ -13,6 +12,8 @@ import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../
 import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 const LM_STUDIO_DEFAULT_BASE_URL = "http://localhost:1234/v1";
 const LM_STUDIO_PLACEHOLDER_KEY = "lm-studio";
 const FALLBACK_MODEL = "local-model";

package/dist/providers/mistral.d.ts

@@ -1,7 +1,7 @@
-import { type LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Mistral AI Provider v2 - BaseProvider Implementation
  * Supports official AI-SDK integration with all Mistral models

package/dist/providers/mistral.js

@@ -1,5 +1,4 @@
 import { createMistral } from "@ai-sdk/mistral";
-import { stepCountIs, streamText } from "ai";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
 import { streamAnalyticsCollector } from "../core/streamAnalytics.js";
@@ -12,6 +11,8 @@ import { createMistralConfig, getProviderModel, validateApiKey, } from "../utils
 import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 // Configuration helpers - now using consolidated utility
 const getMistralApiKey = () => {
     return validateApiKey(createMistralConfig());

package/dist/providers/nvidiaNim.d.ts

@@ -1,7 +1,7 @@
-import { type LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * NVIDIA NIM Provider
  * Wraps NVIDIA's hosted (or self-hosted) inference endpoints via OpenAI-compat.

package/dist/providers/nvidiaNim.js

@@ -1,5 +1,4 @@
 import { createOpenAICompatible } from "@ai-sdk/openai-compatible";
-import { stepCountIs, streamText } from "ai";
 import { NvidiaNimModels } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
@@ -14,6 +13,8 @@ import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../
 import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 /**
  * Decide whether a NIM 400 response body is a rejection of the named
  * field (as opposed to an unrelated 400 that happens to mention the

package/dist/providers/ollama.d.ts

@@ -1,7 +1,7 @@
-import type { LanguageModel, Schema } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { StreamOptions, StreamResult, ZodUnknownSchema } from "../types/index.js";
+import type { LanguageModel, Schema } from "../types/index.js";
 /**
  * Ollama Provider v2 - BaseProvider Implementation
  *

package/dist/providers/ollama.js

@@ -706,8 +706,7 @@ export class OllamaProvider extends BaseProvider {
             let iteration = 0;
             // Get all available tools (direct + MCP + external)
             // BaseProvider.stream() pre-merges base tools + external tools into options.tools
-            const allTools = options.tools ||
-                (await this.getAllTools());
+            const allTools = options.tools || (await this.getAllTools());
             // Convert tools to Ollama format
             const ollamaTools = this.convertToolsToOllamaFormat(allTools);
             span.setAttribute(ATTR.NL_TOOL_COUNT, ollamaTools.length);

package/dist/providers/openAI.d.ts

@@ -1,8 +1,8 @@
-import { type LanguageModel } from "ai";
 import { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeuroLink } from "../neurolink.js";
 import type { EnhancedGenerateResult, TextGenerationOptions, ValidationSchema, StreamOptions, StreamResult } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * OpenAI Provider v2 - BaseProvider Implementation
  * Migrated to use factory pattern with exact Google AI provider pattern

package/dist/providers/openAI.js

@@ -1,6 +1,5 @@
 import { createOpenAI } from "@ai-sdk/openai";
 import { SpanKind, SpanStatusCode, trace } from "@opentelemetry/api";
-import { embed, embedMany, NoOutputGeneratedError, stepCountIs, streamText, } from "ai";
 import { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
@@ -18,6 +17,9 @@ import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
 import { MAX_IMAGE_BYTES, readBoundedBuffer } from "../utils/sizeGuard.js";
 import { assertSafeUrl } from "../utils/ssrfGuard.js";
 import { getModelId } from "./providerTypeUtils.js";
+import { NoOutputGeneratedError } from "../utils/generationErrors.js";
+import { stepCountIs } from "../utils/tool.js";
+import { embed, embedMany, streamText } from "../utils/generation.js";
 /**
  * Retrieve a tool's schema, handling both AI SDK v6 (`inputSchema`) and
  * legacy v4 (`parameters`) field names.

package/dist/providers/openRouter.d.ts

@@ -1,8 +1,8 @@
-import { type LanguageModel, type Schema } from "ai";
 import type { ZodType } from "zod";
 import { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { StreamOptions, StreamResult } from "../types/index.js";
+import type { LanguageModel, Schema } from "../types/index.js";
 /**
  * OpenRouter Provider - BaseProvider Implementation
  * Provides access to 300+ models from 60+ providers via OpenRouter unified gateway

package/dist/providers/openRouter.js

@@ -1,5 +1,4 @@
 import { createOpenRouter } from "@openrouter/ai-sdk-provider";
-import { NoOutputGeneratedError, Output, stepCountIs, streamText, } from "ai";
 import { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
@@ -13,6 +12,9 @@ import { buildNoOutputSentinel, detectPostStreamNoOutput, stampNoOutputSpan, } f
 import { getProviderModel } from "../utils/providerConfig.js";
 import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
+import { NoOutputGeneratedError } from "../utils/generationErrors.js";
+import { Output, stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 // Constants
 const MODELS_DISCOVERY_TIMEOUT_MS = 5000; // 5 seconds for model discovery
 // Configuration helpers

package/dist/providers/openaiCompatible.d.ts

@@ -1,7 +1,7 @@
-import { type LanguageModel, type Schema } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { StreamOptions, StreamResult, ZodUnknownSchema } from "../types/index.js";
+import type { LanguageModel, Schema } from "../types/index.js";
 /**
  * OpenAI Compatible Provider - BaseProvider Implementation
  * Provides access to one of the OpenAI-compatible endpoint (OpenRouter, vLLM, LiteLLM, etc.)

package/dist/providers/openaiCompatible.js

@@ -1,5 +1,4 @@
 import { createOpenAI } from "@ai-sdk/openai";
-import { NoOutputGeneratedError, stepCountIs, streamText, } from "ai";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
 import { streamAnalyticsCollector } from "../core/streamAnalytics.js";
@@ -11,6 +10,9 @@ import { buildNoOutputSentinel, detectPostStreamNoOutput, stampNoOutputSpan, } f
 import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../utils/timeout.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
+import { NoOutputGeneratedError } from "../utils/generationErrors.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 // Constants
 const FALLBACK_OPENAI_COMPATIBLE_MODEL = "gpt-3.5-turbo";
 // Configuration helpers

package/dist/providers/perplexity.d.ts

@@ -1,7 +1,7 @@
-import { type LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Perplexity Provider
  *

package/dist/providers/perplexity.js

@@ -1,5 +1,4 @@
 import { createOpenAI } from "@ai-sdk/openai";
-import { stepCountIs, streamText } from "ai";
 import { PerplexityModels } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
@@ -14,6 +13,8 @@ import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../
 import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 const PERPLEXITY_DEFAULT_BASE_URL = "https://api.perplexity.ai";
 const getPerplexityApiKey = () => validateApiKey(createPerplexityConfig());
 const getDefaultPerplexityModel = () => getProviderModel("PERPLEXITY_MODEL", PerplexityModels.SONAR);

package/dist/providers/providerTypeUtils.d.ts

@@ -1,11 +1,6 @@
-/**
- * Shared type utilities for AI SDK provider integration.
- *
- * Provides type-safe helpers to bridge between the AI SDK's generic types
- * and NeuroLink's internal type system without resorting to `as any` casts.
- */
-import type { LanguageModel, streamText } from "ai";
 import type { StreamTextResult } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
+import type { streamText } from "../utils/generation.js";
 /**
  * Extract the model identifier from a LanguageModel value.
  *

package/dist/providers/providerTypeUtils.js

@@ -1,9 +1,3 @@
-/**
- * Shared type utilities for AI SDK provider integration.
- *
- * Provides type-safe helpers to bridge between the AI SDK's generic types
- * and NeuroLink's internal type system without resorting to `as any` casts.
- */
 /**
  * Type guard: checks whether a LanguageModel value is an object with `modelId`
  * (i.e. LanguageModelV2 or LanguageModelV3) rather than a bare string ID.

package/dist/providers/recraft.d.ts

@@ -1,7 +1,7 @@
-import type { LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { EnhancedGenerateResult, NeurolinkCredentials, StreamOptions, StreamResult, TextGenerationOptions, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Recraft Provider — image generation with vector / illustration focus.
  *

package/dist/providers/replicate.d.ts

@@ -1,7 +1,7 @@
-import type { LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { EnhancedGenerateResult, NeurolinkCredentials, StreamOptions, StreamResult, TextGenerationOptions, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Replicate LLM Provider — predict-then-stream pattern.
  *

package/dist/providers/stability.d.ts

@@ -1,7 +1,7 @@
-import type { LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { EnhancedGenerateResult, NeurolinkCredentials, StreamOptions, StreamResult, TextGenerationOptions, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Stability AI Provider — direct image generation.
  *

package/dist/providers/togetherAi.d.ts

@@ -1,7 +1,7 @@
-import { type LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Together AI Provider
  *

package/dist/providers/togetherAi.js

@@ -1,5 +1,4 @@
 import { createOpenAI } from "@ai-sdk/openai";
-import { stepCountIs, streamText } from "ai";
 import { TogetherAIModels } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
@@ -14,6 +13,8 @@ import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../
 import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 const TOGETHER_DEFAULT_BASE_URL = "https://api.together.xyz/v1";
 const getTogetherApiKey = () => validateApiKey(createTogetherAIConfig());
 const getDefaultTogetherModel = () => getProviderModel("TOGETHER_MODEL", TogetherAIModels.LLAMA_3_3_70B_INSTRUCT_TURBO);

package/dist/providers/voyage.d.ts

@@ -1,7 +1,7 @@
-import type { LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * Voyage AI Provider — embedding-only.
  *

package/dist/providers/xai.d.ts

@@ -1,7 +1,7 @@
-import { type LanguageModel } from "ai";
 import type { AIProviderName } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import type { NeurolinkCredentials, StreamOptions, StreamResult, ValidationSchema } from "../types/index.js";
+import type { LanguageModel } from "../types/index.js";
 /**
  * xAI Grok Provider
  *

package/dist/providers/xai.js

@@ -1,5 +1,4 @@
 import { createOpenAI } from "@ai-sdk/openai";
-import { stepCountIs, streamText } from "ai";
 import { XaiModels } from "../constants/enums.js";
 import { BaseProvider } from "../core/baseProvider.js";
 import { DEFAULT_MAX_STEPS } from "../core/constants.js";
@@ -14,6 +13,8 @@ import { composeAbortSignals, createTimeoutController, TimeoutError, } from "../
 import { emitToolEndFromStepFinish } from "../utils/toolEndEmitter.js";
 import { resolveToolChoice } from "../utils/toolChoice.js";
 import { toAnalyticsStreamResult } from "./providerTypeUtils.js";
+import { stepCountIs } from "../utils/tool.js";
+import { streamText } from "../utils/generation.js";
 /**
  * Logging fetch wrapper — masks the proxy URL on non-2xx responses so
  * stack traces and log lines don't leak the upstream's internal token /

package/dist/proxy/claudeFormat.d.ts

@@ -1,23 +1,8 @@
-/**
- * Claude Messages API format conversion layer.
- *
- * Provides a request parser (Claude -> NeuroLink), a response serializer
- * (NeuroLink -> Claude), a streaming SSE state machine, and an error
- * envelope helper.  Together they allow NeuroLink to act as a
- * drop-in Claude API proxy.
- *
- * Reference: https://docs.anthropic.com/en/api/messages
- */
 import type { ClaudeErrorResponse, ClaudeRequest, ClaudeResponse, ContentBlockType, InternalResult, ParsedClaudeRequest, StreamLifecycleState } from "../types/index.js";
 /** Generate a unique message id in the Claude format. */
 export declare function generateMessageId(): string;
 /** Generate a Claude-format tool use ID (`toolu_` + 24 random chars). */
 export declare function generateToolUseId(): string;
-/**
- * Reset the internal id counter (useful in tests).
- * @internal
- */
-export declare function _resetIdCounter(): void;
 /**
  * Parse an incoming Claude Messages API request into an intermediate
  * representation consumable by NeuroLink's generate/stream pipeline.

package/dist/proxy/claudeFormat.js

@@ -1,16 +1,6 @@
-/**
- * Claude Messages API format conversion layer.
- *
- * Provides a request parser (Claude -> NeuroLink), a response serializer
- * (NeuroLink -> Claude), a streaming SSE state machine, and an error
- * envelope helper.  Together they allow NeuroLink to act as a
- * drop-in Claude API proxy.
- *
- * Reference: https://docs.anthropic.com/en/api/messages
- */
-import { jsonSchema, tool } from "ai";
 import { randomBytes } from "crypto";
 import { normalizeJsonSchemaObject } from "../utils/schemaConversion.js";
+import { jsonSchema, tool } from "../utils/tool.js";
 // ---------------------------------------------------------------------------
 // Helpers
 // ---------------------------------------------------------------------------

package/dist/rag/ragIntegration.d.ts

@@ -1,11 +1,3 @@
-/**
- * RAG Integration for generate() and stream()
- *
- * Provides automatic RAG pipeline setup when `rag` config is provided
- * in GenerateOptions or StreamOptions. Handles file loading, chunking,
- * embedding generation, vector storage, and tool creation internally
- * so developers only need to pass `rag: { files: [...] }`.
- */
 import type { RAGConfig, VectorQueryResult, RAGPreparedTool } from "../types/index.js";
 /**
  * Generate deterministic embeddings for chunks.
@@ -35,7 +27,4 @@ declare function diversifyResults(results: VectorQueryResult[], topK: number): V
  * @returns Prepared RAG tool to inject into the tools record
  */
 export declare function prepareRAGTool(ragConfig: RAGConfig, fallbackProvider?: string): Promise<RAGPreparedTool>;
-/** @internal Exported for testing only */
-export { generateSimpleEmbedding as _generateSimpleEmbedding };
-/** @internal Exported for testing only */
-export { diversifyResults as _diversifyResults };
+export {};

package/dist/rag/ragIntegration.js

@@ -1,11 +1,3 @@
-/**
- * RAG Integration for generate() and stream()
- *
- * Provides automatic RAG pipeline setup when `rag` config is provided
- * in GenerateOptions or StreamOptions. Handles file loading, chunking,
- * embedding generation, vector storage, and tool creation internally
- * so developers only need to pass `rag: { files: [...] }`.
- */
 import { existsSync, readFileSync } from "fs";
 import { extname, resolve } from "path";
 import { z } from "zod";