@jxburros/ai-nugget 0.4.1 → 0.7.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.
Files changed (187) hide show
  1. package/README.md +158 -382
  2. package/UPGRADING.md +182 -0
  3. package/dist/adapters/engines/anthropic.d.ts +7 -1
  4. package/dist/adapters/engines/anthropic.js +82 -37
  5. package/dist/adapters/engines/base.d.ts +29 -2
  6. package/dist/adapters/engines/base.js +52 -8
  7. package/dist/adapters/engines/google.d.ts +7 -1
  8. package/dist/adapters/engines/google.js +80 -29
  9. package/dist/adapters/engines/ollama.d.ts +2 -1
  10. package/dist/adapters/engines/ollama.js +57 -15
  11. package/dist/adapters/engines/openaiChat.d.ts +8 -1
  12. package/dist/adapters/engines/openaiChat.js +109 -29
  13. package/dist/adapters/profiles.d.ts +21 -1
  14. package/dist/adapters/profiles.js +12 -2
  15. package/dist/agent/loop.d.ts +45 -1
  16. package/dist/agent/loop.js +117 -12
  17. package/dist/cjs/adapters/engines/anthropic.js +326 -0
  18. package/dist/cjs/adapters/engines/base.js +147 -0
  19. package/dist/cjs/adapters/engines/google.js +291 -0
  20. package/dist/cjs/adapters/engines/ollama.js +205 -0
  21. package/dist/cjs/adapters/engines/openaiChat.js +312 -0
  22. package/dist/cjs/adapters/index.js +22 -0
  23. package/dist/cjs/adapters/profiles.js +125 -0
  24. package/dist/cjs/agent/index.js +18 -0
  25. package/dist/cjs/agent/loop.js +373 -0
  26. package/dist/cjs/agent/tools.js +51 -0
  27. package/dist/cjs/connect.js +27 -0
  28. package/dist/cjs/errors.js +110 -0
  29. package/dist/cjs/handler.js +586 -0
  30. package/dist/cjs/index.js +28 -0
  31. package/dist/cjs/json.js +121 -0
  32. package/dist/cjs/keys.js +81 -0
  33. package/dist/cjs/package.json +3 -0
  34. package/dist/cjs/policy.js +42 -0
  35. package/dist/cjs/reasoning.js +159 -0
  36. package/dist/cjs/redact.js +78 -0
  37. package/dist/cjs/tokens.js +22 -0
  38. package/dist/cjs/transport.js +172 -0
  39. package/dist/cjs/types.js +6 -0
  40. package/dist/cjs/util.js +126 -0
  41. package/dist/errors.js +25 -1
  42. package/dist/handler.d.ts +67 -1
  43. package/dist/handler.js +266 -61
  44. package/dist/index.d.ts +1 -0
  45. package/dist/index.js +1 -0
  46. package/dist/reasoning.d.ts +45 -0
  47. package/dist/reasoning.js +154 -0
  48. package/dist/redact.d.ts +18 -0
  49. package/dist/redact.js +29 -1
  50. package/dist/transport.d.ts +3 -1
  51. package/dist/transport.js +21 -1
  52. package/dist/types.d.ts +112 -2
  53. package/dist/util.d.ts +22 -0
  54. package/dist/util.js +59 -0
  55. package/nugget/VERSION.txt +2 -2
  56. package/nugget/dist/adapters/engines/anthropic.d.ts +14 -0
  57. package/nugget/dist/adapters/engines/anthropic.d.ts.map +1 -0
  58. package/nugget/dist/adapters/engines/anthropic.js +323 -0
  59. package/nugget/dist/adapters/engines/anthropic.js.map +1 -0
  60. package/nugget/dist/adapters/engines/base.d.ts +50 -0
  61. package/nugget/dist/adapters/engines/base.d.ts.map +1 -0
  62. package/nugget/dist/adapters/engines/base.js +137 -0
  63. package/nugget/dist/adapters/engines/base.js.map +1 -0
  64. package/nugget/dist/adapters/engines/google.d.ts +24 -0
  65. package/nugget/dist/adapters/engines/google.d.ts.map +1 -0
  66. package/nugget/dist/adapters/engines/google.js +288 -0
  67. package/nugget/dist/adapters/engines/google.js.map +1 -0
  68. package/nugget/dist/adapters/engines/ollama.d.ts +11 -0
  69. package/nugget/dist/adapters/engines/ollama.d.ts.map +1 -0
  70. package/nugget/dist/adapters/engines/ollama.js +202 -0
  71. package/nugget/dist/adapters/engines/ollama.js.map +1 -0
  72. package/nugget/dist/adapters/engines/openaiChat.d.ts +22 -0
  73. package/nugget/dist/adapters/engines/openaiChat.d.ts.map +1 -0
  74. package/nugget/dist/adapters/engines/openaiChat.js +308 -0
  75. package/nugget/dist/adapters/engines/openaiChat.js.map +1 -0
  76. package/nugget/dist/adapters/index.d.ts +5 -0
  77. package/nugget/dist/adapters/index.d.ts.map +1 -0
  78. package/nugget/dist/adapters/index.js +17 -0
  79. package/nugget/dist/adapters/index.js.map +1 -0
  80. package/nugget/dist/adapters/profiles.d.ts +62 -0
  81. package/nugget/dist/adapters/profiles.d.ts.map +1 -0
  82. package/nugget/dist/adapters/profiles.js +121 -0
  83. package/nugget/dist/adapters/profiles.js.map +1 -0
  84. package/nugget/dist/agent/index.d.ts +3 -0
  85. package/nugget/dist/agent/index.d.ts.map +1 -0
  86. package/nugget/dist/agent/index.js +3 -0
  87. package/nugget/dist/agent/index.js.map +1 -0
  88. package/nugget/dist/agent/loop.d.ts +114 -0
  89. package/nugget/dist/agent/loop.d.ts.map +1 -0
  90. package/nugget/dist/agent/loop.js +371 -0
  91. package/nugget/dist/agent/loop.js.map +1 -0
  92. package/nugget/dist/agent/tools.d.ts +29 -0
  93. package/nugget/dist/agent/tools.d.ts.map +1 -0
  94. package/nugget/dist/agent/tools.js +48 -0
  95. package/nugget/dist/agent/tools.js.map +1 -0
  96. package/nugget/dist/cjs/adapters/engines/anthropic.js +326 -0
  97. package/nugget/dist/cjs/adapters/engines/base.js +147 -0
  98. package/nugget/dist/cjs/adapters/engines/google.js +291 -0
  99. package/nugget/dist/cjs/adapters/engines/ollama.js +205 -0
  100. package/nugget/dist/cjs/adapters/engines/openaiChat.js +312 -0
  101. package/nugget/dist/cjs/adapters/index.js +22 -0
  102. package/nugget/dist/cjs/adapters/profiles.js +125 -0
  103. package/nugget/dist/cjs/agent/index.js +18 -0
  104. package/nugget/dist/cjs/agent/loop.js +373 -0
  105. package/nugget/dist/cjs/agent/tools.js +51 -0
  106. package/nugget/dist/cjs/connect.js +27 -0
  107. package/nugget/dist/cjs/errors.js +110 -0
  108. package/nugget/dist/cjs/handler.js +586 -0
  109. package/nugget/dist/cjs/index.js +28 -0
  110. package/nugget/dist/cjs/json.js +121 -0
  111. package/nugget/dist/cjs/keys.js +81 -0
  112. package/nugget/dist/cjs/package.json +3 -0
  113. package/nugget/dist/cjs/policy.js +42 -0
  114. package/nugget/dist/cjs/reasoning.js +159 -0
  115. package/nugget/dist/cjs/redact.js +78 -0
  116. package/nugget/dist/cjs/tokens.js +22 -0
  117. package/nugget/dist/cjs/transport.js +172 -0
  118. package/nugget/dist/cjs/types.js +6 -0
  119. package/nugget/dist/cjs/util.js +126 -0
  120. package/nugget/dist/connect.d.ts +31 -0
  121. package/nugget/dist/connect.d.ts.map +1 -0
  122. package/nugget/dist/connect.js +25 -0
  123. package/nugget/dist/connect.js.map +1 -0
  124. package/nugget/dist/errors.d.ts +22 -0
  125. package/nugget/dist/errors.d.ts.map +1 -0
  126. package/nugget/dist/errors.js +104 -0
  127. package/nugget/dist/errors.js.map +1 -0
  128. package/nugget/dist/handler.d.ts +114 -0
  129. package/nugget/dist/handler.d.ts.map +1 -0
  130. package/nugget/dist/handler.js +582 -0
  131. package/nugget/dist/handler.js.map +1 -0
  132. package/nugget/dist/index.d.ts +13 -0
  133. package/nugget/dist/index.d.ts.map +1 -0
  134. package/nugget/dist/index.js +13 -0
  135. package/nugget/dist/index.js.map +1 -0
  136. package/nugget/dist/json.d.ts +7 -0
  137. package/nugget/dist/json.d.ts.map +1 -0
  138. package/nugget/dist/json.js +114 -0
  139. package/nugget/dist/json.js.map +1 -0
  140. package/nugget/dist/keys.d.ts +7 -0
  141. package/nugget/dist/keys.d.ts.map +1 -0
  142. package/nugget/dist/keys.js +75 -0
  143. package/nugget/dist/keys.js.map +1 -0
  144. package/nugget/dist/policy.d.ts +6 -0
  145. package/nugget/dist/policy.d.ts.map +1 -0
  146. package/nugget/dist/policy.js +37 -0
  147. package/nugget/dist/policy.js.map +1 -0
  148. package/nugget/dist/reasoning.d.ts +45 -0
  149. package/nugget/dist/reasoning.d.ts.map +1 -0
  150. package/nugget/dist/reasoning.js +154 -0
  151. package/nugget/dist/reasoning.js.map +1 -0
  152. package/nugget/dist/redact.d.ts +26 -0
  153. package/nugget/dist/redact.d.ts.map +1 -0
  154. package/nugget/dist/redact.js +74 -0
  155. package/nugget/dist/redact.js.map +1 -0
  156. package/nugget/dist/tokens.d.ts +5 -0
  157. package/nugget/dist/tokens.d.ts.map +1 -0
  158. package/nugget/dist/tokens.js +18 -0
  159. package/nugget/dist/tokens.js.map +1 -0
  160. package/nugget/dist/transport.d.ts +30 -0
  161. package/nugget/dist/transport.d.ts.map +1 -0
  162. package/nugget/dist/transport.js +165 -0
  163. package/nugget/dist/transport.js.map +1 -0
  164. package/nugget/dist/types.d.ts +301 -0
  165. package/nugget/dist/types.d.ts.map +1 -0
  166. package/nugget/dist/types.js +4 -0
  167. package/nugget/dist/types.js.map +1 -0
  168. package/nugget/dist/util.d.ts +31 -0
  169. package/nugget/dist/util.d.ts.map +1 -0
  170. package/nugget/dist/util.js +114 -0
  171. package/nugget/dist/util.js.map +1 -0
  172. package/nugget/src/adapters/engines/anthropic.ts +78 -37
  173. package/nugget/src/adapters/engines/base.ts +50 -7
  174. package/nugget/src/adapters/engines/google.ts +79 -29
  175. package/nugget/src/adapters/engines/ollama.ts +56 -17
  176. package/nugget/src/adapters/engines/openaiChat.ts +104 -30
  177. package/nugget/src/adapters/profiles.ts +53 -3
  178. package/nugget/src/agent/loop.ts +156 -13
  179. package/nugget/src/errors.ts +25 -1
  180. package/nugget/src/handler.ts +308 -59
  181. package/nugget/src/index.ts +1 -0
  182. package/nugget/src/reasoning.ts +174 -0
  183. package/nugget/src/redact.ts +32 -1
  184. package/nugget/src/transport.ts +20 -1
  185. package/nugget/src/types.ts +113 -1
  186. package/nugget/src/util.ts +64 -0
  187. package/package.json +18 -5
