@pwshub/aisdk 0.0.3 → 0.0.5

package/src/config.test.js

@@ -0,0 +1,142 @@
+/**
+ * @fileoverview Tests for config module.
+ */
+
+import {
+  describe, it,
+} from 'node:test'
+import assert from 'node:assert'
+import {
+  normalizeConfig, getWireMap,
+} from '../src/config.js'
+
+describe('config', () => {
+  describe('getWireMap', () => {
+    it('should return wire map for openai', () => {
+      const wireMap = getWireMap('openai')
+      assert.ok(wireMap)
+      assert.ok(wireMap.temperature)
+      assert.strictEqual(wireMap.temperature.wireKey, 'temperature')
+    })
+
+    it('should return wire map for anthropic', () => {
+      const wireMap = getWireMap('anthropic')
+      assert.ok(wireMap)
+      assert.ok(wireMap.temperature)
+      assert.strictEqual(wireMap.temperature.wireKey, 'temperature')
+    })
+
+    it('should return wire map for google', () => {
+      const wireMap = getWireMap('google')
+      assert.ok(wireMap)
+      assert.ok(wireMap.temperature)
+      assert.strictEqual(wireMap.temperature.wireKey, 'temperature')
+      assert.strictEqual(wireMap.temperature.scope, 'generationConfig')
+    })
+
+    it('should return wire map for dashscope', () => {
+      const wireMap = getWireMap('dashscope')
+      assert.ok(wireMap)
+      assert.ok(wireMap.temperature)
+      assert.strictEqual(wireMap.temperature.wireKey, 'temperature')
+    })
+
+    it('should return wire map for deepseek', () => {
+      const wireMap = getWireMap('deepseek')
+      assert.ok(wireMap)
+      assert.ok(wireMap.temperature)
+      assert.strictEqual(wireMap.temperature.wireKey, 'temperature')
+    })
+
+    it('should return undefined for unknown provider', () => {
+      const wireMap = getWireMap('unknown')
+      assert.strictEqual(wireMap, undefined)
+    })
+  })
+
+  describe('normalizeConfig', () => {
+    it('should normalize openai config', () => {
+      const config = {
+        temperature: 0.5,
+        maxTokens: 100,
+        topP: 0.9,
+      }
+      const supportedParams = ['temperature', 'maxTokens', 'topP']
+      const result = normalizeConfig(config, 'openai', supportedParams, 'gpt-4o')
+
+      assert.strictEqual(result.temperature, 0.5)
+      assert.strictEqual(result.max_completion_tokens, 100)
+      assert.strictEqual(result.top_p, 0.9)
+    })
+
+    it('should normalize anthropic config', () => {
+      const config = {
+        temperature: 0.5,
+        maxTokens: 100,
+        topK: 50,
+      }
+      const supportedParams = ['temperature', 'maxTokens', 'topK']
+      const result = normalizeConfig(config, 'anthropic', supportedParams, 'claude-sonnet')
+
+      assert.strictEqual(result.temperature, 0.5)
+      assert.strictEqual(result.max_tokens, 100)
+      assert.strictEqual(result.top_k, 50)
+    })
+
+    it('should normalize google config with generationConfig nesting', () => {
+      const config = {
+        temperature: 0.5,
+        maxTokens: 100,
+        topP: 0.9,
+      }
+      const supportedParams = ['temperature', 'maxTokens', 'topP']
+      const result = normalizeConfig(config, 'google', supportedParams, 'gemini-2.0-flash')
+
+      assert.strictEqual(result.temperature, undefined)
+      assert.strictEqual(result.maxTokens, undefined)
+      assert.strictEqual(result.topP, undefined)
+      assert.ok(result.generationConfig)
+      assert.strictEqual(result.generationConfig.temperature, 0.5)
+      assert.strictEqual(result.generationConfig.maxOutputTokens, 100)
+      assert.strictEqual(result.generationConfig.topP, 0.9)
+    })
+
+    it('should skip unsupported params', () => {
+      const config = {
+        temperature: 0.5,
+        topK: 50, // openai doesn't support topK
+      }
+      const supportedParams = ['temperature']
+      const result = normalizeConfig(config, 'openai', supportedParams, 'gpt-4o')
+
+      assert.strictEqual(result.temperature, 0.5)
+      assert.strictEqual(result.top_k, undefined)
+    })
+
+    it('should skip null/undefined values', () => {
+      const config = {
+        temperature: 0.5,
+        maxTokens: null,
+        topP: undefined,
+      }
+      const supportedParams = ['temperature', 'maxTokens', 'topP']
+      const result = normalizeConfig(config, 'openai', supportedParams, 'gpt-4o')
+
+      assert.strictEqual(result.temperature, 0.5)
+      assert.strictEqual(result.max_completion_tokens, undefined)
+      assert.strictEqual(result.top_p, undefined)
+    })
+
+    it('should return empty object when no config provided', () => {
+      const result = normalizeConfig({}, 'openai', ['temperature'], 'gpt-4o')
+      assert.deepStrictEqual(result, {})
+    })
+
+    it('should return empty object when no supported params match', () => {
+      const config = { unknownParam: 123 }
+      const supportedParams = ['temperature']
+      const result = normalizeConfig(config, 'openai', supportedParams, 'gpt-4o')
+      assert.deepStrictEqual(result, {})
+    })
+  })
+})

