npm-ai-hooks 2.0.0 → 2.0.2

package/dist/esm/utils/providerInfo.js

@@ -0,0 +1,91 @@
+// Provider display names
+export const PROVIDER_NAMES = {
+    openai: "OpenAI",
+    claude: "Claude",
+    gemini: "Gemini",
+    groq: "Groq",
+    deepseek: "DeepSeek",
+    xai: "xAI",
+    mistral: "Mistral",
+    perplexity: "Perplexity",
+    openrouter: "OpenRouter"
+};
+// Get all available models for a provider
+export function getProviderModels(provider) {
+    // These are the actual model types from the library
+    const modelMap = {
+        openai: [
+            "gpt-4.1",
+            "gpt-4.1-mini",
+            "gpt-4o",
+            "gpt-4o-mini",
+            "gpt-4-turbo",
+            "gpt-4",
+            "gpt-3.5-turbo"
+        ],
+        claude: [
+            "claude-3-5-sonnet-20241022",
+            "claude-3-5-haiku-20241022",
+            "claude-3-opus-20240229",
+            "claude-3-sonnet-20240229",
+            "claude-3-haiku-20240307"
+        ],
+        gemini: [
+            "gemini-2.0-flash-exp",
+            "gemini-2.0-flash-thinking-exp-01-21",
+            "gemini-exp-1206",
+            "gemini-1.5-pro",
+            "gemini-1.5-flash",
+            "gemini-1.5-flash-8b"
+        ],
+        groq: [
+            "llama-3.3-70b-versatile",
+            "llama-3.1-8b-instant",
+            "gemma2-9b-it"
+        ],
+        deepseek: [
+            "deepseek-chat",
+            "deepseek-reasoner"
+        ],
+        xai: [
+            "grok-4-0131",
+            "grok-beta",
+            "grok-2-1212"
+        ],
+        mistral: [
+            "mistral-large-latest",
+            "mistral-small-latest",
+            "pixtral-large-latest",
+            "ministral-8b-latest",
+            "mistral-nemo"
+        ],
+        perplexity: [
+            "llama-3.1-sonar-large-128k-online",
+            "llama-3.1-sonar-small-128k-online",
+            "llama-3.1-sonar-large-128k-chat",
+            "llama-3.1-sonar-small-128k-chat"
+        ],
+        openrouter: [
+            "openai/gpt-4o",
+            "anthropic/claude-3.5-sonnet",
+            "google/gemini-2.0-flash-exp",
+            "meta-llama/llama-3.3-70b-instruct",
+            "deepseek/deepseek-chat"
+        ]
+    };
+    return modelMap[provider] || [];
+}
+// Get all supported providers
+export function getAllProviders() {
+    return Object.keys(PROVIDER_NAMES);
+}
+export function getProviderInfo(provider) {
+    return {
+        key: provider,
+        name: PROVIDER_NAMES[provider],
+        models: getProviderModels(provider)
+    };
+}
+export function getAllProviderInfo() {
+    return getAllProviders().map(getProviderInfo);
+}

package/dist/esm/wrap.js