@@ -0,0 +1,288 @@
1
+ import { AIError } from '../../errors.js';
2
+ import { estimatedUsage } from '../../tokens.js';
3
+ import { fetchJson, postResponse, sseLines } from '../../transport.js';
4
+ import { applyProviderOptions, REASONING_BUDGET_TOKENS, asNumber, asRecord, asString, joinUrl, textFromMessages } from '../../util.js';
5
+ import { DEFAULT_TIMEOUT_MS, randomId, safeParse, streamAnomaly, streamError, streamTimeout } from './base.js';
6
+ export class GoogleAdapter {
7
+ provider;
8
+ constructor(provider) {
9
+ this.provider = provider;
10
+ }
11
+ async chat(conn, req) {
12
+ let result;
13
+ for await (const event of this.stream(conn, req)) {
14
+ if (event.type === 'done')
15
+ result = event.result;
16
+ if (event.type === 'error')
17
+ throw event.error;
18
+ }
19
+ if (!result)
20
+ throw new AIError('Google stream ended without a result', { kind: 'invalid_response', provider: conn.provider });
21
+ return result;
22
+ }
23
+ async *stream(conn, req) {
24
+ const started = Date.now();
25
+ let firstTokenMs = null;
26
+ let text = '';
27
+ let inputTokens;
28
+ let outputTokens;
29
+ let finish;
30
+ const toolCalls = [];
31
+ const timeout = streamTimeout(conn, req.signal);
32
+ yield { type: 'start', callId: '', provider: conn.provider, model: req.model };
33
+ // Gemini rejects `responseMimeType: application/json` combined with function
34
+ // declarations, so `body()` drops JSON mode when tools are present. Surface
35
+ // that downgrade instead of silently returning unstructured text.
36
+ if (jsonModeDowngraded(req)) {
37
+ yield {
38
+ type: 'context',
39
+ kind: 'json_mode_downgraded',
40
+ data: { reason: 'Gemini does not accept JSON response mode together with tools; the request was sent without JSON mode', provider: conn.provider },
41
+ };
42
+ }
43
+ try {
44
+ const streamUrl = `${conn.baseUrl}/v1beta/models/${encodeURIComponent(req.model)}:streamGenerateContent?alt=sse`;
45
+ const res = await postResponse(streamUrl, body(req), conn.headers, timeout.signal, conn.provider);
46
+ const contentType = res.headers.get('content-type') ?? '';
47
+ const chunks = contentType.includes('text/event-stream') ? sseLines(res) : singleJsonLine(res);
48
+ for await (const line of chunks) {
49
+ timeout.bump();
50
+ const parsed = safeParse(line);
51
+ const record = asRecord(parsed);
52
+ const promptFeedback = asRecord(record?.promptFeedback);
53
+ if (!firstCandidate(record) && promptFeedback?.blockReason)
54
+ finish = 'SAFETY';
55
+ for (const part of candidateParts(record)) {
56
+ const partText = asString(asRecord(part)?.text);
57
+ if (partText) {
58
+ // Gemini marks reasoning parts with `thought: true`; route those to
59
+ // the reasoning channel so they don't blend into the answer text.
60
+ if (asRecord(part)?.thought === true) {
61
+ yield { type: 'reasoning', text: partText };
62
+ continue;
63
+ }
64
+ if (firstTokenMs === null)
65
+ firstTokenMs = Date.now() - started;
66
+ text += partText;
67
+ yield { type: 'delta', text: partText };
68
+ }
69
+ const fnCall = asRecord(asRecord(part)?.functionCall);
70
+ if (fnCall) {
71
+ const call = {
72
+ id: randomId(),
73
+ name: asString(fnCall.name) ?? 'unknown',
74
+ arguments: fnCall.args ?? {},
75
+ raw: JSON.stringify(fnCall.args ?? {}),
76
+ };
77
+ toolCalls.push(call);
78
+ yield { type: 'tool_call', call };
79
+ }
80
+ }
81
+ finish = asString(asRecord(firstCandidate(record))?.finishReason) ?? finish;
82
+ const usage = asRecord(record?.usageMetadata);
83
+ inputTokens = asNumber(usage?.promptTokenCount) ?? inputTokens;
84
+ outputTokens = asNumber(usage?.candidatesTokenCount) ?? outputTokens;
85
+ }
86
+ if (!finish)
87
+ yield streamAnomaly('stream ended without a finishReason');
88
+ const hasTools = toolCalls.length > 0;
89
+ yield { type: 'done', result: {
90
+ text,
91
+ toolCalls: hasTools ? toolCalls : undefined,
92
+ finishReason: mapFinish(finish, hasTools),
93
+ usage: inputTokens !== undefined || outputTokens !== undefined ? { inputTokens, outputTokens, estimated: false } : estimatedUsage(textFromMessages(req.messages), text),
94
+ timing: { firstTokenMs, totalMs: Date.now() - started },
95
+ model: req.model,
96
+ source: { provider: conn.provider, connectionId: conn.id, baseUrl: conn.baseUrl },
97
+ } };
98
+ }
99
+ catch (error) {
100
+ throw streamError(error, timeout, conn.provider);
101
+ }
102
+ finally {
103
+ timeout.done();
104
+ }
105
+ }
106
+ /**
107
+ * `GoogleAdapter` has no `listModels`, so without a `health` probe
108
+ * `AIHandler.testConnection` would report `ok: true` without ever making a
109
+ * network call. `/v1beta/models` is a lightweight, key-authenticated GET
110
+ * that gives an honest connectivity/auth check.
111
+ */
112
+ async health(conn) {
113
+ try {
114
+ await fetchJson(joinUrl(conn.baseUrl, '/v1beta/models'), {
115
+ method: 'GET',
116
+ headers: conn.headers,
117
+ timeoutMs: Math.min(conn.timeoutMs ?? DEFAULT_TIMEOUT_MS, 10_000),
118
+ provider: conn.provider,
119
+ });
120
+ return { ok: true };
121
+ }
122
+ catch (error) {
123
+ return { ok: false, detail: error instanceof Error ? error.message : 'Health check failed' };
124
+ }
125
+ }
126
+ /**
127
+ * Lists models from Google's `/v1beta/models`. Strips the `models/` name
128
+ * prefix, and maps `inputTokenLimit` → `contextWindow` and
129
+ * `supportedGenerationMethods` → `capabilities`.
130
+ */
131
+ async listModels(conn) {
132
+ const { data } = await fetchJson(joinUrl(conn.baseUrl, '/v1beta/models'), {
133
+ method: 'GET',
134
+ headers: conn.headers,
135
+ timeoutMs: Math.min(conn.timeoutMs ?? DEFAULT_TIMEOUT_MS, 15_000),
136
+ provider: conn.provider,
137
+ });
138
+ const rows = Array.isArray(asRecord(data)?.models) ? asRecord(data).models : [];
139
+ const models = [];
140
+ for (const value of rows) {
141
+ const row = asRecord(value);
142
+ const name = asString(row?.name);
143
+ if (!name)
144
+ continue;
145
+ const id = name.startsWith('models/') ? name.slice('models/'.length) : name;
146
+ const model = { id, source: { provider: conn.provider, connectionId: conn.id, baseUrl: conn.baseUrl } };
147
+ const contextWindow = asNumber(row?.inputTokenLimit);
148
+ if (contextWindow !== undefined)
149
+ model.contextWindow = contextWindow;
150
+ const methods = row?.supportedGenerationMethods;
151
+ if (Array.isArray(methods) && methods.every((m) => typeof m === 'string'))
152
+ model.capabilities = methods;
153
+ models.push(model);
154
+ }
155
+ return models;
156
+ }
157
+ }
158
+ /** True when the caller asked for JSON mode but tools force it off. */
159
+ function jsonModeDowngraded(req) {
160
+ return req.responseFormat?.type === 'json' && !!req.tools?.length;
161
+ }
162
+ function body(req) {
163
+ const systemText = req.messages.filter((m) => m.role === 'system').map((m) => textContent(m.content)).join('\n\n');
164
+ const jsonMode = req.responseFormat?.type === 'json' && !req.tools?.length;
165
+ const responseSchema = req.responseFormat?.type === 'json' ? req.responseFormat.schema : undefined;
166
+ const payload = {
167
+ contents: toGoogleContents(req.messages.filter((m) => m.role !== 'system')),
168
+ generationConfig: {
169
+ temperature: req.temperature,
170
+ topP: req.topP,
171
+ maxOutputTokens: req.maxTokens,
172
+ stopSequences: req.stopSequences,
173
+ responseMimeType: jsonMode ? 'application/json' : undefined,
174
+ responseSchema: jsonMode ? responseSchema : undefined,
175
+ },
176
+ };
177
+ if (systemText)
178
+ payload.systemInstruction = { parts: [{ text: systemText }] };
179
+ if (req.tools?.length) {
180
+ payload.tools = [{ functionDeclarations: req.tools.map((tool) => ({ name: tool.name, description: tool.description, parameters: tool.parameters })) }];
181
+ const mode = req.toolChoice === 'none' ? 'NONE' : typeof req.toolChoice === 'object' ? 'ANY' : 'AUTO';
182
+ const config = { mode };
183
+ if (typeof req.toolChoice === 'object')
184
+ config.allowedFunctionNames = [req.toolChoice.name];
185
+ payload.toolConfig = { functionCallingConfig: config };
186
+ }
187
+ if (req.reasoningEffort !== undefined) {
188
+ // First-class reasoning effort → Gemini's thinking budget. 0 disables
189
+ // thinking on models that allow it; others treat it as a minimum.
190
+ payload.generationConfig.thinkingConfig = {
191
+ thinkingBudget: REASONING_BUDGET_TOKENS[req.reasoningEffort],
192
+ };
193
+ }
194
+ // providerOptions reaches Google-native fields: top-level `safetySettings`,
195
+ // `cachedContent`, and `generationConfig` extras (`thinkingConfig`,
196
+ // `responseModalities`, …) merged one level deep into generationConfig.
197
+ return applyProviderOptions(payload, req.providerOptions, ['generationConfig']);
198
+ }
199
+ function toGoogleContents(messages) {
200
+ const contents = [];
201
+ for (let index = 0; index < messages.length; index += 1) {
202
+ const message = messages[index];
203
+ if (message.role !== 'tool') {
204
+ contents.push(toGoogleContent(message));
205
+ continue;
206
+ }
207
+ const parts = [];
208
+ while (index < messages.length && messages[index]?.role === 'tool') {
209
+ const toolMessage = messages[index];
210
+ parts.push({ functionResponse: { name: toolMessage.name ?? 'unknown', response: toolResponseObject(toolMessage) } });
211
+ index += 1;
212
+ }
213
+ index -= 1;
214
+ contents.push({ role: 'user', parts });
215
+ }
216
+ return contents;
217
+ }
218
+ function toGoogleContent(m) {
219
+ // Tool-role messages are handled by toGoogleContents' batching loop and never
220
+ // reach here, so no tool branch is needed.
221
+ if (m.role === 'assistant' && m.toolCalls?.length) {
222
+ const parts = [...googleParts(m.content)];
223
+ for (const call of m.toolCalls)
224
+ parts.push({ functionCall: { name: call.name, args: call.arguments ?? {} } });
225
+ return { role: 'model', parts };
226
+ }
227
+ return {
228
+ role: m.role === 'assistant' ? 'model' : 'user',
229
+ parts: googleParts(m.content),
230
+ };
231
+ }
232
+ /** Maps text/image content — a string or a ContentPart[] — onto Google parts. */
233
+ function googleParts(content) {
234
+ if (typeof content === 'string')
235
+ return content ? [{ text: content }] : [];
236
+ return content.map((part) => part.type === 'image'
237
+ ? { inlineData: { mimeType: part.mimeType ?? 'image/png', data: part.imageBase64 ?? '' } }
238
+ : { text: part.text ?? '' });
239
+ }
240
+ function textContent(content) {
241
+ if (typeof content === 'string')
242
+ return content;
243
+ if (!content)
244
+ return '';
245
+ return content.filter((part) => part.type === 'text').map((part) => part.text ?? '').join('\n');
246
+ }
247
+ function toolResponseObject(message) {
248
+ if (typeof message.content !== 'string')
249
+ return { content: textContent(message.content) };
250
+ try {
251
+ const parsed = JSON.parse(message.content);
252
+ if (parsed !== null && typeof parsed === 'object' && !Array.isArray(parsed))
253
+ return parsed;
254
+ }
255
+ catch {
256
+ // keep string fallback
257
+ }
258
+ return { content: message.content };
259
+ }
260
+ async function* singleJsonLine(res) {
261
+ const data = await res.json();
262
+ // Non-SSE responses may be a single object or an array of streamed chunks.
263
+ if (Array.isArray(data)) {
264
+ for (const item of data)
265
+ yield JSON.stringify(item);
266
+ }
267
+ else {
268
+ yield JSON.stringify(data);
269
+ }
270
+ }
271
+ function firstCandidate(record) {
272
+ const candidates = Array.isArray(record?.candidates) ? record.candidates : [];
273
+ return candidates[0];
274
+ }
275
+ function candidateParts(record) {
276
+ const content = asRecord(asRecord(firstCandidate(record))?.content);
277
+ return Array.isArray(content?.parts) ? content.parts : [];
278
+ }
279
+ function mapFinish(finish, hasTools) {
280
+ if (hasTools)
281
+ return 'tool';
282
+ if (finish === 'MAX_TOKENS')
283
+ return 'length';
284
+ if (finish === 'SAFETY' || finish === 'RECITATION' || finish === 'BLOCKLIST' || finish === 'PROHIBITED_CONTENT')
285
+ return 'content_filter';
286
+ return 'stop';
287
+ }
288
+ //# sourceMappingURL=google.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"google.js","sourceRoot":"","sources":["../../../src/adapters/engines/google.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAC1C,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAEvE,OAAO,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACvI,OAAO,EAAE,kBAAkB,EAAE,QAAQ,EAAE,SAAS,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAE/G,MAAM,OAAO,aAAa;IACf,QAAQ,CAAS;IAC1B,YAAY,QAAgB;QAC1B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,IAAwB,EAAE,GAAgB;QACnD,IAAI,MAA8B,CAAC;QACnC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;YACjD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;gBAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;YACjD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;gBAAE,MAAM,KAAK,CAAC,KAAK,CAAC;QAChD,CAAC;QACD,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,OAAO,CAAC,sCAAsC,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9H,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,IAAwB,EAAE,GAAgB;QACtD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3B,IAAI,YAAY,GAAkB,IAAI,CAAC;QACvC,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,WAA+B,CAAC;QACpC,IAAI,YAAgC,CAAC;QACrC,IAAI,MAA0B,CAAC;QAC/B,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC;QAC/E,6EAA6E;QAC7E,4EAA4E;QAC5E,kEAAkE;QAClE,IAAI,kBAAkB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM;gBACJ,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,sBAAsB;gBAC5B,IAAI,EAAE,EAAE,MAAM,EAAE,uGAAuG,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE;aACnJ,CAAC;QACJ,CAAC;QACD,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,GAAG,IAAI,CAAC,OAAO,kBAAkB,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,gCAAgC,CAAC;YACjH,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClG,MAAM,WAAW,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;YAC1D,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;YAC/F,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;gBAChC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACf,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;gBAC/B,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;gBAChC,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;gBACxD,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,cAAc,EAAE,WAAW;oBAAE,MAAM,GAAG,QAAQ,CAAC;gBAC9E,KAAK,MAAM,IAAI,IAAI,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC1C,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;oBAChD,IAAI,QAAQ,EAAE,CAAC;wBACb,oEAAoE;wBACpE,kEAAkE;wBAClE,IAAI,QAAQ,CAAC,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,EAAE,CAAC;4BACrC,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;4BAC5C,SAAS;wBACX,CAAC;wBACD,IAAI,YAAY,KAAK,IAAI;4BAAE,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;wBAC/D,IAAI,IAAI,QAAQ,CAAC;wBACjB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;oBAC1C,CAAC;oBACD,MAAM,MAAM,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,YAAY,CAAC,CAAC;oBACtD,IAAI,MAAM,EAAE,CAAC;wBACX,MAAM,IAAI,GAAa;4BACrB,EAAE,EAAE,QAAQ,EAAE;4BACd,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,SAAS;4BACxC,SAAS,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;4BAC5B,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;yBACvC,CAAC;wBACF,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBACrB,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;oBACpC,CAAC;gBACH,CAAC;gBACD,MAAM,GAAG,QAAQ,CAAC,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC;gBAC5E,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;gBAC9C,WAAW,GAAG,QAAQ,CAAC,KAAK,EAAE,gBAAgB,CAAC,IAAI,WAAW,CAAC;gBAC/D,YAAY,GAAG,QAAQ,CAAC,KAAK,EAAE,oBAAoB,CAAC,IAAI,YAAY,CAAC;YACvE,CAAC;YACD,IAAI,CAAC,MAAM;gBAAE,MAAM,aAAa,CAAC,qCAAqC,CAAC,CAAC;YACxE,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;YACtC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE;oBAC5B,IAAI;oBACJ,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;oBAC3C,YAAY,EAAE,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC;oBACzC,KAAK,EAAE,WAAW,KAAK,SAAS,IAAI,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,YAAY,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC;oBACvK,MAAM,EAAE,EAAE,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,EAAE;oBACvD,KAAK,EAAE,GAAG,CAAC,KAAK;oBAChB,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;iBAClF,EAAE,CAAC;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC;gBAAS,CAAC;YACT,OAAO,CAAC,IAAI,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,MAAM,CAAC,IAAwB;QACnC,IAAI,CAAC;YACH,MAAM,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,gBAAgB,CAAC,EAAE;gBACvD,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI,kBAAkB,EAAE,MAAM,CAAC;gBACjE,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC,CAAC;YACH,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB,EAAE,CAAC;QAC/F,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,UAAU,CAAC,IAAwB;QACvC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,gBAAgB,CAAC,EAAE;YACxE,MAAM,EAAE,KAAK;YACb,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI,kBAAkB,EAAE,MAAM,CAAC;YACjE,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAE,CAAC,MAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9F,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;YACzB,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC5B,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YACjC,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC5E,MAAM,KAAK,GAAc,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YACnH,MAAM,aAAa,GAAG,QAAQ,CAAC,GAAG,EAAE,eAAe,CAAC,CAAC;YACrD,IAAI,aAAa,KAAK,SAAS;gBAAE,KAAK,CAAC,aAAa,GAAG,aAAa,CAAC;YACrE,MAAM,OAAO,GAAG,GAAG,EAAE,0BAA0B,CAAC;YAChD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC;gBAAE,KAAK,CAAC,YAAY,GAAG,OAAmB,CAAC;YACpH,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED,uEAAuE;AACvE,SAAS,kBAAkB,CAAC,GAAgB;IAC1C,OAAO,GAAG,CAAC,cAAc,EAAE,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC;AACpE,CAAC;AAED,SAAS,IAAI,CAAC,GAAgB;IAC5B,MAAM,UAAU,GAAG,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACnH,MAAM,QAAQ,GAAG,GAAG,CAAC,cAAc,EAAE,IAAI,KAAK,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC;IAC3E,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;IACnG,MAAM,OAAO,GAA4B;QACvC,QAAQ,EAAE,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAC3E,gBAAgB,EAAE;YAChB,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,eAAe,EAAE,GAAG,CAAC,SAAS;YAC9B,aAAa,EAAE,GAAG,CAAC,aAAa;YAChC,gBAAgB,EAAE,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,SAAS;YAC3D,cAAc,EAAE,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS;SACtD;KACF,CAAC;IACF,IAAI,UAAU;QAAE,OAAO,CAAC,iBAAiB,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;IAC9E,IAAI,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;QACtB,OAAO,CAAC,KAAK,GAAG,CAAC,EAAE,oBAAoB,EAAE,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QACvJ,MAAM,IAAI,GAAG,GAAG,CAAC,UAAU,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,GAAG,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;QACtG,MAAM,MAAM,GAA4B,EAAE,IAAI,EAAE,CAAC;QACjD,IAAI,OAAO,GAAG,CAAC,UAAU,KAAK,QAAQ;YAAE,MAAM,CAAC,oBAAoB,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAC5F,OAAO,CAAC,UAAU,GAAG,EAAE,qBAAqB,EAAE,MAAM,EAAE,CAAC;IACzD,CAAC;IACD,IAAI,GAAG,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;QACtC,sEAAsE;QACtE,kEAAkE;QACjE,OAAO,CAAC,gBAA4C,CAAC,cAAc,GAAG;YACrE,cAAc,EAAE,uBAAuB,CAAC,GAAG,CAAC,eAAe,CAAC;SAC7D,CAAC;IACJ,CAAC;IACD,4EAA4E;IAC5E,oEAAoE;IACpE,wEAAwE;IACxE,OAAO,oBAAoB,CAAC,OAAO,EAAE,GAAG,CAAC,eAAe,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC;AAClF,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAuB;IAC/C,MAAM,QAAQ,GAA8B,EAAE,CAAC;IAC/C,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QACxD,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAE,CAAC;QACjC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC;YACxC,SAAS;QACX,CAAC;QACD,MAAM,KAAK,GAAc,EAAE,CAAC;QAC5B,OAAO,KAAK,GAAG,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YACnE,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAE,CAAC;YACrC,KAAK,CAAC,IAAI,CAAC,EAAE,gBAAgB,EAAE,EAAE,IAAI,EAAE,WAAW,CAAC,IAAI,IAAI,SAAS,EAAE,QAAQ,EAAE,kBAAkB,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,CAAC;YACrH,KAAK,IAAI,CAAC,CAAC;QACb,CAAC;QACD,KAAK,IAAI,CAAC,CAAC;QACX,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IACzC,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,eAAe,CAAC,CAAc;IACrC,8EAA8E;IAC9E,2CAA2C;IAC3C,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;QAClD,MAAM,KAAK,GAAc,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QACrD,KAAK,MAAM,IAAI,IAAI,CAAC,CAAC,SAAS;YAAE,KAAK,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9G,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAClC,CAAC;IACD,OAAO;QACL,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;QAC/C,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC;KAC9B,CAAC;AACJ,CAAC;AAED,iFAAiF;AACjF,SAAS,WAAW,CAAC,OAA+B;IAClD,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3E,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO;QAChD,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE,EAAE,EAAE;QAC1F,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,WAAW,CAAC,OAA+B;IAClD,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC;IAChD,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAC;IACxB,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAClG,CAAC;AAED,SAAS,kBAAkB,CAAC,OAAoB;IAC9C,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,EAAE,OAAO,EAAE,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAC1F,IAAI,CAAC;QACH,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACpD,IAAI,MAAM,KAAK,IAAI,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YAAE,OAAO,MAAiC,CAAC;IACxH,CAAC;IAAC,MAAM,CAAC;QACP,uBAAuB;IACzB,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;AACtC,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,cAAc,CAAC,GAAa;IAC1C,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAa,CAAC;IACzC,2EAA2E;IAC3E,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,KAAK,MAAM,IAAI,IAAI,IAAI;YAAE,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,MAA2C;IACjE,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;IAC9E,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC;AACvB,CAAC;AAED,SAAS,cAAc,CAAC,MAA2C;IACjE,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IACpE,OAAO,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;AAC5D,CAAC;AAED,SAAS,SAAS,CAAC,MAA0B,EAAE,QAAiB;IAC9D,IAAI,QAAQ;QAAE,OAAO,MAAM,CAAC;IAC5B,IAAI,MAAM,KAAK,YAAY;QAAE,OAAO,QAAQ,CAAC;IAC7C,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,YAAY,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,oBAAoB;QAAE,OAAO,gBAAgB,CAAC;IACzI,OAAO,MAAM,CAAC;AAChB,CAAC"}
@@ -0,0 +1,11 @@
1
+ import type { ChatRequest, ChatResult, EmbedRequest, EmbedResult, ModelInfo, ProviderAdapter, ResolvedConnection, StreamEvent } from '../../types.js';
2
+ export declare class OllamaAdapter implements ProviderAdapter {
3
+ readonly provider: string;
4
+ constructor(provider: string);
5
+ chat(conn: ResolvedConnection, req: ChatRequest): Promise<ChatResult>;
6
+ stream(conn: ResolvedConnection, req: ChatRequest): AsyncIterable<StreamEvent>;
7
+ listModels(conn: ResolvedConnection): Promise<ModelInfo[]>;
8
+ embed(conn: ResolvedConnection, req: EmbedRequest): Promise<EmbedResult>;
9
+ private showModel;
10
+ }
11
+ //# sourceMappingURL=ollama.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../../src/adapters/engines/ollama.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAe,WAAW,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,kBAAkB,EAAE,WAAW,EAAY,MAAM,gBAAgB,CAAC;AAO7K,qBAAa,aAAc,YAAW,eAAe;IACnD,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;gBACd,QAAQ,EAAE,MAAM;IAItB,IAAI,CAAC,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;IAUpE,MAAM,CAAC,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,WAAW,GAAG,aAAa,CAAC,WAAW,CAAC;IAgE/E,UAAU,CAAC,IAAI,EAAE,kBAAkB,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAqB1D,KAAK,CAAC,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;YAsBhE,SAAS;CAuBxB"}
@@ -0,0 +1,202 @@
1
+ import { AIError } from '../../errors.js';
2
+ import { estimatedUsage } from '../../tokens.js';
3
+ import { fetchJson, ndjsonLines, postResponse } from '../../transport.js';
4
+ import { applyProviderOptions, asNumber, asRecord, asString, joinUrl, mapWithConcurrency, textFromMessages } from '../../util.js';
5
+ import { DEFAULT_TIMEOUT_MS, parseArgs, randomId, streamAnomaly, streamError, streamTimeout } from './base.js';
6
+ /** Max concurrent /api/show probes when listing models — enough to be fast, few enough to be polite to a local daemon. */
7
+ const SHOW_CONCURRENCY = 6;
8
+ export class OllamaAdapter {
9
+ provider;
10
+ constructor(provider) {
11
+ this.provider = provider;
12
+ }
13
+ async chat(conn, req) {
14
+ let result;
15
+ for await (const event of this.stream(conn, req)) {
16
+ if (event.type === 'done')
17
+ result = event.result;
18
+ if (event.type === 'error')
19
+ throw event.error;
20
+ }
21
+ if (!result)
22
+ throw new AIError('Ollama stream ended without a result', { kind: 'invalid_response', provider: conn.provider });
23
+ return result;
24
+ }
25
+ async *stream(conn, req) {
26
+ const started = Date.now();
27
+ let firstTokenMs = null;
28
+ let text = '';
29
+ let inputTokens;
30
+ let outputTokens;
31
+ let doneReason;
32
+ let sawTerminal = false;
33
+ const toolCalls = [];
34
+ const timeout = streamTimeout(conn, req.signal);
35
+ yield { type: 'start', callId: '', provider: conn.provider, model: req.model };
36
+ try {
37
+ const res = await postResponse(joinUrl(conn.baseUrl, '/api/chat'), body(req), conn.headers, timeout.signal, conn.provider);
38
+ for await (const value of ndjsonLines(res)) {
39
+ timeout.bump();
40
+ const record = asRecord(value);
41
+ const message = asRecord(record?.message);
42
+ const thinking = asString(message?.thinking);
43
+ if (thinking)
44
+ yield { type: 'reasoning', text: thinking };
45
+ const piece = asString(message?.content);
46
+ if (piece) {
47
+ if (firstTokenMs === null)
48
+ firstTokenMs = Date.now() - started;
49
+ text += piece;
50
+ yield { type: 'delta', text: piece };
51
+ }
52
+ inputTokens = asNumber(record?.prompt_eval_count) ?? inputTokens;
53
+ outputTokens = asNumber(record?.eval_count) ?? outputTokens;
54
+ doneReason = asString(record?.done_reason) ?? doneReason;
55
+ if (record?.done === true)
56
+ sawTerminal = true;
57
+ const calls = Array.isArray(message?.tool_calls) ? message.tool_calls : [];
58
+ for (const callValue of calls) {
59
+ const fn = asRecord(asRecord(callValue)?.function);
60
+ const args = fn?.arguments ?? {};
61
+ const toolCall = {
62
+ id: randomId(),
63
+ name: asString(fn?.name) ?? 'unknown',
64
+ // Ollama itself sends a native object, but llama.cpp-style backends
65
+ // behind the same protocol send a JSON *string* — coerce either into
66
+ // the object `validateToolArgs` expects.
67
+ arguments: parseArgs(args),
68
+ raw: typeof args === 'string' ? args : JSON.stringify(args),
69
+ };
70
+ toolCalls.push(toolCall);
71
+ yield { type: 'tool_call', call: toolCall };
72
+ }
73
+ }
74
+ if (!sawTerminal)
75
+ yield streamAnomaly('NDJSON stream ended without a done record');
76
+ const hasTools = toolCalls.length > 0;
77
+ yield { type: 'done', result: {
78
+ text,
79
+ toolCalls: hasTools ? toolCalls : undefined,
80
+ finishReason: hasTools ? 'tool' : doneReason === 'length' ? 'length' : 'stop',
81
+ usage: inputTokens !== undefined || outputTokens !== undefined ? { inputTokens, outputTokens, estimated: false } : estimatedUsage(textFromMessages(req.messages), text),
82
+ timing: { firstTokenMs, totalMs: Date.now() - started },
83
+ model: req.model,
84
+ source: { provider: conn.provider, connectionId: conn.id, baseUrl: conn.baseUrl },
85
+ } };
86
+ }
87
+ catch (error) {
88
+ throw streamError(error, timeout, conn.provider);
89
+ }
90
+ finally {
91
+ timeout.done();
92
+ }
93
+ }
94
+ async listModels(conn) {
95
+ const { data } = await fetchJson(`${conn.baseUrl}/api/tags`, {
96
+ method: 'GET',
97
+ headers: conn.headers,
98
+ timeoutMs: conn.timeoutMs ?? DEFAULT_TIMEOUT_MS,
99
+ provider: conn.provider,
100
+ });
101
+ const models = Array.isArray(asRecord(data)?.models) ? asRecord(data).models : [];
102
+ const ids = models.map((model) => asString(asRecord(model)?.name) ?? asString(asRecord(model)?.model) ?? '').filter(Boolean);
103
+ // Probe /api/show per model for context window + capabilities (best effort),
104
+ // with bounded concurrency so a large model list resolves quickly without
105
+ // opening an unbounded number of sockets against a local daemon.
106
+ return mapWithConcurrency(ids, SHOW_CONCURRENCY, async (id) => {
107
+ const info = { id, source: { provider: conn.provider, connectionId: conn.id, baseUrl: conn.baseUrl } };
108
+ const probed = await this.showModel(conn, id).catch(() => undefined);
109
+ if (probed?.contextWindow !== undefined)
110
+ info.contextWindow = probed.contextWindow;
111
+ if (probed?.capabilities)
112
+ info.capabilities = probed.capabilities;
113
+ return info;
114
+ });
115
+ }
116
+ async embed(conn, req) {
117
+ const inputs = Array.isArray(req.input) ? req.input : [req.input];
118
+ const { data } = await fetchJson(joinUrl(conn.baseUrl, '/api/embed'), {
119
+ method: 'POST',
120
+ headers: { 'content-type': 'application/json', ...conn.headers },
121
+ body: JSON.stringify(applyProviderOptions({ model: req.model, input: inputs }, req.providerOptions)),
122
+ timeoutMs: conn.timeoutMs ?? DEFAULT_TIMEOUT_MS,
123
+ provider: conn.provider,
124
+ });
125
+ const record = asRecord(data);
126
+ const rows = Array.isArray(record?.embeddings) ? record.embeddings : [];
127
+ const embeddings = rows.map((row) => (Array.isArray(row) ? row.filter((n) => typeof n === 'number') : []));
128
+ const inputTokens = asNumber(record?.prompt_eval_count);
129
+ return {
130
+ embeddings,
131
+ model: req.model,
132
+ usage: { inputTokens, outputTokens: 0, estimated: inputTokens === undefined },
133
+ source: { provider: conn.provider, connectionId: conn.id, baseUrl: conn.baseUrl },
134
+ raw: data,
135
+ };
136
+ }
137
+ async showModel(conn, model) {
138
+ const { data } = await fetchJson(`${conn.baseUrl}/api/show`, {
139
+ method: 'POST',
140
+ headers: { 'content-type': 'application/json', ...conn.headers },
141
+ body: JSON.stringify({ model }),
142
+ timeoutMs: Math.min(conn.timeoutMs ?? DEFAULT_TIMEOUT_MS, 15_000),
143
+ provider: conn.provider,
144
+ });
145
+ const record = asRecord(data);
146
+ const modelInfo = asRecord(record?.model_info);
147
+ let contextWindow;
148
+ if (modelInfo) {
149
+ for (const [key, value] of Object.entries(modelInfo)) {
150
+ if (key.endsWith('.context_length') && typeof value === 'number') {
151
+ contextWindow = value;
152
+ break;
153
+ }
154
+ }
155
+ }
156
+ const caps = record?.capabilities;
157
+ const capabilities = Array.isArray(caps) && caps.every((c) => typeof c === 'string') ? caps : undefined;
158
+ const result = {};
159
+ if (contextWindow !== undefined)
160
+ result.contextWindow = contextWindow;
161
+ if (capabilities)
162
+ result.capabilities = capabilities;
163
+ return result;
164
+ }
165
+ }
166
+ function body(req) {
167
+ const base = {
168
+ model: req.model,
169
+ messages: req.messages.map(toOllamaMessage),
170
+ stream: true,
171
+ options: { temperature: req.temperature, num_predict: req.maxTokens, top_p: req.topP, stop: req.stopSequences },
172
+ format: req.responseFormat?.type === 'json' ? (req.responseFormat.schema ?? 'json') : undefined,
173
+ tools: req.tools?.map((tool) => ({ type: 'function', function: { name: tool.name, description: tool.description, parameters: tool.parameters } })),
174
+ };
175
+ // First-class reasoning effort → Ollama's `think` flag (reasoning arrives in
176
+ // `message.thinking`, which this engine already routes to `reasoning` events).
177
+ // Ollama has no graded effort for most models, so anything but `'none'` is
178
+ // `true`. Only sent when asked for: a model without thinking support rejects it.
179
+ if (req.reasoningEffort !== undefined)
180
+ base.think = req.reasoningEffort !== 'none';
181
+ // providerOptions reaches Ollama's real request fields — `options.num_ctx`,
182
+ // `options.num_keep`, top-level `keep_alive`, `think`, etc. `options` is
183
+ // merged one level deep so num_ctx joins the samplers above rather than
184
+ // replacing them.
185
+ return applyProviderOptions(base, req.providerOptions, ['options']);
186
+ }
187
+ function toOllamaMessage(m) {
188
+ const images = typeof m.content === 'string' ? undefined : m.content.filter((part) => part.type === 'image').map((part) => part.imageBase64).filter(Boolean);
189
+ const message = {
190
+ role: m.role,
191
+ content: typeof m.content === 'string' ? m.content : m.content.filter((part) => part.type === 'text').map((part) => part.text ?? '').join('\n'),
192
+ };
193
+ if (images && images.length)
194
+ message.images = images;
195
+ if (m.role === 'tool' && m.name)
196
+ message.tool_name = m.name;
197
+ if (m.role === 'assistant' && m.toolCalls?.length) {
198
+ message.tool_calls = m.toolCalls.map((call) => ({ function: { name: call.name, arguments: call.arguments ?? {} } }));
199
+ }
200
+ return message;
201
+ }
202
+ //# sourceMappingURL=ollama.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../../src/adapters/engines/ollama.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAC1C,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAE1E,OAAO,EAAE,oBAAoB,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAClI,OAAO,EAAE,kBAAkB,EAAE,SAAS,EAAE,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAE/G,0HAA0H;AAC1H,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAE3B,MAAM,OAAO,aAAa;IACf,QAAQ,CAAS;IAC1B,YAAY,QAAgB;QAC1B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,IAAwB,EAAE,GAAgB;QACnD,IAAI,MAA8B,CAAC;QACnC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;YACjD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;gBAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;YACjD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;gBAAE,MAAM,KAAK,CAAC,KAAK,CAAC;QAChD,CAAC;QACD,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,OAAO,CAAC,sCAAsC,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9H,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,IAAwB,EAAE,GAAgB;QACtD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3B,IAAI,YAAY,GAAkB,IAAI,CAAC;QACvC,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,WAA+B,CAAC;QACpC,IAAI,YAAgC,CAAC;QACrC,IAAI,UAA8B,CAAC;QACnC,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC;QAC/E,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC3H,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC3C,OAAO,CAAC,IAAI,EAAE,CAAC;gBACf,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;gBAC/B,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC1C,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;gBAC7C,IAAI,QAAQ;oBAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gBAC1D,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBACzC,IAAI,KAAK,EAAE,CAAC;oBACV,IAAI,YAAY,KAAK,IAAI;wBAAE,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;oBAC/D,IAAI,IAAI,KAAK,CAAC;oBACd,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;gBACvC,CAAC;gBACD,WAAW,GAAG,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC,IAAI,WAAW,CAAC;gBACjE,YAAY,GAAG,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAC,IAAI,YAAY,CAAC;gBAC5D,UAAU,GAAG,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,IAAI,UAAU,CAAC;gBACzD,IAAI,MAAM,EAAE,IAAI,KAAK,IAAI;oBAAE,WAAW,GAAG,IAAI,CAAC;gBAC9C,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3E,KAAK,MAAM,SAAS,IAAI,KAAK,EAAE,CAAC;oBAC9B,MAAM,EAAE,GAAG,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,QAAQ,CAAC,CAAC;oBACnD,MAAM,IAAI,GAAG,EAAE,EAAE,SAAS,IAAI,EAAE,CAAC;oBACjC,MAAM,QAAQ,GAAa;wBACzB,EAAE,EAAE,QAAQ,EAAE;wBACd,IAAI,EAAE,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,SAAS;wBACrC,oEAAoE;wBACpE,qEAAqE;wBACrE,yCAAyC;wBACzC,SAAS,EAAE,SAAS,CAAC,IAAI,CAAC;wBAC1B,GAAG,EAAE,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;qBAC5D,CAAC;oBACF,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACzB,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gBAC9C,CAAC;YACH,CAAC;YACD,IAAI,CAAC,WAAW;gBAAE,MAAM,aAAa,CAAC,2CAA2C,CAAC,CAAC;YACnF,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;YACtC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE;oBAC5B,IAAI;oBACJ,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;oBAC3C,YAAY,EAAE,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM;oBAC7E,KAAK,EAAE,WAAW,KAAK,SAAS,IAAI,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,YAAY,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC;oBACvK,MAAM,EAAE,EAAE,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,EAAE;oBACvD,KAAK,EAAE,GAAG,CAAC,KAAK;oBAChB,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;iBAClF,EAAE,CAAC;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC;gBAAS,CAAC;YACT,OAAO,CAAC,IAAI,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAwB;QACvC,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;YAC3D,MAAM,EAAE,KAAK;YACb,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,kBAAkB;YAC/C,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAE,CAAC,MAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;QAChG,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC7H,6EAA6E;QAC7E,0EAA0E;QAC1E,iEAAiE;QACjE,OAAO,kBAAkB,CAAC,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YAC5D,MAAM,IAAI,GAAc,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YAClH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACrE,IAAI,MAAM,EAAE,aAAa,KAAK,SAAS;gBAAE,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;YACnF,IAAI,MAAM,EAAE,YAAY;gBAAE,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;YAClE,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAwB,EAAE,GAAiB;QACrD,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAClE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,EAAE;YACpE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE;YAChE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,GAAG,CAAC,eAAe,CAAC,CAAC;YACpG,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,kBAAkB;YAC/C,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QACxE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACxH,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;QACxD,OAAO;YACL,UAAU;YACV,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,KAAK,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,KAAK,SAAS,EAAE;YAC7E,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;YACjF,GAAG,EAAE,IAAI;SACV,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,IAAwB,EAAE,KAAa;QAC7D,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;YAC3D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE;YAChE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC;YAC/B,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI,kBAAkB,EAAE,MAAM,CAAC;YACjE,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAC/C,IAAI,aAAiC,CAAC;QACtC,IAAI,SAAS,EAAE,CAAC;YACd,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrD,IAAI,GAAG,CAAC,QAAQ,CAAC,iBAAiB,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAAC,aAAa,GAAG,KAAK,CAAC;oBAAC,MAAM;gBAAC,CAAC;YACrG,CAAC;QACH,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,EAAE,YAAY,CAAC;QAClC,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAgB,CAAC,CAAC,CAAC,SAAS,CAAC;QACpH,MAAM,MAAM,GAAwD,EAAE,CAAC;QACvE,IAAI,aAAa,KAAK,SAAS;YAAE,MAAM,CAAC,aAAa,GAAG,aAAa,CAAC;QACtE,IAAI,YAAY;YAAE,MAAM,CAAC,YAAY,GAAG,YAAY,CAAC;QACrD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED,SAAS,IAAI,CAAC,GAAgB;IAC5B,MAAM,IAAI,GAA4B;QACpC,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,QAAQ,EAAE,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,eAAe,CAAC;QAC3C,MAAM,EAAE,IAAI;QACZ,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,WAAW,EAAE,GAAG,CAAC,SAAS,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,aAAa,EAAE;QAC/G,MAAM,EAAE,GAAG,CAAC,cAAc,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,MAAM,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;QAC/F,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;KACnJ,CAAC;IACF,6EAA6E;IAC7E,+EAA+E;IAC/E,2EAA2E;IAC3E,iFAAiF;IACjF,IAAI,GAAG,CAAC,eAAe,KAAK,SAAS;QAAE,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,eAAe,KAAK,MAAM,CAAC;IACnF,4EAA4E;IAC5E,yEAAyE;IACzE,wEAAwE;IACxE,kBAAkB;IAClB,OAAO,oBAAoB,CAAC,IAAI,EAAE,GAAG,CAAC,eAAe,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;AACtE,CAAC;AAED,SAAS,eAAe,CAAC,CAAc;IACrC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC7J,MAAM,OAAO,GAA4B;QACvC,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,OAAO,EAAE,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;KAChJ,CAAC;IACF,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM;QAAE,OAAO,CAAC,MAAM,GAAG,MAAM,CAAC;IACrD,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI;QAAE,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC,IAAI,CAAC;IAC5D,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;QAClD,OAAO,CAAC,UAAU,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IACvH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC"}
@@ -0,0 +1,22 @@
1
+ import type { ChatRequest, ChatResult, EmbedRequest, EmbedResult, ProviderAdapter, ResolvedConnection, StreamEvent } from '../../types.js';
2
+ import type { ProviderProfile } from '../profiles.js';
3
+ export declare class OpenAIChatAdapter implements ProviderAdapter {
4
+ private profile;
5
+ readonly provider: string;
6
+ constructor(provider: string, profile: ProviderProfile);
7
+ chat(conn: ResolvedConnection, req: ChatRequest): Promise<ChatResult>;
8
+ stream(conn: ResolvedConnection, req: ChatRequest): AsyncIterable<StreamEvent>;
9
+ listModels(conn: ResolvedConnection): Promise<import("../../types.js").ModelInfo[]>;
10
+ health(conn: ResolvedConnection): Promise<{
11
+ ok: boolean;
12
+ detail?: string;
13
+ }>;
14
+ embed(conn: ResolvedConnection, req: EmbedRequest): Promise<EmbedResult>;
15
+ }
16
+ /**
17
+ * True when a `/chat/completions` 400 is the "function tools with reasoning_effort
18
+ * are not supported" refusal AND the request actually carried tools AND the caller
19
+ * did not already pin `reasoning_effort` (in which case the choice is theirs).
20
+ */
21
+ export declare function shouldRetryWithoutReasoningEffort(error: unknown, req: ChatRequest): boolean;
22
+ //# sourceMappingURL=openaiChat.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"openaiChat.d.ts","sourceRoot":"","sources":["../../../src/adapters/engines/openaiChat.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAe,WAAW,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,eAAe,EAAE,kBAAkB,EAAE,WAAW,EAAY,MAAM,gBAAgB,CAAC;AAElK,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAGtD,qBAAa,iBAAkB,YAAW,eAAe;IAEzB,OAAO,CAAC,OAAO;IAD7C,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;gBACd,QAAQ,EAAE,MAAM,EAAU,OAAO,EAAE,eAAe;IAIxD,IAAI,CAAC,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;IAUpE,MAAM,CAAC,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,WAAW,GAAG,aAAa,CAAC,WAAW,CAAC;IA4GrF,UAAU,CAAC,IAAI,EAAE,kBAAkB;IAInC,MAAM,CAAC,IAAI,EAAE,kBAAkB;;;;IAIzB,KAAK,CAAC,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;CAyB/E;AAoHD;;;;GAIG;AACH,wBAAgB,iCAAiC,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,GAAG,OAAO,CAU3F"}