@keo-ai/axiom 0.2.6 → 0.2.8

package/README.md

@@ -194,16 +194,21 @@ for await (const chunk of LLM.streamPredict({ model: 'qwen-max', prompt: '讲个
 |------|----------|-----------|------|
 | `qwen-max` | ❌ 不支持 | ✅ | **默认模型** |
 | `qwen3.7-max` | ✅ 映射支持 | ✅ | `low`/`medium`/`high` 映射为 `enable_thinking` |
+| `qwen3.7-flash` | ✅ 映射支持 | ✅ | 同上 |
 | `qwen-plus` | ✅ 映射支持 | ✅ | 同上 |
 | `qwen-turbo` | ✅ 映射支持 | ✅ | 同上 |
-| `deepseek-v4-pro` | ✅ 原生支持 | ✅ | `low`/`medium`/`high` 直接传递 |
-| `deepseek-v4-flash` | ✅ 原生支持 | ✅ | 同上 |
-| `kimi-k2.6` | ✅ 原生支持 | ✅ | 同上 |
+| `deepseek-v4-pro` | ✅ 映射支持 | ✅ | 同上（百炼走「混合思考模式」，由 `enable_thinking` 控制） |
+| `deepseek-v4-flash` | ✅ 映射支持 | ✅ | 同上 |
+| `kimi-k2.6` | ✅ 映射支持 | ✅ | `low`/`medium`/`high` 映射为 `extra_body.thinking.type` |
 | `qwq-plus` | ❌ 不支持 | ✅ | 固定推理行为，不可调节 |
 | `glm-5.1` | ❌ 不支持 | ❌ | 不支持推理参数，不支持 json 模式 |
 | `qwen-vl-plus` | ❌ 不支持 | ✅ | 视觉模型，不支持推理参数 |
 
-> 推理深度的支持方式由 Provider 内部维护。百炼 Provider 中，Qwen 系列通过 `extra_body.enable_thinking` 映射实现（`low` 关闭，`medium`/`high` 开启），DeepSeek / Kimi 则原生透传 `reasoning_effort`。
+> 推理深度的支持方式由 Provider 内部维护。百炼 Provider 中：
+> - Qwen 系列与 DeepSeek v4 系列均通过 `extra_body.enable_thinking` 映射实现（`low` 关闭，`medium`/`high` 开启）；
+> - Kimi 则通过 `extra_body.thinking.type` 映射（`low` 关闭，`medium`/`high` 开启）。
+>
+> 两种模式在百炼兼容接口下都退化为布尔开关，`medium` / `high` 之间的差异无法体现。
 >
 > 目前所有模型均路由到百炼 Provider。后续接入其他厂商时，通过 `MODEL_REGISTRY` 扩展映射即可。
 

package/dist/embedding_search/embed.js

@@ -13,7 +13,7 @@ exports.embed = embed;
 exports.embedBatch = embedBatch;
 const EMBEDDING_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1/embeddings';
 const EMBEDDING_MODEL = 'text-embedding-v4';
-const BATCH_SIZE_LIMIT = 25; // DashScope 兼容接口的单次上限
+const BATCH_SIZE_LIMIT = 10; // DashScope 兼容接口的单次上限
 function getApiKey() {
     const apiKey = process.env.BAILIAN_API_KEY;
     if (!apiKey) {

package/dist/embedding_search/index.d.ts

@@ -34,7 +34,7 @@ export declare class EmbeddingSearch {
     /**
      * 批量获取文本向量。
      *
-     * 调用百炼 embedding 接口，一次最多 25 条文本。
+     * 调用百炼 embedding 接口，一次最多 10 条文本。
      * 返回向量的顺序与输入顺序一致。
      *
      * @example

package/dist/embedding_search/index.js

@@ -179,7 +179,7 @@ class EmbeddingSearch {
     /**
      * 批量获取文本向量。
      *
-     * 调用百炼 embedding 接口，一次最多 25 条文本。
+     * 调用百炼 embedding 接口，一次最多 10 条文本。
      * 返回向量的顺序与输入顺序一致。
      *
      * @example

package/dist/function_call_loop/provider.js

@@ -129,7 +129,7 @@ function createLLMCaller(options) {
                             'Content-Type': 'application/json',
                             Authorization: `Bearer ${options.apiKey}`,
                         },
-                        body: JSON.stringify(body),
+                        body: (0, llm_provider_1.serializeChatRequest)(body),
                     }));
                 }
                 catch (cause) {

package/dist/index.d.ts

@@ -37,7 +37,7 @@ export type { PredictConfig, PredictWithMessagesConfig } from './predict';
  * - `EmbeddingSearch.query()` — 文本 → embedding → pgvector 检索 → 可选 rerank
  * - `EmbeddingSearch.vectorSearch()` — 已有向量 → pgvector 近邻检索
  * - `EmbeddingSearch.embed()` — 单条文本转向量
- * - `EmbeddingSearch.batchEmbed()` — 批量文本转向量（单次最多 25 条）
+ * - `EmbeddingSearch.batchEmbed()` — 批量文本转向量（单次最多 10 条）
  */
 export { EmbeddingSearch } from './embedding_search';
 export type { EmbeddingSearchConfig, SearchResult } from './embedding_search';

package/dist/index.js

@@ -73,7 +73,7 @@ Object.defineProperty(exports, "MODEL_REGISTRY", { enumerable: true, get: functi
  * - `EmbeddingSearch.query()` — 文本 → embedding → pgvector 检索 → 可选 rerank
  * - `EmbeddingSearch.vectorSearch()` — 已有向量 → pgvector 近邻检索
  * - `EmbeddingSearch.embed()` — 单条文本转向量
- * - `EmbeddingSearch.batchEmbed()` — 批量文本转向量（单次最多 25 条）
+ * - `EmbeddingSearch.batchEmbed()` — 批量文本转向量（单次最多 10 条）
  */
 var embedding_search_1 = require("./embedding_search");
 Object.defineProperty(exports, "EmbeddingSearch", { enumerable: true, get: function () { return embedding_search_1.EmbeddingSearch; } });

package/dist/llm_provider/bailian.js

@@ -56,22 +56,21 @@ class BailianProvider {
         if (((_b = body.response_format) === null || _b === void 0 ? void 0 : _b.type) === 'json_object' && !caps.jsonMode) {
             throw new Error(`[adapt] Model "${body.model}" does not support response_format json_object`);
         }
-        // 百炼平台：Qwen 系列通过 enable_thinking 实现 reasoning_effort
-        // Kimi（Moonshot）通过 extra_body.thinking.type 实现（默认开启）
+        // 百炼平台：Qwen 系列与 DeepSeek v4 系列均为「混合思考模式」，
+        // 通过 extra_body.enable_thinking 控制思考开关（low 关闭，medium/high 开启）；
+        // Kimi（Moonshot）则通过 extra_body.thinking.type 实现（默认开启）。
         if (body.reasoning_effort !== undefined) {
-            if (['qwen-plus', 'qwen-turbo', 'qwen3.7-max'].includes(body.model)) {
+            if (['qwen-plus', 'qwen-turbo', 'qwen3.7-max', 'qwen3.7-flash', 'deepseek-v4-pro', 'deepseek-v4-flash'].includes(body.model)) {
                 const { reasoning_effort } = body, rest = __rest(body, ["reasoning_effort"]);
                 return Object.assign(Object.assign({}, rest), { extra_body: Object.assign(Object.assign({}, ((_c = rest.extra_body) !== null && _c !== void 0 ? _c : {})), { enable_thinking: reasoning_effort !== 'low' }) });
             }
             if (body.model === 'kimi-k2.6') {
                 const { reasoning_effort } = body, rest = __rest(body, ["reasoning_effort"]);
-                // Kimi 默认开启思考；只有 low 显式禁用，medium/high 走默认（不传 thinking 字段）
-                if (reasoning_effort === 'low') {
-                    return Object.assign(Object.assign({}, rest), { extra_body: Object.assign(Object.assign({}, ((_d = rest.extra_body) !== null && _d !== void 0 ? _d : {})), { thinking: { type: 'disabled' } }) });
-                }
-                return rest;
+                // 百炼兼容模式下 Kimi 默认不输出 reasoning_content，
+                // 需要显式传 thinking.type：low 禁用，medium/high 显式开启
+                return Object.assign(Object.assign({}, rest), { extra_body: Object.assign(Object.assign({}, ((_d = rest.extra_body) !== null && _d !== void 0 ? _d : {})), { thinking: { type: reasoning_effort === 'low' ? 'disabled' : 'enabled' } }) });
             }
-            // deepseek-v4-pro / deepseek-v4-flash 原生支持 reasoning_effort，直接传递
+            // 其它不支持 reasoning_effort 的模型会由前面的能力校验抛异常
         }
         return body;
     }
@@ -121,7 +120,7 @@ class BailianProvider {
                         'Content-Type': 'application/json',
                         Authorization: `Bearer ${this.config.apiKey}`,
                     },
-                    body: JSON.stringify(body),
+                    body: (0, index_1.serializeChatRequest)(body),
                 }));
             }
             catch (cause) {
@@ -214,6 +213,7 @@ class BailianProvider {
 exports.BailianProvider = BailianProvider;
 BailianProvider.SUPPORTED_MODELS = new Set([
     'qwen3.7-max',
+    'qwen3.7-flash',
     'qwen-plus',
     'qwen-turbo',
     'qwq-plus',
@@ -226,9 +226,12 @@ BailianProvider.SUPPORTED_MODELS = new Set([
 BailianProvider.MODEL_CAPABILITIES = {
     'qwen-max': { jsonMode: true, reasoningEffort: false },
     'qwen3.7-max': { jsonMode: true, reasoningEffort: true },
+    'qwen3.7-flash': { jsonMode: true, reasoningEffort: true },
     'qwen-plus': { jsonMode: true, reasoningEffort: true },
     'qwen-turbo': { jsonMode: true, reasoningEffort: true },
     'qwq-plus': { jsonMode: true, reasoningEffort: false },
+    // DeepSeek v4 系列在百炼走「混合思考模式」：由 enable_thinking 控制开关，
+    // 通过 adaptRequest 映射为 extra_body.enable_thinking（与 Qwen 同一分支）。
     'deepseek-v4-pro': { jsonMode: true, reasoningEffort: true },
     'deepseek-v4-flash': { jsonMode: true, reasoningEffort: true },
     'kimi-k2.6': { jsonMode: true, reasoningEffort: true },

package/dist/llm_provider/index.d.ts

@@ -69,6 +69,12 @@ export interface OpenAIChatResponse {
     };
     model?: string;
 }
+/**
+ * 序列化 chat completion 请求体。
+ * `extra_body` 是 OpenAI SDK 的客户端概念，在 HTTP 协议层面不存在，
+ * 必须将其内容展平到请求体顶层，否则服务端会忽略这些扩展参数。
+ */
+export declare function serializeChatRequest(body: OpenAIChatRequest): string;
 /**
  * 发送 OpenAI 兼容的 chat completions 请求。
  * 只做 HTTP 层：构造请求、fetch、错误处理、基础解析。

package/dist/llm_provider/index.js

@@ -12,9 +12,30 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
+var __rest = (this && this.__rest) || function (s, e) {
+    var t = {};
+    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
+        t[p] = s[p];
+    if (s != null && typeof Object.getOwnPropertySymbols === "function")
+        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
+            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
+                t[p[i]] = s[p[i]];
+        }
+    return t;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BailianProvider = exports.MODEL_REGISTRY = void 0;
+exports.serializeChatRequest = serializeChatRequest;
 exports.callChatCompletions = callChatCompletions;
+/**
+ * 序列化 chat completion 请求体。
+ * `extra_body` 是 OpenAI SDK 的客户端概念，在 HTTP 协议层面不存在，
+ * 必须将其内容展平到请求体顶层，否则服务端会忽略这些扩展参数。
+ */
+function serializeChatRequest(body) {
+    const { extra_body } = body, rest = __rest(body, ["extra_body"]);
+    return JSON.stringify(Object.assign(Object.assign({}, rest), (extra_body !== null && extra_body !== void 0 ? extra_body : {})));
+}
 /**
  * 发送 OpenAI 兼容的 chat completions 请求。
  * 只做 HTTP 层：构造请求、fetch、错误处理、基础解析。
@@ -32,7 +53,7 @@ function callChatCompletions(baseUrl_1, apiKey_1, body_1) {
                     'Content-Type': 'application/json',
                     Authorization: `Bearer ${apiKey}`,
                 },
-                body: JSON.stringify(body),
+                body: serializeChatRequest(body),
             });
         }
         catch (cause) {

package/dist/llm_provider/models.d.ts

@@ -1,7 +1,7 @@
 /**
  * 支持的模型枚举。项目层通过此枚举选择模型，Axiom 内部路由到对应 Provider。
  */
-export type Model = 'qwen3.7-max' | 'qwen-plus' | 'qwen-turbo' | 'qwq-plus' | 'deepseek-v4-pro' | 'deepseek-v4-flash' | 'kimi-k2.6' | 'glm-5.1' | 'qwen-vl-plus';
+export type Model = 'qwen3.7-max' | 'qwen3.7-flash' | 'qwen-plus' | 'qwen-turbo' | 'qwq-plus' | 'deepseek-v4-pro' | 'deepseek-v4-flash' | 'kimi-k2.6' | 'glm-5.1' | 'qwen-vl-plus';
 /** 模型到 Provider 的映射配置 */
 export interface ModelConfig {
     readonly model: string;
@@ -16,7 +16,10 @@ export declare const MODEL_REGISTRY: Readonly<Record<Model, ReadonlyArray<ModelC
 export interface ModelCapabilities {
     /** 是否原生支持 response_format: { type: 'json_object' } */
     readonly jsonMode: boolean;
-    /** 是否支持 reasoning_effort 参数 */
+    /**
+     * 是否支持调节推理深度。开启后由 Provider 的 `adaptRequest` 决定如何映射到底层字段
+     * （如 Qwen / DeepSeek → `extra_body.enable_thinking`、Kimi → `extra_body.thinking.type`）。
+     */
     readonly reasoningEffort: boolean;
 }
 /**

package/dist/llm_provider/models.js

@@ -7,6 +7,7 @@ exports.MODEL_REGISTRY = void 0;
  */
 exports.MODEL_REGISTRY = {
     'qwen3.7-max': [{ model: 'qwen3.7-max', provider: 'bailian' }],
+    'qwen3.7-flash': [{ model: 'qwen3.7-flash', provider: 'bailian' }],
     'qwen-plus': [{ model: 'qwen-plus', provider: 'bailian' }],
     'qwen-turbo': [{ model: 'qwen-turbo', provider: 'bailian' }],
     'qwq-plus': [{ model: 'qwq-plus', provider: 'bailian' }],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@keo-ai/axiom",
-  "version": "0.2.6",
+  "version": "0.2.8",
   "description": "基于 LLM 的预测与推理库，支持多 Provider 切换",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -10,7 +10,9 @@
   "license": "MIT",
   "scripts": {
     "build": "tsc",
-    "test": "vitest run --exclude '**/*.integration.test.ts'"
+    "typecheck": "tsc --noEmit",
+    "test": "vitest run --exclude '**/*.integration.test.ts'",
+    "test:integration": "vitest run --config vitest.integration.config.ts"
   },
   "dependencies": {
     "dotenv": "^17.4.2"