kimi-proxy 0.0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (91) hide show
  1. package/LICENSE +24 -0
  2. package/README.md +177 -0
  3. package/dist/adapters/anthropicAdapter.d.ts +138 -0
  4. package/dist/adapters/anthropicAdapter.js +184 -0
  5. package/dist/adapters/anthropicAdapter.js.map +1 -0
  6. package/dist/config.d.ts +27 -0
  7. package/dist/config.js +228 -0
  8. package/dist/config.js.map +1 -0
  9. package/dist/core/converters/anthropicToOpenAI.d.ts +11 -0
  10. package/dist/core/converters/anthropicToOpenAI.js +18 -0
  11. package/dist/core/converters/anthropicToOpenAI.js.map +1 -0
  12. package/dist/core/converters/openaiPassthrough.d.ts +18 -0
  13. package/dist/core/converters/openaiPassthrough.js +794 -0
  14. package/dist/core/converters/openaiPassthrough.js.map +1 -0
  15. package/dist/core/converters/types.d.ts +14 -0
  16. package/dist/core/converters/types.js +16 -0
  17. package/dist/core/converters/types.js.map +1 -0
  18. package/dist/core/ensureToolCall.d.ts +24 -0
  19. package/dist/core/ensureToolCall.js +93 -0
  20. package/dist/core/ensureToolCall.js.map +1 -0
  21. package/dist/core/modelRegistry.d.ts +46 -0
  22. package/dist/core/modelRegistry.js +92 -0
  23. package/dist/core/modelRegistry.js.map +1 -0
  24. package/dist/core/pipeline.d.ts +17 -0
  25. package/dist/core/pipeline.js +109 -0
  26. package/dist/core/pipeline.js.map +1 -0
  27. package/dist/core/pipelineControl.d.ts +12 -0
  28. package/dist/core/pipelineControl.js +51 -0
  29. package/dist/core/pipelineControl.js.map +1 -0
  30. package/dist/core/providers/openRouterProvider.d.ts +52 -0
  31. package/dist/core/providers/openRouterProvider.js +158 -0
  32. package/dist/core/providers/openRouterProvider.js.map +1 -0
  33. package/dist/core/providers/openaiProvider.d.ts +22 -0
  34. package/dist/core/providers/openaiProvider.js +79 -0
  35. package/dist/core/providers/openaiProvider.js.map +1 -0
  36. package/dist/core/providers/types.d.ts +24 -0
  37. package/dist/core/providers/types.js +14 -0
  38. package/dist/core/providers/types.js.map +1 -0
  39. package/dist/core/providers/vertexProvider.d.ts +37 -0
  40. package/dist/core/providers/vertexProvider.js +167 -0
  41. package/dist/core/providers/vertexProvider.js.map +1 -0
  42. package/dist/core/syntheticResponse.d.ts +6 -0
  43. package/dist/core/syntheticResponse.js +36 -0
  44. package/dist/core/syntheticResponse.js.map +1 -0
  45. package/dist/core/transforms/request/ClampMaxTokensTransform.d.ts +7 -0
  46. package/dist/core/transforms/request/ClampMaxTokensTransform.js +29 -0
  47. package/dist/core/transforms/request/ClampMaxTokensTransform.js.map +1 -0
  48. package/dist/core/transforms/request/DisableStreamingTransform.d.ts +7 -0
  49. package/dist/core/transforms/request/DisableStreamingTransform.js +15 -0
  50. package/dist/core/transforms/request/DisableStreamingTransform.js.map +1 -0
  51. package/dist/core/transforms/request/EnsureToolCallRequestTransform.d.ts +12 -0
  52. package/dist/core/transforms/request/EnsureToolCallRequestTransform.js +120 -0
  53. package/dist/core/transforms/request/EnsureToolCallRequestTransform.js.map +1 -0
  54. package/dist/core/transforms/response/EnsureToolCallResponseTransform.d.ts +9 -0
  55. package/dist/core/transforms/response/EnsureToolCallResponseTransform.js +223 -0
  56. package/dist/core/transforms/response/EnsureToolCallResponseTransform.js.map +1 -0
  57. package/dist/core/transforms/response/KimiResponseTransform.d.ts +7 -0
  58. package/dist/core/transforms/response/KimiResponseTransform.js +32 -0
  59. package/dist/core/transforms/response/KimiResponseTransform.js.map +1 -0
  60. package/dist/core/types.d.ts +80 -0
  61. package/dist/core/types.js +18 -0
  62. package/dist/core/types.js.map +1 -0
  63. package/dist/index.d.ts +1 -0
  64. package/dist/index.js +17 -0
  65. package/dist/index.js.map +1 -0
  66. package/dist/persistence/logStore.d.ts +43 -0
  67. package/dist/persistence/logStore.js +115 -0
  68. package/dist/persistence/logStore.js.map +1 -0
  69. package/dist/server.d.ts +3 -0
  70. package/dist/server.js +385 -0
  71. package/dist/server.js.map +1 -0
  72. package/dist/services/kimiFixer.d.ts +13 -0
  73. package/dist/services/kimiFixer.js +160 -0
  74. package/dist/services/kimiFixer.js.map +1 -0
  75. package/dist/services/streaming.d.ts +9 -0
  76. package/dist/services/streaming.js +513 -0
  77. package/dist/services/streaming.js.map +1 -0
  78. package/dist/utils/envResolver.d.ts +26 -0
  79. package/dist/utils/envResolver.js +64 -0
  80. package/dist/utils/envResolver.js.map +1 -0
  81. package/dist/utils/ids.d.ts +1 -0
  82. package/dist/utils/ids.js +7 -0
  83. package/dist/utils/ids.js.map +1 -0
  84. package/dist/utils/logger.d.ts +3 -0
  85. package/dist/utils/logger.js +12 -0
  86. package/dist/utils/logger.js.map +1 -0
  87. package/frontend/dist/assets/JSONViewer-97138fd7.js +3 -0
  88. package/frontend/dist/assets/index-0e7c091b.css +1 -0
  89. package/frontend/dist/assets/index-4b354ce2.js +40 -0
  90. package/frontend/dist/index.html +22 -0
  91. package/package.json +60 -0
