@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 +21 -0
- package/dist/contract.d.ts +3 -0
- package/dist/contract.d.ts.map +1 -0
- package/dist/contract.js +3 -0
- package/dist/contract.js.map +1 -0
- package/dist/ids.d.ts +2 -0
- package/dist/ids.d.ts.map +1 -0
- package/dist/ids.js +6 -0
- package/dist/ids.js.map +1 -0
- package/dist/index.d.ts +6 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +5 -0
- package/dist/index.js.map +1 -0
- package/dist/mock.d.ts +7 -0
- package/dist/mock.d.ts.map +1 -0
- package/dist/mock.js +133 -0
- package/dist/mock.js.map +1 -0
- package/dist/openai.d.ts +8 -0
- package/dist/openai.d.ts.map +1 -0
- package/dist/openai.js +219 -0
- package/dist/openai.js.map +1 -0
- package/dist/plugin.d.ts +10 -0
- package/dist/plugin.d.ts.map +1 -0
- package/dist/plugin.js +98 -0
- package/dist/plugin.js.map +1 -0
- package/dist/types.d.ts +69 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/package.json +45 -0
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 @@
|
|
|
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"}
|
package/dist/contract.js
ADDED
|
@@ -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 @@
|
|
|
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
package/dist/ids.js.map
ADDED
|
@@ -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"}
|
package/dist/index.d.ts
ADDED
|
@@ -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 @@
|
|
|
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
|
package/dist/mock.js.map
ADDED
|
@@ -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"}
|
package/dist/openai.d.ts
ADDED
|
@@ -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"}
|
package/dist/plugin.d.ts
ADDED
|
@@ -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"}
|
package/dist/types.d.ts
ADDED
|
@@ -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 @@
|
|
|
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
|
+
}
|