@@ -31,16 +31,43 @@ export function wrap(fn, options) {
     }
     return async (...args) => {
         try {
-            const input = fn(...args);
+            // Await the function result to handle both sync and async functions
+            const rawInput = fn(...args);
+            const input = rawInput instanceof Promise ? await rawInput : rawInput;
+            // Check if input is multimodal
+            const isMultimodal = typeof input === 'object' && input !== null && ('image' in input || 'file' in input);
+            const textInput = isMultimodal ? input.text || '' : String(input);
+            const imageData = isMultimodal ? input.image : undefined;
+            const fileData = isMultimodal ? input.file : undefined;
             // Step 1: get provider function and the actual provider name
             const { fn: providerFn, provider: providerKey } = getProvider(options.provider);
             // Step 2: pick model: passed model or provider-specific default
             const model = options.model || (providerKey in DEFAULT_MODEL ? DEFAULT_MODEL[providerKey] : undefined);
             if (!model) {
-                throw new AIHookError("NO_MODEL_FOUND", "No model found: You must specify a provider or pass a valid model.\n\nAt least one provider API key is required in your .env file.\n\nPlease add one of the following to your .env (see .env.example for details):\n  - AI_HOOK_OPENAI_KEY\n  - AI_HOOK_OPENROUTER_KEY\n  - AI_HOOK_GROQ_KEY\n", options.provider, "Reference .env.example for setup instructions.");
+                throw new AIHookError("NO_MODEL_FOUND", "No model found: You must specify a provider or pass a valid model.\n\nAt least one provider API key is required in your .env file.\n\nPlease add one of the following to your .env (see .env.example for details):\n  - OPENAI_KEY\n  - OPENROUTER_KEY\n  - GROQ_KEY\n", options.provider, "Reference .env.example for setup instructions.");
+            }
+            // Step 3: build prompt with multimodal support
+            let prompt;
+            if (options.customPrompt) {
+                // Use custom prompt if provided
+                prompt = `${options.customPrompt}\n\n${textInput}`;
+                if (imageData) {
+                    prompt = `${prompt}\n\n[Image attached]`;
+                }
+                if (fileData) {
+                    prompt = `${prompt}\n\n[File: ${fileData.name}]`;
+                }
+            }
+            else {
+                // Use built-in task prompt
+                prompt = buildPrompt(options.task, textInput, options.targetLanguage);
+                if (imageData) {
+                    prompt = `${prompt}\n\n[Image attached]`;
+                }
+                if (fileData) {
+                    prompt = `${prompt}\n\n[File: ${fileData.name}]`;
+                }
             }
-            // Step 3: build prompt
-            const prompt = buildPrompt(options.task, input, options.targetLanguage);
             const startTime = Date.now();
             let output;
             try {

package/dist/index.d.ts

@@ -1,2 +1,7 @@
 export { wrap } from "./wrap";
+export type { MultimodalInput } from "./wrap";
 export { initAIHooks, addProvider, removeProvider, getAvailableProviders, getProvider, isInitialized, reset } from "./providers";
+export type { Provider, ProviderModels, TaskType } from "./types";
+export type { OpenAIModel, ClaudeModel, GeminiModel, GroqModel, DeepSeekModel, XAIModel, MistralModel, PerplexityModel, OpenRouterModel } from "./types";
+export { PROVIDER_NAMES, getProviderModels, getAllProviders, getProviderInfo, getAllProviderInfo, type ProviderInfo } from "./utils/providerInfo";
+export { validateApiKey, validateApiKeys, quickValidateKeyFormat, type KeyValidationResult } from "./utils/keyValidator";

package/dist/providers/base/SpecializedProviders.d.ts

@@ -9,7 +9,7 @@ export declare class GeminiProvider extends BaseProvider {
     protected buildRequestConfig(prompt: string, model: string, apiKey: string): {
         url: string;
         headers: {};
-        method?: import("axios").Method | string;
+        method?: (string & {}) | import("axios").Method;
         baseURL?: string;
         allowAbsoluteUrls?: boolean;
         transformRequest?: import("axios").AxiosRequestTransformer | import("axios").AxiosRequestTransformer[];
@@ -20,10 +20,10 @@ export declare class GeminiProvider extends BaseProvider {
         timeout?: number;
         timeoutErrorMessage?: string;
         withCredentials?: boolean;
-        adapter?: (import("axios").AxiosAdapter | ("fetch" | "xhr" | "http" | (string & {}))) | (import("axios").AxiosAdapter | ("fetch" | "xhr" | "http" | (string & {})))[];
+        adapter?: (import("axios").AxiosAdapter | ((string & {}) | "xhr" | "http" | "fetch")) | (import("axios").AxiosAdapter | ((string & {}) | "xhr" | "http" | "fetch"))[];
         auth?: import("axios").AxiosBasicCredentials;
         responseType?: import("axios").ResponseType;
-        responseEncoding?: import("axios").responseEncoding | string;
+        responseEncoding?: (string & {}) | import("axios").responseEncoding;
         xsrfCookieName?: string;
         xsrfHeaderName?: string;
         onUploadProgress?: (progressEvent: import("axios").AxiosProgressEvent) => void;
@@ -36,13 +36,18 @@ export declare class GeminiProvider extends BaseProvider {
         beforeRedirect?: (options: Record<string, any>, responseDetails: {
             headers: Record<string, string>;
             statusCode: import("axios").HttpStatusCode;
+        }, requestDetails: {
+            headers: Record<string, string>;
+            url: string;
+            method: string;
         }) => void;
         socketPath?: string | null;
+        allowedSocketPaths?: string | string[] | null;
         transport?: any;
         httpAgent?: any;
         httpsAgent?: any;
         proxy?: import("axios").AxiosProxyConfig | false;
-        cancelToken?: import("axios").CancelToken;
+        cancelToken?: import("axios").CancelToken | undefined;
         decompress?: boolean;
         transitional?: import("axios").TransitionalOptions;
         signal?: import("axios").GenericAbortSignal;
@@ -57,8 +62,16 @@ export declare class GeminiProvider extends BaseProvider {
         family?: import("axios").AddressFamily;
         lookup?: ((hostname: string, options: object, cb: (err: Error | null, address: import("axios").LookupAddress | import("axios").LookupAddress[], family?: import("axios").AddressFamily) => void) => void) | ((hostname: string, options: object) => Promise<[address: import("axios").LookupAddressEntry | import("axios").LookupAddressEntry[], family?: import("axios").AddressFamily] | import("axios").LookupAddress>);
         withXSRFToken?: boolean | ((config: import("axios").InternalAxiosRequestConfig) => boolean | undefined);
-        parseReviver?: (this: any, key: string, value: any) => any;
+        parseReviver?: (this: any, key: string, value: any, context?: {
+            source?: string;
+        }) => any;
         fetchOptions?: Omit<RequestInit, "body" | "headers" | "method" | "signal"> | Record<string, any>;
+        httpVersion?: 1 | 2;
+        http2Options?: Record<string, any> & {
+            sessionTimeout?: number;
+        };
+        formDataHeaderPolicy?: "legacy" | "content-only";
+        redact?: string[];
     };
     protected buildAuthHeaders(): Record<string, string>;
 }

package/dist/types/index.d.ts

@@ -7,6 +7,15 @@ import { DeepSeekModel } from "./deepseek";
 import { XAIModel } from "./xai";
 import { PerplexityModel } from "./perplexity";
 import { MistralModel } from "./mistral";
+export type { OpenAIModel } from "./openai";
+export type { OpenRouterModel } from "./openrouter";
+export type { GroqModel } from "./groq";
+export type { GeminiModel } from "./gemini";
+export type { ClaudeModel } from "./claude";
+export type { DeepSeekModel } from "./deepseek";
+export type { XAIModel } from "./xai";
+export type { PerplexityModel } from "./perplexity";
+export type { MistralModel } from "./mistral";
 export type Provider = "openrouter" | "groq" | "openai" | "gemini" | "claude" | "deepseek" | "xai" | "perplexity" | "mistral";
 export type ProviderModels = {
     openrouter: OpenRouterModel;
@@ -26,11 +35,13 @@ export type TaskType = "summarize" | "translate" | "explain" | "rewrite" | "sent
 export type WrapOptions<P extends Provider | undefined = undefined> = {
     provider: P extends Provider ? P : never;
     model?: P extends Provider ? ProviderModels[P] : never;
-    task: TaskType;
+    task?: TaskType;
     targetLanguage?: string;
+    customPrompt?: string;
 } | {
     provider?: never;
     model?: never;
-    task: TaskType;
+    task?: TaskType;
     targetLanguage?: string;
+    customPrompt?: string;
 };

package/dist/utils/keyValidator.d.ts

@@ -0,0 +1,19 @@
+import type { Provider } from "../types";
+export interface KeyValidationResult {
+    valid: boolean;
+    provider: Provider;
+    error?: string;
+    message: string;
+}
+/**
+ * Validates an API key by making a minimal test request to the provider
+ */
+export declare function validateApiKey(provider: Provider, apiKey: string): Promise<KeyValidationResult>;
+/**
+ * Validates multiple API keys
+ */
+export declare function validateApiKeys(keys: Record<Provider, string>): Promise<Record<Provider, KeyValidationResult>>;
+/**
+ * Quick check if API key format looks valid (doesn't make API call)
+ */
+export declare function quickValidateKeyFormat(provider: Provider, apiKey: string): KeyValidationResult;

package/dist/utils/providerInfo.d.ts

@@ -0,0 +1,11 @@
+import type { Provider } from "../types";
+export declare const PROVIDER_NAMES: Record<Provider, string>;
+export declare function getProviderModels(provider: Provider): string[];
+export declare function getAllProviders(): Provider[];
+export interface ProviderInfo {
+    key: Provider;
+    name: string;
+    models: string[];
+}
+export declare function getProviderInfo(provider: Provider): ProviderInfo;
+export declare function getAllProviderInfo(): ProviderInfo[];

package/dist/wrap.d.ts

@@ -1,4 +1,13 @@
 import { WrapOptions, Provider } from "./types";
+export interface MultimodalInput {
+    text?: string;
+    image?: string;
+    file?: {
+        name: string;
+        data: string;
+        type: string;
+    };
+}
 export declare function wrap<T extends (...args: any[]) => any, P extends Provider | undefined = undefined>(fn: T, options: WrapOptions<P>): (...args: Parameters<T>) => Promise<{
     output: string;
     meta: any;

package/package.json

@@ -1,107 +1,94 @@
-{
-  "name": "npm-ai-hooks",
-  "version": "2.0.0",
-  "description": "Universal AI Hook Layer for Node.js and React – one wrapper for all AI providers. Inject LLM-like behavior into any JavaScript or TypeScript function with a single line, without writing prompts, handling SDKs, or locking into any provider.",
-  "main": "dist/cjs/index.js",
-  "module": "dist/esm/index.js",
-  "types": "dist/index.d.ts",
-  "exports": {
-    ".": {
-      "types": "./dist/index.d.ts",
-      "import": "./dist/esm/index.js",
-      "require": "./dist/cjs/index.js"
-    }
-  },
-  "files": [
-    "dist",
-    "Readme.md",
-    "LICENSE"
-  ],
-  "scripts": {
-    "dev": "ts-node src/index.ts",
-    "clean": "rimraf dist",
-    "build": "npm run clean && npm run build:esm && npm run build:cjs",
-    "build:esm": "tsc --project tsconfig.esm.json",
-    "build:cjs": "tsc --project tsconfig.cjs.json",
-    "test": "jest --verbose",
-    "test:watch": "jest --watch",
-    "test:coverage": "jest --coverage",
-    "test:ci": "jest --ci --coverage --watchAll=false",
-    "test:providers": "jest tests/providers.test.ts --verbose",
-    "test:tasks": "jest tests/tasks.test.ts --verbose",
-    "test:errors": "jest tests/error-handling.test.ts --verbose",
-    "test:integration": "jest tests/integration.test.ts --verbose",
-    "test:performance": "jest tests/performance.test.ts --verbose",
-    "test:env": "jest --testNamePattern=\"Real API|Environment-based\" --verbose",
-    "test:mock": "jest --testNamePattern=\"Provider Detection|Task Tests\" --verbose",
-    "lint": "eslint . --ext .ts",
-    "format": "prettier --write .",
-    "prepare": "npm run build",
-    "demo": "npx ts-node examples/demo.ts",
-    "setup:dev": "bash scripts/setup-dev.sh",
-    "setup:dev:win": "scripts\\setup-dev.bat",
-    "setup:dev:ps": "powershell -ExecutionPolicy Bypass -File scripts\\setup-dev.ps1"
-  },
-  "keywords": [
-    "ai",
-    "hooks",
-    "llm",
-    "nodejs",
-    "typescript",
-    "openai",
-    "gemini",
-    "claude",
-    "deepseek",
-    "groq",
-    "mistral"
-  ],
-  "author": "AteebNoOne <ateebnoone@gmail.com>",
-  "repository": {
-    "type": "git",
-    "url": "https://github.com/iTeebot/npm-ai-hooks.git"
-  },
-  "license": "MIT",
-  "dependencies": {
-    "axios": "^1.12.2"
-  },
-  "devDependencies": {
-    "@types/jest": "^30.0.0",
-    "@types/node": "^24.7.0",
-    "dotenv": "^17.2.3",
-    "eslint": "^9.37.0",
-    "jest": "^30.2.0",
-    "prettier": "^3.6.2",
-    "rimraf": "^6.0.1",
-    "ts-jest": "^29.4.4",
-    "ts-node": "^10.9.2",
-    "typescript": "^5.9.3"
-  },
-  "keywords": [
-    "ai",
-    "llm",
-    "openai",
-    "claude",
-    "gemini",
-    "groq",
-    "openrouter",
-    "deepseek",
-    "mistral",
-    "xai",
-    "perplexity",
-    "react",
-    "vite",
-    "express",
-    "nodejs",
-    "typescript",
-    "hooks",
-    "wrapper",
-    "universal",
-    "cross-platform"
-  ],
-  "engines": {
-    "node": ">=18"
-  },
-  "publishConfig": {
-    "access": "public"
-  }
-}
+{
+  "name": "npm-ai-hooks",
+  "version": "2.0.2",
+  "description": "Universal AI Hook Layer for Node.js and React – one wrapper for all AI providers. Inject LLM-like behavior into any JavaScript or TypeScript function with a single line, without writing prompts, handling SDKs, or locking into any provider.",
+  "main": "dist/cjs/index.js",
+  "module": "dist/esm/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/esm/index.js",
+      "require": "./dist/cjs/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "Readme.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "dev": "ts-node src/index.ts",
+    "clean": "rimraf dist",
+    "build": "npm run clean && npm run build:esm && npm run build:cjs",
+    "build:esm": "tsc --project tsconfig.esm.json",
+    "build:cjs": "tsc --project tsconfig.cjs.json",
+    "test": "jest --verbose",
+    "test:watch": "jest --watch",
+    "test:coverage": "jest --coverage",
+    "test:ci": "jest --ci --coverage --watchAll=false",
+    "test:providers": "jest tests/providers.test.ts --verbose",
+    "test:tasks": "jest tests/tasks.test.ts --verbose",
+    "test:errors": "jest tests/error-handling.test.ts --verbose",
+    "test:integration": "jest tests/integration.test.ts --verbose",
+    "test:performance": "jest tests/performance.test.ts --verbose",
+    "test:env": "jest --testNamePattern=\"Real API|Environment-based\" --verbose",
+    "test:mock": "jest --testNamePattern=\"Provider Detection|Task Tests\" --verbose",
+    "lint": "eslint . --ext .ts",
+    "format": "prettier --write .",
+    "prepare": "npm run build",
+    "demo": "npx ts-node examples/demo.ts",
+    "setup:dev": "bash scripts/setup-dev.sh",
+    "setup:dev:win": "scripts\\setup-dev.bat",
+    "setup:dev:ps": "powershell -ExecutionPolicy Bypass -File scripts\\setup-dev.ps1"
+  },
+  "author": "AteebNoOne <ateebnoone@gmail.com>",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/iTeebot/npm-ai-hooks.git"
+  },
+  "license": "MIT",
+  "dependencies": {
+    "axios": "^1.12.2"
+  },
+  "devDependencies": {
+    "@types/jest": "^30.0.0",
+    "@types/node": "^24.7.0",
+    "dotenv": "^17.2.3",
+    "eslint": "^9.37.0",
+    "jest": "^30.2.0",
+    "prettier": "^3.6.2",
+    "rimraf": "^6.0.1",
+    "ts-jest": "^29.4.4",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.9.3"
+  },
+  "keywords": [
+    "ai",
+    "llm",
+    "openai",
+    "claude",
+    "gemini",
+    "groq",
+    "openrouter",
+    "deepseek",
+    "mistral",
+    "xai",
+    "perplexity",
+    "react",
+    "vite",
+    "express",
+    "nodejs",
+    "typescript",
+    "hooks",
+    "wrapper",
+    "universal",
+    "cross-platform"
+  ],
+  "engines": {
+    "node": ">=18"
+  },
+  "publishConfig": {
+    "access": "public"
+  }
+}