@@ -0,0 +1,115 @@
1
+ import Database from "better-sqlite3";
2
+ import fs from "node:fs";
3
+ import path from "node:path";
4
+ import { logger } from "../utils/logger.js";
5
+ export class LogStore {
6
+ dbPath;
7
+ db;
8
+ insertStmt;
9
+ constructor(dbPath) {
10
+ this.dbPath = dbPath;
11
+ const dir = path.dirname(dbPath);
12
+ fs.mkdirSync(dir, { recursive: true });
13
+ this.db = new Database(dbPath);
14
+ this.db.pragma("journal_mode = WAL");
15
+ this.db
16
+ .prepare(`
17
+ CREATE TABLE IF NOT EXISTS request_logs (
18
+ id INTEGER PRIMARY KEY AUTOINCREMENT,
19
+ timestamp TEXT DEFAULT (datetime('now')),
20
+ method TEXT NOT NULL,
21
+ url TEXT NOT NULL,
22
+ status_code INTEGER NOT NULL,
23
+ model TEXT,
24
+ request_body TEXT,
25
+ response_body TEXT,
26
+ provider_request_body TEXT,
27
+ provider_response_body TEXT
28
+ )
29
+ `)
30
+ .run();
31
+ this.ensureColumn("provider_request_body");
32
+ this.ensureColumn("provider_response_body");
33
+ this.insertStmt = this.db.prepare(`
34
+ INSERT INTO request_logs (timestamp, method, url, status_code, model, request_body, response_body, provider_request_body, provider_response_body)
35
+ VALUES (@timestamp, @method, @url, @statusCode, @model, @requestBody, @responseBody, @providerRequestBody, @providerResponseBody)
36
+ `);
37
+ }
38
+ append(entry) {
39
+ try {
40
+ this.insertStmt.run({
41
+ timestamp: new Date().toISOString(),
42
+ method: entry.method,
43
+ url: entry.url,
44
+ statusCode: entry.statusCode,
45
+ model: entry.model ?? null,
46
+ requestBody: JSON.stringify(entry.requestBody ?? null),
47
+ responseBody: JSON.stringify(entry.responseBody ?? null),
48
+ providerRequestBody: JSON.stringify(entry.providerRequestBody ?? null),
49
+ providerResponseBody: JSON.stringify(entry.providerResponseBody ?? null),
50
+ });
51
+ }
52
+ catch (error) {
53
+ logger.error({ err: error }, "Failed to append log entry");
54
+ }
55
+ }
56
+ list(query) {
57
+ const page = Math.max(1, query.page);
58
+ const pageSize = Math.max(1, Math.min(200, query.pageSize));
59
+ const offset = (page - 1) * pageSize;
60
+ const params = {};
61
+ let whereClause = "";
62
+ if (query.search) {
63
+ whereClause = `WHERE method LIKE @search OR url LIKE @search OR model LIKE @search OR CAST(status_code AS TEXT) LIKE @search OR request_body LIKE @search OR response_body LIKE @search`;
64
+ params.search = `%${query.search}%`;
65
+ }
66
+ const totalRow = this.db
67
+ .prepare(`SELECT COUNT(*) as count FROM request_logs ${whereClause}`)
68
+ .get(params);
69
+ const total = totalRow?.count ?? 0;
70
+ const rows = this.db
71
+ .prepare(`
72
+ SELECT * FROM request_logs
73
+ ${whereClause}
74
+ ORDER BY datetime(timestamp) DESC
75
+ LIMIT @limit OFFSET @offset
76
+ `)
77
+ .all({ ...params, limit: pageSize, offset });
78
+ const items = rows.map((row) => ({
79
+ id: row.id,
80
+ timestamp: new Date(row.timestamp).toISOString(),
81
+ method: row.method,
82
+ url: row.url,
83
+ status_code: row.status_code,
84
+ model: row.model,
85
+ request_body: safeJsonParse(row.request_body),
86
+ response_body: safeJsonParse(row.response_body),
87
+ provider_request_body: safeJsonParse(row.provider_request_body),
88
+ provider_response_body: safeJsonParse(row.provider_response_body),
89
+ }));
90
+ return { items, total: Number(total) || 0, page, pageSize };
91
+ }
92
+ ensureColumn(column) {
93
+ const info = this.db
94
+ .prepare(`PRAGMA table_info(request_logs)`)
95
+ .all();
96
+ const hasColumn = info.some((entry) => entry.name === column);
97
+ if (!hasColumn) {
98
+ this.db
99
+ .prepare(`ALTER TABLE request_logs ADD COLUMN ${column} TEXT`)
100
+ .run();
101
+ }
102
+ }
103
+ }
104
+ function safeJsonParse(payload) {
105
+ if (!payload) {
106
+ return null;
107
+ }
108
+ try {
109
+ return JSON.parse(payload);
110
+ }
111
+ catch {
112
+ return payload;
113
+ }
114
+ }
115
+ //# sourceMappingURL=logStore.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"logStore.js","sourceRoot":"","sources":["../../src/persistence/logStore.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAqD5C,MAAM,OAAO,QAAQ;IAIU;IAHrB,EAAE,CAAoB;IACtB,UAAU,CAAqB;IAEvC,YAA6B,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACjC,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC/B,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QACrC,IAAI,CAAC,EAAE;aACJ,OAAO,CACN;;;;;;;;;;;;;OAaD,CACA;aACA,GAAG,EAAE,CAAC;QAET,IAAI,CAAC,YAAY,CAAC,uBAAuB,CAAC,CAAC;QAC3C,IAAI,CAAC,YAAY,CAAC,wBAAwB,CAAC,CAAC;QAE5C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;KAGjC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,KAAqB;QAC1B,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAClB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,GAAG,EAAE,KAAK,CAAC,GAAG;gBACd,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,IAAI;gBAC1B,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC;gBACtD,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,YAAY,IAAI,IAAI,CAAC;gBACxD,mBAAmB,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,mBAAmB,IAAI,IAAI,CAAC;gBACtE,oBAAoB,EAAE,IAAI,CAAC,SAAS,CAClC,KAAK,CAAC,oBAAoB,IAAI,IAAI,CACnC;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,4BAA4B,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,IAAI,CAAC,KAAoB;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC;QAErC,MAAM,MAAM,GAAoC,EAAE,CAAC;QACnD,IAAI,WAAW,GAAG,EAAE,CAAC;QAErB,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjB,WAAW,GAAG,0KAA0K,CAAC;YACzL,MAAM,CAAC,MAAM,GAAG,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC;QACtC,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE;aACrB,OAAO,CAAC,8CAA8C,WAAW,EAAE,CAAC;aACpE,GAAG,CAAC,MAAM,CAAmC,CAAC;QACjD,MAAM,KAAK,GAAG,QAAQ,EAAE,KAAK,IAAI,CAAC,CAAC;QAEnC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACjB,OAAO,CACN;;UAEE,WAAW;;;OAGd,CACA;aACA,GAAG,CAAC,EAAE,GAAG,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAa,CAAC;QAE3D,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC/B,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE;YAChD,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,YAAY,EAAE,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC;YAC7C,aAAa,EAAE,aAAa,CAAC,GAAG,CAAC,aAAa,CAAC;YAC/C,qBAAqB,EAAE,aAAa,CAAC,GAAG,CAAC,qBAAqB,CAAC;YAC/D,sBAAsB,EAAE,aAAa,CAAC,GAAG,CAAC,sBAAsB,CAAC;SAClE,CAAC,CAAC,CAAC;QAEJ,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;IAC9D,CAAC;IAEO,YAAY,CAAC,MAAc;QACjC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE;aACjB,OAAO,CAAC,iCAAiC,CAAC;aAC1C,GAAG,EAA8B,CAAC;QACrC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC9D,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,EAAE;iBACJ,OAAO,CAAC,uCAAuC,MAAM,OAAO,CAAC;iBAC7D,GAAG,EAAE,CAAC;QACX,CAAC;IACH,CAAC;CACF;AAED,SAAS,aAAa,CAAC,OAAsB;IAC3C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAc,CAAC;IAC1C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,OAAO,CAAC;IACjB,CAAC;AACH,CAAC"}
@@ -0,0 +1,3 @@
1
+ import { FastifyInstance } from "fastify";
2
+ import { AppConfig } from "./config.js";
3
+ export declare function createServer(config: AppConfig): Promise<FastifyInstance>;
package/dist/server.js ADDED
@@ -0,0 +1,385 @@
1
+ import Fastify from "fastify";
2
+ import cors from "@fastify/cors";
3
+ import FastifyStatic from "@fastify/static";
4
+ import { fileURLToPath } from "node:url";
5
+ import { dirname, join } from "node:path";
6
+ import { existsSync } from "node:fs";
7
+ import { pipeline } from "node:stream/promises";
8
+ import { LogStore } from "./persistence/logStore.js";
9
+ import { LLMProxyPipeline } from "./core/pipeline.js";
10
+ import { ProviderRegistry } from "./core/providers/types.js";
11
+ import { OpenAIProvider } from "./core/providers/openaiProvider.js";
12
+ import { OpenRouterProvider } from "./core/providers/openRouterProvider.js";
13
+ import { VertexMaaSProvider } from "./core/providers/vertexProvider.js";
14
+ import { ConverterRegistry } from "./core/converters/types.js";
15
+ import { OpenAIChatPassthroughConverter, OpenAIResponsesPassthroughConverter, } from "./core/converters/openaiPassthrough.js";
16
+ import { AnthropicToOpenAIConverter } from "./core/converters/anthropicToOpenAI.js";
17
+ import { DisableStreamingTransform } from "./core/transforms/request/DisableStreamingTransform.js";
18
+ import { ClampMaxTokensTransform } from "./core/transforms/request/ClampMaxTokensTransform.js";
19
+ import { EnsureToolCallRequestTransform } from "./core/transforms/request/EnsureToolCallRequestTransform.js";
20
+ import { KimiResponseTransform } from "./core/transforms/response/KimiResponseTransform.js";
21
+ import { EnsureToolCallResponseTransform } from "./core/transforms/response/EnsureToolCallResponseTransform.js";
22
+ import { ClientFormat, ProxyOperation, } from "./core/types.js";
23
+ import { openAIStreamFromResponse, anthropicStreamFromResponse, openAIResponsesStreamFromResponse, } from "./services/streaming.js";
24
+ import { createRequestId } from "./utils/ids.js";
25
+ import { logger } from "./utils/logger.js";
26
+ import { resolveEnvVarsDeep } from "./utils/envResolver.js";
27
+ import { ModelRegistry } from "./core/modelRegistry.js";
28
+ import { createEnsureToolCallState, getEnsureToolCallState, } from "./core/ensureToolCall.js";
29
+ import { setPipelineMaxAttempts } from "./core/pipelineControl.js";
30
+ const TOOL_CALL_ENFORCER_ATTEMPTS = resolveEnsureToolCallAttemptLimit();
31
+ export async function createServer(config) {
32
+ const server = Fastify({ logger: false });
33
+ await server.register(cors, { origin: true });
34
+ const currentDir = dirname(fileURLToPath(import.meta.url));
35
+ const uiDist = join(currentDir, "..", "frontend", "dist");
36
+ const indexPath = join(uiDist, "index.html");
37
+ const hasStaticAssets = existsSync(indexPath);
38
+ if (hasStaticAssets) {
39
+ await server.register(FastifyStatic, {
40
+ root: uiDist,
41
+ prefix: "/",
42
+ wildcard: false,
43
+ decorateReply: true,
44
+ maxAge: 86400000,
45
+ });
46
+ server.get("/*", async (req, reply) => {
47
+ const file = req.params["*"] ?? "";
48
+ const isAsset = file.includes(".");
49
+ if (!isAsset) {
50
+ reply.sendFile("index.html");
51
+ return;
52
+ }
53
+ reply.sendFile(file);
54
+ });
55
+ }
56
+ else {
57
+ logger.warn(`Static frontend assets not found at ${indexPath}; skipping static serving.`);
58
+ }
59
+ const logStore = new LogStore(config.logging.dbPath);
60
+ const providerRegistry = new ProviderRegistry();
61
+ if (config.providers.openai) {
62
+ providerRegistry.register(new OpenAIProvider(config.providers.openai));
63
+ }
64
+ if (config.providers.openrouter) {
65
+ providerRegistry.register(new OpenRouterProvider(config.providers.openrouter));
66
+ }
67
+ if (config.providers.vertex) {
68
+ providerRegistry.register(new VertexMaaSProvider(config.providers.vertex));
69
+ }
70
+ const modelRegistry = new ModelRegistry(config.models.definitions, config.models.defaultStrategy);
71
+ const converterRegistry = new ConverterRegistry();
72
+ converterRegistry.register(new OpenAIChatPassthroughConverter());
73
+ converterRegistry.register(new OpenAIResponsesPassthroughConverter());
74
+ converterRegistry.register(new AnthropicToOpenAIConverter());
75
+ const requestTransforms = [
76
+ new DisableStreamingTransform(),
77
+ new ClampMaxTokensTransform(),
78
+ new EnsureToolCallRequestTransform(),
79
+ ];
80
+ const responseTransforms = [
81
+ new KimiResponseTransform(),
82
+ new EnsureToolCallResponseTransform(),
83
+ ];
84
+ const pipeline = new LLMProxyPipeline({
85
+ converterRegistry,
86
+ requestTransforms,
87
+ responseTransforms,
88
+ });
89
+ const pipelineDeps = {
90
+ pipeline,
91
+ providerRegistry,
92
+ modelRegistry,
93
+ logStore,
94
+ config,
95
+ };
96
+ server.post("/v1/chat/completions", async (req, reply) => {
97
+ const body = requestBodyAsObject(req);
98
+ await handleProxyRequest(req, reply, body, pipelineDeps, {
99
+ operation: ProxyOperation.ChatCompletions,
100
+ clientFormat: ClientFormat.OpenAIChatCompletions,
101
+ });
102
+ });
103
+ server.post("/v1/messages", async (req, reply) => {
104
+ const body = requestBodyAsObject(req);
105
+ await handleAnthropicMessages(req, reply, body, pipelineDeps);
106
+ });
107
+ server.post("/v1/responses", async (req, reply) => {
108
+ const body = requestBodyAsObject(req);
109
+ await handleProxyRequest(req, reply, body, pipelineDeps, {
110
+ operation: ProxyOperation.Responses,
111
+ clientFormat: ClientFormat.OpenAIResponses,
112
+ });
113
+ });
114
+ server.get("/v1/models", (_req, reply) => {
115
+ const summaries = modelRegistry.list();
116
+ const data = summaries.map((summary) => ({
117
+ id: summary.name,
118
+ object: "model",
119
+ owned_by: "system",
120
+ name: summary.name,
121
+ metadata: {
122
+ strategy: summary.strategy,
123
+ variants: summary.variants,
124
+ },
125
+ }));
126
+ reply.send({ object: "list", data });
127
+ });
128
+ server.get("/api/logs", (req, reply) => {
129
+ const { page = "1", page_size = "20", search, } = req.query;
130
+ const result = logStore.list({
131
+ page: Number(page),
132
+ pageSize: Number(page_size),
133
+ search,
134
+ });
135
+ reply.send(result);
136
+ });
137
+ return server;
138
+ }
139
+ function requestBodyAsObject(req) {
140
+ const body = req.body;
141
+ if (!body || typeof body !== "object") {
142
+ return {};
143
+ }
144
+ return body;
145
+ }
146
+ async function handleProxyRequest(req, reply, body, deps, options) {
147
+ const model = body.model;
148
+ if (!model || typeof model !== "string") {
149
+ reply.status(400).send({ error: { message: "model is required" } });
150
+ return;
151
+ }
152
+ let provider;
153
+ let resolvedModel;
154
+ let providerConfig;
155
+ try {
156
+ resolvedModel = deps.modelRegistry.resolve(model);
157
+ provider = deps.providerRegistry.get(resolvedModel.provider);
158
+ if (resolvedModel.providerConfig) {
159
+ providerConfig = resolveEnvVarsDeep(resolvedModel.providerConfig);
160
+ }
161
+ }
162
+ catch (error) {
163
+ logger.error({ err: error, model }, "Model resolution failed");
164
+ deps.logStore.append({
165
+ method: req.method,
166
+ url: req.url,
167
+ statusCode: 400,
168
+ model: typeof model === "string" ? model : undefined,
169
+ requestBody: body,
170
+ responseBody: { error: { message: error.message } },
171
+ providerRequestBody: null,
172
+ providerResponseBody: null,
173
+ });
174
+ reply.status(400).send({ error: { message: error.message } });
175
+ return;
176
+ }
177
+ const effectiveBody = deepClone(body);
178
+ effectiveBody.model = resolvedModel.upstreamModel;
179
+ const request = {
180
+ id: createRequestId(),
181
+ operation: options.operation,
182
+ clientFormat: options.clientFormat,
183
+ model,
184
+ body: effectiveBody,
185
+ headers: normalizeHeaders(req.headers),
186
+ stream: Boolean(body.stream),
187
+ state: {
188
+ originalStream: Boolean(body.stream),
189
+ },
190
+ };
191
+ request.state.resolvedModel = serializeResolvedModel(resolvedModel);
192
+ maybeEnableToolCallEnforcement(request, resolvedModel.ensureToolCall === true);
193
+ try {
194
+ const result = await deps.pipeline.execute(provider, request, providerConfig);
195
+ deps.logStore.append({
196
+ method: req.method,
197
+ url: req.url,
198
+ statusCode: result.statusCode,
199
+ model,
200
+ requestBody: body,
201
+ responseBody: result.responseBody,
202
+ providerRequestBody: result.providerRequestBody,
203
+ providerResponseBody: result.providerResponse?.body,
204
+ });
205
+ if (result.isError) {
206
+ reply.status(result.statusCode).send(result.responseBody);
207
+ return;
208
+ }
209
+ await sendResponse(reply, request, result.responseBody, deps.config.streaming);
210
+ }
211
+ catch (error) {
212
+ logger.error({ err: error }, "Pipeline execution failed");
213
+ reply.status(500).send({ error: { message: "Internal proxy error" } });
214
+ }
215
+ }
216
+ async function handleAnthropicMessages(req, reply, body, deps) {
217
+ const model = body.model;
218
+ if (!model || typeof model !== "string") {
219
+ reply.status(400).send({ error: { message: "model is required" } });
220
+ return;
221
+ }
222
+ let provider;
223
+ let resolvedModel;
224
+ let providerConfig;
225
+ try {
226
+ resolvedModel = deps.modelRegistry.resolve(model);
227
+ provider = deps.providerRegistry.get(resolvedModel.provider);
228
+ if (resolvedModel.providerConfig) {
229
+ providerConfig = resolveEnvVarsDeep(resolvedModel.providerConfig);
230
+ }
231
+ }
232
+ catch (error) {
233
+ logger.error({ err: error, model }, "Model resolution failed");
234
+ deps.logStore.append({
235
+ method: req.method,
236
+ url: req.url,
237
+ statusCode: 400,
238
+ model: typeof model === "string" ? model : undefined,
239
+ requestBody: body,
240
+ responseBody: { error: { message: error.message } },
241
+ providerRequestBody: null,
242
+ providerResponseBody: null,
243
+ });
244
+ reply.status(400).send({ error: { message: error.message } });
245
+ return;
246
+ }
247
+ const effectiveBody = deepClone(body);
248
+ effectiveBody.model = resolvedModel.upstreamModel;
249
+ const request = {
250
+ id: createRequestId(),
251
+ operation: ProxyOperation.Messages,
252
+ clientFormat: ClientFormat.AnthropicMessages,
253
+ model,
254
+ body: effectiveBody,
255
+ headers: normalizeHeaders(req.headers),
256
+ stream: Boolean(body.stream),
257
+ state: {
258
+ originalStream: Boolean(body.stream),
259
+ },
260
+ };
261
+ request.state.resolvedModel = serializeResolvedModel(resolvedModel);
262
+ maybeEnableToolCallEnforcement(request, resolvedModel.ensureToolCall === true);
263
+ try {
264
+ const result = await deps.pipeline.execute(provider, request, providerConfig);
265
+ deps.logStore.append({
266
+ method: req.method,
267
+ url: req.url,
268
+ statusCode: result.statusCode,
269
+ model,
270
+ requestBody: body,
271
+ responseBody: result.responseBody,
272
+ providerRequestBody: result.providerRequestBody,
273
+ providerResponseBody: result.providerResponse?.body,
274
+ });
275
+ if (result.isError) {
276
+ reply.status(result.statusCode).send(result.responseBody);
277
+ return;
278
+ }
279
+ await sendResponse(reply, request, result.responseBody, deps.config.streaming);
280
+ }
281
+ catch (error) {
282
+ logger.error({ err: error }, "Anthropic messages handling failed");
283
+ reply.status(500).send({ error: { message: "Internal proxy error" } });
284
+ }
285
+ }
286
+ function serializeResolvedModel(variant) {
287
+ return {
288
+ name: variant.name,
289
+ provider: variant.provider,
290
+ upstreamModel: variant.upstreamModel,
291
+ weight: variant.weight,
292
+ strategy: variant.strategy ?? "first",
293
+ ensureToolCall: variant.ensureToolCall,
294
+ };
295
+ }
296
+ function maybeEnableToolCallEnforcement(request, enabled) {
297
+ if (!enabled) {
298
+ return;
299
+ }
300
+ if (!getEnsureToolCallState(request.state)) {
301
+ createEnsureToolCallState(request.state);
302
+ }
303
+ setPipelineMaxAttempts(request.state, TOOL_CALL_ENFORCER_ATTEMPTS);
304
+ }
305
+ async function sendResponse(reply, request, payload, streamingOptions) {
306
+ const shouldStream = Boolean(request.state.originalStream);
307
+ if (!shouldStream) {
308
+ reply.send(payload);
309
+ return;
310
+ }
311
+ if (!isJsonObject(payload)) {
312
+ reply.send(payload);
313
+ return;
314
+ }
315
+ if (request.clientFormat === ClientFormat.AnthropicMessages) {
316
+ reply.header("Content-Type", "text/event-stream");
317
+ reply.header("Cache-Control", "no-cache");
318
+ reply.header("Connection", "keep-alive");
319
+ reply.send(anthropicStreamFromResponse(payload, streamingOptions));
320
+ return;
321
+ }
322
+ if (request.clientFormat === ClientFormat.OpenAIResponses) {
323
+ // Using reply.hijack() to bypass Fastify's default response handling which
324
+ // incorrectly sets Content-Length: 0 for streaming responses in some environments.
325
+ // This allows us to pipe the stream directly to the raw socket.
326
+ reply.hijack();
327
+ reply.raw.setHeader("Content-Type", "text/event-stream");
328
+ reply.raw.setHeader("Cache-Control", "no-cache");
329
+ reply.raw.setHeader("Connection", "keep-alive");
330
+ reply.removeHeader("Content-Length");
331
+ const stream = openAIResponsesStreamFromResponse(payload, streamingOptions);
332
+ try {
333
+ await pipeline(stream, reply.raw);
334
+ }
335
+ catch (error) {
336
+ if (isPrematureCloseError(error)) {
337
+ logger.debug({ err: error }, "Client disconnected before streaming completed");
338
+ }
339
+ else {
340
+ throw error;
341
+ }
342
+ }
343
+ return;
344
+ }
345
+ reply.header("Content-Type", "text/event-stream");
346
+ reply.header("Cache-Control", "no-cache");
347
+ reply.header("Connection", "keep-alive");
348
+ reply.send(openAIStreamFromResponse(payload, streamingOptions));
349
+ }
350
+ function normalizeHeaders(headers) {
351
+ const normalized = {};
352
+ for (const [key, value] of Object.entries(headers)) {
353
+ if (typeof value === "string") {
354
+ normalized[key.toLowerCase()] = value;
355
+ }
356
+ }
357
+ return normalized;
358
+ }
359
+ function deepClone(value) {
360
+ if (value === undefined || value === null) {
361
+ return value;
362
+ }
363
+ return JSON.parse(JSON.stringify(value));
364
+ }
365
+ function isPrematureCloseError(error) {
366
+ if (!error || typeof error !== "object") {
367
+ return false;
368
+ }
369
+ const err = error;
370
+ if (err.code === "ERR_STREAM_PREMATURE_CLOSE") {
371
+ return true;
372
+ }
373
+ return (typeof err.message === "string" && err.message.includes("Premature close"));
374
+ }
375
+ function isJsonObject(value) {
376
+ return typeof value === "object" && value !== null && !Array.isArray(value);
377
+ }
378
+ function resolveEnsureToolCallAttemptLimit() {
379
+ const raw = Number(process.env.ENSURE_TOOL_CALL_MAX_ATTEMPTS ?? "3");
380
+ if (!Number.isFinite(raw) || raw < 1) {
381
+ return 3;
382
+ }
383
+ return Math.min(5, Math.floor(raw));
384
+ }
385
+ //# sourceMappingURL=server.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,OAIN,MAAM,SAAS,CAAC;AACjB,OAAO,IAAI,MAAM,eAAe,CAAC;AACjC,OAAO,aAAa,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAEhD,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAe,MAAM,2BAA2B,CAAC;AAC1E,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,wCAAwC,CAAC;AAC5E,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EACL,8BAA8B,EAC9B,mCAAmC,GACpC,MAAM,wCAAwC,CAAC;AAChD,OAAO,EAAE,0BAA0B,EAAE,MAAM,wCAAwC,CAAC;AACpF,OAAO,EAAE,yBAAyB,EAAE,MAAM,wDAAwD,CAAC;AACnG,OAAO,EAAE,uBAAuB,EAAE,MAAM,sDAAsD,CAAC;AAC/F,OAAO,EAAE,8BAA8B,EAAE,MAAM,6DAA6D,CAAC;AAC7G,OAAO,EAAE,qBAAqB,EAAE,MAAM,qDAAqD,CAAC;AAC5F,OAAO,EAAE,+BAA+B,EAAE,MAAM,+DAA+D,CAAC;AAChH,OAAO,EACL,YAAY,EACZ,cAAc,GAIf,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,wBAAwB,EACxB,2BAA2B,EAC3B,iCAAiC,GAElC,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAE5D,OAAO,EAAE,aAAa,EAAgB,MAAM,yBAAyB,CAAC;AACtE,OAAO,EACL,yBAAyB,EACzB,sBAAsB,GACvB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AAenE,MAAM,2BAA2B,GAAG,iCAAiC,EAAE,CAAC;AAExE,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,MAAiB;IAEjB,MAAM,MAAM,GAAG,OAAO,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1C,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAE9C,MAAM,UAAU,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAC1D,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAC7C,MAAM,eAAe,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IAE9C,IAAI,eAAe,EAAE,CAAC;QACpB,MAAM,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE;YACnC,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,GAAG;YACX,QAAQ,EAAE,KAAK;YACf,aAAa,EAAE,IAAI;YACnB,MAAM,EAAE,QAAQ;SACjB,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;YACpC,MAAM,IAAI,GAAI,GAAG,CAAC,MAAiC,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YAC/D,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YACnC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBAC7B,OAAO;YACT,CAAC;YACD,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,MAAM,CAAC,IAAI,CACT,uCAAuC,SAAS,4BAA4B,CAC7E,CAAC;IACJ,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,EAAE,CAAC;IAEhD,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;QAC5B,gBAAgB,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;IACzE,CAAC;IAED,IAAI,MAAM,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;QAChC,gBAAgB,CAAC,QAAQ,CACvB,IAAI,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,CACpD,CAAC;IACJ,CAAC;IAED,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;QAC5B,gBAAgB,CAAC,QAAQ,CAAC,IAAI,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7E,CAAC;IAED,MAAM,aAAa,GAAG,IAAI,aAAa,CACrC,MAAM,CAAC,MAAM,CAAC,WAAW,EACzB,MAAM,CAAC,MAAM,CAAC,eAAe,CAC9B,CAAC;IAEF,MAAM,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAC;IAClD,iBAAiB,CAAC,QAAQ,CAAC,IAAI,8BAA8B,EAAE,CAAC,CAAC;IACjE,iBAAiB,CAAC,QAAQ,CAAC,IAAI,mCAAmC,EAAE,CAAC,CAAC;IACtE,iBAAiB,CAAC,QAAQ,CAAC,IAAI,0BAA0B,EAAE,CAAC,CAAC;IAE7D,MAAM,iBAAiB,GAAuB;QAC5C,IAAI,yBAAyB,EAAE;QAC/B,IAAI,uBAAuB,EAAE;QAC7B,IAAI,8BAA8B,EAAE;KACrC,CAAC;IACF,MAAM,kBAAkB,GAAwB;QAC9C,IAAI,qBAAqB,EAAE;QAC3B,IAAI,+BAA+B,EAAE;KACtC,CAAC;IAEF,MAAM,QAAQ,GAAG,IAAI,gBAAgB,CAAC;QACpC,iBAAiB;QACjB,iBAAiB;QACjB,kBAAkB;KACnB,CAAC,CAAC;IAEH,MAAM,YAAY,GAAoB;QACpC,QAAQ;QACR,gBAAgB;QAChB,aAAa;QACb,QAAQ;QACR,MAAM;KACP,CAAC;IAEF,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;QACvD,MAAM,IAAI,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,kBAAkB,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,YAAY,EAAE;YACvD,SAAS,EAAE,cAAc,CAAC,eAAe;YACzC,YAAY,EAAE,YAAY,CAAC,qBAAqB;SACjD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;QAC/C,MAAM,IAAI,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,uBAAuB,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;QAChD,MAAM,IAAI,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,kBAAkB,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,YAAY,EAAE;YACvD,SAAS,EAAE,cAAc,CAAC,SAAS;YACnC,YAAY,EAAE,YAAY,CAAC,eAAe;SAC3C,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACvC,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC;QACvC,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACvC,EAAE,EAAE,OAAO,CAAC,IAAI;YAChB,MAAM,EAAE,OAAO;YACf,QAAQ,EAAE,QAAQ;YAClB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,QAAQ,EAAE;gBACR,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;aAC3B;SACF,CAAC,CAAC,CAAC;QACJ,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;QACrC,MAAM,EACJ,IAAI,GAAG,GAAG,EACV,SAAS,GAAG,IAAI,EAChB,MAAM,GACP,GAAG,GAAG,CAAC,KAA+B,CAAC;QACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC;YAC3B,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC;YAClB,QAAQ,EAAE,MAAM,CAAC,SAAS,CAAC;YAC3B,MAAM;SACP,CAAC,CAAC;QACH,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,mBAAmB,CAAC,GAAmB;IAC9C,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;IACtB,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QACtC,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,IAAkB,CAAC;AAC5B,CAAC;AAED,KAAK,UAAU,kBAAkB,CAC/B,GAAmB,EACnB,KAAmB,EACnB,IAAgB,EAChB,IAAqB,EACrB,OAA4B;IAE5B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;IACzB,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,mBAAmB,EAAE,EAAE,CAAC,CAAC;QACpE,OAAO;IACT,CAAC;IAED,IAAI,QAAqB,CAAC;IAC1B,IAAI,aAA2B,CAAC;IAChC,IAAI,cAAsC,CAAC;IAC3C,IAAI,CAAC;QACH,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAClD,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAE7D,IAAI,aAAa,CAAC,cAAc,EAAE,CAAC;YACjC,cAAc,GAAG,kBAAkB,CACjC,aAAa,CAAC,cAAc,CACf,CAAC;QAClB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,yBAAyB,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;YACnB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,UAAU,EAAE,GAAG;YACf,KAAK,EAAE,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;YACpD,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,OAAO,EAAG,KAAe,CAAC,OAAO,EAAE,EAAE;YAC9D,mBAAmB,EAAE,IAAI;YACzB,oBAAoB,EAAE,IAAI;SAC3B,CAAC,CAAC;QACH,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAG,KAAe,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QACzE,OAAO;IACT,CAAC;IAED,MAAM,aAAa,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;IACtC,aAAa,CAAC,KAAK,GAAG,aAAa,CAAC,aAAa,CAAC;IAElD,MAAM,OAAO,GAAiB;QAC5B,EAAE,EAAE,eAAe,EAAE;QACrB,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,YAAY,EAAE,OAAO,CAAC,YAAY;QAClC,KAAK;QACL,IAAI,EAAE,aAAa;QACnB,OAAO,EAAE,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC;QACtC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC;QAC5B,KAAK,EAAE;YACL,cAAc,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC;SACrC;KACF,CAAC;IACF,OAAO,CAAC,KAAK,CAAC,aAAa,GAAG,sBAAsB,CAAC,aAAa,CAAC,CAAC;IACpE,8BAA8B,CAC5B,OAAO,EACP,aAAa,CAAC,cAAc,KAAK,IAAI,CACtC,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CACxC,QAAQ,EACR,OAAO,EACP,cAAc,CACf,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;YACnB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,KAAK;YACL,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;YAC/C,oBAAoB,EAAE,MAAM,CAAC,gBAAgB,EAAE,IAAI;SACpD,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAC1D,OAAO;QACT,CAAC;QAED,MAAM,YAAY,CAChB,KAAK,EACL,OAAO,EACP,MAAM,CAAC,YAAY,EACnB,IAAI,CAAC,MAAM,CAAC,SAAS,CACtB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,2BAA2B,CAAC,CAAC;QAC1D,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAE,EAAE,CAAC,CAAC;IACzE,CAAC;AACH,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,GAAmB,EACnB,KAAmB,EACnB,IAAgB,EAChB,IAAqB;IAErB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;IACzB,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,mBAAmB,EAAE,EAAE,CAAC,CAAC;QACpE,OAAO;IACT,CAAC;IAED,IAAI,QAAqB,CAAC;IAC1B,IAAI,aAA2B,CAAC;IAChC,IAAI,cAAsC,CAAC;IAC3C,IAAI,CAAC;QACH,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAClD,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAE7D,IAAI,aAAa,CAAC,cAAc,EAAE,CAAC;YACjC,cAAc,GAAG,kBAAkB,CACjC,aAAa,CAAC,cAAc,CACf,CAAC;QAClB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,yBAAyB,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;YACnB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,UAAU,EAAE,GAAG;YACf,KAAK,EAAE,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;YACpD,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,OAAO,EAAG,KAAe,CAAC,OAAO,EAAE,EAAE;YAC9D,mBAAmB,EAAE,IAAI;YACzB,oBAAoB,EAAE,IAAI;SAC3B,CAAC,CAAC;QACH,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAG,KAAe,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QACzE,OAAO;IACT,CAAC;IAED,MAAM,aAAa,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;IACtC,aAAa,CAAC,KAAK,GAAG,aAAa,CAAC,aAAa,CAAC;IAElD,MAAM,OAAO,GAAiB;QAC5B,EAAE,EAAE,eAAe,EAAE;QACrB,SAAS,EAAE,cAAc,CAAC,QAAQ;QAClC,YAAY,EAAE,YAAY,CAAC,iBAAiB;QAC5C,KAAK;QACL,IAAI,EAAE,aAAa;QACnB,OAAO,EAAE,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC;QACtC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC;QAC5B,KAAK,EAAE;YACL,cAAc,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC;SACrC;KACF,CAAC;IACF,OAAO,CAAC,KAAK,CAAC,aAAa,GAAG,sBAAsB,CAAC,aAAa,CAAC,CAAC;IACpE,8BAA8B,CAC5B,OAAO,EACP,aAAa,CAAC,cAAc,KAAK,IAAI,CACtC,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CACxC,QAAQ,EACR,OAAO,EACP,cAAc,CACf,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;YACnB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,KAAK;YACL,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;YAC/C,oBAAoB,EAAE,MAAM,CAAC,gBAAgB,EAAE,IAAI;SACpD,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAC1D,OAAO;QACT,CAAC;QAED,MAAM,YAAY,CAChB,KAAK,EACL,OAAO,EACP,MAAM,CAAC,YAAY,EACnB,IAAI,CAAC,MAAM,CAAC,SAAS,CACtB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,oCAAoC,CAAC,CAAC;QACnE,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAE,EAAE,CAAC,CAAC;IACzE,CAAC;AACH,CAAC;AAED,SAAS,sBAAsB,CAAC,OAAqB;IACnD,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,aAAa,EAAE,OAAO,CAAC,aAAa;QACpC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,OAAO;QACrC,cAAc,EAAE,OAAO,CAAC,cAAc;KACvC,CAAC;AACJ,CAAC;AAED,SAAS,8BAA8B,CACrC,OAAqB,EACrB,OAAgB;IAEhB,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO;IACT,CAAC;IACD,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC3C,yBAAyB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC3C,CAAC;IACD,sBAAsB,CAAC,OAAO,CAAC,KAAK,EAAE,2BAA2B,CAAC,CAAC;AACrE,CAAC;AAED,KAAK,UAAU,YAAY,CACzB,KAAmB,EACnB,OAAgE,EAChE,OAAkB,EAClB,gBAAkC;IAElC,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAC3D,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpB,OAAO;IACT,CAAC;IAED,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpB,OAAO;IACT,CAAC;IAED,IAAI,OAAO,CAAC,YAAY,KAAK,YAAY,CAAC,iBAAiB,EAAE,CAAC;QAC5D,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;QAClD,KAAK,CAAC,MAAM,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;QAC1C,KAAK,CAAC,MAAM,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QACzC,KAAK,CAAC,IAAI,CAAC,2BAA2B,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC;QACnE,OAAO;IACT,CAAC;IAED,IAAI,OAAO,CAAC,YAAY,KAAK,YAAY,CAAC,eAAe,EAAE,CAAC;QAC1D,2EAA2E;QAC3E,mFAAmF;QACnF,gEAAgE;QAChE,KAAK,CAAC,MAAM,EAAE,CAAC;QACf,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;QACzD,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;QACjD,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAChD,KAAK,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;QAErC,MAAM,MAAM,GAAG,iCAAiC,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;QAC5E,IAAI,CAAC;YACH,MAAM,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,MAAM,CAAC,KAAK,CACV,EAAE,GAAG,EAAE,KAAK,EAAE,EACd,gDAAgD,CACjD,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,OAAO;IACT,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;IAClD,KAAK,CAAC,MAAM,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;IAC1C,KAAK,CAAC,MAAM,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;IACzC,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,gBAAgB,CACvB,OAAkC;IAElC,MAAM,UAAU,GAA2B,EAAE,CAAC;IAC9C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACnD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,UAAU,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,GAAG,KAAK,CAAC;QACxC,CAAC;IACH,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,SAAS,CAAI,KAAQ;IAC5B,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC1C,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAM,CAAC;AAChD,CAAC;AACD,SAAS,qBAAqB,CAAC,KAAc;IAC3C,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,GAAG,GAAG,KAA4C,CAAC;IACzD,IAAI,GAAG,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;QAC9C,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,CACL,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAC3E,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAgB;IACpC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,iCAAiC;IACxC,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,6BAA6B,IAAI,GAAG,CAAC,CAAC;IACrE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,EAAE,CAAC;QACrC,OAAO,CAAC,CAAC;IACX,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;AACtC,CAAC"}
@@ -0,0 +1,13 @@
1
+ import type { JsonObject } from "../core/types.js";
2
+ export interface KimiFixMetadata extends JsonObject {
3
+ extractedToolCalls: number;
4
+ extractedFromReasoning: number;
5
+ extractedFromContent: number;
6
+ cleanedReasoningContent: boolean;
7
+ cleanedMessageContent: boolean;
8
+ }
9
+ export interface KimiFixResult {
10
+ response: JsonObject;
11
+ metadata: KimiFixMetadata;
12
+ }
13
+ export declare function fixKimiResponse(response: JsonObject): KimiFixResult;