@asterxsk/kiln 0.1.0 → 0.2.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 (203) hide show
  1. package/LICENSE +21 -21
  2. package/README.md +170 -170
  3. package/agent/AGENTS.md +67 -67
  4. package/agent/README.md +5 -5
  5. package/agent/extensions/AGENTS.md +68 -68
  6. package/agent/extensions/ask-user/index.ts +418 -418
  7. package/agent/extensions/ask-user/package-lock.json +769 -769
  8. package/agent/extensions/ask-user/package.json +19 -19
  9. package/agent/extensions/ask-user/prompt.ts +45 -45
  10. package/agent/extensions/ask-user/tsconfig.json +7 -7
  11. package/agent/extensions/background-terminals/docs/implementation-guide.md +942 -942
  12. package/agent/extensions/background-terminals/index.ts +627 -627
  13. package/agent/extensions/background-terminals/manager.test.ts +735 -735
  14. package/agent/extensions/background-terminals/output.test.ts +109 -109
  15. package/agent/extensions/background-terminals/package-lock.json +769 -769
  16. package/agent/extensions/background-terminals/package.json +17 -17
  17. package/agent/extensions/background-terminals/prompt.test.ts +125 -125
  18. package/agent/extensions/background-terminals/ps.test.ts +82 -82
  19. package/agent/extensions/background-terminals/result-delivery.test.ts +44 -44
  20. package/agent/extensions/background-terminals/src/domain.ts +87 -87
  21. package/agent/extensions/background-terminals/src/manager.ts +907 -907
  22. package/agent/extensions/background-terminals/src/output.ts +84 -84
  23. package/agent/extensions/background-terminals/src/prompt.ts +142 -142
  24. package/agent/extensions/background-terminals/src/result-delivery.ts +27 -27
  25. package/agent/extensions/background-terminals/src/runtime.ts +36 -36
  26. package/agent/extensions/background-terminals/src/ui/output-view.ts +79 -79
  27. package/agent/extensions/background-terminals/src/ui/ps.ts +621 -621
  28. package/agent/extensions/background-terminals/tsconfig.json +7 -7
  29. package/agent/extensions/file-search/index.spec.ts +443 -443
  30. package/agent/extensions/file-search/index.ts +459 -459
  31. package/agent/extensions/file-search/package-lock.json +2253 -2253
  32. package/agent/extensions/file-search/package.json +23 -23
  33. package/agent/extensions/file-search/src/args.ts +122 -122
  34. package/agent/extensions/file-search/src/binaries.ts +422 -422
  35. package/agent/extensions/file-search/src/output.ts +126 -126
  36. package/agent/extensions/file-search/src/process.ts +146 -146
  37. package/agent/extensions/file-search/src/prompt.ts +52 -52
  38. package/agent/extensions/file-search/tsconfig.json +7 -7
  39. package/agent/extensions/modelconf/PLAN.md +915 -915
  40. package/agent/extensions/modelconf/index.ts +296 -296
  41. package/agent/extensions/modelconf/src/ui/ModelConfView.ts +1101 -1101
  42. package/agent/extensions/pi-web-access/CHANGELOG.md +690 -690
  43. package/agent/extensions/pi-web-access/LICENSE +21 -21
  44. package/agent/extensions/pi-web-access/README.md +470 -470
  45. package/agent/extensions/pi-web-access/SECURITY.md +5 -5
  46. package/agent/extensions/pi-web-access/activity.ts +101 -101
  47. package/agent/extensions/pi-web-access/auth-fetch.ts +148 -148
  48. package/agent/extensions/pi-web-access/brightdata-unlocker.ts +272 -272
  49. package/agent/extensions/pi-web-access/chrome-cookies.ts +669 -669
  50. package/agent/extensions/pi-web-access/content-find.ts +139 -139
  51. package/agent/extensions/pi-web-access/credential-source.ts +191 -191
  52. package/agent/extensions/pi-web-access/data-uri-sanitize.ts +406 -406
  53. package/agent/extensions/pi-web-access/datalab-pdf-extract.ts +568 -568
  54. package/agent/extensions/pi-web-access/declared-web-links.ts +173 -173
  55. package/agent/extensions/pi-web-access/evidence/CONTRACT-EVIDENCE.md +496 -496
  56. package/agent/extensions/pi-web-access/evidence/contract-probe.mjs +140 -140
  57. package/agent/extensions/pi-web-access/exa.ts +526 -526
  58. package/agent/extensions/pi-web-access/extract.ts +1196 -1196
  59. package/agent/extensions/pi-web-access/feature-config.ts +29 -29
  60. package/agent/extensions/pi-web-access/fetch-params.ts +111 -111
  61. package/agent/extensions/pi-web-access/gemini-adc.ts +298 -298
  62. package/agent/extensions/pi-web-access/gemini-api.ts +353 -353
  63. package/agent/extensions/pi-web-access/gemini-pdf-extract.ts +108 -108
  64. package/agent/extensions/pi-web-access/gemini-url-context.ts +128 -128
  65. package/agent/extensions/pi-web-access/gemini-web-config.ts +101 -101
  66. package/agent/extensions/pi-web-access/gemini-web.ts +487 -487
  67. package/agent/extensions/pi-web-access/github-api.ts +197 -197
  68. package/agent/extensions/pi-web-access/github-extract.ts +746 -746
  69. package/agent/extensions/pi-web-access/github-issue-pr.ts +700 -700
  70. package/agent/extensions/pi-web-access/index.ts +1737 -1737
  71. package/agent/extensions/pi-web-access/package-lock.json +5808 -5808
  72. package/agent/extensions/pi-web-access/package.json +64 -64
  73. package/agent/extensions/pi-web-access/page-query.ts +96 -96
  74. package/agent/extensions/pi-web-access/pdf-extract.ts +409 -409
  75. package/agent/extensions/pi-web-access/promise-try.d.ts +7 -7
  76. package/agent/extensions/pi-web-access/query-rewrite.ts +51 -51
  77. package/agent/extensions/pi-web-access/render-search-error.ts +170 -170
  78. package/agent/extensions/pi-web-access/rsc-extract.ts +338 -338
  79. package/agent/extensions/pi-web-access/source-check.ts +282 -282
  80. package/agent/extensions/pi-web-access/ssrf-protection.ts +526 -526
  81. package/agent/extensions/pi-web-access/storage.ts +521 -521
  82. package/agent/extensions/pi-web-access/summary-model-scope.ts +125 -125
  83. package/agent/extensions/pi-web-access/test/auth-fetch.test.mjs +208 -208
  84. package/agent/extensions/pi-web-access/test/brightdata-unlocker.test.mjs +840 -840
  85. package/agent/extensions/pi-web-access/test/chrome-cookie-extraction.test.mjs +441 -441
  86. package/agent/extensions/pi-web-access/test/config-path.test.mjs +283 -283
  87. package/agent/extensions/pi-web-access/test/content-find.test.mjs +25 -25
  88. package/agent/extensions/pi-web-access/test/credential-source.test.mjs +118 -118
  89. package/agent/extensions/pi-web-access/test/data-uri-sanitize.test.mjs +210 -210
  90. package/agent/extensions/pi-web-access/test/datalab-pdf-extract.test.mjs +552 -552
  91. package/agent/extensions/pi-web-access/test/declared-web-links.test.mjs +212 -212
  92. package/agent/extensions/pi-web-access/test/fetch-answer-storage.test.mjs +40 -40
  93. package/agent/extensions/pi-web-access/test/fetch-cache-storage.test.mjs +334 -334
  94. package/agent/extensions/pi-web-access/test/fetch-content-domain-policy.test.mjs +95 -95
  95. package/agent/extensions/pi-web-access/test/fetch-modes.test.mjs +53 -53
  96. package/agent/extensions/pi-web-access/test/fetch-not-found-guidance.test.mjs +92 -92
  97. package/agent/extensions/pi-web-access/test/fetch-params.test.mjs +86 -86
  98. package/agent/extensions/pi-web-access/test/fetch-render-call.test.mjs +34 -34
  99. package/agent/extensions/pi-web-access/test/fetch-routing.test.mjs +173 -173
  100. package/agent/extensions/pi-web-access/test/gemini-adc-auth.test.mjs +257 -257
  101. package/agent/extensions/pi-web-access/test/gemini-api-transport.test.mjs +170 -170
  102. package/agent/extensions/pi-web-access/test/gemini-pdf-extract.test.mjs +133 -133
  103. package/agent/extensions/pi-web-access/test/gemini-web-cookie-opt-in.test.mjs +178 -178
  104. package/agent/extensions/pi-web-access/test/gemini-web-header-overflow.test.mjs +148 -148
  105. package/agent/extensions/pi-web-access/test/get-search-content.test.mjs +223 -223
  106. package/agent/extensions/pi-web-access/test/github-extract.test.mjs +378 -378
  107. package/agent/extensions/pi-web-access/test/github-issue-pr.test.mjs +565 -565
  108. package/agent/extensions/pi-web-access/test/inline-content-config.test.mjs +99 -99
  109. package/agent/extensions/pi-web-access/test/lazy-extract-load.test.mjs +118 -118
  110. package/agent/extensions/pi-web-access/test/local-video-oversize.test.mjs +52 -52
  111. package/agent/extensions/pi-web-access/test/package-typebox-dependency.test.mjs +50 -50
  112. package/agent/extensions/pi-web-access/test/page-query.test.mjs +51 -51
  113. package/agent/extensions/pi-web-access/test/pdf-config.test.mjs +140 -140
  114. package/agent/extensions/pi-web-access/test/pdf-extract.test.mjs +500 -500
  115. package/agent/extensions/pi-web-access/test/proxy-transport.test.mjs +286 -286
  116. package/agent/extensions/pi-web-access/test/query-rewrite.test.mjs +52 -52
  117. package/agent/extensions/pi-web-access/test/rsc-fallback.test.mjs +102 -102
  118. package/agent/extensions/pi-web-access/test/search-error-render.test.mjs +152 -152
  119. package/agent/extensions/pi-web-access/test/search-providers.test.mjs +274 -274
  120. package/agent/extensions/pi-web-access/test/source-check.test.mjs +179 -179
  121. package/agent/extensions/pi-web-access/test/ssrf-allow-ranges-config.test.mjs +205 -205
  122. package/agent/extensions/pi-web-access/test/ssrf-protection.test.mjs +456 -456
  123. package/agent/extensions/pi-web-access/test/tool-registration-config.test.mjs +182 -182
  124. package/agent/extensions/pi-web-access/test/youtube-extract-errors.test.mjs +64 -64
  125. package/agent/extensions/pi-web-access/tsconfig.json +11 -11
  126. package/agent/extensions/pi-web-access/utils.ts +451 -451
  127. package/agent/extensions/pi-web-access/video-extract.ts +392 -392
  128. package/agent/extensions/pi-web-access/youtube-extract.ts +328 -328
  129. package/agent/extensions/shared/activity-status.ts +31 -31
  130. package/agent/extensions/shared/child-session.test.ts +270 -270
  131. package/agent/extensions/shared/child-session.ts +148 -148
  132. package/agent/extensions/shared/context-utilization.test.ts +48 -48
  133. package/agent/extensions/shared/context-utilization.ts +47 -47
  134. package/agent/extensions/shared/dashboard-state.ts +99 -99
  135. package/agent/extensions/shared/tool-call-timeout.test.ts +117 -117
  136. package/agent/extensions/shared/tool-call-timeout.ts +104 -104
  137. package/agent/extensions/subagents/by-the-way.test.ts +29 -29
  138. package/agent/extensions/subagents/claude.test.ts +119 -119
  139. package/agent/extensions/subagents/codex.test.ts +102 -102
  140. package/agent/extensions/subagents/context-usage.test.ts +107 -107
  141. package/agent/extensions/subagents/docs/design-plan.md +568 -568
  142. package/agent/extensions/subagents/docs/effect-v4-extension-guide.md +354 -354
  143. package/agent/extensions/subagents/docs/effect-v4-notes.md +571 -571
  144. package/agent/extensions/subagents/index.ts +779 -779
  145. package/agent/extensions/subagents/manager.test.ts +276 -276
  146. package/agent/extensions/subagents/package-lock.json +2244 -2244
  147. package/agent/extensions/subagents/package.json +19 -19
  148. package/agent/extensions/subagents/result-delivery.test.ts +27 -27
  149. package/agent/extensions/subagents/src/backend.ts +73 -73
  150. package/agent/extensions/subagents/src/backends/claude.ts +701 -701
  151. package/agent/extensions/subagents/src/backends/codex.ts +1060 -1060
  152. package/agent/extensions/subagents/src/backends/pi.ts +575 -575
  153. package/agent/extensions/subagents/src/backends/stub.ts +300 -300
  154. package/agent/extensions/subagents/src/by-the-way.ts +21 -21
  155. package/agent/extensions/subagents/src/domain.ts +253 -253
  156. package/agent/extensions/subagents/src/format.ts +74 -74
  157. package/agent/extensions/subagents/src/manager.ts +736 -736
  158. package/agent/extensions/subagents/src/prompt.ts +92 -92
  159. package/agent/extensions/subagents/src/result-delivery.ts +20 -20
  160. package/agent/extensions/subagents/src/runtime.ts +53 -53
  161. package/agent/extensions/subagents/src/ui/takeover.ts +583 -583
  162. package/agent/extensions/subagents/src/ui/transcript.ts +201 -201
  163. package/agent/extensions/subagents/takeover.test.ts +29 -29
  164. package/agent/extensions/subagents/tsconfig.json +7 -7
  165. package/agent/extensions/todo/AGENTS.md +38 -38
  166. package/agent/extensions/todo/LICENSE +21 -21
  167. package/agent/extensions/todo/config.ts +55 -55
  168. package/agent/extensions/todo/index.ts +151 -151
  169. package/agent/extensions/todo/locales/de.json +17 -17
  170. package/agent/extensions/todo/locales/en.json +15 -15
  171. package/agent/extensions/todo/locales/es.json +17 -17
  172. package/agent/extensions/todo/locales/fr.json +17 -17
  173. package/agent/extensions/todo/locales/pt-BR.json +17 -17
  174. package/agent/extensions/todo/locales/pt.json +17 -17
  175. package/agent/extensions/todo/locales/ru.json +17 -17
  176. package/agent/extensions/todo/locales/uk.json +17 -17
  177. package/agent/extensions/todo/locales/zh.json +17 -17
  178. package/agent/extensions/todo/package-lock.json +3358 -3358
  179. package/agent/extensions/todo/package.json +67 -67
  180. package/agent/extensions/todo/state/i18n-bridge.ts +64 -64
  181. package/agent/extensions/todo/state/invariants.ts +20 -20
  182. package/agent/extensions/todo/state/replay.ts +38 -38
  183. package/agent/extensions/todo/state/selectors.ts +107 -107
  184. package/agent/extensions/todo/state/state-reducer.ts +326 -326
  185. package/agent/extensions/todo/state/state.ts +18 -18
  186. package/agent/extensions/todo/state/store.ts +82 -82
  187. package/agent/extensions/todo/state/task-graph.ts +57 -57
  188. package/agent/extensions/todo/todo-overlay.ts +200 -200
  189. package/agent/extensions/todo/todo.ts +155 -155
  190. package/agent/extensions/todo/tool/response-envelope.ts +109 -109
  191. package/agent/extensions/todo/tool/types.ts +206 -206
  192. package/agent/extensions/todo/view/format.ts +177 -177
  193. package/agent/install.ps1 +637 -527
  194. package/agent/install.sh +620 -511
  195. package/agent/keybindings.json +7 -7
  196. package/bin/kiln.js +124 -11
  197. package/package.json +8 -2
  198. package/agent/extensions/taste/index.ts +0 -443
  199. package/agent/extensions/taste/install.ps1 +0 -23
  200. package/agent/extensions/taste/install.sh +0 -21
  201. /package/agent/extensions/{status line → statusline}/index.ts +0 -0
  202. /package/agent/extensions/{status line → statusline}/install.ps1 +0 -0
  203. /package/agent/extensions/{status line → statusline}/install.sh +0 -0
