@jaypie/testkit 1.2.41 → 1.2.43

package/dist/mock/index.d.ts

@@ -7,29 +7,48 @@ import * as original from '@jaypie/dynamodb';
 import { ParentReference, ScanTableOptions, StorableEntity, CreateTableOptions, CreateTableResult, ExportResult, DynamoClientConfig, QueryResult, SeedOptions, SeedResult } from '@jaypie/dynamodb';
 export { BaseQueryOptions, CreateTableOptions, CreateTableResult, ExportResult, ParentReference, QueryParams, QueryResult, ScanTableOptions, SeedOptions, SeedResult, StorableEntity } from '@jaypie/dynamodb';
 import * as original$1 from '@jaypie/express';
+import * as _vitest_spy from '@vitest/spy';
 import * as original$2 from '@jaypie/llm';
 export { FORMAT, LEVEL, Logger, log, redactAuth, sanitizeAuth } from '@jaypie/logger';
 export { IndexableModel } from '@jaypie/fabric';
 export { mongoose } from '@jaypie/mongoose';
 
-declare const getMessages: vitest.Mock<(...args: any[]) => any>;
+declare const getMessages: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const getS3FileBuffer: (({ bucket, key }: {
     bucket: string;
     key: string;
 }) => Promise<Buffer>) & {
     mock: any;
 };
-declare const getSecret: vitest.Mock<(...args: any[]) => any>;
-declare const sendMessage: vitest.Mock<(...args: any[]) => any>;
+declare const getSecret: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const sendMessage: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const getEnvSecret: ((key: string) => Promise<string>) & {
     mock: any;
 };
-declare const loadEnvSecrets: vitest.Mock<(...args: any[]) => any>;
-declare const getSingletonMessage: vitest.Mock<(...args: any[]) => any>;
+declare const loadEnvSecrets: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const getSingletonMessage: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const getTextractJob: ((jobId: string) => Promise<any>) & {
     mock: any;
 };
