@nodetool-ai/runtime 0.7.0-rc.11
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/dist/bridge-protocol.d.ts +39 -0
- package/dist/bridge-protocol.d.ts.map +1 -0
- package/dist/bridge-protocol.js +39 -0
- package/dist/bridge-protocol.js.map +1 -0
- package/dist/comfy-executor.d.ts +41 -0
- package/dist/comfy-executor.d.ts.map +1 -0
- package/dist/comfy-executor.js +305 -0
- package/dist/comfy-executor.js.map +1 -0
- package/dist/context-packer.d.ts +22 -0
- package/dist/context-packer.d.ts.map +1 -0
- package/dist/context-packer.js +66 -0
- package/dist/context-packer.js.map +1 -0
- package/dist/context.d.ts +405 -0
- package/dist/context.d.ts.map +1 -0
- package/dist/context.js +1455 -0
- package/dist/context.js.map +1 -0
- package/dist/index.d.ts +18 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +15 -0
- package/dist/index.js.map +1 -0
- package/dist/node-executor.d.ts +53 -0
- package/dist/node-executor.d.ts.map +1 -0
- package/dist/node-executor.js +9 -0
- package/dist/node-executor.js.map +1 -0
- package/dist/providers/aki-manifest.json +34 -0
- package/dist/providers/aki-provider.d.ts +65 -0
- package/dist/providers/aki-provider.d.ts.map +1 -0
- package/dist/providers/aki-provider.js +517 -0
- package/dist/providers/aki-provider.js.map +1 -0
- package/dist/providers/anthropic-provider.d.ts +55 -0
- package/dist/providers/anthropic-provider.d.ts.map +1 -0
- package/dist/providers/anthropic-provider.js +563 -0
- package/dist/providers/anthropic-provider.js.map +1 -0
- package/dist/providers/base-provider.d.ts +194 -0
- package/dist/providers/base-provider.d.ts.map +1 -0
- package/dist/providers/base-provider.js +515 -0
- package/dist/providers/base-provider.js.map +1 -0
- package/dist/providers/cerebras-provider.d.ts +20 -0
- package/dist/providers/cerebras-provider.d.ts.map +1 -0
- package/dist/providers/cerebras-provider.js +52 -0
- package/dist/providers/cerebras-provider.js.map +1 -0
- package/dist/providers/claude-agent-provider.d.ts +107 -0
- package/dist/providers/claude-agent-provider.d.ts.map +1 -0
- package/dist/providers/claude-agent-provider.js +586 -0
- package/dist/providers/claude-agent-provider.js.map +1 -0
- package/dist/providers/cohere-provider.d.ts +36 -0
- package/dist/providers/cohere-provider.d.ts.map +1 -0
- package/dist/providers/cohere-provider.js +116 -0
- package/dist/providers/cohere-provider.js.map +1 -0
- package/dist/providers/cost-calculator.d.ts +109 -0
- package/dist/providers/cost-calculator.d.ts.map +1 -0
- package/dist/providers/cost-calculator.js +475 -0
- package/dist/providers/cost-calculator.js.map +1 -0
- package/dist/providers/deepseek-provider.d.ts +25 -0
- package/dist/providers/deepseek-provider.d.ts.map +1 -0
- package/dist/providers/deepseek-provider.js +58 -0
- package/dist/providers/deepseek-provider.js.map +1 -0
- package/dist/providers/fake-provider.d.ts +52 -0
- package/dist/providers/fake-provider.d.ts.map +1 -0
- package/dist/providers/fake-provider.js +107 -0
- package/dist/providers/fake-provider.js.map +1 -0
- package/dist/providers/fal-provider.d.ts +32 -0
- package/dist/providers/fal-provider.d.ts.map +1 -0
- package/dist/providers/fal-provider.js +445 -0
- package/dist/providers/fal-provider.js.map +1 -0
- package/dist/providers/gemini-provider.d.ts +109 -0
- package/dist/providers/gemini-provider.d.ts.map +1 -0
- package/dist/providers/gemini-provider.js +943 -0
- package/dist/providers/gemini-provider.js.map +1 -0
- package/dist/providers/groq-provider.d.ts +20 -0
- package/dist/providers/groq-provider.d.ts.map +1 -0
- package/dist/providers/groq-provider.js +52 -0
- package/dist/providers/groq-provider.js.map +1 -0
- package/dist/providers/huggingface-provider.d.ts +59 -0
- package/dist/providers/huggingface-provider.d.ts.map +1 -0
- package/dist/providers/huggingface-provider.js +305 -0
- package/dist/providers/huggingface-provider.js.map +1 -0
- package/dist/providers/index.d.ts +69 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +137 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/jina-provider.d.ts +37 -0
- package/dist/providers/jina-provider.d.ts.map +1 -0
- package/dist/providers/jina-provider.js +118 -0
- package/dist/providers/jina-provider.js.map +1 -0
- package/dist/providers/kie-provider.d.ts +19 -0
- package/dist/providers/kie-provider.d.ts.map +1 -0
- package/dist/providers/kie-provider.js +118 -0
- package/dist/providers/kie-provider.js.map +1 -0
- package/dist/providers/llama-provider.d.ts +49 -0
- package/dist/providers/llama-provider.d.ts.map +1 -0
- package/dist/providers/llama-provider.js +381 -0
- package/dist/providers/llama-provider.js.map +1 -0
- package/dist/providers/lmstudio-provider.d.ts +22 -0
- package/dist/providers/lmstudio-provider.d.ts.map +1 -0
- package/dist/providers/lmstudio-provider.js +63 -0
- package/dist/providers/lmstudio-provider.js.map +1 -0
- package/dist/providers/manifest-models.d.ts +12 -0
- package/dist/providers/manifest-models.d.ts.map +1 -0
- package/dist/providers/manifest-models.js +96 -0
- package/dist/providers/manifest-models.js.map +1 -0
- package/dist/providers/meshy-provider.d.ts +53 -0
- package/dist/providers/meshy-provider.d.ts.map +1 -0
- package/dist/providers/meshy-provider.js +230 -0
- package/dist/providers/meshy-provider.js.map +1 -0
- package/dist/providers/minimax-provider.d.ts +66 -0
- package/dist/providers/minimax-provider.d.ts.map +1 -0
- package/dist/providers/minimax-provider.js +531 -0
- package/dist/providers/minimax-provider.js.map +1 -0
- package/dist/providers/mistral-provider.d.ts +26 -0
- package/dist/providers/mistral-provider.d.ts.map +1 -0
- package/dist/providers/mistral-provider.js +74 -0
- package/dist/providers/mistral-provider.js.map +1 -0
- package/dist/providers/moonshot-provider.d.ts +24 -0
- package/dist/providers/moonshot-provider.d.ts.map +1 -0
- package/dist/providers/moonshot-provider.js +43 -0
- package/dist/providers/moonshot-provider.js.map +1 -0
- package/dist/providers/ollama-provider.d.ts +71 -0
- package/dist/providers/ollama-provider.d.ts.map +1 -0
- package/dist/providers/ollama-provider.js +480 -0
- package/dist/providers/ollama-provider.js.map +1 -0
- package/dist/providers/openai-provider.d.ts +103 -0
- package/dist/providers/openai-provider.d.ts.map +1 -0
- package/dist/providers/openai-provider.js +917 -0
- package/dist/providers/openai-provider.js.map +1 -0
- package/dist/providers/openrouter-provider.d.ts +49 -0
- package/dist/providers/openrouter-provider.d.ts.map +1 -0
- package/dist/providers/openrouter-provider.js +147 -0
- package/dist/providers/openrouter-provider.js.map +1 -0
- package/dist/providers/provider-registry.d.ts +24 -0
- package/dist/providers/provider-registry.d.ts.map +1 -0
- package/dist/providers/provider-registry.js +124 -0
- package/dist/providers/provider-registry.js.map +1 -0
- package/dist/providers/python-provider.d.ts +75 -0
- package/dist/providers/python-provider.d.ts.map +1 -0
- package/dist/providers/python-provider.js +157 -0
- package/dist/providers/python-provider.js.map +1 -0
- package/dist/providers/replicate-provider.d.ts +73 -0
- package/dist/providers/replicate-provider.d.ts.map +1 -0
- package/dist/providers/replicate-provider.js +484 -0
- package/dist/providers/replicate-provider.js.map +1 -0
- package/dist/providers/rodin-provider.d.ts +41 -0
- package/dist/providers/rodin-provider.d.ts.map +1 -0
- package/dist/providers/rodin-provider.js +236 -0
- package/dist/providers/rodin-provider.js.map +1 -0
- package/dist/providers/scripted-provider.d.ts +141 -0
- package/dist/providers/scripted-provider.d.ts.map +1 -0
- package/dist/providers/scripted-provider.js +246 -0
- package/dist/providers/scripted-provider.js.map +1 -0
- package/dist/providers/together-provider.d.ts +20 -0
- package/dist/providers/together-provider.d.ts.map +1 -0
- package/dist/providers/together-provider.js +56 -0
- package/dist/providers/together-provider.js.map +1 -0
- package/dist/providers/types.d.ts +202 -0
- package/dist/providers/types.d.ts.map +1 -0
- package/dist/providers/types.js +2 -0
- package/dist/providers/types.js.map +1 -0
- package/dist/providers/vllm-provider.d.ts +22 -0
- package/dist/providers/vllm-provider.d.ts.map +1 -0
- package/dist/providers/vllm-provider.js +66 -0
- package/dist/providers/vllm-provider.js.map +1 -0
- package/dist/providers/voyage-provider.d.ts +37 -0
- package/dist/providers/voyage-provider.d.ts.map +1 -0
- package/dist/providers/voyage-provider.js +124 -0
- package/dist/providers/voyage-provider.js.map +1 -0
- package/dist/providers/xai-provider.d.ts +24 -0
- package/dist/providers/xai-provider.d.ts.map +1 -0
- package/dist/providers/xai-provider.js +57 -0
- package/dist/providers/xai-provider.js.map +1 -0
- package/dist/python-bridge-types.d.ts +70 -0
- package/dist/python-bridge-types.d.ts.map +1 -0
- package/dist/python-bridge-types.js +2 -0
- package/dist/python-bridge-types.js.map +1 -0
- package/dist/python-node-executor.d.ts +20 -0
- package/dist/python-node-executor.d.ts.map +1 -0
- package/dist/python-node-executor.js +219 -0
- package/dist/python-node-executor.js.map +1 -0
- package/dist/python-stdio-bridge.d.ts +67 -0
- package/dist/python-stdio-bridge.d.ts.map +1 -0
- package/dist/python-stdio-bridge.js +704 -0
- package/dist/python-stdio-bridge.js.map +1 -0
- package/dist/recommended-models.d.ts +17 -0
- package/dist/recommended-models.d.ts.map +1 -0
- package/dist/recommended-models.js +139 -0
- package/dist/recommended-models.js.map +1 -0
- package/dist/telemetry.d.ts +61 -0
- package/dist/telemetry.d.ts.map +1 -0
- package/dist/telemetry.js +128 -0
- package/dist/telemetry.js.map +1 -0
- package/dist/trace-exporters.d.ts +72 -0
- package/dist/trace-exporters.d.ts.map +1 -0
- package/dist/trace-exporters.js +216 -0
- package/dist/trace-exporters.js.map +1 -0
- package/dist/tracing-helpers.d.ts +99 -0
- package/dist/tracing-helpers.d.ts.map +1 -0
- package/dist/tracing-helpers.js +217 -0
- package/dist/tracing-helpers.js.map +1 -0
- package/package.json +64 -0
|
@@ -0,0 +1,517 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* AKI.IO provider — wraps the official `@aki-io/aki-io` SDK.
|
|
3
|
+
*
|
|
4
|
+
* AKI models are exposed as endpoints (e.g. `llama3_chat`). Each request
|
|
5
|
+
* targets one endpoint, so we spin up an `AkiClient` per model. Messages map
|
|
6
|
+
* onto AKI's `chat_context` (system/user/assistant) and the last attached
|
|
7
|
+
* image rides along in the `image` field.
|
|
8
|
+
*
|
|
9
|
+
* Docs: https://aki.io/docs
|
|
10
|
+
*/
|
|
11
|
+
import { randomUUID } from "node:crypto";
|
|
12
|
+
import { readFileSync } from "node:fs";
|
|
13
|
+
import { AkiClient, decodeBinary } from "@aki-io/aki-io";
|
|
14
|
+
import { createLogger } from "@nodetool-ai/config";
|
|
15
|
+
import { BaseProvider } from "./base-provider.js";
|
|
16
|
+
const log = createLogger("nodetool.runtime.providers.aki");
|
|
17
|
+
function isStringRecord(value) {
|
|
18
|
+
return (!!value &&
|
|
19
|
+
typeof value === "object" &&
|
|
20
|
+
Object.values(value).every((v) => typeof v === "string"));
|
|
21
|
+
}
|
|
22
|
+
function isAkiManifestEntry(value) {
|
|
23
|
+
return (!!value &&
|
|
24
|
+
typeof value === "object" &&
|
|
25
|
+
typeof value.endpointId === "string" &&
|
|
26
|
+
typeof value.title === "string" &&
|
|
27
|
+
typeof value.outputType === "string" &&
|
|
28
|
+
(value.paramNames === undefined ||
|
|
29
|
+
isStringRecord(value.paramNames)));
|
|
30
|
+
}
|
|
31
|
+
const AKI_MANIFEST_URL = new URL("./aki-manifest.json", import.meta.url);
|
|
32
|
+
const AKI_FALLBACK_IMAGE_MODELS = [
|
|
33
|
+
{
|
|
34
|
+
id: "sdxl_img",
|
|
35
|
+
name: "SDXL",
|
|
36
|
+
provider: "aki",
|
|
37
|
+
supportedTasks: ["text_to_image"]
|
|
38
|
+
},
|
|
39
|
+
{
|
|
40
|
+
id: "flux-text2img",
|
|
41
|
+
name: "FLUX Text to Image",
|
|
42
|
+
provider: "aki",
|
|
43
|
+
supportedTasks: ["text_to_image"]
|
|
44
|
+
},
|
|
45
|
+
{
|
|
46
|
+
id: "flux-img2img",
|
|
47
|
+
name: "FLUX Image to Image",
|
|
48
|
+
provider: "aki",
|
|
49
|
+
supportedTasks: ["image_to_image"]
|
|
50
|
+
}
|
|
51
|
+
];
|
|
52
|
+
const AKI_FALLBACK_LANGUAGE_MODELS = [
|
|
53
|
+
{ id: "llama3_chat", name: "Llama 3 Chat", provider: "aki" }
|
|
54
|
+
];
|
|
55
|
+
let akiManifestCache = null;
|
|
56
|
+
function uint8ToBase64(data) {
|
|
57
|
+
return Buffer.from(data).toString("base64");
|
|
58
|
+
}
|
|
59
|
+
function dataUriToBase64(uri) {
|
|
60
|
+
const commaIndex = uri.indexOf(",");
|
|
61
|
+
if (commaIndex < 0) {
|
|
62
|
+
return uri;
|
|
63
|
+
}
|
|
64
|
+
const header = uri.slice(0, commaIndex);
|
|
65
|
+
const payload = uri.slice(commaIndex + 1);
|
|
66
|
+
if (header.includes(";base64")) {
|
|
67
|
+
return payload;
|
|
68
|
+
}
|
|
69
|
+
return Buffer.from(decodeURIComponent(payload), "utf8").toString("base64");
|
|
70
|
+
}
|
|
71
|
+
function loadAkiManifest() {
|
|
72
|
+
if (akiManifestCache) {
|
|
73
|
+
return akiManifestCache;
|
|
74
|
+
}
|
|
75
|
+
try {
|
|
76
|
+
const raw = Buffer.from(readFileSync(AKI_MANIFEST_URL)).toString("utf8");
|
|
77
|
+
const parsed = JSON.parse(raw);
|
|
78
|
+
if (!Array.isArray(parsed)) {
|
|
79
|
+
akiManifestCache = [];
|
|
80
|
+
return akiManifestCache;
|
|
81
|
+
}
|
|
82
|
+
akiManifestCache = parsed.filter(isAkiManifestEntry);
|
|
83
|
+
return akiManifestCache;
|
|
84
|
+
}
|
|
85
|
+
catch (error) {
|
|
86
|
+
log.warn("Failed to load AKI manifest, falling back to defaults", error);
|
|
87
|
+
akiManifestCache = [];
|
|
88
|
+
return akiManifestCache;
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
function getManifestEntry(endpointId) {
|
|
92
|
+
return loadAkiManifest().find((entry) => entry.endpointId === endpointId);
|
|
93
|
+
}
|
|
94
|
+
function manifestLanguageModels() {
|
|
95
|
+
const models = loadAkiManifest()
|
|
96
|
+
.filter((entry) => entry.outputType === "text")
|
|
97
|
+
.map((entry) => ({
|
|
98
|
+
id: entry.endpointId,
|
|
99
|
+
name: entry.title,
|
|
100
|
+
provider: "aki"
|
|
101
|
+
}));
|
|
102
|
+
return models.length > 0 ? models : AKI_FALLBACK_LANGUAGE_MODELS;
|
|
103
|
+
}
|
|
104
|
+
function manifestImageModels() {
|
|
105
|
+
const models = loadAkiManifest()
|
|
106
|
+
.filter((entry) => entry.outputType === "image")
|
|
107
|
+
.map((entry) => ({
|
|
108
|
+
id: entry.endpointId,
|
|
109
|
+
name: entry.title,
|
|
110
|
+
provider: "aki",
|
|
111
|
+
supportedTasks: entry.supportedTasks && entry.supportedTasks.length > 0
|
|
112
|
+
? entry.supportedTasks
|
|
113
|
+
: undefined
|
|
114
|
+
}));
|
|
115
|
+
return models.length > 0 ? models : AKI_FALLBACK_IMAGE_MODELS;
|
|
116
|
+
}
|
|
117
|
+
function remapRequestParams(endpointId, request) {
|
|
118
|
+
const paramNames = getManifestEntry(endpointId)?.paramNames;
|
|
119
|
+
if (!paramNames || Object.keys(paramNames).length === 0) {
|
|
120
|
+
return request;
|
|
121
|
+
}
|
|
122
|
+
const remapped = {};
|
|
123
|
+
for (const [key, value] of Object.entries(request)) {
|
|
124
|
+
remapped[paramNames[key] ?? key] = value;
|
|
125
|
+
}
|
|
126
|
+
return remapped;
|
|
127
|
+
}
|
|
128
|
+
function withPromptParam(request) {
|
|
129
|
+
if (!("prompt_input" in request)) {
|
|
130
|
+
return request;
|
|
131
|
+
}
|
|
132
|
+
const remapped = { ...request };
|
|
133
|
+
remapped.prompt = remapped.prompt_input;
|
|
134
|
+
delete remapped.prompt_input;
|
|
135
|
+
return remapped;
|
|
136
|
+
}
|
|
137
|
+
function shouldRetryWithPromptParam(error) {
|
|
138
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
139
|
+
const normalized = message.toLowerCase();
|
|
140
|
+
return (normalized.includes("invalid input parameter(s): prompt_input") &&
|
|
141
|
+
normalized.includes("missing required argument: prompt"));
|
|
142
|
+
}
|
|
143
|
+
function responseImageToBytes(images) {
|
|
144
|
+
const firstValue = Array.isArray(images) ? images[0] : images;
|
|
145
|
+
if (!firstValue) {
|
|
146
|
+
return null;
|
|
147
|
+
}
|
|
148
|
+
if (firstValue instanceof Uint8Array) {
|
|
149
|
+
return firstValue;
|
|
150
|
+
}
|
|
151
|
+
if (Buffer.isBuffer(firstValue)) {
|
|
152
|
+
return new Uint8Array(firstValue);
|
|
153
|
+
}
|
|
154
|
+
if (typeof firstValue === "string") {
|
|
155
|
+
const [, decoded] = decodeBinary(firstValue);
|
|
156
|
+
return decoded ? new Uint8Array(decoded) : null;
|
|
157
|
+
}
|
|
158
|
+
return null;
|
|
159
|
+
}
|
|
160
|
+
export class AkiProvider extends BaseProvider {
|
|
161
|
+
static requiredSecrets() {
|
|
162
|
+
return ["AKI_API_KEY"];
|
|
163
|
+
}
|
|
164
|
+
apiKey;
|
|
165
|
+
_clientFactory;
|
|
166
|
+
constructor(secrets, options = {}) {
|
|
167
|
+
super("aki");
|
|
168
|
+
const apiKey = secrets.AKI_API_KEY;
|
|
169
|
+
if (!apiKey || !apiKey.trim()) {
|
|
170
|
+
throw new Error("AKI_API_KEY is not configured");
|
|
171
|
+
}
|
|
172
|
+
this.apiKey = apiKey;
|
|
173
|
+
this._clientFactory =
|
|
174
|
+
options.clientFactory ?? ((config) => new AkiClient(config));
|
|
175
|
+
}
|
|
176
|
+
getContainerEnv() {
|
|
177
|
+
return { AKI_API_KEY: this.apiKey };
|
|
178
|
+
}
|
|
179
|
+
async hasToolSupport(_model) {
|
|
180
|
+
return true;
|
|
181
|
+
}
|
|
182
|
+
/** Create a fresh AkiClient targeting the given endpoint (= model id). */
|
|
183
|
+
makeClient(endpointName) {
|
|
184
|
+
const config = {
|
|
185
|
+
endpointName,
|
|
186
|
+
apiKey: this.apiKey,
|
|
187
|
+
outputBinaryFormat: "byteString",
|
|
188
|
+
raiseExceptions: true,
|
|
189
|
+
returnToolCallDict: true
|
|
190
|
+
};
|
|
191
|
+
return this._clientFactory(config);
|
|
192
|
+
}
|
|
193
|
+
/**
|
|
194
|
+
* Convert NodeTool `Message[]` into AKI's chat format. AKI does not model
|
|
195
|
+
* `tool` messages or multi-part content, so image parts are collapsed into
|
|
196
|
+
* the single `image` field (last image wins) and text parts are joined.
|
|
197
|
+
*/
|
|
198
|
+
async toChatContext(messages) {
|
|
199
|
+
const chatContext = [];
|
|
200
|
+
let image;
|
|
201
|
+
for (const msg of messages) {
|
|
202
|
+
if (msg.role === "tool")
|
|
203
|
+
continue;
|
|
204
|
+
const role = msg.role === "system"
|
|
205
|
+
? "system"
|
|
206
|
+
: msg.role === "assistant"
|
|
207
|
+
? "assistant"
|
|
208
|
+
: "user";
|
|
209
|
+
let content = "";
|
|
210
|
+
if (typeof msg.content === "string") {
|
|
211
|
+
content = msg.content;
|
|
212
|
+
}
|
|
213
|
+
else if (Array.isArray(msg.content)) {
|
|
214
|
+
const parts = [];
|
|
215
|
+
for (const part of msg.content) {
|
|
216
|
+
if (part.type === "text") {
|
|
217
|
+
parts.push(part.text);
|
|
218
|
+
}
|
|
219
|
+
else if (part.type === "image_url") {
|
|
220
|
+
const data = part.image.data;
|
|
221
|
+
if (typeof data === "string") {
|
|
222
|
+
image = data.startsWith("data:")
|
|
223
|
+
? dataUriToBase64(data)
|
|
224
|
+
: data;
|
|
225
|
+
}
|
|
226
|
+
else if (data instanceof Uint8Array) {
|
|
227
|
+
image = uint8ToBase64(data);
|
|
228
|
+
}
|
|
229
|
+
else if (part.image.uri) {
|
|
230
|
+
image = await this.resolveImageUri(part.image.uri);
|
|
231
|
+
}
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
content = parts.join("\n");
|
|
235
|
+
}
|
|
236
|
+
chatContext.push({ role, content });
|
|
237
|
+
}
|
|
238
|
+
return { chatContext, image };
|
|
239
|
+
}
|
|
240
|
+
formatTools(tools) {
|
|
241
|
+
return tools.map((tool) => {
|
|
242
|
+
if (tool.type === "code_interpreter" ||
|
|
243
|
+
tool.name === "code_interpreter") {
|
|
244
|
+
return { type: "code_interpreter" };
|
|
245
|
+
}
|
|
246
|
+
return {
|
|
247
|
+
type: "function",
|
|
248
|
+
function: {
|
|
249
|
+
name: tool.name,
|
|
250
|
+
description: tool.description ?? "",
|
|
251
|
+
parameters: tool.inputSchema ?? { type: "object", properties: {} }
|
|
252
|
+
}
|
|
253
|
+
};
|
|
254
|
+
});
|
|
255
|
+
}
|
|
256
|
+
parseToolCalls(raw) {
|
|
257
|
+
if (raw == null) {
|
|
258
|
+
return [];
|
|
259
|
+
}
|
|
260
|
+
const candidates = Array.isArray(raw) ? raw : [raw];
|
|
261
|
+
const toolCalls = [];
|
|
262
|
+
for (const candidate of candidates) {
|
|
263
|
+
let parsed = candidate;
|
|
264
|
+
if (typeof parsed === "string") {
|
|
265
|
+
try {
|
|
266
|
+
parsed = JSON.parse(parsed);
|
|
267
|
+
}
|
|
268
|
+
catch {
|
|
269
|
+
continue;
|
|
270
|
+
}
|
|
271
|
+
}
|
|
272
|
+
if (!parsed || typeof parsed !== "object") {
|
|
273
|
+
continue;
|
|
274
|
+
}
|
|
275
|
+
const record = parsed;
|
|
276
|
+
const nestedFunction = record.function && typeof record.function === "object"
|
|
277
|
+
? record.function
|
|
278
|
+
: null;
|
|
279
|
+
const name = typeof record.name === "string"
|
|
280
|
+
? record.name
|
|
281
|
+
: typeof nestedFunction?.name === "string"
|
|
282
|
+
? nestedFunction.name
|
|
283
|
+
: "";
|
|
284
|
+
if (!name) {
|
|
285
|
+
continue;
|
|
286
|
+
}
|
|
287
|
+
const rawArgs = record.arguments ?? nestedFunction?.arguments ?? record.args ?? {};
|
|
288
|
+
let args = {};
|
|
289
|
+
if (typeof rawArgs === "string") {
|
|
290
|
+
args = this.parseToolCallArgs(rawArgs);
|
|
291
|
+
}
|
|
292
|
+
else if (rawArgs && typeof rawArgs === "object" && !Array.isArray(rawArgs)) {
|
|
293
|
+
args = rawArgs;
|
|
294
|
+
}
|
|
295
|
+
toolCalls.push({
|
|
296
|
+
id: typeof record.id === "string"
|
|
297
|
+
? record.id
|
|
298
|
+
: typeof record.tool_call_id === "string"
|
|
299
|
+
? record.tool_call_id
|
|
300
|
+
: randomUUID(),
|
|
301
|
+
name,
|
|
302
|
+
args
|
|
303
|
+
});
|
|
304
|
+
}
|
|
305
|
+
return toolCalls;
|
|
306
|
+
}
|
|
307
|
+
async buildParams(args) {
|
|
308
|
+
const { chatContext, image } = await this.toChatContext(args.messages);
|
|
309
|
+
const params = { chat_context: chatContext };
|
|
310
|
+
if (image)
|
|
311
|
+
params.image = image;
|
|
312
|
+
if (args.tools && args.tools.length > 0) {
|
|
313
|
+
params.tools = this.formatTools(args.tools);
|
|
314
|
+
}
|
|
315
|
+
// AKI accepts tool definitions but currently rejects `tool_choice`, so we
|
|
316
|
+
// intentionally do not send it.
|
|
317
|
+
if (args.maxTokens != null)
|
|
318
|
+
params.max_gen_tokens = args.maxTokens;
|
|
319
|
+
if (args.temperature != null)
|
|
320
|
+
params.temperature = args.temperature;
|
|
321
|
+
if (args.topP != null)
|
|
322
|
+
params.top_p = args.topP;
|
|
323
|
+
return params;
|
|
324
|
+
}
|
|
325
|
+
async resolveImageUri(uri) {
|
|
326
|
+
const resolved = await this.resolveUri(uri);
|
|
327
|
+
if (resolved.startsWith("data:")) {
|
|
328
|
+
return dataUriToBase64(resolved);
|
|
329
|
+
}
|
|
330
|
+
if (resolved.startsWith("http://") || resolved.startsWith("https://")) {
|
|
331
|
+
try {
|
|
332
|
+
const response = await fetch(resolved);
|
|
333
|
+
if (!response.ok) {
|
|
334
|
+
return undefined;
|
|
335
|
+
}
|
|
336
|
+
return Buffer.from(await response.arrayBuffer()).toString("base64");
|
|
337
|
+
}
|
|
338
|
+
catch (error) {
|
|
339
|
+
log.debug("Failed to fetch remote AKI image URI", {
|
|
340
|
+
uri: resolved,
|
|
341
|
+
error
|
|
342
|
+
});
|
|
343
|
+
// Best-effort URI normalization; keep request flowing without image.
|
|
344
|
+
return undefined;
|
|
345
|
+
}
|
|
346
|
+
}
|
|
347
|
+
return resolved;
|
|
348
|
+
}
|
|
349
|
+
async generateMessage(args) {
|
|
350
|
+
const client = this.makeClient(args.model);
|
|
351
|
+
const params = await this.buildParams(args);
|
|
352
|
+
const response = await client.doApiRequest(params);
|
|
353
|
+
if (!response.success) {
|
|
354
|
+
throw new Error(response.error ??
|
|
355
|
+
`Aki request failed (code ${response.error_code ?? "unknown"})`);
|
|
356
|
+
}
|
|
357
|
+
if (typeof response.num_generated_tokens === "number") {
|
|
358
|
+
this.trackUsage(args.model, {
|
|
359
|
+
inputTokens: response.prompt_length ?? 0,
|
|
360
|
+
outputTokens: response.num_generated_tokens
|
|
361
|
+
});
|
|
362
|
+
}
|
|
363
|
+
return {
|
|
364
|
+
role: "assistant",
|
|
365
|
+
content: response.text ?? "",
|
|
366
|
+
toolCalls: this.parseToolCalls(response.tool_calls)
|
|
367
|
+
};
|
|
368
|
+
}
|
|
369
|
+
async *generateMessages(args) {
|
|
370
|
+
const client = this.makeClient(args.model);
|
|
371
|
+
const params = await this.buildParams(args);
|
|
372
|
+
const stream = client.getApiRequestGenerator(params);
|
|
373
|
+
let emitted = "";
|
|
374
|
+
let errorMessage;
|
|
375
|
+
let promptTokens;
|
|
376
|
+
let generatedTokens;
|
|
377
|
+
const emittedToolCallIds = new Set();
|
|
378
|
+
for await (const update of stream) {
|
|
379
|
+
const u = update;
|
|
380
|
+
if (u.success === false && typeof u.error === "string") {
|
|
381
|
+
errorMessage = u.error;
|
|
382
|
+
break;
|
|
383
|
+
}
|
|
384
|
+
const progressData = (u.progress_data ?? u.result_data);
|
|
385
|
+
const promptLength = progressData?.["prompt_length"] ?? u["prompt_length"];
|
|
386
|
+
if (typeof promptLength === "number") {
|
|
387
|
+
promptTokens = promptLength;
|
|
388
|
+
}
|
|
389
|
+
const generatedLength = progressData?.["num_generated_tokens"] ?? u["num_generated_tokens"];
|
|
390
|
+
if (typeof generatedLength === "number") {
|
|
391
|
+
generatedTokens = Math.max(generatedTokens ?? 0, generatedLength);
|
|
392
|
+
}
|
|
393
|
+
const text = typeof progressData?.["text"] === "string"
|
|
394
|
+
? progressData["text"]
|
|
395
|
+
: undefined;
|
|
396
|
+
if (text && text.length > emitted.length) {
|
|
397
|
+
const delta = text.slice(emitted.length);
|
|
398
|
+
emitted = text;
|
|
399
|
+
yield { type: "chunk", content: delta, done: false };
|
|
400
|
+
}
|
|
401
|
+
const toolCalls = this.parseToolCalls(progressData?.["tool_calls"] ?? u["tool_calls"]);
|
|
402
|
+
for (const toolCall of toolCalls) {
|
|
403
|
+
if (emittedToolCallIds.has(toolCall.id)) {
|
|
404
|
+
continue;
|
|
405
|
+
}
|
|
406
|
+
emittedToolCallIds.add(toolCall.id);
|
|
407
|
+
yield toolCall;
|
|
408
|
+
}
|
|
409
|
+
}
|
|
410
|
+
if (errorMessage)
|
|
411
|
+
throw new Error(errorMessage);
|
|
412
|
+
if (typeof generatedTokens === "number") {
|
|
413
|
+
this.trackUsage(args.model, {
|
|
414
|
+
inputTokens: promptTokens ?? 0,
|
|
415
|
+
outputTokens: generatedTokens
|
|
416
|
+
});
|
|
417
|
+
}
|
|
418
|
+
yield { type: "chunk", content: "", done: true };
|
|
419
|
+
}
|
|
420
|
+
buildImageRequest(prompt, params, image) {
|
|
421
|
+
const request = {
|
|
422
|
+
prompt_input: prompt
|
|
423
|
+
};
|
|
424
|
+
if (image) {
|
|
425
|
+
request.image = uint8ToBase64(image);
|
|
426
|
+
}
|
|
427
|
+
const width = "width" in params
|
|
428
|
+
? params.width
|
|
429
|
+
: params.targetWidth;
|
|
430
|
+
const height = "height" in params
|
|
431
|
+
? params.height
|
|
432
|
+
: params.targetHeight;
|
|
433
|
+
if (width != null) {
|
|
434
|
+
request.width = width;
|
|
435
|
+
}
|
|
436
|
+
if (height != null) {
|
|
437
|
+
request.height = height;
|
|
438
|
+
}
|
|
439
|
+
if (params.negativePrompt) {
|
|
440
|
+
request.negative_prompt = params.negativePrompt;
|
|
441
|
+
}
|
|
442
|
+
if (params.guidanceScale != null) {
|
|
443
|
+
request.guidance_scale = params.guidanceScale;
|
|
444
|
+
}
|
|
445
|
+
if (params.numInferenceSteps != null) {
|
|
446
|
+
request.num_inference_steps = params.numInferenceSteps;
|
|
447
|
+
}
|
|
448
|
+
if (params.seed != null && params.seed !== -1) {
|
|
449
|
+
request.seed = params.seed;
|
|
450
|
+
}
|
|
451
|
+
if (params.scheduler) {
|
|
452
|
+
request.scheduler = params.scheduler;
|
|
453
|
+
}
|
|
454
|
+
if ("safetyCheck" in params && params.safetyCheck != null) {
|
|
455
|
+
request.safety_check = params.safetyCheck;
|
|
456
|
+
}
|
|
457
|
+
if (params.quality) {
|
|
458
|
+
request.quality = params.quality;
|
|
459
|
+
}
|
|
460
|
+
if ("strength" in params && params.strength != null) {
|
|
461
|
+
request.strength = params.strength;
|
|
462
|
+
}
|
|
463
|
+
return request;
|
|
464
|
+
}
|
|
465
|
+
async runImageRequest(modelId, request) {
|
|
466
|
+
const client = this.makeClient(modelId);
|
|
467
|
+
let response;
|
|
468
|
+
const initialRequest = remapRequestParams(modelId, request);
|
|
469
|
+
try {
|
|
470
|
+
response = await client.doApiRequest(initialRequest);
|
|
471
|
+
}
|
|
472
|
+
catch (error) {
|
|
473
|
+
if (!shouldRetryWithPromptParam(error)) {
|
|
474
|
+
throw error;
|
|
475
|
+
}
|
|
476
|
+
log.info("Retrying AKI image request with prompt param alias", {
|
|
477
|
+
modelId
|
|
478
|
+
});
|
|
479
|
+
response = await client.doApiRequest(withPromptParam(initialRequest));
|
|
480
|
+
}
|
|
481
|
+
if (!response.success) {
|
|
482
|
+
throw new Error(response.error ??
|
|
483
|
+
`Aki image request failed (code ${response.error_code ?? "unknown"})`);
|
|
484
|
+
}
|
|
485
|
+
const bytes = responseImageToBytes(response.images);
|
|
486
|
+
if (!bytes) {
|
|
487
|
+
throw new Error("AKI image generation returned no image data");
|
|
488
|
+
}
|
|
489
|
+
return bytes;
|
|
490
|
+
}
|
|
491
|
+
async textToImage(params) {
|
|
492
|
+
const prompt = params.prompt.trim();
|
|
493
|
+
if (!prompt) {
|
|
494
|
+
throw new Error("Prompt is required");
|
|
495
|
+
}
|
|
496
|
+
const request = this.buildImageRequest(prompt, params);
|
|
497
|
+
return this.runImageRequest(params.model.id, request);
|
|
498
|
+
}
|
|
499
|
+
async imageToImage(image, params) {
|
|
500
|
+
const prompt = params.prompt.trim();
|
|
501
|
+
if (!prompt) {
|
|
502
|
+
throw new Error("Prompt is required");
|
|
503
|
+
}
|
|
504
|
+
if (!image.length) {
|
|
505
|
+
throw new Error("Image is required");
|
|
506
|
+
}
|
|
507
|
+
const request = this.buildImageRequest(prompt, params, image);
|
|
508
|
+
return this.runImageRequest(params.model.id, request);
|
|
509
|
+
}
|
|
510
|
+
async getAvailableLanguageModels() {
|
|
511
|
+
return manifestLanguageModels();
|
|
512
|
+
}
|
|
513
|
+
async getAvailableImageModels() {
|
|
514
|
+
return manifestImageModels();
|
|
515
|
+
}
|
|
516
|
+
}
|
|
517
|
+
//# sourceMappingURL=aki-provider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"aki-provider.js","sourceRoot":"","sources":["../../src/providers/aki-provider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAOzD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAEnD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAYlD,MAAM,GAAG,GAAG,YAAY,CAAC,gCAAgC,CAAC,CAAC;AAU3D,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,CACL,CAAC,CAAC,KAAK;QACP,OAAO,KAAK,KAAK,QAAQ;QACzB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CACzD,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAc;IACxC,OAAO,CACL,CAAC,CAAC,KAAK;QACP,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAQ,KAA0B,CAAC,UAAU,KAAK,QAAQ;QAC1D,OAAQ,KAA0B,CAAC,KAAK,KAAK,QAAQ;QACrD,OAAQ,KAA0B,CAAC,UAAU,KAAK,QAAQ;QAC1D,CAAE,KAA0B,CAAC,UAAU,KAAK,SAAS;YACnD,cAAc,CAAE,KAA0B,CAAC,UAAU,CAAC,CAAC,CAC1D,CAAC;AACJ,CAAC;AAED,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,qBAAqB,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAEzE,MAAM,yBAAyB,GAAiB;IAC9C;QACE,EAAE,EAAE,UAAU;QACd,IAAI,EAAE,MAAM;QACZ,QAAQ,EAAE,KAAK;QACf,cAAc,EAAE,CAAC,eAAe,CAAC;KAClC;IACD;QACE,EAAE,EAAE,eAAe;QACnB,IAAI,EAAE,oBAAoB;QAC1B,QAAQ,EAAE,KAAK;QACf,cAAc,EAAE,CAAC,eAAe,CAAC;KAClC;IACD;QACE,EAAE,EAAE,cAAc;QAClB,IAAI,EAAE,qBAAqB;QAC3B,QAAQ,EAAE,KAAK;QACf,cAAc,EAAE,CAAC,gBAAgB,CAAC;KACnC;CACF,CAAC;AAEF,MAAM,4BAA4B,GAAoB;IACpD,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,KAAK,EAAE;CAC7D,CAAC;AACF,IAAI,gBAAgB,GAA8B,IAAI,CAAC;AAMvD,SAAS,aAAa,CAAC,IAAgB;IACrC,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AAC9C,CAAC;AAED,SAAS,eAAe,CAAC,GAAW;IAClC,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACpC,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;QACnB,OAAO,GAAG,CAAC;IACb,CAAC;IACD,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IACxC,MAAM,OAAO,GAAG,GAAG,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;IAC1C,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC/B,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AAC7E,CAAC;AAED,SAAS,eAAe;IACtB,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACzE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAY,CAAC;QAC1C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,gBAAgB,GAAG,EAAE,CAAC;YACtB,OAAO,gBAAgB,CAAC;QAC1B,CAAC;QACD,gBAAgB,GAAG,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;QACrD,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,IAAI,CAAC,uDAAuD,EAAE,KAAK,CAAC,CAAC;QACzE,gBAAgB,GAAG,EAAE,CAAC;QACtB,OAAO,gBAAgB,CAAC;IAC1B,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,UAAkB;IAC1C,OAAO,eAAe,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,KAAK,UAAU,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,sBAAsB;IAC7B,MAAM,MAAM,GAAG,eAAe,EAAE;SAC7B,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,KAAK,MAAM,CAAC;SAC9C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACf,EAAE,EAAE,KAAK,CAAC,UAAU;QACpB,IAAI,EAAE,KAAK,CAAC,KAAK;QACjB,QAAQ,EAAE,KAAc;KACzB,CAAC,CAAC,CAAC;IAEN,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,4BAA4B,CAAC;AACnE,CAAC;AAED,SAAS,mBAAmB;IAC1B,MAAM,MAAM,GAAG,eAAe,EAAE;SAC7B,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,KAAK,OAAO,CAAC;SAC/C,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACf,EAAE,EAAE,KAAK,CAAC,UAAU;QACpB,IAAI,EAAE,KAAK,CAAC,KAAK;QACjB,QAAQ,EAAE,KAAc;QACxB,cAAc,EACZ,KAAK,CAAC,cAAc,IAAI,KAAK,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC;YACrD,CAAC,CAAC,KAAK,CAAC,cAAc;YACtB,CAAC,CAAC,SAAS;KAChB,CAAC,CAAC,CAAC;IAEN,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,yBAAyB,CAAC;AAChE,CAAC;AAED,SAAS,kBAAkB,CACzB,UAAkB,EAClB,OAAyB;IAEzB,MAAM,UAAU,GAAG,gBAAgB,CAAC,UAAU,CAAC,EAAE,UAAU,CAAC;IAC5D,IAAI,CAAC,UAAU,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,QAAQ,GAAqB,EAAE,CAAC;IACtC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACnD,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,GAAG,KAAK,CAAC;IAC3C,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,eAAe,CAAC,OAAyB;IAChD,IAAI,CAAC,CAAC,cAAc,IAAI,OAAO,CAAC,EAAE,CAAC;QACjC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,QAAQ,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;IAChC,QAAQ,CAAC,MAAM,GAAG,QAAQ,CAAC,YAAY,CAAC;IACxC,OAAO,QAAQ,CAAC,YAAY,CAAC;IAC7B,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,0BAA0B,CAAC,KAAc;IAChD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACvE,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IACzC,OAAO,CACL,UAAU,CAAC,QAAQ,CAAC,0CAA0C,CAAC;QAC/D,UAAU,CAAC,QAAQ,CAAC,mCAAmC,CAAC,CACzD,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,MAA6B;IACzD,MAAM,UAAU,GAAY,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IACvE,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,UAAU,YAAY,UAAU,EAAE,CAAC;QACrC,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QAChC,OAAO,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;IACpC,CAAC;IACD,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;QACnC,MAAM,CAAC,EAAE,OAAO,CAAC,GAAG,YAAY,CAAC,UAAU,CAAC,CAAC;QAC7C,OAAO,OAAO,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAClD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,OAAO,WAAY,SAAQ,YAAY;IAE3C,MAAM,CAAU,eAAe;QAC7B,OAAO,CAAC,aAAa,CAAC,CAAC;IACzB,CAAC;IAEkB,MAAM,CAAS;IACjB,cAAc,CAAyC;IAExE,YACE,OAAiC,EACjC,UAA8B,EAAE;QAEhC,KAAK,CAAC,KAAK,CAAC,CAAC;QACb,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;QACnC,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,cAAc;YACjB,OAAO,CAAC,aAAa,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;IACjE,CAAC;IAEQ,eAAe;QACtB,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;IACtC,CAAC;IAEQ,KAAK,CAAC,cAAc,CAAC,MAAc;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,0EAA0E;IAClE,UAAU,CAAC,YAAoB;QACrC,MAAM,MAAM,GAAoB;YAC9B,YAAY;YACZ,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,kBAAkB,EAAE,YAAY;YAChC,eAAe,EAAE,IAAI;YACrB,kBAAkB,EAAE,IAAI;SACzB,CAAC;QACF,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,aAAa,CAAC,QAAmB;QAI7C,MAAM,WAAW,GAAkB,EAAE,CAAC;QACtC,IAAI,KAAyB,CAAC;QAE9B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM;gBAAE,SAAS;YAClC,MAAM,IAAI,GACR,GAAG,CAAC,IAAI,KAAK,QAAQ;gBACnB,CAAC,CAAC,QAAQ;gBACV,CAAC,CAAC,GAAG,CAAC,IAAI,KAAK,WAAW;oBACxB,CAAC,CAAC,WAAW;oBACb,CAAC,CAAC,MAAM,CAAC;YAEf,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACpC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YACxB,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACtC,MAAM,KAAK,GAAa,EAAE,CAAC;gBAC3B,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACzB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACxB,CAAC;yBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBACrC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;wBAC7B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;4BAC7B,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;gCAC9B,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC;gCACvB,CAAC,CAAC,IAAI,CAAC;wBACX,CAAC;6BAAM,IAAI,IAAI,YAAY,UAAU,EAAE,CAAC;4BACtC,KAAK,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;wBAC9B,CAAC;6BAAM,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;4BAC1B,KAAK,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBACrD,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;YAED,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QACtC,CAAC;QAED,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;IAChC,CAAC;IAEO,WAAW,CAAC,KAAqB;QACvC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IACE,IAAI,CAAC,IAAI,KAAK,kBAAkB;gBAChC,IAAI,CAAC,IAAI,KAAK,kBAAkB,EAChC,CAAC;gBACD,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;YACtC,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;oBACnC,UAAU,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;iBACnE;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,GAAY;QACjC,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;YAChB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACpD,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,IAAI,MAAM,GAAG,SAAS,CAAC;YACvB,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAY,CAAC;gBACzC,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAS;gBACX,CAAC;YACH,CAAC;YACD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC1C,SAAS;YACX,CAAC;YAED,MAAM,MAAM,GAAG,MAAiC,CAAC;YACjD,MAAM,cAAc,GAClB,MAAM,CAAC,QAAQ,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ;gBACpD,CAAC,CAAE,MAAM,CAAC,QAAoC;gBAC9C,CAAC,CAAC,IAAI,CAAC;YACX,MAAM,IAAI,GACR,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ;gBAC7B,CAAC,CAAC,MAAM,CAAC,IAAI;gBACb,CAAC,CAAC,OAAO,cAAc,EAAE,IAAI,KAAK,QAAQ;oBACxC,CAAC,CAAC,cAAc,CAAC,IAAI;oBACrB,CAAC,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GACX,MAAM,CAAC,SAAS,IAAI,cAAc,EAAE,SAAS,IAAI,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACrE,IAAI,IAAI,GAA4B,EAAE,CAAC;YACvC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAChC,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACzC,CAAC;iBAAM,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC7E,IAAI,GAAG,OAAkC,CAAC;YAC5C,CAAC;YAED,SAAS,CAAC,IAAI,CAAC;gBACb,EAAE,EACA,OAAO,MAAM,CAAC,EAAE,KAAK,QAAQ;oBAC3B,CAAC,CAAC,MAAM,CAAC,EAAE;oBACX,CAAC,CAAC,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ;wBACvC,CAAC,CAAC,MAAM,CAAC,YAAY;wBACrB,CAAC,CAAC,UAAU,EAAE;gBACpB,IAAI;gBACJ,IAAI;aACL,CAAC,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,IAOzB;QACC,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvE,MAAM,MAAM,GAAqB,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;QAC/D,IAAI,KAAK;YAAE,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;QAChC,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxC,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9C,CAAC;QACD,0EAA0E;QAC1E,gCAAgC;QAChC,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI;YAAE,MAAM,CAAC,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC;QACnE,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI;YAAE,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACpE,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI;YAAE,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC;QAChD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,GAAW;QACvC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QAC5C,IAAI,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACjC,OAAO,eAAe,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YACtE,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,CAAC;gBACvC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,OAAO,SAAS,CAAC;gBACnB,CAAC;gBACD,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,KAAK,CAAC,sCAAsC,EAAE;oBAChD,GAAG,EAAE,QAAQ;oBACb,KAAK;iBACN,CAAC,CAAC;gBACH,qEAAqE;gBACrE,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAQrB;QACC,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAE5C,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACnD,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,QAAQ,CAAC,KAAK;gBACZ,4BAA4B,QAAQ,CAAC,UAAU,IAAI,SAAS,GAAG,CAClE,CAAC;QACJ,CAAC;QACD,IAAI,OAAO,QAAQ,CAAC,oBAAoB,KAAK,QAAQ,EAAE,CAAC;YACtD,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE;gBAC1B,WAAW,EAAE,QAAQ,CAAC,aAAa,IAAI,CAAC;gBACxC,YAAY,EAAE,QAAQ,CAAC,oBAAoB;aAC5C,CAAC,CAAC;QACL,CAAC;QACD,OAAO;YACL,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,QAAQ,CAAC,IAAI,IAAI,EAAE;YAC5B,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC;SACpD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,gBAAgB,CAAC,IAQvB;QACC,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAE5C,MAAM,MAAM,GAAG,MAAM,CAAC,sBAAsB,CAAC,MAAM,CAIlD,CAAC;QAEF,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,YAAgC,CAAC;QACrC,IAAI,YAAgC,CAAC;QACrC,IAAI,eAAmC,CAAC;QACxC,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAU,CAAC;QAE7C,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,MAAM,EAAE,CAAC;YAClC,MAAM,CAAC,GAAG,MAAiC,CAAC;YAC5C,IAAI,CAAC,CAAC,OAAO,KAAK,KAAK,IAAI,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACvD,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC;gBACvB,MAAM;YACR,CAAC;YACD,MAAM,YAAY,GAAG,CAAC,CAAC,CAAC,aAAa,IAAI,CAAC,CAAC,WAAW,CAEzC,CAAC;YACd,MAAM,YAAY,GAAG,YAAY,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,eAAe,CAAC,CAAC;YAC3E,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;gBACrC,YAAY,GAAG,YAAY,CAAC;YAC9B,CAAC;YACD,MAAM,eAAe,GACnB,YAAY,EAAE,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,sBAAsB,CAAC,CAAC;YACtE,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;gBACxC,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,eAAe,IAAI,CAAC,EAAE,eAAe,CAAC,CAAC;YACpE,CAAC;YACD,MAAM,IAAI,GACR,OAAO,YAAY,EAAE,CAAC,MAAM,CAAC,KAAK,QAAQ;gBACxC,CAAC,CAAE,YAAY,CAAC,MAAM,CAAY;gBAClC,CAAC,CAAC,SAAS,CAAC;YAChB,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;gBACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACzC,OAAO,GAAG,IAAI,CAAC;gBACf,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAW,CAAC;YAChE,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CACnC,YAAY,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAChD,CAAC;YACF,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBACjC,IAAI,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;oBACxC,SAAS;gBACX,CAAC;gBACD,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACpC,MAAM,QAAQ,CAAC;YACjB,CAAC;QACH,CAAC;QAED,IAAI,YAAY;YAAE,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAChD,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;YACxC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE;gBAC1B,WAAW,EAAE,YAAY,IAAI,CAAC;gBAC9B,YAAY,EAAE,eAAe;aAC9B,CAAC,CAAC;QACL,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAW,CAAC;IAC5D,CAAC;IAEO,iBAAiB,CACvB,MAAc,EACd,MAEsB,EACtB,KAAkB;QAElB,MAAM,OAAO,GAAqB;YAChC,YAAY,EAAE,MAAM;SACrB,CAAC;QACF,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,CAAC,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,KAAK,GACT,OAAO,IAAI,MAAM;YACf,CAAC,CAAC,MAAM,CAAC,KAAK;YACd,CAAC,CAAE,MAA6B,CAAC,WAAW,CAAC;QACjD,MAAM,MAAM,GACV,QAAQ,IAAI,MAAM;YAChB,CAAC,CAAC,MAAM,CAAC,MAAM;YACf,CAAC,CAAE,MAA6B,CAAC,YAAY,CAAC;QAClD,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;YAAA,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;QAAA,CAAC;QAC3C,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;YAAA,OAAO,CAAC,MAAM,GAAG,MAAM,CAAC;QAAA,CAAC;QAC9C,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;YAAA,OAAO,CAAC,eAAe,GAAG,MAAM,CAAC,cAAc,CAAC;QAAA,CAAC;QAC7E,IAAI,MAAM,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;YAAA,OAAO,CAAC,cAAc,GAAG,MAAM,CAAC,aAAa,CAAC;QAAA,CAAC;QAClF,IAAI,MAAM,CAAC,iBAAiB,IAAI,IAAI,EAAE,CAAC;YACrC,OAAO,CAAC,mBAAmB,GAAG,MAAM,CAAC,iBAAiB,CAAC;QACzD,CAAC;QACD,IAAI,MAAM,CAAC,IAAI,IAAI,IAAI,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,CAAC,EAAE,CAAC;YAAA,OAAO,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QAAA,CAAC;QAC5E,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YAAA,OAAO,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAAA,CAAC;QAC7D,IAAI,aAAa,IAAI,MAAM,IAAI,MAAM,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;YAC1D,OAAO,CAAC,YAAY,GAAG,MAAM,CAAC,WAAW,CAAC;QAC5C,CAAC;QACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YAAA,OAAO,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAAA,CAAC;QACvD,IAAI,UAAU,IAAI,MAAM,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;YACpD,OAAO,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACrC,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,OAAe,EACf,OAAyB;QAEzB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAExC,IAAI,QAAqB,CAAC;QAC1B,MAAM,cAAc,GAAG,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC5D,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvC,MAAM,KAAK,CAAC;YACd,CAAC;YAED,GAAG,CAAC,IAAI,CAAC,oDAAoD,EAAE;gBAC7D,OAAO;aACR,CAAC,CAAC;YACH,QAAQ,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,QAAQ,CAAC,KAAK;gBACZ,kCAAkC,QAAQ,CAAC,UAAU,IAAI,SAAS,GAAG,CACxE,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,oBAAoB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACpD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEQ,KAAK,CAAC,WAAW,CAAC,MAAyB;QAClD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACpC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACxD,CAAC;IAEQ,KAAK,CAAC,YAAY,CACzB,KAAiB,EACjB,MAA0B;QAE1B,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACpC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACxD,CAAC;IAEQ,KAAK,CAAC,0BAA0B;QACvC,OAAO,sBAAsB,EAAE,CAAC;IAClC,CAAC;IAEQ,KAAK,CAAC,uBAAuB;QACpC,OAAO,mBAAmB,EAAE,CAAC;IAC/B,CAAC;CACF"}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
import Anthropic from "@anthropic-ai/sdk";
|
|
2
|
+
import { BaseProvider } from "./base-provider.js";
|
|
3
|
+
import type { LanguageModel, Message, ProviderStreamItem, ProviderTool } from "./types.js";
|
|
4
|
+
interface AnthropicProviderOptions {
|
|
5
|
+
client?: Anthropic;
|
|
6
|
+
clientFactory?: (apiKey: string) => Anthropic;
|
|
7
|
+
fetchFn?: typeof fetch;
|
|
8
|
+
}
|
|
9
|
+
export declare class AnthropicProvider extends BaseProvider {
|
|
10
|
+
static requiredSecrets(): string[];
|
|
11
|
+
readonly apiKey: string;
|
|
12
|
+
private _client;
|
|
13
|
+
private _clientFactory;
|
|
14
|
+
private _fetch;
|
|
15
|
+
constructor(secrets: {
|
|
16
|
+
ANTHROPIC_API_KEY?: string;
|
|
17
|
+
}, options?: AnthropicProviderOptions);
|
|
18
|
+
getContainerEnv(): Record<string, string>;
|
|
19
|
+
getClient(): Anthropic;
|
|
20
|
+
hasToolSupport(_model: string): Promise<boolean>;
|
|
21
|
+
getAvailableLanguageModels(): Promise<LanguageModel[]>;
|
|
22
|
+
private prepareJsonSchema;
|
|
23
|
+
convertMessage(message: Message): Promise<Record<string, unknown> | null>;
|
|
24
|
+
formatTools(tools: ProviderTool[]): Array<Record<string, unknown>>;
|
|
25
|
+
private extractSystemMessage;
|
|
26
|
+
generateMessages(args: {
|
|
27
|
+
messages: Message[];
|
|
28
|
+
model: string;
|
|
29
|
+
tools?: ProviderTool[];
|
|
30
|
+
toolChoice?: string | "any";
|
|
31
|
+
maxTokens?: number;
|
|
32
|
+
temperature?: number;
|
|
33
|
+
topP?: number;
|
|
34
|
+
presencePenalty?: number;
|
|
35
|
+
frequencyPenalty?: number;
|
|
36
|
+
audio?: Record<string, unknown>;
|
|
37
|
+
thinkingBudget?: number;
|
|
38
|
+
}): AsyncGenerator<ProviderStreamItem>;
|
|
39
|
+
generateMessage(args: {
|
|
40
|
+
messages: Message[];
|
|
41
|
+
model: string;
|
|
42
|
+
tools?: ProviderTool[];
|
|
43
|
+
toolChoice?: string | "any";
|
|
44
|
+
maxTokens?: number;
|
|
45
|
+
temperature?: number;
|
|
46
|
+
topP?: number;
|
|
47
|
+
presencePenalty?: number;
|
|
48
|
+
frequencyPenalty?: number;
|
|
49
|
+
thinkingBudget?: number;
|
|
50
|
+
}): Promise<Message>;
|
|
51
|
+
asHttpStatusError(error: unknown): Error;
|
|
52
|
+
isContextLengthError(error: unknown): boolean;
|
|
53
|
+
}
|
|
54
|
+
export {};
|
|
55
|
+
//# sourceMappingURL=anthropic-provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic-provider.d.ts","sourceRoot":"","sources":["../../src/providers/anthropic-provider.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAO1C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGlD,OAAO,KAAK,EACV,aAAa,EACb,OAAO,EAIP,kBAAkB,EAClB,YAAY,EAEb,MAAM,YAAY,CAAC;AAEpB,UAAU,wBAAwB;IAChC,MAAM,CAAC,EAAE,SAAS,CAAC;IACnB,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,SAAS,CAAC;IAC9C,OAAO,CAAC,EAAE,OAAO,KAAK,CAAC;CACxB;AAwCD,qBAAa,iBAAkB,SAAQ,YAAY;IACjD,MAAM,CAAC,eAAe,IAAI,MAAM,EAAE;IAIlC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,OAAO,CAAC,OAAO,CAAmB;IAClC,OAAO,CAAC,cAAc,CAAgC;IACtD,OAAO,CAAC,MAAM,CAAe;gBAG3B,OAAO,EAAE;QAAE,iBAAiB,CAAC,EAAE,MAAM,CAAA;KAAE,EACvC,OAAO,GAAE,wBAA6B;IAoBxC,eAAe,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAIzC,SAAS,IAAI,SAAS;IAOhB,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAIhD,0BAA0B,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;IAyF5D,OAAO,CAAC,iBAAiB;IA8DnB,cAAc,CAClB,OAAO,EAAE,OAAO,GACf,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAoK1C,WAAW,CAAC,KAAK,EAAE,YAAY,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAWlE,OAAO,CAAC,oBAAoB;IAoBrB,gBAAgB,CAAC,IAAI,EAAE;QAC5B,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,YAAY,EAAE,CAAC;QACvB,UAAU,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;QAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChC,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB,GAAG,cAAc,CAAC,kBAAkB,CAAC;IAsKhC,eAAe,CAAC,IAAI,EAAE;QAC1B,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,YAAY,EAAE,CAAC;QACvB,UAAU,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;QAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB,GAAG,OAAO,CAAC,OAAO,CAAC;IA2EpB,iBAAiB,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK;IAIxC,oBAAoB,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO;CAU9C"}
|