@@ -1,353 +1,353 @@
1
- import { existsSync, readFileSync } from "node:fs";
2
- import { hasCredentialSource, redactCredential, resolveCredential } from "./credential-source.ts";
3
- import { getWebSearchConfigPath } from "./utils.ts";
4
- import {
5
- isGeminiAdcAvailable,
6
- getAdcAccessToken,
7
- getAdcProject,
8
- getAdcLocation,
9
- getVertexApiBase,
10
- isVertexHost,
11
- } from "./gemini-adc.ts";
12
-
13
- const DEFAULT_API_HOST = "https://generativelanguage.googleapis.com";
14
- const API_VERSION = "v1beta";
15
- export const API_BASE = `${DEFAULT_API_HOST}/${API_VERSION}`;
16
- const CONFIG_PATH = getWebSearchConfigPath();
17
- export const DEFAULT_MODEL = "gemini-3.6-flash";
18
-
19
- interface GeminiApiConfig {
20
- geminiApiKey?: unknown;
21
- geminiBaseUrl?: unknown;
22
- cloudflareApiKey?: unknown;
23
- }
24
-
25
- let cachedConfig: GeminiApiConfig | null = null;
26
-
27
- function loadConfig(): GeminiApiConfig {
28
- if (cachedConfig) return cachedConfig;
29
- if (!existsSync(CONFIG_PATH)) {
30
- cachedConfig = {};
31
- return cachedConfig;
32
- }
33
-
34
- const raw = readFileSync(CONFIG_PATH, "utf-8");
35
- try {
36
- cachedConfig = JSON.parse(raw) as GeminiApiConfig;
37
- return cachedConfig;
38
- } catch (err) {
39
- const message = err instanceof Error ? err.message : String(err);
40
- throw new Error(`Failed to parse ${CONFIG_PATH}: ${message}`);
41
- }
42
- }
43
-
44
- function withTimeout(signal: AbortSignal | undefined, timeoutMs: number): AbortSignal {
45
- const timeout = AbortSignal.timeout(timeoutMs);
46
- return signal ? AbortSignal.any([signal, timeout]) : timeout;
47
- }
48
-
49
- function normalizeApiKey(value: unknown): string | null {
50
- if (typeof value !== "string") return null;
51
- const normalized = value.trim();
52
- return normalized.length > 0 ? normalized : null;
53
- }
54
-
55
- function normalizeBaseUrl(value: unknown): string | null {
56
- if (typeof value !== "string") return null;
57
- const normalized = value.trim().replace(/\/+$/, "");
58
- return normalized.length > 0 ? normalized : null;
59
- }
60
-
61
- function isCloudflareGateway(): boolean {
62
- return getApiHost().includes("gateway.ai.cloudflare.com");
63
- }
64
-
65
- export async function getApiKey(signal?: AbortSignal): Promise<string | null> {
66
- return resolveCredential({
67
- provider: "Gemini",
68
- configuredValue: loadConfig().geminiApiKey,
69
- environmentValue: process.env.GEMINI_API_KEY,
70
- signal,
71
- });
72
- }
73
-
74
- export function getApiHost(): string {
75
- return (
76
- normalizeBaseUrl(process.env.GOOGLE_GEMINI_BASE_URL) ??
77
- normalizeBaseUrl(loadConfig().geminiBaseUrl) ??
78
- DEFAULT_API_HOST
79
- );
80
- }
81
-
82
- export function getVersionedApiBase(): string {
83
- const project = getAdcProject();
84
- const location = getAdcLocation();
85
- if (isGeminiAdcAvailable() && project && location) {
86
- return getVertexApiBase(project, location);
87
- }
88
- return `${getApiHost()}/${API_VERSION}`;
89
- }
90
-
91
- export function getUploadBase(): string {
92
- return `${getApiHost()}/upload/${API_VERSION}`;
93
- }
94
-
95
- function getLegacyCloudflareApiKey(): string | null {
96
- return normalizeApiKey(process.env.CLOUDFLARE_API_KEY) ?? normalizeApiKey(loadConfig().cloudflareApiKey);
97
- }
98
-
99
- async function resolveCloudflareApiKey(signal?: AbortSignal): Promise<string | null> {
100
- return resolveCredential({
101
- provider: "Cloudflare",
102
- configuredValue: loadConfig().cloudflareApiKey,
103
- environmentValue: process.env.CLOUDFLARE_API_KEY,
104
- signal,
105
- });
106
- }
107
-
108
- export function getCloudflareApiKey(): string | null {
109
- return getLegacyCloudflareApiKey();
110
- }
111
-
112
- export function isGatewayConfigured(): boolean {
113
- return isCloudflareGateway() && hasCredentialSource({
114
- provider: "Cloudflare",
115
- configuredValue: loadConfig().cloudflareApiKey,
116
- environmentValue: process.env.CLOUDFLARE_API_KEY,
117
- });
118
- }
119
-
120
- export function buildAuthHeaders(apiKey: string | null = null, cloudflareApiKey: string | null = getLegacyCloudflareApiKey()): Record<string, string> {
121
- if (!isCloudflareGateway()) return apiKey ? { "x-goog-api-key": apiKey } : {};
122
- return cloudflareApiKey ? { "cf-aig-authorization": `Bearer ${cloudflareApiKey}` } : {};
123
- }
124
-
125
- function redactGeminiCredentials(text: string, apiKey: string | null | undefined, cloudflareApiKey: string | null | undefined, adcToken?: string | null): string {
126
- let out = redactCredential(redactCredential(text, apiKey), cloudflareApiKey);
127
- if (adcToken) out = redactCredential(out, adcToken);
128
- return out;
129
- }
130
-
131
- const responseCredentials = new WeakMap<Response, {
132
- apiKey: string | null | undefined;
133
- cloudflareApiKey: string | null | undefined;
134
- adcToken?: string | null;
135
- }>();
136
-
137
- export function redactGeminiApiResponse(response: Response, text: string, apiKey?: string | null, adcToken?: string | null): string {
138
- const credentials = responseCredentials.get(response);
139
- return redactGeminiCredentials(text, credentials?.apiKey ?? apiKey, credentials?.cloudflareApiKey, credentials?.adcToken ?? adcToken);
140
- }
141
-
142
- export async function fetchGeminiApi(
143
- url: string | URL,
144
- init: RequestInit = {},
145
- apiKey?: string | null,
146
- ): Promise<Response> {
147
- const parsedUrl = new URL(url);
148
- for (const name of parsedUrl.searchParams.keys()) {
149
- if (["key", "api_key"].includes(name.toLowerCase())) {
150
- throw new Error("Gemini API credential query parameters are not allowed");
151
- }
152
- }
153
- const project = getAdcProject();
154
- const location = getAdcLocation();
155
- const adcMode = isGeminiAdcAvailable() && !!project && !!location && isVertexHost(parsedUrl.origin);
156
- let adcToken: string | null = null;
157
- if (adcMode) {
158
- adcToken = await getAdcAccessToken(init.signal ?? undefined);
159
- }
160
- const resolvedApiKey = adcMode ? null : (apiKey === undefined ? await getApiKey(init.signal ?? undefined) : apiKey);
161
- const cloudflareApiKey = isCloudflareGateway() ? await resolveCloudflareApiKey(init.signal ?? undefined) : null;
162
- const allowedOrigins = new Set([
163
- new URL(getApiHost()).origin,
164
- new URL(DEFAULT_API_HOST).origin,
165
- ...(adcMode ? [parsedUrl.origin] : []),
166
- ]);
167
- const needsAuth = resolvedApiKey || isGatewayConfigured() || adcMode;
168
- if (needsAuth && !allowedOrigins.has(parsedUrl.origin)) {
169
- throw new Error("Gemini API request host is not allowed");
170
- }
171
- const headers = new Headers(init.headers);
172
- headers.delete("x-goog-api-key");
173
- headers.delete("cf-aig-authorization");
174
- headers.delete("authorization");
175
- if (adcMode && adcToken) {
176
- headers.set("Authorization", `Bearer ${adcToken}`);
177
- } else {
178
- for (const [name, value] of Object.entries(buildAuthHeaders(resolvedApiKey, cloudflareApiKey))) {
179
- headers.set(name, value);
180
- }
181
- }
182
- try {
183
- const response = await fetch(parsedUrl, { ...init, headers });
184
- responseCredentials.set(response, { apiKey: resolvedApiKey, cloudflareApiKey, adcToken });
185
- return response;
186
- } catch (error) {
187
- const message = error instanceof Error ? error.message : String(error);
188
- const redactedMessage = redactGeminiCredentials(message, resolvedApiKey, cloudflareApiKey, adcToken);
189
- if (redactedMessage === message) throw error;
190
- const redactedError = new Error(redactedMessage);
191
- if (error instanceof Error) redactedError.name = error.name;
192
- throw redactedError;
193
- }
194
- }
195
-
196
- function hasGeminiApiKeySource(): boolean {
197
- return hasCredentialSource({
198
- provider: "Gemini",
199
- configuredValue: loadConfig().geminiApiKey,
200
- environmentValue: process.env.GEMINI_API_KEY,
201
- });
202
- }
203
-
204
- export function isGeminiApiAvailable(): boolean {
205
- return isGeminiAdcAvailable() || hasGeminiApiKeySource() || isGatewayConfigured();
206
- }
207
-
208
- /**
209
- * Whether the Gemini Files-API paths (YouTube and local video analysis) can
210
- * authenticate. These use the Gemini Files API and metadata endpoints, which do
211
- * not support ADC, so they still require an API key or a Cloudflare gateway.
212
- */
213
- export function isGeminiApiAvailableWithVideo(): boolean {
214
- return hasGeminiApiKeySource() || isGatewayConfigured();
215
- }
216
-
217
- export interface GeminiApiOptions {
218
- apiKey?: string;
219
- model?: string;
220
- mimeType?: string;
221
- signal?: AbortSignal;
222
- timeoutMs?: number;
223
- }
224
-
225
- export interface GeminiGenerateContentResult {
226
- text: string;
227
- finishReason?: string;
228
- blockReason?: string;
229
- }
230
-
231
- export async function queryGeminiApiWithInlineData(
232
- prompt: string,
233
- data: string,
234
- mimeType: string,
235
- options: GeminiApiOptions = {},
236
- ): Promise<GeminiGenerateContentResult> {
237
- const signal = withTimeout(options.signal, options.timeoutMs ?? 120000);
238
- const apiKey = isGeminiAdcAvailable() ? null : (options.apiKey ?? await getApiKey(signal));
239
- if (!apiKey && !isGatewayConfigured() && !isGeminiAdcAvailable()) {
240
- throw new Error(
241
- "Gemini API not configured. Either:\n" +
242
- ` 1. Configure geminiApiKey in ${CONFIG_PATH} or set GEMINI_API_KEY\n` +
243
- " 2. Set GOOGLE_GEMINI_BASE_URL + CLOUDFLARE_API_KEY for Cloudflare AI Gateway routing\n" +
244
- " 3. Set geminiAuth to \"adc\" with a Google Cloud ADC (GOOGLE_APPLICATION_CREDENTIALS) + project/location"
245
- );
246
- }
247
-
248
- const model = options.model ?? DEFAULT_MODEL;
249
- const url = `${getVersionedApiBase()}/models/${model}:generateContent`;
250
- const body = {
251
- contents: [
252
- {
253
- role: "user",
254
- parts: [
255
- { inlineData: { mimeType, data } },
256
- { text: prompt },
257
- ],
258
- },
259
- ],
260
- };
261
-
262
- const res = await fetchGeminiApi(url, {
263
- method: "POST",
264
- headers: { "Content-Type": "application/json" },
265
- body: JSON.stringify(body),
266
- signal,
267
- }, apiKey);
268
-
269
- if (!res.ok) {
270
- const errorText = redactGeminiApiResponse(res, await res.text(), apiKey);
271
- throw new Error(`Gemini API error ${res.status}: ${errorText.slice(0, 300)}`);
272
- }
273
-
274
- const response = (await res.json()) as GenerateContentResponse;
275
- const candidate = response.candidates?.[0];
276
- const text = candidate?.content?.parts
277
- ?.map((part) => part.text)
278
- .filter((part): part is string => typeof part === "string" && part.length > 0)
279
- .join("\n") ?? "";
280
-
281
- return {
282
- text,
283
- ...(candidate?.finishReason ? { finishReason: candidate.finishReason } : {}),
284
- ...(response.promptFeedback?.blockReason ? { blockReason: response.promptFeedback.blockReason } : {}),
285
- };
286
- }
287
-
288
- export async function queryGeminiApiWithVideo(
289
- prompt: string,
290
- videoUri: string,
291
- options: GeminiApiOptions = {},
292
- ): Promise<string> {
293
- const signal = withTimeout(options.signal, options.timeoutMs ?? 120000);
294
- const apiKey = options.apiKey ?? await getApiKey(signal);
295
- if (!apiKey && !isGatewayConfigured()) {
296
- throw new Error(
297
- "Gemini API not configured. Either:\n" +
298
- ` 1. Configure geminiApiKey in ${CONFIG_PATH} or set GEMINI_API_KEY\n` +
299
- " 2. Set GOOGLE_GEMINI_BASE_URL + CLOUDFLARE_API_KEY for Cloudflare AI Gateway routing"
300
- );
301
- }
302
-
303
- const model = options.model ?? DEFAULT_MODEL;
304
- const url = `${getVersionedApiBase()}/models/${model}:generateContent`;
305
-
306
- const fileData: Record<string, string> = { fileUri: videoUri };
307
- if (options.mimeType) fileData.mimeType = options.mimeType;
308
-
309
- const body = {
310
- contents: [
311
- {
312
- role: "user",
313
- parts: [
314
- { fileData },
315
- { text: prompt },
316
- ],
317
- },
318
- ],
319
- };
320
-
321
- const res = await fetchGeminiApi(url, {
322
- method: "POST",
323
- headers: { "Content-Type": "application/json" },
324
- body: JSON.stringify(body),
325
- signal,
326
- }, apiKey);
327
-
328
- if (!res.ok) {
329
- const errorText = redactGeminiApiResponse(res, await res.text(), apiKey);
330
- throw new Error(`Gemini API error ${res.status}: ${errorText.slice(0, 300)}`);
331
- }
332
-
333
- const data = (await res.json()) as GenerateContentResponse;
334
- const text = data.candidates?.[0]?.content?.parts
335
- ?.map((p) => p.text)
336
- .filter(Boolean)
337
- .join("\n");
338
-
339
- if (!text) throw new Error("Gemini API returned empty response");
340
- return text;
341
- }
342
-
343
- interface GenerateContentResponse {
344
- candidates?: Array<{
345
- content?: {
346
- parts?: Array<{ text?: string }>;
347
- };
348
- finishReason?: string;
349
- }>;
350
- promptFeedback?: {
351
- blockReason?: string;
352
- };
353
- }
1
+ import { existsSync, readFileSync } from "node:fs";
2
+ import { hasCredentialSource, redactCredential, resolveCredential } from "./credential-source.ts";
3
+ import { getWebSearchConfigPath } from "./utils.ts";
4
+ import {
5
+ isGeminiAdcAvailable,
6
+ getAdcAccessToken,
7
+ getAdcProject,
8
+ getAdcLocation,
9
+ getVertexApiBase,
10
+ isVertexHost,
11
+ } from "./gemini-adc.ts";
12
+
13
+ const DEFAULT_API_HOST = "https://generativelanguage.googleapis.com";
14
+ const API_VERSION = "v1beta";
15
+ export const API_BASE = `${DEFAULT_API_HOST}/${API_VERSION}`;
16
+ const CONFIG_PATH = getWebSearchConfigPath();
17
+ export const DEFAULT_MODEL = "gemini-3.6-flash";
18
+
19
+ interface GeminiApiConfig {
20
+ geminiApiKey?: unknown;
21
+ geminiBaseUrl?: unknown;
22
+ cloudflareApiKey?: unknown;
23
+ }
24
+
25
+ let cachedConfig: GeminiApiConfig | null = null;
26
+
27
+ function loadConfig(): GeminiApiConfig {
28
+ if (cachedConfig) return cachedConfig;
29
+ if (!existsSync(CONFIG_PATH)) {
30
+ cachedConfig = {};
31
+ return cachedConfig;
32
+ }
33
+
34
+ const raw = readFileSync(CONFIG_PATH, "utf-8");
35
+ try {
36
+ cachedConfig = JSON.parse(raw) as GeminiApiConfig;
37
+ return cachedConfig;
38
+ } catch (err) {
39
+ const message = err instanceof Error ? err.message : String(err);
40
+ throw new Error(`Failed to parse ${CONFIG_PATH}: ${message}`);
41
+ }
42
+ }
43
+
44
+ function withTimeout(signal: AbortSignal | undefined, timeoutMs: number): AbortSignal {
45
+ const timeout = AbortSignal.timeout(timeoutMs);
46
+ return signal ? AbortSignal.any([signal, timeout]) : timeout;
47
+ }
48
+
49
+ function normalizeApiKey(value: unknown): string | null {
50
+ if (typeof value !== "string") return null;
51
+ const normalized = value.trim();
52
+ return normalized.length > 0 ? normalized : null;
53
+ }
54
+
55
+ function normalizeBaseUrl(value: unknown): string | null {
56
+ if (typeof value !== "string") return null;
57
+ const normalized = value.trim().replace(/\/+$/, "");
58
+ return normalized.length > 0 ? normalized : null;
59
+ }
60
+
61
+ function isCloudflareGateway(): boolean {
62
+ return getApiHost().includes("gateway.ai.cloudflare.com");
63
+ }
64
+
65
+ export async function getApiKey(signal?: AbortSignal): Promise<string | null> {
66
+ return resolveCredential({
67
+ provider: "Gemini",
68
+ configuredValue: loadConfig().geminiApiKey,
69
+ environmentValue: process.env.GEMINI_API_KEY,
70
+ signal,
71
+ });
72
+ }
73
+
74
+ export function getApiHost(): string {
75
+ return (
76
+ normalizeBaseUrl(process.env.GOOGLE_GEMINI_BASE_URL) ??
77
+ normalizeBaseUrl(loadConfig().geminiBaseUrl) ??
78
+ DEFAULT_API_HOST
79
+ );
80
+ }
81
+
82
+ export function getVersionedApiBase(): string {
83
+ const project = getAdcProject();
84
+ const location = getAdcLocation();
85
+ if (isGeminiAdcAvailable() && project && location) {
86
+ return getVertexApiBase(project, location);
87
+ }
88
+ return `${getApiHost()}/${API_VERSION}`;
89
+ }
90
+
91
+ export function getUploadBase(): string {
92
+ return `${getApiHost()}/upload/${API_VERSION}`;
93
+ }
94
+
95
+ function getLegacyCloudflareApiKey(): string | null {
96
+ return normalizeApiKey(process.env.CLOUDFLARE_API_KEY) ?? normalizeApiKey(loadConfig().cloudflareApiKey);
97
+ }
98
+
99
+ async function resolveCloudflareApiKey(signal?: AbortSignal): Promise<string | null> {
100
+ return resolveCredential({
101
+ provider: "Cloudflare",
102
+ configuredValue: loadConfig().cloudflareApiKey,
103
+ environmentValue: process.env.CLOUDFLARE_API_KEY,
104
+ signal,
105
+ });
106
+ }
107
+
108
+ export function getCloudflareApiKey(): string | null {
109
+ return getLegacyCloudflareApiKey();
110
+ }
111
+
112
+ export function isGatewayConfigured(): boolean {
113
+ return isCloudflareGateway() && hasCredentialSource({
114
+ provider: "Cloudflare",
115
+ configuredValue: loadConfig().cloudflareApiKey,
116
+ environmentValue: process.env.CLOUDFLARE_API_KEY,
117
+ });
118
+ }
119
+
120
+ export function buildAuthHeaders(apiKey: string | null = null, cloudflareApiKey: string | null = getLegacyCloudflareApiKey()): Record<string, string> {
121
+ if (!isCloudflareGateway()) return apiKey ? { "x-goog-api-key": apiKey } : {};
122
+ return cloudflareApiKey ? { "cf-aig-authorization": `Bearer ${cloudflareApiKey}` } : {};
123
+ }
124
+
125
+ function redactGeminiCredentials(text: string, apiKey: string | null | undefined, cloudflareApiKey: string | null | undefined, adcToken?: string | null): string {
126
+ let out = redactCredential(redactCredential(text, apiKey), cloudflareApiKey);
127
+ if (adcToken) out = redactCredential(out, adcToken);
128
+ return out;
129
+ }
130
+
131
+ const responseCredentials = new WeakMap<Response, {
132
+ apiKey: string | null | undefined;
133
+ cloudflareApiKey: string | null | undefined;
134
+ adcToken?: string | null;
135
+ }>();
136
+
137
+ export function redactGeminiApiResponse(response: Response, text: string, apiKey?: string | null, adcToken?: string | null): string {
138
+ const credentials = responseCredentials.get(response);
139
+ return redactGeminiCredentials(text, credentials?.apiKey ?? apiKey, credentials?.cloudflareApiKey, credentials?.adcToken ?? adcToken);
140
+ }
141
+
142
+ export async function fetchGeminiApi(
143
+ url: string | URL,
144
+ init: RequestInit = {},
145
+ apiKey?: string | null,
146
+ ): Promise<Response> {
147
+ const parsedUrl = new URL(url);
148
+ for (const name of parsedUrl.searchParams.keys()) {
149
+ if (["key", "api_key"].includes(name.toLowerCase())) {
150
+ throw new Error("Gemini API credential query parameters are not allowed");
151
+ }
152
+ }
153
+ const project = getAdcProject();
154
+ const location = getAdcLocation();
155
+ const adcMode = isGeminiAdcAvailable() && !!project && !!location && isVertexHost(parsedUrl.origin);
156
+ let adcToken: string | null = null;
157
+ if (adcMode) {
158
+ adcToken = await getAdcAccessToken(init.signal ?? undefined);
159
+ }
160
+ const resolvedApiKey = adcMode ? null : (apiKey === undefined ? await getApiKey(init.signal ?? undefined) : apiKey);
161
+ const cloudflareApiKey = isCloudflareGateway() ? await resolveCloudflareApiKey(init.signal ?? undefined) : null;
162
+ const allowedOrigins = new Set([
163
+ new URL(getApiHost()).origin,
164
+ new URL(DEFAULT_API_HOST).origin,
165
+ ...(adcMode ? [parsedUrl.origin] : []),
166
+ ]);
167
+ const needsAuth = resolvedApiKey || isGatewayConfigured() || adcMode;
168
+ if (needsAuth && !allowedOrigins.has(parsedUrl.origin)) {
169
+ throw new Error("Gemini API request host is not allowed");
170
+ }
171
+ const headers = new Headers(init.headers);
172
+ headers.delete("x-goog-api-key");
173
+ headers.delete("cf-aig-authorization");
174
+ headers.delete("authorization");
175
+ if (adcMode && adcToken) {
176
+ headers.set("Authorization", `Bearer ${adcToken}`);
177
+ } else {
178
+ for (const [name, value] of Object.entries(buildAuthHeaders(resolvedApiKey, cloudflareApiKey))) {
179
+ headers.set(name, value);
180
+ }
181
+ }
182
+ try {
183
+ const response = await fetch(parsedUrl, { ...init, headers });
184
+ responseCredentials.set(response, { apiKey: resolvedApiKey, cloudflareApiKey, adcToken });
185
+ return response;
186
+ } catch (error) {
187
+ const message = error instanceof Error ? error.message : String(error);
188
+ const redactedMessage = redactGeminiCredentials(message, resolvedApiKey, cloudflareApiKey, adcToken);
189
+ if (redactedMessage === message) throw error;
190
+ const redactedError = new Error(redactedMessage);
191
+ if (error instanceof Error) redactedError.name = error.name;
192
+ throw redactedError;
193
+ }
194
+ }
195
+
196
+ function hasGeminiApiKeySource(): boolean {
197
+ return hasCredentialSource({
198
+ provider: "Gemini",
199
+ configuredValue: loadConfig().geminiApiKey,
200
+ environmentValue: process.env.GEMINI_API_KEY,
201
+ });
202
+ }
203
+
204
+ export function isGeminiApiAvailable(): boolean {
205
+ return isGeminiAdcAvailable() || hasGeminiApiKeySource() || isGatewayConfigured();
206
+ }
207
+
208
+ /**
209
+ * Whether the Gemini Files-API paths (YouTube and local video analysis) can
210
+ * authenticate. These use the Gemini Files API and metadata endpoints, which do
211
+ * not support ADC, so they still require an API key or a Cloudflare gateway.
212
+ */
213
+ export function isGeminiApiAvailableWithVideo(): boolean {
214
+ return hasGeminiApiKeySource() || isGatewayConfigured();
215
+ }
216
+
217
+ export interface GeminiApiOptions {
218
+ apiKey?: string;
219
+ model?: string;
220
+ mimeType?: string;
221
+ signal?: AbortSignal;
222
+ timeoutMs?: number;
223
+ }
224
+
225
+ export interface GeminiGenerateContentResult {
226
+ text: string;
227
+ finishReason?: string;
228
+ blockReason?: string;
229
+ }
230
+
231
+ export async function queryGeminiApiWithInlineData(
232
+ prompt: string,
233
+ data: string,
234
+ mimeType: string,
235
+ options: GeminiApiOptions = {},
236
+ ): Promise<GeminiGenerateContentResult> {
237
+ const signal = withTimeout(options.signal, options.timeoutMs ?? 120000);
238
+ const apiKey = isGeminiAdcAvailable() ? null : (options.apiKey ?? await getApiKey(signal));
239
+ if (!apiKey && !isGatewayConfigured() && !isGeminiAdcAvailable()) {
240
+ throw new Error(
241
+ "Gemini API not configured. Either:\n" +
242
+ ` 1. Configure geminiApiKey in ${CONFIG_PATH} or set GEMINI_API_KEY\n` +
243
+ " 2. Set GOOGLE_GEMINI_BASE_URL + CLOUDFLARE_API_KEY for Cloudflare AI Gateway routing\n" +
244
+ " 3. Set geminiAuth to \"adc\" with a Google Cloud ADC (GOOGLE_APPLICATION_CREDENTIALS) + project/location"
245
+ );
246
+ }
247
+
248
+ const model = options.model ?? DEFAULT_MODEL;
249
+ const url = `${getVersionedApiBase()}/models/${model}:generateContent`;
250
+ const body = {
251
+ contents: [
252
+ {
253
+ role: "user",
254
+ parts: [
255
+ { inlineData: { mimeType, data } },
256
+ { text: prompt },
257
+ ],
258
+ },
259
+ ],
260
+ };
261
+
262
+ const res = await fetchGeminiApi(url, {
263
+ method: "POST",
264
+ headers: { "Content-Type": "application/json" },
265
+ body: JSON.stringify(body),
266
+ signal,
267
+ }, apiKey);
268
+
269
+ if (!res.ok) {
270
+ const errorText = redactGeminiApiResponse(res, await res.text(), apiKey);
271
+ throw new Error(`Gemini API error ${res.status}: ${errorText.slice(0, 300)}`);
272
+ }
273
+
274
+ const response = (await res.json()) as GenerateContentResponse;
275
+ const candidate = response.candidates?.[0];
276
+ const text = candidate?.content?.parts
277
+ ?.map((part) => part.text)
278
+ .filter((part): part is string => typeof part === "string" && part.length > 0)
279
+ .join("\n") ?? "";
280
+
281
+ return {
282
+ text,
283
+ ...(candidate?.finishReason ? { finishReason: candidate.finishReason } : {}),
284
+ ...(response.promptFeedback?.blockReason ? { blockReason: response.promptFeedback.blockReason } : {}),
285
+ };
286
+ }
287
+
288
+ export async function queryGeminiApiWithVideo(
289
+ prompt: string,
290
+ videoUri: string,
291
+ options: GeminiApiOptions = {},
292
+ ): Promise<string> {
293
+ const signal = withTimeout(options.signal, options.timeoutMs ?? 120000);
294
+ const apiKey = options.apiKey ?? await getApiKey(signal);
295
+ if (!apiKey && !isGatewayConfigured()) {
296
+ throw new Error(
297
+ "Gemini API not configured. Either:\n" +
298
+ ` 1. Configure geminiApiKey in ${CONFIG_PATH} or set GEMINI_API_KEY\n` +
299
+ " 2. Set GOOGLE_GEMINI_BASE_URL + CLOUDFLARE_API_KEY for Cloudflare AI Gateway routing"
300
+ );
301
+ }
302
+
303
+ const model = options.model ?? DEFAULT_MODEL;
304
+ const url = `${getVersionedApiBase()}/models/${model}:generateContent`;
305
+
306
+ const fileData: Record<string, string> = { fileUri: videoUri };
307
+ if (options.mimeType) fileData.mimeType = options.mimeType;
308
+
309
+ const body = {
310
+ contents: [
311
+ {
312
+ role: "user",
313
+ parts: [
314
+ { fileData },
315
+ { text: prompt },
316
+ ],
317
+ },
318
+ ],
319
+ };
320
+
321
+ const res = await fetchGeminiApi(url, {
322
+ method: "POST",
323
+ headers: { "Content-Type": "application/json" },
324
+ body: JSON.stringify(body),
325
+ signal,
326
+ }, apiKey);
327
+
328
+ if (!res.ok) {
329
+ const errorText = redactGeminiApiResponse(res, await res.text(), apiKey);
330
+ throw new Error(`Gemini API error ${res.status}: ${errorText.slice(0, 300)}`);
331
+ }
332
+
333
+ const data = (await res.json()) as GenerateContentResponse;
334
+ const text = data.candidates?.[0]?.content?.parts
335
+ ?.map((p) => p.text)
336
+ .filter(Boolean)
337
+ .join("\n");
338
+
339
+ if (!text) throw new Error("Gemini API returned empty response");
340
+ return text;
341
+ }
342
+
343
+ interface GenerateContentResponse {
344
+ candidates?: Array<{
345
+ content?: {
346
+ parts?: Array<{ text?: string }>;
347
+ };
348
+ finishReason?: string;
349
+ }>;
350
+ promptFeedback?: {
351
+ blockReason?: string;
352
+ };
353
+ }