package/src/errors.js

@@ -35,7 +35,7 @@ export class ProviderError extends Error {
    * @param {object} meta
    * @param {number} meta.status        - HTTP status code
    * @param {string} meta.provider      - Provider ID
-   * @param {string} meta.model         - Model ID that was called
+   * @param {string} meta.model         - Model name that was called
    * @param {string} [meta.raw]         - Raw response body from provider
    */
   constructor(message, {

package/src/index.js

@@ -6,7 +6,7 @@
  *
  * const ai = createAi()
  * const result = await ai.ask({
- *   model: 'claude-sonnet-4-20250514',
+ *   model: 'anthropic/claude-sonnet-4-20250514',
  *   apikey: 'your-api-key',
  *   prompt: 'What is the capital of Vietnam?',
  *   temperature: 0.5,
@@ -16,18 +16,18 @@
  *
  * @example With fallbacks
  * const result = await ai.ask({
- *   model: 'gpt-4o',
+ *   model: 'openai/gpt-4o',
  *   apikey: 'your-openai-key',
  *   prompt: '...',
- *   fallbacks: ['gpt-4o-mini', 'claude-haiku-4-5-20251001'],
+ *   fallbacks: ['openai/gpt-4o-mini', 'anthropic/claude-haiku-4-5-20251001'],
  * })
- * if (result.model !== 'gpt-4o') {
+ * if (result.model !== 'openai/gpt-4o') {
  *   console.warn('Fell back to', result.model)
  * }
  *
  * @example Google provider-specific options
  * const result = await ai.ask({
- *   model: 'gemini-2.0-flash',
+ *   model: 'google/gemini-2.0-flash',
  *   apikey: 'your-google-key',
  *   prompt: '...',
  *   providerOptions: {
@@ -38,10 +38,21 @@
  *   },
  * })
  *
+ * @example Using messages array for multi-turn conversations
+ * const result = await ai.ask({
+ *   model: 'anthropic/claude-sonnet-4-20250514',
+ *   apikey: 'your-api-key',
+ *   messages: [
+ *     { role: 'user', content: 'What is the capital of Vietnam?' },
+ *     { role: 'assistant', content: 'The capital of Vietnam is Hanoi.' },
+ *     { role: 'user', content: 'What is its population?' },
+ *   ],
+ * })
+ *
  */
 
 import {
-  getModel, listModels, setModels,
+  getModel, listModels, setModels, addModels,
 } from './registry.js'
 import { normalizeConfig } from './config.js'
 import { coerceConfig } from './coerce.js'
@@ -62,11 +73,12 @@ export {
 
 /**
  * @typedef {Object} AskParams
- * @property {string} model                       - Model ID (must be registered via setModels())
+ * @property {string} model                       - Model name or 'provider/name' format (e.g., 'gpt-4o', 'ollama/llama3.2')
  * @property {string} apikey                      - API key for the provider
- * @property {string} prompt                      - The user message
- * @property {string} [system]                    - Optional system prompt
- * @property {string[]} [fallbacks]               - Ordered list of fallback model IDs
+ * @property {string} [prompt]                    - The user message (alternative to messages)
+ * @property {string} [system]                    - Optional system prompt (used with prompt)
+ * @property {import('./providers.js').Message[]} [messages] - Array of messages with role and content (alternative to prompt)
+ * @property {string[]} [fallbacks]               - Ordered list of fallback models (same format as model)
  * @property {Record<string, unknown>} [providerOptions] - Provider-specific options merged into body
  * @property {number} [temperature]
  * @property {number} [maxTokens]
@@ -152,11 +164,11 @@ const callModel = async (modelId, params, gatewayUrl) => {
   const normalizedConfig = normalizeConfig(coerced, providerId, supportedParams, modelId)
 
   const {
-    prompt, system, providerOptions = {},
+    prompt, system, messages, providerOptions = {},
   } = params
 
   /** @type {import('./providers.js').Message[]} */
-  const messages = [
+  const messageList = messages ?? [
     ...(system ? [{
       role: 'system', content: system,
     }] : []),
@@ -166,7 +178,7 @@ const callModel = async (modelId, params, gatewayUrl) => {
   ]
 
   const url = gatewayUrl ?? adapter.url(modelName, apikey)
-  const body = adapter.buildBody(modelName, messages, normalizedConfig, providerOptions)
+  const body = adapter.buildBody(modelName, messageList, normalizedConfig, providerOptions)
 
   let res
   try {
@@ -267,4 +279,4 @@ export const createAi = (opts = {}) => {
   }
 }
 
-export { setModels }
+export { addModels, setModels, listModels }

package/src/models.js

@@ -0,0 +1,401 @@
+/**
+ * @fileoverview Default model registry for @pwshub/aisdk.
+ *
+ * This module exports a default list of models that are loaded automatically
+ * when the library is imported. Users can modify this list via addModels()
+ * and setModels() from the main export.
+ */
+
+/**
+ * @typedef {import('./registry.js').ModelRecord} ModelRecord
+ */
+
+/** @type {ModelRecord[]} */
+export const DEFAULT_MODELS = [
+  {
+    id: 'claude-haiku-4-5',
+    name: 'claude-haiku-4-5',
+    provider: 'anthropic',
+    input_price: 1,
+    output_price: 5,
+    cache_price: 0,
+    max_in: 200000,
+    max_out: 64000,
+    enable: true,
+  },
+  {
+    id: 'claude-sonnet-4-6',
+    name: 'claude-sonnet-4-6',
+    provider: 'anthropic',
+    input_price: 3,
+    output_price: 15,
+    cache_price: 0,
+    max_in: 200000,
+    max_out: 64000,
+    enable: true,
+  },
+  {
+    id: 'claude-sonnet-4-5',
+    name: 'claude-sonnet-4-5',
+    provider: 'anthropic',
+    input_price: 3,
+    output_price: 15,
+    cache_price: 0,
+    max_in: 200000,
+    max_out: 1000000,
+    enable: true,
+  },
+  {
+    id: 'claude-opus-4-6',
+    name: 'claude-opus-4-6',
+    provider: 'anthropic',
+    input_price: 5,
+    output_price: 25,
+    cache_price: 0,
+    max_in: 200000,
+    max_out: 128000,
+    enable: true,
+  },
+  {
+    id: 'gemini-2.5-flash',
+    name: 'gemini-2.5-flash',
+    provider: 'google',
+    input_price: 0.3,
+    output_price: 2.5,
+    cache_price: 0.03,
+    max_in: 1048576,
+    max_out: 65536,
+    enable: true,
+  },
+  {
+    id: 'gemini-2.5-flash-lite',
+    name: 'gemini-2.5-flash-lite',
+    provider: 'google',
+    input_price: 0.1,
+    output_price: 0.4,
+    cache_price: 0.01,
+    max_in: 1048576,
+    max_out: 65536,
+    enable: true,
+  },
+  {
+    id: 'gemini-2.5-pro',
+    name: 'gemini-2.5-pro',
+    provider: 'google',
+    input_price: 1.25,
+    output_price: 10,
+    cache_price: 0.125,
+    max_in: 1048576,
+    max_out: 65536,
+    enable: true,
+  },
+  {
+    id: 'gemini-3.1-pro-preview',
+    name: 'gemini-3.1-pro-preview',
+    provider: 'google',
+    input_price: 2,
+    output_price: 12,
+    cache_price: 0.2,
+    max_in: 1048576,
+    max_out: 65536,
+    enable: true,
+  },
+  {
+    id: 'gemini-3.1-flash-lite-preview',
+    name: 'gemini-3.1-flash-lite-preview',
+    provider: 'google',
+    input_price: 0.25,
+    output_price: 1.5,
+    cache_price: 0.025,
+    max_in: 1048576,
+    max_out: 65536,
+    enable: true,
+  },
+  {
+    id: 'gpt-4.1-nano',
+    name: 'gpt-4.1-nano',
+    provider: 'openai',
+    input_price: 0.1,
+    output_price: 0.4,
+    cache_price: 0.025,
+    max_in: 1047576,
+    max_out: 32768,
+    enable: true,
+  },
+  {
+    id: 'gpt-4.1-mini',
+    name: 'gpt-4.1-mini',
+    provider: 'openai',
+    input_price: 0.4,
+    output_price: 1.6,
+    cache_price: 0.1,
+    max_in: 1047576,
+    max_out: 32768,
+    enable: true,
+  },
+  {
+    id: 'gpt-4.1',
+    name: 'gpt-4.1',
+    provider: 'openai',
+    input_price: 2,
+    output_price: 8,
+    cache_price: 0.5,
+    max_in: 1047576,
+    max_out: 32768,
+    enable: true,
+  },
+  {
+    id: 'gpt-4o',
+    name: 'gpt-4o',
+    provider: 'openai',
+    input_price: 2.5,
+    output_price: 10,
+    cache_price: 1.25,
+    max_in: 128000,
+    max_out: 16384,
+    enable: true,
+  },
+  {
+    id: 'gpt-4o-mini',
+    name: 'gpt-4o-mini',
+    provider: 'openai',
+    input_price: 0.15,
+    output_price: 0.6,
+    cache_price: 0.075,
+    max_in: 128000,
+    max_out: 16384,
+    enable: true,
+  },
+  {
+    id: 'gpt-5',
+    name: 'gpt-5',
+    provider: 'openai',
+    input_price: 1.25,
+    output_price: 10,
+    cache_price: 0.125,
+    max_in: 400000,
+    max_out: 128000,
+    enable: true,
+  },
+  {
+    id: 'gpt-5-mini',
+    name: 'gpt-5-mini',
+    provider: 'openai',
+    input_price: 0.25,
+    output_price: 2,
+    cache_price: 0.025,
+    max_in: 400000,
+    max_out: 128000,
+    enable: true,
+  },
+  {
+    id: 'gpt-5-nano',
+    name: 'gpt-5-nano',
+    provider: 'openai',
+    input_price: 0.05,
+    output_price: 0.4,
+    cache_price: 0.005,
+    max_in: 400000,
+    max_out: 128000,
+    enable: true,
+  },
+  {
+    id: 'gpt-5.1',
+    name: 'gpt-5.1',
+    provider: 'openai',
+    input_price: 1.25,
+    output_price: 10,
+    cache_price: 0.125,
+    max_in: 400000,
+    max_out: 128000,
+    enable: true,
+  },
+  {
+    id: 'gpt-5.2',
+    name: 'gpt-5.2',
+    provider: 'openai',
+    input_price: 1.75,
+    output_price: 14,
+    cache_price: 0.175,
+    max_in: 400000,
+    max_out: 128000,
+    enable: true,
+  },
+  {
+    id: 'gpt-5.4',
+    name: 'gpt-5.4',
+    provider: 'openai',
+    input_price: 2.5,
+    output_price: 15,
+    cache_price: 0.25,
+    max_in: 1050000,
+    max_out: 128000,
+    enable: true,
+  },
+  {
+    id: 'o3-mini',
+    name: 'o3-mini',
+    provider: 'openai',
+    input_price: 1.1,
+    output_price: 4.4,
+    cache_price: 0.55,
+    max_in: 200000,
+    max_out: 100000,
+    enable: true,
+  },
+  {
+    id: 'o4-mini',
+    name: 'o4-mini',
+    provider: 'openai',
+    input_price: 1.1,
+    output_price: 4.4,
+    cache_price: 0.275,
+    max_in: 200000,
+    max_out: 100000,
+    enable: true,
+  },
+  {
+    id: 'deepseek-chat',
+    name: 'deepseek-chat',
+    provider: 'deepseek',
+    input_price: 0.28,
+    output_price: 0.42,
+    cache_price: 0.028,
+    max_in: 128000,
+    max_out: 8000,
+    enable: true,
+  },
+  {
+    id: 'deepseek-reasoner',
+    name: 'deepseek-reasoner',
+    provider: 'deepseek',
+    input_price: 0.28,
+    output_price: 0.42,
+    cache_price: 0.028,
+    max_in: 128000,
+    max_out: 64000,
+    enable: true,
+  },
+  {
+    id: 'qwen-flash',
+    name: 'qwen-flash',
+    provider: 'dashscope',
+    input_price: 0.05,
+    output_price: 0.4,
+    cache_price: 0,
+    max_in: 995904,
+    max_out: 32768,
+    enable: true,
+  },
+  {
+    id: 'qwen3.5-flash',
+    name: 'qwen3.5-flash',
+    provider: 'dashscope',
+    input_price: 0.1,
+    output_price: 0.4,
+    cache_price: 0,
+    max_in: 983616,
+    max_out: 65536,
+    enable: true,
+  },
+  {
+    id: 'qwen-plus',
+    name: 'qwen-plus',
+    provider: 'dashscope',
+    input_price: 0.4,
+    output_price: 1.2,
+    cache_price: 0,
+    max_in: 997952,
+    max_out: 32768,
+    enable: true,
+  },
+  {
+    id: 'qwen3.5-plus',
+    name: 'qwen3.5-plus',
+    provider: 'dashscope',
+    input_price: 0.4,
+    output_price: 2.4,
+    cache_price: 0,
+    max_in: 991808,
+    max_out: 65536,
+    enable: true,
+  },
+  {
+    id: 'qwen-max',
+    name: 'qwen-max',
+    provider: 'dashscope',
+    input_price: 1.6,
+    output_price: 6.4,
+    cache_price: 0,
+    max_in: 30720,
+    max_out: 8192,
+    enable: true,
+  },
+  {
+    id: 'qwen3-max',
+    name: 'qwen3-max',
+    provider: 'dashscope',
+    input_price: 1.2,
+    output_price: 6,
+    cache_price: 0,
+    max_in: 258048,
+    max_out: 65536,
+    enable: true,
+  },
+  // Mistral models
+  {
+    id: 'mistral-large-latest',
+    name: 'mistral-large-latest',
+    provider: 'mistral',
+    input_price: 0.5,
+    output_price: 1.5,
+    cache_price: 0,
+    max_in: 128000,
+    max_out: 128000,
+    enable: true,
+  },
+  {
+    id: 'mistral-medium-latest',
+    name: 'mistral-medium-latest',
+    provider: 'mistral',
+    input_price: 0.4,
+    output_price: 2,
+    cache_price: 0,
+    max_in: 64000,
+    max_out: 64000,
+    enable: true,
+  },
+  {
+    id: 'mistral-small-latest',
+    name: 'mistral-small-latest',
+    provider: 'mistral',
+    input_price: 0.15,
+    output_price: 0.6,
+    cache_price: 0,
+    max_in: 128000,
+    max_out: 128000,
+    enable: true,
+  },
+  {
+    id: 'magistral-medium-latest',
+    name: 'magistral-medium-latest',
+    provider: 'mistral',
+    input_price: 2,
+    output_price: 5,
+    cache_price: 0,
+    max_in: 64000,
+    max_out: 64000,
+    enable: true,
+  },
+  {
+    id: 'magistral-small-latest',
+    name: 'magistral-small-latest',
+    provider: 'mistral',
+    input_price: 0.5,
+    output_price: 1.5,
+    cache_price: 0,
+    max_in: 64000,
+    max_out: 64000,
+    enable: true,
+  },
+]