@typescript-agent-harness/llm 0.15.0

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.
package/LICENSE ADDED
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2026 typescript-agent-harness contributors
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
@@ -0,0 +1,3 @@
1
+ import type { LLMService } from "./types.js";
2
+ export declare const LLM: import("@typescript-agent-harness/core").ServiceKey<LLMService>;
3
+ //# sourceMappingURL=contract.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"contract.d.ts","sourceRoot":"","sources":["../src/contract.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE7C,eAAO,MAAM,GAAG,iEAAsC,CAAC"}
@@ -0,0 +1,3 @@
1
+ import { createServiceKey } from "@typescript-agent-harness/core";
2
+ export const LLM = createServiceKey("llm");
3
+ //# sourceMappingURL=contract.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"contract.js","sourceRoot":"","sources":["../src/contract.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAGlE,MAAM,CAAC,MAAM,GAAG,GAAG,gBAAgB,CAAa,KAAK,CAAC,CAAC"}
package/dist/ids.d.ts ADDED
@@ -0,0 +1,2 @@
1
+ export declare function nextId(prefix: string): string;
2
+ //# sourceMappingURL=ids.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ids.d.ts","sourceRoot":"","sources":["../src/ids.ts"],"names":[],"mappings":"AAEA,wBAAgB,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAG7C"}
package/dist/ids.js ADDED
@@ -0,0 +1,6 @@
1
+ let n = 0;
2
+ export function nextId(prefix) {
3
+ n += 1;
4
+ return `${prefix}_${n}_${Date.now().toString(36)}`;
5
+ }
6
+ //# sourceMappingURL=ids.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ids.js","sourceRoot":"","sources":["../src/ids.ts"],"names":[],"mappings":"AAAA,IAAI,CAAC,GAAG,CAAC,CAAC;AAEV,MAAM,UAAU,MAAM,CAAC,MAAc;IACnC,CAAC,IAAI,CAAC,CAAC;IACP,OAAO,GAAG,MAAM,IAAI,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;AACrD,CAAC"}
@@ -0,0 +1,6 @@
1
+ export { LLM } from "./contract.js";
2
+ export { llmPlugin, type LLMPluginOptions } from "./plugin.js";
3
+ export { createMockLLM } from "./mock.js";
4
+ export { createOpenAICompatLLM, type OpenAICompatConfig } from "./openai.js";
5
+ export type { ChatMessage, FinishReason, JSONSchema, LLMRequest, LLMResponse, LLMService, LLMStreamPart, TokenUsage, ToolCall, ToolSchema, } from "./types.js";
6
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,eAAe,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,KAAK,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,qBAAqB,EAAE,KAAK,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAC7E,YAAY,EACV,WAAW,EACX,YAAY,EACZ,UAAU,EACV,UAAU,EACV,WAAW,EACX,UAAU,EACV,aAAa,EACb,UAAU,EACV,QAAQ,EACR,UAAU,GACX,MAAM,YAAY,CAAC"}
package/dist/index.js ADDED
@@ -0,0 +1,5 @@
1
+ export { LLM } from "./contract.js";
2
+ export { llmPlugin } from "./plugin.js";
3
+ export { createMockLLM } from "./mock.js";
4
+ export { createOpenAICompatLLM } from "./openai.js";
5
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,eAAe,CAAC;AACpC,OAAO,EAAE,SAAS,EAAyB,MAAM,aAAa,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,qBAAqB,EAA2B,MAAM,aAAa,CAAC"}
package/dist/mock.d.ts ADDED
@@ -0,0 +1,7 @@
1
+ import type { LLMService } from "./types.js";
2
+ /**
3
+ * Deterministic provider for demos and tests.
4
+ * list_files → (optional) read README.md → short project summary.
5
+ */
6
+ export declare function createMockLLM(defaultModel?: string): LLMService;
7
+ //# sourceMappingURL=mock.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"mock.d.ts","sourceRoot":"","sources":["../src/mock.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAIV,UAAU,EAGX,MAAM,YAAY,CAAC;AAwFpB;;;GAGG;AACH,wBAAgB,aAAa,CAAC,YAAY,SAAS,GAAG,UAAU,CA0D/D"}
package/dist/mock.js ADDED
@@ -0,0 +1,133 @@
1
+ import { nextId } from "./ids.js";
2
+ function toolNameByCallId(messages) {
3
+ const map = new Map();
4
+ for (const m of messages) {
5
+ if (m.role === "assistant" && m.toolCalls) {
6
+ for (const c of m.toolCalls)
7
+ map.set(c.id, c.name);
8
+ }
9
+ }
10
+ return map;
11
+ }
12
+ function toolOutputs(messages) {
13
+ const names = toolNameByCallId(messages);
14
+ const out = new Map();
15
+ for (const m of messages) {
16
+ if (m.role !== "tool")
17
+ continue;
18
+ const name = names.get(m.toolCallId);
19
+ if (!name)
20
+ continue;
21
+ try {
22
+ out.set(name, JSON.parse(m.content));
23
+ }
24
+ catch {
25
+ out.set(name, m.content);
26
+ }
27
+ }
28
+ return out;
29
+ }
30
+ function call(name, args) {
31
+ const toolCalls = [{ id: nextId("call"), name, arguments: args }];
32
+ return {
33
+ toolCalls,
34
+ response: {
35
+ id: nextId("llm"),
36
+ model: "",
37
+ message: { role: "assistant", toolCalls },
38
+ toolCalls,
39
+ finishReason: "tool_calls",
40
+ },
41
+ };
42
+ }
43
+ function summarize(outputs) {
44
+ const listing = outputs.get("list_files");
45
+ const read = outputs.get("read_file");
46
+ const entries = listing?.entries ?? [];
47
+ const dirs = entries.filter((e) => e.type === "dir").map((e) => e.name);
48
+ const files = entries.filter((e) => e.type === "file").map((e) => e.name);
49
+ const lines = [
50
+ "这是一个 TypeScript Agent Runtime 仓库(typescript-agent-harness)。",
51
+ dirs.length
52
+ ? `顶层目录:${dirs.join("、")}。`
53
+ : "顶层没有子目录。",
54
+ files.length
55
+ ? `顶层文件:${files.join("、")}。`
56
+ : "",
57
+ ];
58
+ if (read?.content) {
59
+ const first = read.content
60
+ .split("\n")
61
+ .map((l) => l.trim())
62
+ .find((l) => {
63
+ if (!l)
64
+ return false;
65
+ if (l.startsWith("#") || l.startsWith("[") || l.startsWith("**"))
66
+ return false;
67
+ if (l.startsWith(">") || l.startsWith("```"))
68
+ return false;
69
+ return l.length > 20;
70
+ });
71
+ if (first)
72
+ lines.push(`README 摘要:${first}`);
73
+ }
74
+ lines.push("核心做法:用 Plugin 往 Runtime 上挂 LLM / Tools / Session,而不是把一切塞进一个 Agent 类。");
75
+ return lines.filter(Boolean).join("\n");
76
+ }
77
+ /**
78
+ * Deterministic provider for demos and tests.
79
+ * list_files → (optional) read README.md → short project summary.
80
+ */
81
+ export function createMockLLM(defaultModel = "mock") {
82
+ async function generate(request) {
83
+ const model = request.model ?? defaultModel;
84
+ const has = (name) => request.tools?.some((t) => t.name === name);
85
+ const outputs = toolOutputs(request.messages);
86
+ const last = request.messages.at(-1);
87
+ if (has("list_files") && !outputs.has("list_files")) {
88
+ const { response } = call("list_files", { directory: "." });
89
+ response.model = model;
90
+ return response;
91
+ }
92
+ const listing = outputs.get("list_files");
93
+ const hasReadme = listing?.entries?.some((e) => e.name.toLowerCase().startsWith("readme"));
94
+ if (has("read_file") && hasReadme && !outputs.has("read_file")) {
95
+ const name = listing?.entries?.find((e) => e.name === "README.zh-CN.md")?.name ??
96
+ listing?.entries?.find((e) => e.name.toLowerCase().startsWith("readme"))?.name ??
97
+ "README.md";
98
+ const { response } = call("read_file", { path: name });
99
+ response.model = model;
100
+ return response;
101
+ }
102
+ if (outputs.size > 0) {
103
+ return {
104
+ id: nextId("llm"),
105
+ model,
106
+ message: { role: "assistant", content: summarize(outputs) },
107
+ finishReason: "stop",
108
+ };
109
+ }
110
+ return {
111
+ id: nextId("llm"),
112
+ model,
113
+ message: {
114
+ role: "assistant",
115
+ content: last?.role === "user" ? `收到:${last.content}` : "OK",
116
+ },
117
+ finishReason: "stop",
118
+ };
119
+ }
120
+ async function* stream(request) {
121
+ const response = await generate(request);
122
+ const text = response.message.content;
123
+ if (text) {
124
+ const size = 8;
125
+ for (let i = 0; i < text.length; i += size) {
126
+ yield { type: "text-delta", text: text.slice(i, i + size) };
127
+ }
128
+ }
129
+ yield { type: "done", response };
130
+ }
131
+ return { generate, stream };
132
+ }
133
+ //# sourceMappingURL=mock.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"mock.js","sourceRoot":"","sources":["../src/mock.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAOlC,SAAS,gBAAgB,CAAC,QAAuB;IAC/C,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;IACtC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,SAAS,EAAE,CAAC;YAC1C,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS;gBAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,WAAW,CAAC,QAAuB;IAC1C,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACzC,MAAM,GAAG,GAAG,IAAI,GAAG,EAAmB,CAAC;IACvC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM;YAAE,SAAS;QAChC,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACrC,IAAI,CAAC,IAAI;YAAE,SAAS;QACpB,IAAI,CAAC;YACH,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAY,CAAC,CAAC;QAClD,CAAC;QAAC,MAAM,CAAC;YACP,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,IAAI,CAAC,IAAY,EAAE,IAAa;IACvC,MAAM,SAAS,GAAe,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9E,OAAO;QACL,SAAS;QACT,QAAQ,EAAE;YACR,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC;YACjB,KAAK,EAAE,EAAE;YACT,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE;YACzC,SAAS;YACT,YAAY,EAAE,YAAY;SAC3B;KACF,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,OAA6B;IAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,CAA2B,CAAC;IACpE,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAEvB,CAAC;IAEd,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;IACvC,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACxE,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAE1E,MAAM,KAAK,GAAG;QACZ,6DAA6D;QAC7D,IAAI,CAAC,MAAM;YACT,CAAC,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG;YAC3B,CAAC,CAAC,UAAU;QACd,KAAK,CAAC,MAAM;YACV,CAAC,CAAC,QAAQ,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG;YAC5B,CAAC,CAAC,EAAE;KACP,CAAC;IAEF,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO;aACvB,KAAK,CAAC,IAAI,CAAC;aACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;aACpB,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;YACV,IAAI,CAAC,CAAC;gBAAE,OAAO,KAAK,CAAC;YACrB,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC/E,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC;gBAAE,OAAO,KAAK,CAAC;YAC3D,OAAO,CAAC,CAAC,MAAM,GAAG,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC;QACL,IAAI,KAAK;YAAE,KAAK,CAAC,IAAI,CAAC,aAAa,KAAK,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,IAAI,CACR,sEAAsE,CACvE,CAAC;IAEF,OAAO,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1C,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,YAAY,GAAG,MAAM;IACjD,KAAK,UAAU,QAAQ,CAAC,OAAmB;QACzC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,YAAY,CAAC;QAC5C,MAAM,GAAG,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAC1E,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAErC,IAAI,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;YACpD,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;YAC5D,QAAQ,CAAC,KAAK,GAAG,KAAK,CAAC;YACvB,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,CAA2B,CAAC;QACpE,MAAM,SAAS,GAAG,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC3F,IAAI,GAAG,CAAC,WAAW,CAAC,IAAI,SAAS,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;YAC/D,MAAM,IAAI,GACR,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAC,EAAE,IAAI;gBACjE,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI;gBAC9E,WAAW,CAAC;YACd,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YACvD,QAAQ,CAAC,KAAK,GAAG,KAAK,CAAC;YACvB,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACrB,OAAO;gBACL,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC;gBACjB,KAAK;gBACL,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,EAAE;gBAC3D,YAAY,EAAE,MAAM;aACrB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC;YACjB,KAAK;YACL,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,IAAI,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI;aAC7D;YACD,YAAY,EAAE,MAAM;SACrB,CAAC;IACJ,CAAC;IAED,KAAK,SAAS,CAAC,CAAC,MAAM,CAAC,OAAmB;QACxC,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC;QACtC,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,IAAI,GAAG,CAAC,CAAC;YACf,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC;gBAC3C,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC;YAC9D,CAAC;QACH,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;IACnC,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;AAC9B,CAAC"}
@@ -0,0 +1,8 @@
1
+ import type { LLMService } from "./types.js";
2
+ export type OpenAICompatConfig = {
3
+ baseURL: string;
4
+ apiKey: string;
5
+ defaultModel: string;
6
+ };
7
+ export declare function createOpenAICompatLLM(config: OpenAICompatConfig): LLMService;
8
+ //# sourceMappingURL=openai.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../src/openai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAKV,UAAU,EAIX,MAAM,YAAY,CAAC;AAGpB,MAAM,MAAM,kBAAkB,GAAG;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;CACtB,CAAC;AA8EF,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,kBAAkB,GAAG,UAAU,CAqL5E"}
package/dist/openai.js ADDED
@@ -0,0 +1,219 @@
1
+ import { nextId } from "./ids.js";
2
+ function toOpenAIMessages(messages) {
3
+ return messages.map((m) => {
4
+ if (m.role === "tool") {
5
+ return { role: "tool", tool_call_id: m.toolCallId, content: m.content };
6
+ }
7
+ if (m.role === "assistant") {
8
+ const out = { role: "assistant", content: m.content ?? null };
9
+ if (m.toolCalls?.length) {
10
+ out.tool_calls = m.toolCalls.map((c) => ({
11
+ id: c.id,
12
+ type: "function",
13
+ function: {
14
+ name: c.name,
15
+ arguments: typeof c.arguments === "string"
16
+ ? c.arguments
17
+ : JSON.stringify(c.arguments ?? {}),
18
+ },
19
+ }));
20
+ }
21
+ return out;
22
+ }
23
+ return { role: m.role, content: m.content };
24
+ });
25
+ }
26
+ function parseArgs(raw) {
27
+ try {
28
+ return JSON.parse(raw);
29
+ }
30
+ catch {
31
+ return raw;
32
+ }
33
+ }
34
+ function mapFinish(reason) {
35
+ if (reason === "tool_calls")
36
+ return "tool_calls";
37
+ if (reason === "length")
38
+ return "length";
39
+ return "stop";
40
+ }
41
+ function requestBody(request, model, stream) {
42
+ const body = {
43
+ model,
44
+ messages: toOpenAIMessages(request.messages),
45
+ stream,
46
+ };
47
+ if (request.temperature !== undefined)
48
+ body.temperature = request.temperature;
49
+ if (request.maxTokens !== undefined)
50
+ body.max_tokens = request.maxTokens;
51
+ if (request.tools?.length) {
52
+ body.tools = request.tools.map((t) => ({
53
+ type: "function",
54
+ function: {
55
+ name: t.name,
56
+ description: t.description,
57
+ parameters: t.inputSchema,
58
+ },
59
+ }));
60
+ }
61
+ return body;
62
+ }
63
+ export function createOpenAICompatLLM(config) {
64
+ const base = config.baseURL.replace(/\/$/, "");
65
+ async function chatCompletions(request, stream) {
66
+ const model = request.model ?? config.defaultModel;
67
+ const init = {
68
+ method: "POST",
69
+ headers: {
70
+ "content-type": "application/json",
71
+ authorization: `Bearer ${config.apiKey}`,
72
+ },
73
+ body: JSON.stringify(requestBody(request, model, stream)),
74
+ };
75
+ if (request.signal)
76
+ init.signal = request.signal;
77
+ const res = await fetch(`${base}/chat/completions`, init);
78
+ if (!res.ok) {
79
+ const text = await res.text();
80
+ throw new Error(`OpenAI-compatible error ${res.status}: ${text}`);
81
+ }
82
+ return res;
83
+ }
84
+ async function generate(request) {
85
+ const model = request.model ?? config.defaultModel;
86
+ const res = await chatCompletions(request, false);
87
+ const json = (await res.json());
88
+ const choice = json.choices?.[0];
89
+ const msg = choice?.message;
90
+ const toolCalls = msg?.tool_calls?.map((c) => ({
91
+ id: c.id,
92
+ name: c.function.name,
93
+ arguments: parseArgs(c.function.arguments),
94
+ }));
95
+ const usage = json.usage
96
+ ? {
97
+ promptTokens: json.usage.prompt_tokens ?? 0,
98
+ completionTokens: json.usage.completion_tokens ?? 0,
99
+ totalTokens: json.usage.total_tokens ?? 0,
100
+ }
101
+ : undefined;
102
+ const assistant = {
103
+ role: "assistant",
104
+ };
105
+ if (msg?.content)
106
+ assistant.content = msg.content;
107
+ if (toolCalls?.length)
108
+ assistant.toolCalls = toolCalls;
109
+ const response = {
110
+ id: json.id ?? nextId("llm"),
111
+ model: json.model ?? model,
112
+ message: assistant,
113
+ finishReason: mapFinish(choice?.finish_reason),
114
+ };
115
+ if (toolCalls?.length)
116
+ response.toolCalls = toolCalls;
117
+ if (usage)
118
+ response.usage = usage;
119
+ return response;
120
+ }
121
+ async function* stream(request) {
122
+ const model = request.model ?? config.defaultModel;
123
+ const res = await chatCompletions(request, true);
124
+ if (!res.body)
125
+ throw new Error("OpenAI-compatible stream: empty body");
126
+ const reader = res.body.getReader();
127
+ const decoder = new TextDecoder();
128
+ let buf = "";
129
+ let content = "";
130
+ let id = nextId("llm");
131
+ let outModel = model;
132
+ let finish = "stop";
133
+ const toolParts = new Map();
134
+ let usage;
135
+ const flushLine = (line) => {
136
+ const trimmed = line.trim();
137
+ if (!trimmed.startsWith("data:"))
138
+ return;
139
+ const data = trimmed.slice(5).trim();
140
+ if (!data || data === "[DONE]")
141
+ return;
142
+ const json = JSON.parse(data);
143
+ if (json.id)
144
+ id = json.id;
145
+ if (json.model)
146
+ outModel = json.model;
147
+ if (json.usage) {
148
+ usage = {
149
+ promptTokens: json.usage.prompt_tokens ?? 0,
150
+ completionTokens: json.usage.completion_tokens ?? 0,
151
+ totalTokens: json.usage.total_tokens ?? 0,
152
+ };
153
+ }
154
+ const choice = json.choices?.[0];
155
+ if (choice?.finish_reason)
156
+ finish = mapFinish(choice.finish_reason);
157
+ const delta = choice?.delta;
158
+ if (delta?.content)
159
+ content += delta.content;
160
+ for (const tc of delta?.tool_calls ?? []) {
161
+ const index = tc.index ?? 0;
162
+ const prev = toolParts.get(index) ?? { id: "", name: "", args: "" };
163
+ if (tc.id)
164
+ prev.id = tc.id;
165
+ if (tc.function?.name)
166
+ prev.name += tc.function.name;
167
+ if (tc.function?.arguments)
168
+ prev.args += tc.function.arguments;
169
+ toolParts.set(index, prev);
170
+ }
171
+ return delta?.content ?? "";
172
+ };
173
+ while (true) {
174
+ const { done, value } = await reader.read();
175
+ if (done)
176
+ break;
177
+ buf += decoder.decode(value, { stream: true });
178
+ const lines = buf.split("\n");
179
+ buf = lines.pop() ?? "";
180
+ for (const line of lines) {
181
+ const piece = flushLine(line);
182
+ if (piece)
183
+ yield { type: "text-delta", text: piece };
184
+ }
185
+ }
186
+ if (buf.trim()) {
187
+ const piece = flushLine(buf);
188
+ if (piece)
189
+ yield { type: "text-delta", text: piece };
190
+ }
191
+ const toolCalls = [...toolParts.entries()]
192
+ .sort((a, b) => a[0] - b[0])
193
+ .map(([, t]) => ({
194
+ id: t.id || nextId("call"),
195
+ name: t.name,
196
+ arguments: parseArgs(t.args),
197
+ }));
198
+ const assistant = {
199
+ role: "assistant",
200
+ };
201
+ if (content)
202
+ assistant.content = content;
203
+ if (toolCalls.length)
204
+ assistant.toolCalls = toolCalls;
205
+ const response = {
206
+ id,
207
+ model: outModel,
208
+ message: assistant,
209
+ finishReason: toolCalls.length ? "tool_calls" : finish,
210
+ };
211
+ if (toolCalls.length)
212
+ response.toolCalls = toolCalls;
213
+ if (usage)
214
+ response.usage = usage;
215
+ yield { type: "done", response };
216
+ }
217
+ return { generate, stream };
218
+ }
219
+ //# sourceMappingURL=openai.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"openai.js","sourceRoot":"","sources":["../src/openai.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAmBlC,SAAS,gBAAgB,CAAC,QAAuB;IAC/C,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACxB,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtB,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;QAC1E,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAkB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;YAC7E,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;gBACxB,GAAG,CAAC,UAAU,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACvC,EAAE,EAAE,CAAC,CAAC,EAAE;oBACR,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACR,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,SAAS,EACP,OAAO,CAAC,CAAC,SAAS,KAAK,QAAQ;4BAC7B,CAAC,CAAC,CAAC,CAAC,SAAS;4BACb,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC;qBACxC;iBACF,CAAC,CAAC,CAAC;YACN,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,SAAS,CAAC,GAAW;IAC5B,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAY,CAAC;IACpC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,GAAG,CAAC;IACb,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,MAAiC;IAClD,IAAI,MAAM,KAAK,YAAY;QAAE,OAAO,YAAY,CAAC;IACjD,IAAI,MAAM,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IACzC,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,WAAW,CAClB,OAAmB,EACnB,KAAa,EACb,MAAe;IAEf,MAAM,IAAI,GAA4B;QACpC,KAAK;QACL,QAAQ,EAAE,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC;QAC5C,MAAM;KACP,CAAC;IACF,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS;QAAE,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;IAC9E,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS;QAAE,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;IACzE,IAAI,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;QAC1B,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACrC,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,UAAU,EAAE,CAAC,CAAC,WAAW;aAC1B;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,MAA0B;IAC9D,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAE/C,KAAK,UAAU,eAAe,CAAC,OAAmB,EAAE,MAAe;QACjE,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,MAAM,CAAC,YAAY,CAAC;QACnD,MAAM,IAAI,GAAgB;YACxB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,MAAM,CAAC,MAAM,EAAE;aACzC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;SAC1D,CAAC;QACF,IAAI,OAAO,CAAC,MAAM;YAAE,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,mBAAmB,EAAE,IAAI,CAAC,CAAC;QAC1D,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC,CAAC;QACpE,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,UAAU,QAAQ,CAAC,OAAmB;QACzC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,MAAM,CAAC,YAAY,CAAC;QACnD,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAClD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAY7B,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,GAAG,GAAG,MAAM,EAAE,OAAO,CAAC;QAC5B,MAAM,SAAS,GAA2B,GAAG,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACrE,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI;YACrB,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC;SAC3C,CAAC,CAAC,CAAC;QAEJ,MAAM,KAAK,GAA2B,IAAI,CAAC,KAAK;YAC9C,CAAC,CAAC;gBACE,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC;gBAC3C,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC;gBACnD,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC;aAC1C;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,SAAS,GAAgD;YAC7D,IAAI,EAAE,WAAW;SAClB,CAAC;QACF,IAAI,GAAG,EAAE,OAAO;YAAE,SAAS,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;QAClD,IAAI,SAAS,EAAE,MAAM;YAAE,SAAS,CAAC,SAAS,GAAG,SAAS,CAAC;QAEvD,MAAM,QAAQ,GAAgB;YAC5B,EAAE,EAAE,IAAI,CAAC,EAAE,IAAI,MAAM,CAAC,KAAK,CAAC;YAC5B,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,KAAK;YAC1B,OAAO,EAAE,SAAS;YAClB,YAAY,EAAE,SAAS,CAAC,MAAM,EAAE,aAAa,CAAC;SAC/C,CAAC;QACF,IAAI,SAAS,EAAE,MAAM;YAAE,QAAQ,CAAC,SAAS,GAAG,SAAS,CAAC;QACtD,IAAI,KAAK;YAAE,QAAQ,CAAC,KAAK,GAAG,KAAK,CAAC;QAClC,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,SAAS,CAAC,CAAC,MAAM,CAAC,OAAmB;QACxC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,MAAM,CAAC,YAAY,CAAC;QACnD,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACjD,IAAI,CAAC,GAAG,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAEvE,MAAM,MAAM,GAAG,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,GAAG,GAAG,EAAE,CAAC;QACb,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QACvB,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,MAAM,GAAiB,MAAM,CAAC;QAClC,MAAM,SAAS,GAAG,IAAI,GAAG,EAGtB,CAAC;QACJ,IAAI,KAA6B,CAAC;QAElC,MAAM,SAAS,GAAG,CAAC,IAAY,EAAE,EAAE;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAC5B,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC;gBAAE,OAAO;YACzC,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACrC,IAAI,CAAC,IAAI,IAAI,IAAI,KAAK,QAAQ;gBAAE,OAAO;YACvC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAmB3B,CAAC;YACF,IAAI,IAAI,CAAC,EAAE;gBAAE,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC;YAC1B,IAAI,IAAI,CAAC,KAAK;gBAAE,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC;YACtC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,KAAK,GAAG;oBACN,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC;oBAC3C,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC;oBACnD,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC;iBAC1C,CAAC;YACJ,CAAC;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACjC,IAAI,MAAM,EAAE,aAAa;gBAAE,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YACpE,MAAM,KAAK,GAAG,MAAM,EAAE,KAAK,CAAC;YAC5B,IAAI,KAAK,EAAE,OAAO;gBAAE,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC;YAC7C,KAAK,MAAM,EAAE,IAAI,KAAK,EAAE,UAAU,IAAI,EAAE,EAAE,CAAC;gBACzC,MAAM,KAAK,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,CAAC;gBAC5B,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;gBACpE,IAAI,EAAE,CAAC,EAAE;oBAAE,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;gBAC3B,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI;oBAAE,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACrD,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS;oBAAE,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAC/D,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAC7B,CAAC;YACD,OAAO,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;QAC9B,CAAC,CAAC;QAEF,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAChB,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC9B,GAAG,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YACxB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;gBAC9B,IAAI,KAAK;oBAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;YACvD,CAAC;QACH,CAAC;QACD,IAAI,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC;YACf,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;YAC7B,IAAI,KAAK;gBAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACvD,CAAC;QAED,MAAM,SAAS,GAAe,CAAC,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;aACnD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;aAC3B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACf,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,MAAM,CAAC,MAAM,CAAC;YAC1B,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;SAC7B,CAAC,CAAC,CAAC;QAEN,MAAM,SAAS,GAAgD;YAC7D,IAAI,EAAE,WAAW;SAClB,CAAC;QACF,IAAI,OAAO;YAAE,SAAS,CAAC,OAAO,GAAG,OAAO,CAAC;QACzC,IAAI,SAAS,CAAC,MAAM;YAAE,SAAS,CAAC,SAAS,GAAG,SAAS,CAAC;QAEtD,MAAM,QAAQ,GAAgB;YAC5B,EAAE;YACF,KAAK,EAAE,QAAQ;YACf,OAAO,EAAE,SAAS;YAClB,YAAY,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM;SACvD,CAAC;QACF,IAAI,SAAS,CAAC,MAAM;YAAE,QAAQ,CAAC,SAAS,GAAG,SAAS,CAAC;QACrD,IAAI,KAAK;YAAE,QAAQ,CAAC,KAAK,GAAG,KAAK,CAAC;QAClC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;IACnC,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;AAC9B,CAAC"}
@@ -0,0 +1,10 @@
1
+ import type { Plugin } from "@typescript-agent-harness/core";
2
+ export type LLMPluginOptions = {
3
+ /** mock (default) | openai-compatible */
4
+ provider?: "mock" | "openai-compatible";
5
+ defaultModel?: string;
6
+ baseURL?: string;
7
+ apiKey?: string;
8
+ };
9
+ export declare function llmPlugin(options?: LLMPluginOptions): Plugin;
10
+ //# sourceMappingURL=plugin.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"plugin.d.ts","sourceRoot":"","sources":["../src/plugin.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAW,MAAM,EAAE,MAAM,gCAAgC,CAAC;AAMtE,MAAM,MAAM,gBAAgB,GAAG;IAC7B,yCAAyC;IACzC,QAAQ,CAAC,EAAE,MAAM,GAAG,mBAAmB,CAAC;IACxC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAgEF,wBAAgB,SAAS,CAAC,OAAO,GAAE,gBAAqB,GAAG,MAAM,CAsChE"}
package/dist/plugin.js ADDED
@@ -0,0 +1,98 @@
1
+ import { LLM } from "./contract.js";
2
+ import { createMockLLM } from "./mock.js";
3
+ import { createOpenAICompatLLM } from "./openai.js";
4
+ function wrapWithEvents(inner, ctx, defaultModel) {
5
+ return {
6
+ async generate(request) {
7
+ const requestId = `${Date.now().toString(36)}`;
8
+ const model = request.model ?? defaultModel;
9
+ await ctx.emit("llm.request", {
10
+ requestId,
11
+ model,
12
+ messageCount: request.messages.length,
13
+ });
14
+ try {
15
+ const response = await inner.generate({ ...request, model });
16
+ await ctx.emit("llm.response", {
17
+ requestId,
18
+ model: response.model,
19
+ finishReason: response.finishReason,
20
+ usage: response.usage,
21
+ });
22
+ return response;
23
+ }
24
+ catch (error) {
25
+ await ctx.emit("llm.error", { requestId, error });
26
+ throw error;
27
+ }
28
+ },
29
+ async *stream(request) {
30
+ const requestId = `${Date.now().toString(36)}`;
31
+ const model = request.model ?? defaultModel;
32
+ await ctx.emit("llm.request", {
33
+ requestId,
34
+ model,
35
+ messageCount: request.messages.length,
36
+ });
37
+ try {
38
+ let doneResponse;
39
+ for await (const part of inner.stream({ ...request, model })) {
40
+ if (part.type === "text-delta") {
41
+ await ctx.emit("llm.stream", { requestId, text: part.text });
42
+ }
43
+ else if (part.type === "done") {
44
+ doneResponse = part.response;
45
+ }
46
+ yield part;
47
+ }
48
+ if (doneResponse) {
49
+ await ctx.emit("llm.response", {
50
+ requestId,
51
+ model: doneResponse.model,
52
+ finishReason: doneResponse.finishReason,
53
+ usage: doneResponse.usage,
54
+ });
55
+ }
56
+ }
57
+ catch (error) {
58
+ await ctx.emit("llm.error", { requestId, error });
59
+ throw error;
60
+ }
61
+ },
62
+ };
63
+ }
64
+ export function llmPlugin(options = {}) {
65
+ const provider = options.provider ?? "mock";
66
+ return {
67
+ name: "llm",
68
+ setup(ctx) {
69
+ let service;
70
+ if (provider === "openai-compatible") {
71
+ const apiKey = options.apiKey ??
72
+ process.env.DEEPSEEK_API_KEY ??
73
+ process.env.OPENAI_API_KEY;
74
+ const baseURL = options.baseURL ??
75
+ process.env.DEEPSEEK_BASE_URL ??
76
+ process.env.OPENAI_BASE_URL ??
77
+ (process.env.DEEPSEEK_API_KEY
78
+ ? "https://api.deepseek.com/v1"
79
+ : undefined);
80
+ const defaultModel = options.defaultModel ??
81
+ process.env.DEEPSEEK_MODEL ??
82
+ process.env.OPENAI_MODEL ??
83
+ (process.env.DEEPSEEK_API_KEY ? "deepseek-chat" : "gpt-4o-mini");
84
+ if (!baseURL || !apiKey) {
85
+ throw new Error("openai-compatible provider requires apiKey + baseURL (DEEPSEEK_API_KEY or OPENAI_API_KEY / OPENAI_BASE_URL)");
86
+ }
87
+ service = createOpenAICompatLLM({ baseURL, apiKey, defaultModel });
88
+ ctx.provide(LLM, wrapWithEvents(service, ctx, defaultModel));
89
+ }
90
+ else {
91
+ const mockModel = options.defaultModel ?? "mock";
92
+ service = createMockLLM(mockModel);
93
+ ctx.provide(LLM, wrapWithEvents(service, ctx, mockModel));
94
+ }
95
+ },
96
+ };
97
+ }
98
+ //# sourceMappingURL=plugin.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"plugin.js","sourceRoot":"","sources":["../src/plugin.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,GAAG,EAAE,MAAM,eAAe,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAWpD,SAAS,cAAc,CACrB,KAAiB,EACjB,GAAY,EACZ,YAAoB;IAEpB,OAAO;QACL,KAAK,CAAC,QAAQ,CAAC,OAAmB;YAChC,MAAM,SAAS,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,YAAY,CAAC;YAC5C,MAAM,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE;gBAC5B,SAAS;gBACT,KAAK;gBACL,YAAY,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM;aACtC,CAAC,CAAC;YACH,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC7D,MAAM,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE;oBAC7B,SAAS;oBACT,KAAK,EAAE,QAAQ,CAAC,KAAK;oBACrB,YAAY,EAAE,QAAQ,CAAC,YAAY;oBACnC,KAAK,EAAE,QAAQ,CAAC,KAAK;iBACtB,CAAC,CAAC;gBACH,OAAO,QAAQ,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,KAAK,CAAC,CAAC,MAAM,CAAC,OAAmB;YAC/B,MAAM,SAAS,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;YAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,YAAY,CAAC;YAC5C,MAAM,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE;gBAC5B,SAAS;gBACT,KAAK;gBACL,YAAY,EAAE,OAAO,CAAC,QAAQ,CAAC,MAAM;aACtC,CAAC,CAAC;YACH,IAAI,CAAC;gBACH,IAAI,YAAqC,CAAC;gBAC1C,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;oBAC7D,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;wBAC/B,MAAM,GAAG,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC/D,CAAC;yBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBAChC,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC;oBAC/B,CAAC;oBACD,MAAM,IAAI,CAAC;gBACb,CAAC;gBACD,IAAI,YAAY,EAAE,CAAC;oBACjB,MAAM,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE;wBAC7B,SAAS;wBACT,KAAK,EAAE,YAAY,CAAC,KAAK;wBACzB,YAAY,EAAE,YAAY,CAAC,YAAY;wBACvC,KAAK,EAAE,YAAY,CAAC,KAAK;qBAC1B,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,UAA4B,EAAE;IACtD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,MAAM,CAAC;IAE5C,OAAO;QACL,IAAI,EAAE,KAAK;QACX,KAAK,CAAC,GAAG;YACP,IAAI,OAAmB,CAAC;YACxB,IAAI,QAAQ,KAAK,mBAAmB,EAAE,CAAC;gBACrC,MAAM,MAAM,GACV,OAAO,CAAC,MAAM;oBACd,OAAO,CAAC,GAAG,CAAC,gBAAgB;oBAC5B,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;gBAC7B,MAAM,OAAO,GACX,OAAO,CAAC,OAAO;oBACf,OAAO,CAAC,GAAG,CAAC,iBAAiB;oBAC7B,OAAO,CAAC,GAAG,CAAC,eAAe;oBAC3B,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB;wBAC3B,CAAC,CAAC,6BAA6B;wBAC/B,CAAC,CAAC,SAAS,CAAC,CAAC;gBACjB,MAAM,YAAY,GAChB,OAAO,CAAC,YAAY;oBACpB,OAAO,CAAC,GAAG,CAAC,cAAc;oBAC1B,OAAO,CAAC,GAAG,CAAC,YAAY;oBACxB,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC;gBACnE,IAAI,CAAC,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC;oBACxB,MAAM,IAAI,KAAK,CACb,6GAA6G,CAC9G,CAAC;gBACJ,CAAC;gBACD,OAAO,GAAG,qBAAqB,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;gBACnE,GAAG,CAAC,OAAO,CAAC,GAAG,EAAE,cAAc,CAAC,OAAO,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC,CAAC;YAC/D,CAAC;iBAAM,CAAC;gBACN,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,IAAI,MAAM,CAAC;gBACjD,OAAO,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;gBACnC,GAAG,CAAC,OAAO,CAAC,GAAG,EAAE,cAAc,CAAC,OAAO,EAAE,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}
@@ -0,0 +1,69 @@
1
+ export type JSONSchema = {
2
+ type?: string;
3
+ description?: string;
4
+ properties?: Record<string, JSONSchema>;
5
+ required?: string[];
6
+ additionalProperties?: boolean;
7
+ items?: JSONSchema;
8
+ };
9
+ export type ToolSchema = {
10
+ name: string;
11
+ description: string;
12
+ inputSchema: JSONSchema;
13
+ };
14
+ export type ToolCall = {
15
+ id: string;
16
+ name: string;
17
+ arguments: unknown;
18
+ };
19
+ export type TokenUsage = {
20
+ promptTokens: number;
21
+ completionTokens: number;
22
+ totalTokens: number;
23
+ };
24
+ export type ChatMessage = {
25
+ role: "system";
26
+ content: string;
27
+ } | {
28
+ role: "user";
29
+ content: string;
30
+ } | {
31
+ role: "assistant";
32
+ content?: string;
33
+ toolCalls?: ToolCall[];
34
+ } | {
35
+ role: "tool";
36
+ toolCallId: string;
37
+ content: string;
38
+ };
39
+ export type FinishReason = "stop" | "tool_calls" | "length" | "cancelled" | "error";
40
+ export type LLMRequest = {
41
+ messages: ChatMessage[];
42
+ model?: string;
43
+ tools?: ToolSchema[];
44
+ temperature?: number;
45
+ maxTokens?: number;
46
+ signal?: AbortSignal;
47
+ };
48
+ export type LLMResponse = {
49
+ id: string;
50
+ model: string;
51
+ message: Extract<ChatMessage, {
52
+ role: "assistant";
53
+ }>;
54
+ toolCalls?: ToolCall[];
55
+ usage?: TokenUsage;
56
+ finishReason: FinishReason;
57
+ };
58
+ export type LLMStreamPart = {
59
+ type: "text-delta";
60
+ text: string;
61
+ } | {
62
+ type: "done";
63
+ response: LLMResponse;
64
+ };
65
+ export type LLMService = {
66
+ generate(request: LLMRequest): Promise<LLMResponse>;
67
+ stream(request: LLMRequest): AsyncIterable<LLMStreamPart>;
68
+ };
69
+ //# sourceMappingURL=types.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,UAAU,GAAG;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACxC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,KAAK,CAAC,EAAE,UAAU,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,UAAU,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,QAAQ,GAAG;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,OAAO,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,WAAW,GACnB;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GACnC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GACjC;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAA;CAAE,GAC/D;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC;AAE1D,MAAM,MAAM,YAAY,GACpB,MAAM,GACN,YAAY,GACZ,QAAQ,GACR,WAAW,GACX,OAAO,CAAC;AAEZ,MAAM,MAAM,UAAU,GAAG;IACvB,QAAQ,EAAE,WAAW,EAAE,CAAC;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE;QAAE,IAAI,EAAE,WAAW,CAAA;KAAE,CAAC,CAAC;IACrD,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;IACvB,KAAK,CAAC,EAAE,UAAU,CAAC;IACnB,YAAY,EAAE,YAAY,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,aAAa,GACrB;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACpC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,WAAW,CAAA;CAAE,CAAC;AAE5C,MAAM,MAAM,UAAU,GAAG;IACvB,QAAQ,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACpD,MAAM,CAAC,OAAO,EAAE,UAAU,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;CAC3D,CAAC"}
package/dist/types.js ADDED
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}
package/package.json ADDED
@@ -0,0 +1,45 @@
1
+ {
2
+ "name": "@typescript-agent-harness/llm",
3
+ "version": "0.15.0",
4
+ "description": "LLM capability: mock + OpenAI-compatible providers",
5
+ "type": "module",
6
+ "main": "./dist/index.js",
7
+ "types": "./dist/index.d.ts",
8
+ "exports": {
9
+ ".": {
10
+ "types": "./dist/index.d.ts",
11
+ "import": "./dist/index.js"
12
+ }
13
+ },
14
+ "files": [
15
+ "dist"
16
+ ],
17
+ "dependencies": {
18
+ "@typescript-agent-harness/core": "0.15.0"
19
+ },
20
+ "devDependencies": {
21
+ "@types/node": "^22.18.0",
22
+ "typescript": "^5.9.2"
23
+ },
24
+ "license": "MIT",
25
+ "engines": {
26
+ "node": ">=22"
27
+ },
28
+ "publishConfig": {
29
+ "access": "public",
30
+ "registry": "https://registry.npmjs.org/"
31
+ },
32
+ "repository": {
33
+ "type": "git",
34
+ "url": "git+https://github.com/MichaelVendor/typescript-agent-harness.git",
35
+ "directory": "packages/llm"
36
+ },
37
+ "homepage": "https://MichaelVendor.github.io/typescript-agent-harness/",
38
+ "bugs": {
39
+ "url": "https://github.com/MichaelVendor/typescript-agent-harness/issues"
40
+ },
41
+ "scripts": {
42
+ "build": "tsc -p tsconfig.json",
43
+ "typecheck": "tsc -p tsconfig.json --noEmit"
44
+ }
45
+ }