-declare const sendBatchMessages: vitest.Mock<(...args: any[]) => any>;
+declare const sendBatchMessages: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const sendTextractJob: (({ bucket, key, featureTypes, }: {
     bucket: string;
     key: string;
@@ -149,26 +168,68 @@ declare const UnavailableError: typeof errors.UnavailableError;
 declare const UnhandledError: typeof errors.UnhandledError;
 declare const UnreachableCodeError: typeof errors.UnreachableCodeError;
 
-declare const cloneDeep: vitest.Mock<(...args: any[]) => any>;
-declare const envBoolean: vitest.Mock<(...args: any[]) => any>;
-declare const envsKey: vitest.Mock<(...args: any[]) => any>;
+declare const cloneDeep: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const envBoolean: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const envsKey: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const errorFromStatusCode: ((statusCode: number, message?: string) => Error) & {
     mock: any;
 };
-declare const formatError: vitest.Mock<(...args: any[]) => any>;
-declare const generateJaypieKey: vitest.Mock<(...args: any[]) => any>;
-declare const hashJaypieKey: vitest.Mock<(...args: any[]) => any>;
-declare const jaypieApiKeyId: vitest.Mock<(...args: any[]) => any>;
+declare const formatError: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const generateJaypieKey: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const hashJaypieKey: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const jaypieApiKeyId: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const jaypieErrorFromStatus: ((statusCode: number, message?: string) => Error) & {
     mock: any;
 };
-declare const getHeaderFrom: vitest.Mock<(...args: any[]) => any>;
-declare const getObjectKeyCaseInsensitive: vitest.Mock<(...args: any[]) => any>;
-declare const isClass: vitest.Mock<(...args: any[]) => any>;
-declare const isJaypieError: vitest.Mock<(...args: any[]) => any>;
-declare const resolveValue: vitest.Mock<(...args: any[]) => any>;
-declare const safeParseFloat: vitest.Mock<(...args: any[]) => any>;
-declare const placeholders: vitest.Mock<(...args: any[]) => any>;
+declare const getHeaderFrom: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const getObjectKeyCaseInsensitive: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const isClass: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const isJaypieError: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const resolveValue: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const safeParseFloat: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const placeholders: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const force: kit.ForceFunction;
 declare const jaypieHandler: ((handler: Function, options?: {
     setup?: Function | Function[];
@@ -178,9 +239,18 @@ declare const jaypieHandler: ((handler: Function, options?: {
 }) => Function) & {
     mock: any;
 };
-declare const sleep: vitest.Mock<(...args: any[]) => any>;
-declare const uuid: vitest.Mock<(...args: any[]) => any>;
-declare const validateJaypieKey: vitest.Mock<(...args: any[]) => any>;
+declare const sleep: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const uuid: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const validateJaypieKey: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const HTTP: {
     readonly ALLOW: {
         readonly ANY: "*";
@@ -348,7 +418,9 @@ declare const DATADOG: {
     ENV: {
         DATADOG_API_KEY: string;
         DATADOG_API_KEY_ARN: string;
+        DD_API_KEY: string;
         DD_API_KEY_SECRET_ARN: string;
+        DD_LLMOBS_ENABLED: string;
         DD_SITE: string;
         SECRET_DATADOG_API_KEY: string;
     };
@@ -362,10 +434,26 @@ declare const DATADOG: {
     };
     SITE: string;
 };
-declare const hasDatadogEnv: vitest.Mock<(...args: any[]) => any>;
-declare const submitDistribution: vitest.Mock<(...args: any[]) => any>;
-declare const submitMetric: vitest.Mock<(...args: any[]) => any>;
-declare const submitMetricSet: vitest.Mock<(...args: any[]) => any>;
+declare const hasDatadogEnv: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const loadDatadogApiKey: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const submitDistribution: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const submitMetric: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const submitMetricSet: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 
 declare const APEX = "@";
 declare const ARCHIVED_SUFFIX = "#archived";
@@ -532,9 +620,18 @@ declare const destroyTable: ((params: {
     mock: any;
 };
 
-declare const createLambdaHandler: vitest.Mock<(...args: any[]) => any>;
-declare const createLambdaStreamHandler: vitest.Mock<(...args: any[]) => any>;
-declare const getCurrentInvoke: vitest.Mock<(...args: any[]) => any>;
+declare const createLambdaHandler: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const createLambdaStreamHandler: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const getCurrentInvoke: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const LambdaRequest: typeof original$1.LambdaRequest;
 declare const LambdaResponseBuffered: typeof original$1.LambdaResponseBuffered;
 declare const LambdaResponseStreaming: typeof original$1.LambdaResponseStreaming;
@@ -545,18 +642,54 @@ declare const EXPRESS: {
         readonly ROOT: RegExp;
     };
 };
-declare const badRequestRoute: vitest.Mock<(...args: any[]) => any>;
-declare const echoRoute: vitest.Mock<(...args: any[]) => any>;
-declare const forbiddenRoute: vitest.Mock<(...args: any[]) => any>;
-declare const goneRoute: vitest.Mock<(...args: any[]) => any>;
-declare const methodNotAllowedRoute: vitest.Mock<(...args: any[]) => any>;
-declare const noContentRoute: vitest.Mock<(...args: any[]) => any>;
-declare const notFoundRoute: vitest.Mock<(...args: any[]) => any>;
-declare const notImplementedRoute: vitest.Mock<(...args: any[]) => any>;
-declare const expressHttpCodeHandler: vitest.Mock<(...args: any[]) => any>;
-declare const cors: vitest.Mock<(...args: any[]) => any>;
-declare const fabricApiResponse: vitest.Mock<(...args: any[]) => any>;
-declare const getCurrentInvokeUuid: vitest.Mock<(...args: any[]) => any>;
+declare const badRequestRoute: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const echoRoute: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const forbiddenRoute: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const goneRoute: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const methodNotAllowedRoute: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const noContentRoute: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const notFoundRoute: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const notImplementedRoute: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const expressHttpCodeHandler: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const cors: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const fabricApiResponse: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const getCurrentInvokeUuid: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 interface ExpressHandlerFunction {
     (req: any, res: any, ...extra: any[]): Promise<any> | any;
 }
@@ -585,9 +718,18 @@ declare const expressHandler: ((handlerOrProps: ExpressHandlerParameter, propsOr
     mock: any;
 };
 
-declare const isLocalEnv: vitest.Mock<(...args: any[]) => any>;
-declare const isNodeTestEnv: vitest.Mock<(...args: any[]) => any>;
-declare const isProductionEnv: vitest.Mock<(...args: any[]) => any>;
+declare const isLocalEnv: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const isNodeTestEnv: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const isProductionEnv: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 
 type HandlerFunction = (...args: unknown[]) => unknown;
 type LifecycleFunction = (...args: unknown[]) => unknown | Promise<unknown>;
@@ -640,9 +782,18 @@ declare const websocketHandler: ((handler: WebSocketHandlerFunction, props?: Lam
 };
 
 declare const LLM: typeof original$2.LLM;
-declare const Llm: vitest.Mock<(...args: any[]) => any> & {
-    operate: vitest.Mock<(...args: any[]) => any>;
-    send: vitest.Mock<(...args: any[]) => any>;
+declare const Llm: vitest.MockInstance<_vitest_spy.Procedure> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {} & {
+    operate: vitest.MockInstance<(...args: any[]) => any> & {
+        (...args: any[]): any;
+        new (...args: any[]): any;
+    } & {};
+    send: vitest.MockInstance<(...args: any[]) => any> & {
+        (...args: any[]): any;
+        new (...args: any[]): any;
+    } & {};
 };
 declare const Toolkit: typeof original$2.Toolkit;
 declare const JaypieToolkit: typeof original$2.JaypieToolkit;
@@ -729,14 +880,26 @@ declare module "vitest" {
   }
 }
 
-declare const createLogger: vitest.Mock<(...args: any[]) => any>;
-declare const _resetDatadogTransport: vitest.Mock<(...args: any[]) => any>;
-declare const getDatadogTransport: vitest.Mock<(...args: any[]) => any>;
-declare const isDatadogForwardingEnabled: vitest.Mock<(...args: any[]) => any>;
+declare const createLogger: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const _resetDatadogTransport: vitest.Mock<_vitest_spy.Procedure>;
+declare const getDatadogTransport: vitest.Mock<_vitest_spy.Procedure>;
+declare const isDatadogForwardingEnabled: vitest.Mock<_vitest_spy.Procedure>;
 
-declare const connect: vitest.Mock<(...args: any[]) => any>;
-declare const connectFromSecretEnv: vitest.Mock<(...args: any[]) => any>;
-declare const disconnect: vitest.Mock<(...args: any[]) => any>;
+declare const connect: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const connectFromSecretEnv: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
+declare const disconnect: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 
 /**
  * Mock for MarkdownPage class from @jaypie/textract
@@ -745,12 +908,18 @@ declare const MarkdownPage: any;
 /**
  * Mock for textractJsonToMarkdown function from @jaypie/textract
  */
-declare const textractJsonToMarkdown: vitest.Mock<(...args: any[]) => any>;
+declare const textractJsonToMarkdown: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 
 declare const expandIncludes: ((...args: any[]) => any) & {
     mock: any;
 };
-declare const isValidAlias: vitest.Mock<(...args: any[]) => any>;
+declare const isValidAlias: vitest.MockInstance<(...args: any[]) => any> & {
+    (...args: any[]): any;
+    new (...args: any[]): any;
+} & {};
 declare const getAlternativeSpellings: ((...args: any[]) => any) & {
     mock: any;
 };
@@ -778,5 +947,5 @@ declare const createMemoryStore: ((...args: any[]) => any) & {
 
 declare const mock: Record<string, any>;
 
-export { APEX, ARCHIVED_SUFFIX, BadGatewayError, BadRequestError, BedrockProvider, ConfigurationError, ConflictError, CorsError, DATADOG, DELETED_SUFFIX, EXPRESS, ForbiddenError, GatewayTimeoutError, GeminiProvider, GoneError, GoogleProvider, HTTP, IllogicalError, InternalError, JAYPIE, JaypieError, JaypieStream, JaypieToolkit, LLM, LambdaRequest, LambdaResponseBuffered, LambdaResponseStreaming, Llm, LlmMessageRole, LlmMessageType, LlmStreamChunkType, MarkdownPage, MethodNotAllowedError, MultiError, NotFoundError, NotImplementedError, OpenRouterProvider, PROJECT, ProjectError, ProjectMultiError, RejectedError, SEPARATOR, TeapotError, TooManyRequestsError, Toolkit, UnauthorizedError, UnavailableError, UnhandledError, UnreachableCodeError, XaiProvider, _resetDatadogTransport, archiveEntity, badRequestRoute, broadcastToConnections, buildCompositeKey, calculateScope, clearWebSocketClientCache, cloneDeep, connect, connectFromSecretEnv, cors, countTable, createEntity, createExpressStream, createJaypieStream, createLambdaHandler, createLambdaStream, createLambdaStreamHandler, createLayeredStore, createLogger, createMarkdownStore, createMemoryStore, createSkillService, createTable, mock as default, deleteEntity, destroyEntity, destroyTable, disconnect, echoRoute, envBoolean, envsKey, errorFromStatusCode, expandIncludes, exportEntities, exportEntitiesToJson, expressHandler, expressHttpCodeHandler, expressStreamHandler, extractReasoning, fabricApiResponse, forbiddenRoute, force, formatError, formatNljson, formatSse, formatStreamError, formatStreamErrorNljson, formatStreamErrorSse, generateJaypieKey, getAlternativeSpellings, getClient, getContentTypeForFormat, getCurrentInvoke, getCurrentInvokeUuid, getDatadogTransport, getDocClient, getEntity, getEnvSecret, getHeaderFrom, getMessages, getObjectKeyCaseInsensitive, getS3FileBuffer, getSecret, getSingletonMessage, getTableName, getTextractJob, goneRoute, hasDatadogEnv, hashJaypieKey, indexEntity, initClient, isClass, isDatadogForwardingEnabled, isInitialized, isJaypieError, isLlmOperateInput, isLlmOperateInputContent, isLlmOperateInputFile, isLlmOperateInputImage, isLocalEnv, isNodeTestEnv, isProductionEnv, isValidAlias, jaypieApiKeyId, jaypieErrorFromStatus, jaypieHandler, lambdaHandler, lambdaStreamHandler, loadEnvSecrets, methodNotAllowedRoute, migrationHandler, noContentRoute, normalizeAlias, notFoundRoute, notImplementedRoute, parseList, placeholders, query, queryByAlias, queryByCategory, queryByScope, queryByType, queryByXid, resetClient, resolveValue, safeParseFloat, scanTable, seedEntities, seedEntityIfNotExists, sendBatchMessages, sendMessage, sendTextractJob, sendToConnection, sleep, streamToSse, submitDistribution, submitMetric, submitMetricSet, textractJsonToMarkdown, toolkit, tools, transactWriteEntities, updateEntity, uuid, validateAlias, validateJaypieKey, websocketHandler };
+export { APEX, ARCHIVED_SUFFIX, BadGatewayError, BadRequestError, BedrockProvider, ConfigurationError, ConflictError, CorsError, DATADOG, DELETED_SUFFIX, EXPRESS, ForbiddenError, GatewayTimeoutError, GeminiProvider, GoneError, GoogleProvider, HTTP, IllogicalError, InternalError, JAYPIE, JaypieError, JaypieStream, JaypieToolkit, LLM, LambdaRequest, LambdaResponseBuffered, LambdaResponseStreaming, Llm, LlmMessageRole, LlmMessageType, LlmStreamChunkType, MarkdownPage, MethodNotAllowedError, MultiError, NotFoundError, NotImplementedError, OpenRouterProvider, PROJECT, ProjectError, ProjectMultiError, RejectedError, SEPARATOR, TeapotError, TooManyRequestsError, Toolkit, UnauthorizedError, UnavailableError, UnhandledError, UnreachableCodeError, XaiProvider, _resetDatadogTransport, archiveEntity, badRequestRoute, broadcastToConnections, buildCompositeKey, calculateScope, clearWebSocketClientCache, cloneDeep, connect, connectFromSecretEnv, cors, countTable, createEntity, createExpressStream, createJaypieStream, createLambdaHandler, createLambdaStream, createLambdaStreamHandler, createLayeredStore, createLogger, createMarkdownStore, createMemoryStore, createSkillService, createTable, mock as default, deleteEntity, destroyEntity, destroyTable, disconnect, echoRoute, envBoolean, envsKey, errorFromStatusCode, expandIncludes, exportEntities, exportEntitiesToJson, expressHandler, expressHttpCodeHandler, expressStreamHandler, extractReasoning, fabricApiResponse, forbiddenRoute, force, formatError, formatNljson, formatSse, formatStreamError, formatStreamErrorNljson, formatStreamErrorSse, generateJaypieKey, getAlternativeSpellings, getClient, getContentTypeForFormat, getCurrentInvoke, getCurrentInvokeUuid, getDatadogTransport, getDocClient, getEntity, getEnvSecret, getHeaderFrom, getMessages, getObjectKeyCaseInsensitive, getS3FileBuffer, getSecret, getSingletonMessage, getTableName, getTextractJob, goneRoute, hasDatadogEnv, hashJaypieKey, indexEntity, initClient, isClass, isDatadogForwardingEnabled, isInitialized, isJaypieError, isLlmOperateInput, isLlmOperateInputContent, isLlmOperateInputFile, isLlmOperateInputImage, isLocalEnv, isNodeTestEnv, isProductionEnv, isValidAlias, jaypieApiKeyId, jaypieErrorFromStatus, jaypieHandler, lambdaHandler, lambdaStreamHandler, loadDatadogApiKey, loadEnvSecrets, methodNotAllowedRoute, migrationHandler, noContentRoute, normalizeAlias, notFoundRoute, notImplementedRoute, parseList, placeholders, query, queryByAlias, queryByCategory, queryByScope, queryByType, queryByXid, resetClient, resolveValue, safeParseFloat, scanTable, seedEntities, seedEntityIfNotExists, sendBatchMessages, sendMessage, sendTextractJob, sendToConnection, sleep, streamToSse, submitDistribution, submitMetric, submitMetricSet, textractJsonToMarkdown, toolkit, tools, transactWriteEntities, updateEntity, uuid, validateAlias, validateJaypieKey, websocketHandler };
 export type { ExpressHandlerFunction, ExpressHandlerOptions, ExpressStreamHandlerFunction, ExpressStreamHandlerOptions, LambdaOptions, WebSocketContext, WebSocketResponse };

package/dist/mock/index.js

@@ -17,7 +17,6 @@ import { fileURLToPath } from 'url';
 import { TextractDocument } from 'amazon-textract-response-parser';
 import * as original$5 from '@jaypie/textract';
 
-/* eslint-disable no-console */
 /**
  * Internal wrapper for vi.fn() that adds _jaypie: true to all mocks
  */
@@ -66,9 +65,11 @@ function createMockWrappedFunction(fn, fallbackOrOptions = "_MOCK_WRAPPED_RESULT
     const fallback = options.fallback ?? "_MOCK_WRAPPED_RESULT";
     const throws = options.throws ?? false;
     const isClass = options.class ?? false;
-    return _createJaypieMock().mockImplementation((...args) => {
+    // Shared behavior for both call and construct: try the real
+    // implementation, fall back (with a warning) when it throws.
+    const invoke = (instantiate, args) => {
         try {
-            return isClass ? new fn(...args) : fn(...args);
+            return instantiate ? new fn(...args) : fn(...args);
         }
         catch (error) {
             if (throws) {
@@ -90,7 +91,17 @@ function createMockWrappedFunction(fn, fallbackOrOptions = "_MOCK_WRAPPED_RESULT
             }
             return fallback;
         }
-    });
+    };
+    // vitest 4 requires a `class`/`function` implementation when the mock is
+    // instantiated with `new`. A plain `function` (not an arrow) can be invoked
+    // either with or without `new`; returning an object replaces the instance in
+    // both cases, so callers receive the wrapped result (or fallback).
+    if (isClass) {
+        return _createJaypieMock().mockImplementation(function (...args) {
+            return invoke(true, args);
+        });
+    }
+    return _createJaypieMock().mockImplementation((...args) => invoke(false, args));
 }
 function createMockWrappedObject(object, fallbackOrOptions = "_MOCK_WRAPPED_RESULT") {
     let returnMock = {};
@@ -635,6 +646,7 @@ var core = /*#__PURE__*/Object.freeze({
 
 const DATADOG = original$1.DATADOG;
 const hasDatadogEnv = createMockWrappedFunction(original$1.hasDatadogEnv, false);
+const loadDatadogApiKey = createMockResolvedFunction(false);
 const submitDistribution = createMockResolvedFunction(true);
 const submitMetric = createMockResolvedFunction(true);
 const submitMetricSet = createMockResolvedFunction(true);
@@ -643,6 +655,7 @@ var datadog = /*#__PURE__*/Object.freeze({
     __proto__: null,
     DATADOG: DATADOG,
     hasDatadogEnv: hasDatadogEnv,
+    loadDatadogApiKey: loadDatadogApiKey,
     submitDistribution: submitDistribution,
     submitMetric: submitMetric,
     submitMetricSet: submitMetricSet
@@ -1135,15 +1148,21 @@ const mockOperate = createMockResolvedFunction({
     ],
 });
 const mockSend = createMockResolvedFunction("_MOCK_LLM_RESPONSE");
-const Llm = Object.assign(vi.fn().mockImplementation((providerName = "_MOCK_LLM_PROVIDER") => ({
-    _provider: providerName,
-    _llm: {
+const Llm = Object.assign(
+// vitest 4 requires a constructable (non-arrow) implementation when the mock
+// is instantiated with `new`. A plain `function` returning an object replaces
+// the instance, preserving the previous mock shape.
+vi.fn().mockImplementation(function (providerName = "_MOCK_LLM_PROVIDER") {
+    return {
+        _provider: providerName,
+        _llm: {
+            operate: mockOperate,
+            send: mockSend,
+        },
         operate: mockOperate,
         send: mockSend,
-    },
-    operate: mockOperate,
-    send: mockSend,
-})), {
+    };
+}), {
     operate: mockOperate,
     send: mockSend,
 });
@@ -1390,5 +1409,5 @@ const mock = {
     ...tildeskill,
 };
 
-export { APEX, ARCHIVED_SUFFIX, BadGatewayError, BadRequestError, BedrockProvider, ConfigurationError, ConflictError, CorsError, DATADOG, DELETED_SUFFIX, EXPRESS, ForbiddenError, GatewayTimeoutError, GeminiProvider, GoneError, GoogleProvider, HTTP$1 as HTTP, IllogicalError, InternalError, JAYPIE, JaypieError, JaypieStream, JaypieToolkit, LLM, LambdaRequest, LambdaResponseBuffered, LambdaResponseStreaming, Llm, LlmMessageRole, LlmMessageType, LlmStreamChunkType, MarkdownPage, MethodNotAllowedError, MultiError, NotFoundError, NotImplementedError, OpenRouterProvider, PROJECT, ProjectError, ProjectMultiError, RejectedError, SEPARATOR, TeapotError, TooManyRequestsError, Toolkit, UnauthorizedError, UnavailableError, UnhandledError, UnreachableCodeError, XaiProvider, _resetDatadogTransport, archiveEntity, badRequestRoute, broadcastToConnections, buildCompositeKey, calculateScope, clearWebSocketClientCache, cloneDeep, connect, connectFromSecretEnv, cors, countTable, createEntity, createExpressStream, createJaypieStream, createLambdaHandler, createLambdaStream, createLambdaStreamHandler, createLayeredStore, createLogger, createMarkdownStore, createMemoryStore, createSkillService, createTable, mock as default, deleteEntity, destroyEntity, destroyTable, disconnect, echoRoute, envBoolean, envsKey, errorFromStatusCode, expandIncludes, exportEntities, exportEntitiesToJson, expressHandler, expressHttpCodeHandler, expressStreamHandler, extractReasoning, fabricApiResponse, forbiddenRoute, force, formatError, formatNljson, formatSse, formatStreamError, formatStreamErrorNljson, formatStreamErrorSse, generateJaypieKey, getAlternativeSpellings, getClient, getContentTypeForFormat, getCurrentInvoke, getCurrentInvokeUuid, getDatadogTransport, getDocClient, getEntity, getEnvSecret, getHeaderFrom, getMessages, getObjectKeyCaseInsensitive, getS3FileBuffer, getSecret, getSingletonMessage, getTableName, getTextractJob, goneRoute, hasDatadogEnv, hashJaypieKey, indexEntity, initClient, isClass, isDatadogForwardingEnabled, isInitialized, isJaypieError, isLlmOperateInput, isLlmOperateInputContent, isLlmOperateInputFile, isLlmOperateInputImage, isLocalEnv, isNodeTestEnv, isProductionEnv, isValidAlias, jaypieApiKeyId, jaypieErrorFromStatus, jaypieHandler, lambdaHandler, lambdaStreamHandler, loadEnvSecrets, methodNotAllowedRoute, migrationHandler, noContentRoute, normalizeAlias, notFoundRoute, notImplementedRoute, parseList, placeholders, query, queryByAlias, queryByCategory, queryByScope, queryByType, queryByXid, resetClient, resolveValue, safeParseFloat, scanTable, seedEntities, seedEntityIfNotExists, sendBatchMessages, sendMessage, sendTextractJob, sendToConnection, sleep, streamToSse, submitDistribution, submitMetric, submitMetricSet, textractJsonToMarkdown, toolkit, tools, transactWriteEntities, updateEntity, uuid, validateAlias, validateJaypieKey, websocketHandler };
+export { APEX, ARCHIVED_SUFFIX, BadGatewayError, BadRequestError, BedrockProvider, ConfigurationError, ConflictError, CorsError, DATADOG, DELETED_SUFFIX, EXPRESS, ForbiddenError, GatewayTimeoutError, GeminiProvider, GoneError, GoogleProvider, HTTP$1 as HTTP, IllogicalError, InternalError, JAYPIE, JaypieError, JaypieStream, JaypieToolkit, LLM, LambdaRequest, LambdaResponseBuffered, LambdaResponseStreaming, Llm, LlmMessageRole, LlmMessageType, LlmStreamChunkType, MarkdownPage, MethodNotAllowedError, MultiError, NotFoundError, NotImplementedError, OpenRouterProvider, PROJECT, ProjectError, ProjectMultiError, RejectedError, SEPARATOR, TeapotError, TooManyRequestsError, Toolkit, UnauthorizedError, UnavailableError, UnhandledError, UnreachableCodeError, XaiProvider, _resetDatadogTransport, archiveEntity, badRequestRoute, broadcastToConnections, buildCompositeKey, calculateScope, clearWebSocketClientCache, cloneDeep, connect, connectFromSecretEnv, cors, countTable, createEntity, createExpressStream, createJaypieStream, createLambdaHandler, createLambdaStream, createLambdaStreamHandler, createLayeredStore, createLogger, createMarkdownStore, createMemoryStore, createSkillService, createTable, mock as default, deleteEntity, destroyEntity, destroyTable, disconnect, echoRoute, envBoolean, envsKey, errorFromStatusCode, expandIncludes, exportEntities, exportEntitiesToJson, expressHandler, expressHttpCodeHandler, expressStreamHandler, extractReasoning, fabricApiResponse, forbiddenRoute, force, formatError, formatNljson, formatSse, formatStreamError, formatStreamErrorNljson, formatStreamErrorSse, generateJaypieKey, getAlternativeSpellings, getClient, getContentTypeForFormat, getCurrentInvoke, getCurrentInvokeUuid, getDatadogTransport, getDocClient, getEntity, getEnvSecret, getHeaderFrom, getMessages, getObjectKeyCaseInsensitive, getS3FileBuffer, getSecret, getSingletonMessage, getTableName, getTextractJob, goneRoute, hasDatadogEnv, hashJaypieKey, indexEntity, initClient, isClass, isDatadogForwardingEnabled, isInitialized, isJaypieError, isLlmOperateInput, isLlmOperateInputContent, isLlmOperateInputFile, isLlmOperateInputImage, isLocalEnv, isNodeTestEnv, isProductionEnv, isValidAlias, jaypieApiKeyId, jaypieErrorFromStatus, jaypieHandler, lambdaHandler, lambdaStreamHandler, loadDatadogApiKey, loadEnvSecrets, methodNotAllowedRoute, migrationHandler, noContentRoute, normalizeAlias, notFoundRoute, notImplementedRoute, parseList, placeholders, query, queryByAlias, queryByCategory, queryByScope, queryByType, queryByXid, resetClient, resolveValue, safeParseFloat, scanTable, seedEntities, seedEntityIfNotExists, sendBatchMessages, sendMessage, sendTextractJob, sendToConnection, sleep, streamToSse, submitDistribution, submitMetric, submitMetricSet, textractJsonToMarkdown, toolkit, tools, transactWriteEntities, updateEntity, uuid, validateAlias, validateJaypieKey, websocketHandler };
 //# sourceMappingURL=index.js.map