@asterxsk/kiln 0.1.0 → 0.3.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +155 -170
- package/agent/AGENTS.md +67 -67
- package/agent/README.md +5 -5
- package/agent/extensions/ask-user/index.ts +418 -418
- package/agent/extensions/ask-user/install.ps1 +4 -0
- package/agent/extensions/ask-user/install.sh +3 -0
- package/agent/extensions/ask-user/package-lock.json +769 -769
- package/agent/extensions/ask-user/package.json +19 -19
- package/agent/extensions/ask-user/prompt.ts +45 -45
- package/agent/extensions/ask-user/tsconfig.json +7 -7
- package/agent/extensions/background-terminals/docs/implementation-guide.md +942 -942
- package/agent/extensions/background-terminals/index.ts +627 -627
- package/agent/extensions/background-terminals/install.ps1 +4 -0
- package/agent/extensions/background-terminals/install.sh +3 -0
- package/agent/extensions/background-terminals/manager.test.ts +735 -735
- package/agent/extensions/background-terminals/output.test.ts +109 -109
- package/agent/extensions/background-terminals/package-lock.json +769 -769
- package/agent/extensions/background-terminals/package.json +17 -17
- package/agent/extensions/background-terminals/prompt.test.ts +125 -125
- package/agent/extensions/background-terminals/ps.test.ts +82 -82
- package/agent/extensions/background-terminals/result-delivery.test.ts +44 -44
- package/agent/extensions/background-terminals/src/domain.ts +87 -87
- package/agent/extensions/background-terminals/src/manager.ts +907 -907
- package/agent/extensions/background-terminals/src/output.ts +84 -84
- package/agent/extensions/background-terminals/src/prompt.ts +142 -142
- package/agent/extensions/background-terminals/src/result-delivery.ts +27 -27
- package/agent/extensions/background-terminals/src/runtime.ts +36 -36
- package/agent/extensions/background-terminals/src/ui/output-view.ts +79 -79
- package/agent/extensions/background-terminals/src/ui/ps.ts +621 -621
- package/agent/extensions/background-terminals/tsconfig.json +7 -7
- package/agent/extensions/destructive/install.ps1 +4 -0
- package/agent/extensions/destructive/install.sh +3 -0
- package/agent/extensions/file-search/index.spec.ts +443 -443
- package/agent/extensions/file-search/index.ts +459 -459
- package/agent/extensions/file-search/install.ps1 +4 -0
- package/agent/extensions/file-search/install.sh +3 -0
- package/agent/extensions/file-search/package-lock.json +2253 -2253
- package/agent/extensions/file-search/package.json +23 -23
- package/agent/extensions/file-search/src/args.ts +122 -122
- package/agent/extensions/file-search/src/binaries.ts +422 -422
- package/agent/extensions/file-search/src/output.ts +126 -126
- package/agent/extensions/file-search/src/process.ts +146 -146
- package/agent/extensions/file-search/src/prompt.ts +52 -52
- package/agent/extensions/file-search/tsconfig.json +7 -7
- package/agent/extensions/goal/install.ps1 +4 -0
- package/agent/extensions/goal/install.sh +3 -0
- package/agent/extensions/herdr-agent-state.ts +257 -0
- package/agent/extensions/kiln-update/README.md +25 -0
- package/agent/extensions/kiln-update/index.ts +99 -0
- package/agent/extensions/kiln-update/install.ps1 +27 -0
- package/agent/extensions/kiln-update/install.sh +24 -0
- package/agent/extensions/modelconf/PLAN.md +915 -915
- package/agent/extensions/modelconf/index.ts +296 -296
- package/agent/extensions/modelconf/install.ps1 +4 -0
- package/agent/extensions/modelconf/install.sh +3 -0
- package/agent/extensions/modelconf/src/ui/ModelConfView.ts +1101 -1101
- package/agent/extensions/pi-web-access/CHANGELOG.md +690 -690
- package/agent/extensions/pi-web-access/LICENSE +21 -21
- package/agent/extensions/pi-web-access/README.md +470 -470
- package/agent/extensions/pi-web-access/SECURITY.md +5 -5
- package/agent/extensions/pi-web-access/activity.ts +101 -101
- package/agent/extensions/pi-web-access/auth-fetch.ts +148 -148
- package/agent/extensions/pi-web-access/brightdata-unlocker.ts +272 -272
- package/agent/extensions/pi-web-access/chrome-cookies.ts +669 -669
- package/agent/extensions/pi-web-access/content-find.ts +139 -139
- package/agent/extensions/pi-web-access/credential-source.ts +191 -191
- package/agent/extensions/pi-web-access/data-uri-sanitize.ts +406 -406
- package/agent/extensions/pi-web-access/datalab-pdf-extract.ts +568 -568
- package/agent/extensions/pi-web-access/declared-web-links.ts +173 -173
- package/agent/extensions/pi-web-access/evidence/CONTRACT-EVIDENCE.md +496 -496
- package/agent/extensions/pi-web-access/evidence/contract-probe.mjs +140 -140
- package/agent/extensions/pi-web-access/exa.ts +526 -526
- package/agent/extensions/pi-web-access/extract.ts +1196 -1196
- package/agent/extensions/pi-web-access/feature-config.ts +29 -29
- package/agent/extensions/pi-web-access/fetch-params.ts +111 -111
- package/agent/extensions/pi-web-access/gemini-adc.ts +298 -298
- package/agent/extensions/pi-web-access/gemini-api.ts +353 -353
- package/agent/extensions/pi-web-access/gemini-pdf-extract.ts +108 -108
- package/agent/extensions/pi-web-access/gemini-url-context.ts +128 -128
- package/agent/extensions/pi-web-access/gemini-web-config.ts +101 -101
- package/agent/extensions/pi-web-access/gemini-web.ts +487 -487
- package/agent/extensions/pi-web-access/github-api.ts +197 -197
- package/agent/extensions/pi-web-access/github-extract.ts +746 -746
- package/agent/extensions/pi-web-access/github-issue-pr.ts +700 -700
- package/agent/extensions/pi-web-access/index.ts +1737 -1737
- package/agent/extensions/pi-web-access/package-lock.json +5808 -5808
- package/agent/extensions/pi-web-access/package.json +64 -64
- package/agent/extensions/pi-web-access/page-query.ts +96 -96
- package/agent/extensions/pi-web-access/pdf-extract.ts +409 -409
- package/agent/extensions/pi-web-access/promise-try.d.ts +7 -7
- package/agent/extensions/pi-web-access/query-rewrite.ts +51 -51
- package/agent/extensions/pi-web-access/render-search-error.ts +170 -170
- package/agent/extensions/pi-web-access/rsc-extract.ts +338 -338
- package/agent/extensions/pi-web-access/source-check.ts +282 -282
- package/agent/extensions/pi-web-access/ssrf-protection.ts +526 -526
- package/agent/extensions/pi-web-access/storage.ts +521 -521
- package/agent/extensions/pi-web-access/summary-model-scope.ts +125 -125
- package/agent/extensions/pi-web-access/test/auth-fetch.test.mjs +208 -208
- package/agent/extensions/pi-web-access/test/brightdata-unlocker.test.mjs +840 -840
- package/agent/extensions/pi-web-access/test/chrome-cookie-extraction.test.mjs +441 -441
- package/agent/extensions/pi-web-access/test/config-path.test.mjs +283 -283
- package/agent/extensions/pi-web-access/test/content-find.test.mjs +25 -25
- package/agent/extensions/pi-web-access/test/credential-source.test.mjs +118 -118
- package/agent/extensions/pi-web-access/test/data-uri-sanitize.test.mjs +210 -210
- package/agent/extensions/pi-web-access/test/datalab-pdf-extract.test.mjs +552 -552
- package/agent/extensions/pi-web-access/test/declared-web-links.test.mjs +212 -212
- package/agent/extensions/pi-web-access/test/fetch-answer-storage.test.mjs +40 -40
- package/agent/extensions/pi-web-access/test/fetch-cache-storage.test.mjs +334 -334
- package/agent/extensions/pi-web-access/test/fetch-content-domain-policy.test.mjs +95 -95
- package/agent/extensions/pi-web-access/test/fetch-modes.test.mjs +53 -53
- package/agent/extensions/pi-web-access/test/fetch-not-found-guidance.test.mjs +92 -92
- package/agent/extensions/pi-web-access/test/fetch-params.test.mjs +86 -86
- package/agent/extensions/pi-web-access/test/fetch-render-call.test.mjs +34 -34
- package/agent/extensions/pi-web-access/test/fetch-routing.test.mjs +173 -173
- package/agent/extensions/pi-web-access/test/gemini-adc-auth.test.mjs +257 -257
- package/agent/extensions/pi-web-access/test/gemini-api-transport.test.mjs +170 -170
- package/agent/extensions/pi-web-access/test/gemini-pdf-extract.test.mjs +133 -133
- package/agent/extensions/pi-web-access/test/gemini-web-cookie-opt-in.test.mjs +178 -178
- package/agent/extensions/pi-web-access/test/gemini-web-header-overflow.test.mjs +148 -148
- package/agent/extensions/pi-web-access/test/get-search-content.test.mjs +223 -223
- package/agent/extensions/pi-web-access/test/github-extract.test.mjs +378 -378
- package/agent/extensions/pi-web-access/test/github-issue-pr.test.mjs +565 -565
- package/agent/extensions/pi-web-access/test/inline-content-config.test.mjs +99 -99
- package/agent/extensions/pi-web-access/test/lazy-extract-load.test.mjs +118 -118
- package/agent/extensions/pi-web-access/test/local-video-oversize.test.mjs +52 -52
- package/agent/extensions/pi-web-access/test/package-typebox-dependency.test.mjs +50 -50
- package/agent/extensions/pi-web-access/test/page-query.test.mjs +51 -51
- package/agent/extensions/pi-web-access/test/pdf-config.test.mjs +140 -140
- package/agent/extensions/pi-web-access/test/pdf-extract.test.mjs +500 -500
- package/agent/extensions/pi-web-access/test/proxy-transport.test.mjs +286 -286
- package/agent/extensions/pi-web-access/test/query-rewrite.test.mjs +52 -52
- package/agent/extensions/pi-web-access/test/rsc-fallback.test.mjs +102 -102
- package/agent/extensions/pi-web-access/test/search-error-render.test.mjs +152 -152
- package/agent/extensions/pi-web-access/test/search-providers.test.mjs +274 -274
- package/agent/extensions/pi-web-access/test/source-check.test.mjs +179 -179
- package/agent/extensions/pi-web-access/test/ssrf-allow-ranges-config.test.mjs +205 -205
- package/agent/extensions/pi-web-access/test/ssrf-protection.test.mjs +456 -456
- package/agent/extensions/pi-web-access/test/tool-registration-config.test.mjs +182 -182
- package/agent/extensions/pi-web-access/test/youtube-extract-errors.test.mjs +64 -64
- package/agent/extensions/pi-web-access/tsconfig.json +11 -11
- package/agent/extensions/pi-web-access/utils.ts +451 -451
- package/agent/extensions/pi-web-access/video-extract.ts +392 -392
- package/agent/extensions/pi-web-access/youtube-extract.ts +328 -328
- package/agent/extensions/shared/activity-status.ts +31 -31
- package/agent/extensions/shared/child-session.test.ts +270 -270
- package/agent/extensions/shared/child-session.ts +148 -148
- package/agent/extensions/shared/context-utilization.test.ts +48 -48
- package/agent/extensions/shared/context-utilization.ts +47 -47
- package/agent/extensions/shared/dashboard-state.ts +99 -99
- package/agent/extensions/shared/install.ps1 +4 -0
- package/agent/extensions/shared/install.sh +3 -0
- package/agent/extensions/shared/tool-call-timeout.test.ts +117 -117
- package/agent/extensions/shared/tool-call-timeout.ts +104 -104
- package/agent/extensions/skillsconf/install.ps1 +4 -0
- package/agent/extensions/skillsconf/install.sh +3 -0
- package/agent/extensions/statusline/index.ts +282 -0
- package/agent/extensions/statusline/install.ps1 +27 -0
- package/agent/extensions/statusline/install.sh +24 -0
- package/agent/extensions/subagents/by-the-way.test.ts +29 -29
- package/agent/extensions/subagents/claude.test.ts +119 -119
- package/agent/extensions/subagents/codex.test.ts +102 -102
- package/agent/extensions/subagents/context-usage.test.ts +107 -107
- package/agent/extensions/subagents/docs/design-plan.md +568 -568
- package/agent/extensions/subagents/docs/effect-v4-extension-guide.md +354 -354
- package/agent/extensions/subagents/docs/effect-v4-notes.md +571 -571
- package/agent/extensions/subagents/index.ts +779 -779
- package/agent/extensions/subagents/install.ps1 +4 -0
- package/agent/extensions/subagents/install.sh +3 -0
- package/agent/extensions/subagents/manager.test.ts +276 -276
- package/agent/extensions/subagents/package-lock.json +2244 -2244
- package/agent/extensions/subagents/package.json +19 -19
- package/agent/extensions/subagents/result-delivery.test.ts +27 -27
- package/agent/extensions/subagents/src/backend.ts +73 -73
- package/agent/extensions/subagents/src/backends/claude.ts +701 -701
- package/agent/extensions/subagents/src/backends/codex.ts +1060 -1060
- package/agent/extensions/subagents/src/backends/pi.ts +575 -575
- package/agent/extensions/subagents/src/backends/stub.ts +300 -300
- package/agent/extensions/subagents/src/by-the-way.ts +21 -21
- package/agent/extensions/subagents/src/domain.ts +253 -253
- package/agent/extensions/subagents/src/format.ts +74 -74
- package/agent/extensions/subagents/src/manager.ts +736 -736
- package/agent/extensions/subagents/src/prompt.ts +92 -92
- package/agent/extensions/subagents/src/result-delivery.ts +20 -20
- package/agent/extensions/subagents/src/runtime.ts +53 -53
- package/agent/extensions/subagents/src/ui/takeover.ts +583 -583
- package/agent/extensions/subagents/src/ui/transcript.ts +201 -201
- package/agent/extensions/subagents/takeover.test.ts +29 -29
- package/agent/extensions/subagents/tsconfig.json +7 -7
- package/agent/extensions/taste/install.ps1 +4 -0
- package/agent/extensions/taste/install.sh +3 -0
- package/agent/extensions/todo/AGENTS.md +38 -38
- package/agent/extensions/todo/LICENSE +21 -21
- package/agent/extensions/todo/config.ts +55 -55
- package/agent/extensions/todo/index.ts +151 -151
- package/agent/extensions/todo/install.ps1 +4 -0
- package/agent/extensions/todo/install.sh +3 -0
- package/agent/extensions/todo/locales/de.json +17 -17
- package/agent/extensions/todo/locales/en.json +15 -15
- package/agent/extensions/todo/locales/es.json +17 -17
- package/agent/extensions/todo/locales/fr.json +17 -17
- package/agent/extensions/todo/locales/pt-BR.json +17 -17
- package/agent/extensions/todo/locales/pt.json +17 -17
- package/agent/extensions/todo/locales/ru.json +17 -17
- package/agent/extensions/todo/locales/uk.json +17 -17
- package/agent/extensions/todo/locales/zh.json +17 -17
- package/agent/extensions/todo/package-lock.json +3358 -3358
- package/agent/extensions/todo/package.json +67 -67
- package/agent/extensions/todo/state/i18n-bridge.ts +64 -64
- package/agent/extensions/todo/state/invariants.ts +20 -20
- package/agent/extensions/todo/state/replay.ts +38 -38
- package/agent/extensions/todo/state/selectors.ts +107 -107
- package/agent/extensions/todo/state/state-reducer.ts +326 -326
- package/agent/extensions/todo/state/state.ts +18 -18
- package/agent/extensions/todo/state/store.ts +82 -82
- package/agent/extensions/todo/state/task-graph.ts +57 -57
- package/agent/extensions/todo/todo-overlay.ts +200 -200
- package/agent/extensions/todo/todo.ts +155 -155
- package/agent/extensions/todo/tool/response-envelope.ts +109 -109
- package/agent/extensions/todo/tool/types.ts +206 -206
- package/agent/extensions/todo/view/format.ts +177 -177
- package/agent/extensions/trim-context/install.ps1 +4 -0
- package/agent/extensions/trim-context/install.sh +3 -0
- package/agent/keybindings.json +7 -7
- package/agent/version.txt +1 -0
- package/agent.bak.20260905-190020/AGENTS.md +67 -0
- package/agent.bak.20260905-190020/README.md +5 -0
- package/agent.bak.20260905-190020/extensions/ask-user/index.ts +418 -0
- package/agent.bak.20260905-190020/extensions/ask-user/package-lock.json +769 -0
- package/agent.bak.20260905-190020/extensions/ask-user/package.json +19 -0
- package/agent.bak.20260905-190020/extensions/ask-user/prompt.ts +45 -0
- package/agent.bak.20260905-190020/extensions/ask-user/tsconfig.json +7 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/docs/implementation-guide.md +942 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/index.ts +629 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/package-lock.json +769 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/package.json +17 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/src/domain.ts +87 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/src/manager.ts +907 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/src/output.ts +84 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/src/prompt.ts +142 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/src/result-delivery.ts +27 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/src/runtime.ts +36 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/src/ui/output-view.ts +79 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/src/ui/ps.ts +621 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/src/widget-order.ts +121 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/test/manager.test.ts +735 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/test/output.test.ts +109 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/test/prompt.test.ts +125 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/test/ps.test.ts +82 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/test/result-delivery.test.ts +44 -0
- package/agent.bak.20260905-190020/extensions/background-terminals/tsconfig.json +7 -0
- package/agent.bak.20260905-190020/extensions/destructive/README.md +34 -0
- package/agent.bak.20260905-190020/extensions/destructive/index.ts +124 -0
- package/agent.bak.20260905-190020/extensions/destructive/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/destructive/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/file-search/index.ts +459 -0
- package/agent.bak.20260905-190020/extensions/file-search/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/file-search/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/file-search/package-lock.json +2253 -0
- package/agent.bak.20260905-190020/extensions/file-search/package.json +23 -0
- package/agent.bak.20260905-190020/extensions/file-search/src/args.ts +122 -0
- package/agent.bak.20260905-190020/extensions/file-search/src/binaries.ts +422 -0
- package/agent.bak.20260905-190020/extensions/file-search/src/output.ts +126 -0
- package/agent.bak.20260905-190020/extensions/file-search/src/process.ts +146 -0
- package/agent.bak.20260905-190020/extensions/file-search/src/prompt.ts +52 -0
- package/agent.bak.20260905-190020/extensions/file-search/test/index.spec.ts +443 -0
- package/agent.bak.20260905-190020/extensions/file-search/tsconfig.json +7 -0
- package/agent.bak.20260905-190020/extensions/goal/README.md +50 -0
- package/agent.bak.20260905-190020/extensions/goal/index.ts +155 -0
- package/agent.bak.20260905-190020/extensions/goal/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/goal/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/kiln-update/README.md +25 -0
- package/agent.bak.20260905-190020/extensions/kiln-update/index.ts +99 -0
- package/agent.bak.20260905-190020/extensions/kiln-update/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/kiln-update/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/modelconf/PLAN.md +915 -0
- package/agent.bak.20260905-190020/extensions/modelconf/README.md +66 -0
- package/agent.bak.20260905-190020/extensions/modelconf/index.ts +296 -0
- package/agent.bak.20260905-190020/extensions/modelconf/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/modelconf/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/modelconf/package-lock.json +1809 -0
- package/agent.bak.20260905-190020/extensions/modelconf/package.json +13 -0
- package/agent.bak.20260905-190020/extensions/modelconf/src/fuzzy.ts +26 -0
- package/agent.bak.20260905-190020/extensions/modelconf/src/glob.ts +13 -0
- package/agent.bak.20260905-190020/extensions/modelconf/src/persistence.test.ts +46 -0
- package/agent.bak.20260905-190020/extensions/modelconf/src/persistence.ts +164 -0
- package/agent.bak.20260905-190020/extensions/modelconf/src/ui/ModelConfView.test.ts +75 -0
- package/agent.bak.20260905-190020/extensions/modelconf/src/ui/ModelConfView.ts +1101 -0
- package/agent.bak.20260905-190020/extensions/modelconf/tsconfig.json +15 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/CHANGELOG.md +690 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/LICENSE +21 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/README.md +470 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/SECURITY.md +5 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/activity.ts +101 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/auth-fetch.ts +148 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/banner.png +0 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/brightdata-unlocker.ts +272 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/chrome-cookies.ts +669 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/content-find.ts +139 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/credential-source.ts +191 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/data-uri-sanitize.ts +406 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/datalab-pdf-extract.ts +568 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/declared-web-links.ts +173 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/evidence/CONTRACT-EVIDENCE.md +496 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/evidence/contract-probe.mjs +140 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/exa.ts +526 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/extract.ts +1196 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/feature-config.ts +29 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/fetch-params.ts +111 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/gemini-adc.ts +298 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/gemini-api.ts +353 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/gemini-pdf-extract.ts +108 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/gemini-search.ts +21 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/gemini-url-context.ts +128 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/gemini-web-config.ts +101 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/gemini-web.ts +487 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/github-api.ts +197 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/github-extract.ts +746 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/github-issue-pr.ts +700 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/index.ts +1737 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/package-lock.json +5808 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/package.json +64 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/page-query.ts +96 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/pdf-extract.ts +409 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/pi-web-fetch-demo.mp4 +0 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/promise-try.d.ts +7 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/query-rewrite.ts +51 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/render-search-error.ts +170 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/rsc-extract.ts +338 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/search-types.ts +20 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/source-check.ts +282 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/ssrf-protection.ts +526 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/storage.ts +521 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/summary-model-scope.ts +125 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/auth-fetch.test.mjs +208 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/brightdata-unlocker.test.mjs +840 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/chrome-cookie-extraction.test.mjs +441 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/config-path.test.mjs +283 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/content-find.test.mjs +25 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/credential-source.test.mjs +118 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/data-uri-sanitize.test.mjs +210 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/datalab-pdf-extract.test.mjs +552 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/declared-web-links.test.mjs +212 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/fetch-answer-storage.test.mjs +40 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/fetch-cache-storage.test.mjs +334 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/fetch-content-domain-policy.test.mjs +95 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/fetch-modes.test.mjs +53 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/fetch-not-found-guidance.test.mjs +92 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/fetch-params.test.mjs +86 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/fetch-render-call.test.mjs +34 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/fetch-routing.test.mjs +173 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/gemini-adc-auth.test.mjs +257 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/gemini-api-transport.test.mjs +170 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/gemini-pdf-extract.test.mjs +133 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/gemini-web-cookie-opt-in.test.mjs +178 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/gemini-web-header-overflow.test.mjs +148 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/get-search-content.test.mjs +223 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/github-extract.test.mjs +378 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/github-issue-pr.test.mjs +565 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/inline-content-config.test.mjs +99 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/lazy-extract-load.test.mjs +118 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/local-video-oversize.test.mjs +52 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/package-typebox-dependency.test.mjs +50 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/page-query.test.mjs +51 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/pdf-config.test.mjs +140 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/pdf-extract.test.mjs +500 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/proxy-transport.test.mjs +286 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/query-rewrite.test.mjs +52 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/rsc-fallback.test.mjs +102 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/search-error-render.test.mjs +152 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/search-providers.test.mjs +274 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/source-check.test.mjs +179 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/ssrf-allow-ranges-config.test.mjs +205 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/ssrf-protection.test.mjs +456 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/summary-model-scope.test.mjs +106 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/tool-registration-config.test.mjs +182 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/web-search-answer-render.test.mjs +66 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/test/youtube-extract-errors.test.mjs +64 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/tsconfig.json +11 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/utils.ts +451 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/video-extract.ts +392 -0
- package/agent.bak.20260905-190020/extensions/pi-web-access/youtube-extract.ts +328 -0
- package/agent.bak.20260905-190020/extensions/shared/activity-status.ts +31 -0
- package/agent.bak.20260905-190020/extensions/shared/child-session.ts +148 -0
- package/agent.bak.20260905-190020/extensions/shared/context-utilization.ts +47 -0
- package/agent.bak.20260905-190020/extensions/shared/dashboard-state.ts +99 -0
- package/agent.bak.20260905-190020/extensions/shared/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/shared/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/shared/test/child-session.test.ts +270 -0
- package/agent.bak.20260905-190020/extensions/shared/test/context-utilization.test.ts +48 -0
- package/agent.bak.20260905-190020/extensions/shared/test/tool-call-timeout.test.ts +117 -0
- package/agent.bak.20260905-190020/extensions/shared/test/widget-order.test.ts +167 -0
- package/agent.bak.20260905-190020/extensions/shared/tool-call-timeout.ts +104 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/README.md +74 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/index.ts +92 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/package-lock.json +1809 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/package.json +18 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/delete-skill.test.ts +64 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/delete-skill.ts +45 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/filter.test.ts +80 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/filter.ts +74 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/fuzzy.ts +26 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/persistence.test.ts +72 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/persistence.ts +169 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/ui/SkillConfView.test.ts +337 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/ui/SkillConfView.ts +758 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/ui/text-input.ts +91 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/src/ui/tui-helpers.ts +144 -0
- package/agent.bak.20260905-190020/extensions/skillsconf/tsconfig.json +15 -0
- package/agent.bak.20260905-190020/extensions/statusline/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/statusline/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/subagents/docs/design-plan.md +568 -0
- package/agent.bak.20260905-190020/extensions/subagents/docs/effect-v4-extension-guide.md +354 -0
- package/agent.bak.20260905-190020/extensions/subagents/docs/effect-v4-notes.md +571 -0
- package/agent.bak.20260905-190020/extensions/subagents/index.ts +826 -0
- package/agent.bak.20260905-190020/extensions/subagents/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/subagents/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/subagents/package-lock.json +2244 -0
- package/agent.bak.20260905-190020/extensions/subagents/package.json +19 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/backend.ts +73 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/backends/claude.ts +701 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/backends/codex.ts +1060 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/backends/pi.ts +575 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/backends/stub.ts +300 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/by-the-way.ts +21 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/domain.ts +253 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/format.ts +74 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/manager.ts +736 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/prompt.ts +92 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/result-delivery.ts +20 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/runtime.ts +53 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/ui/takeover.ts +583 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/ui/transcript.ts +201 -0
- package/agent.bak.20260905-190020/extensions/subagents/src/widget-order.ts +121 -0
- package/agent.bak.20260905-190020/extensions/subagents/test/by-the-way.test.ts +29 -0
- package/agent.bak.20260905-190020/extensions/subagents/test/claude.test.ts +119 -0
- package/agent.bak.20260905-190020/extensions/subagents/test/codex.test.ts +102 -0
- package/agent.bak.20260905-190020/extensions/subagents/test/context-usage.test.ts +107 -0
- package/agent.bak.20260905-190020/extensions/subagents/test/manager.test.ts +276 -0
- package/agent.bak.20260905-190020/extensions/subagents/test/result-delivery.test.ts +27 -0
- package/agent.bak.20260905-190020/extensions/subagents/test/takeover.test.ts +29 -0
- package/agent.bak.20260905-190020/extensions/subagents/tsconfig.json +7 -0
- package/agent.bak.20260905-190020/extensions/taste/index.ts +443 -0
- package/agent.bak.20260905-190020/extensions/taste/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/taste/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/todo/AGENTS.md +38 -0
- package/agent.bak.20260905-190020/extensions/todo/LICENSE +21 -0
- package/agent.bak.20260905-190020/extensions/todo/config.ts +55 -0
- package/agent.bak.20260905-190020/extensions/todo/index.ts +151 -0
- package/agent.bak.20260905-190020/extensions/todo/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/todo/install.sh +21 -0
- package/agent.bak.20260905-190020/extensions/todo/locales/de.json +17 -0
- package/agent.bak.20260905-190020/extensions/todo/locales/en.json +15 -0
- package/agent.bak.20260905-190020/extensions/todo/locales/es.json +17 -0
- package/agent.bak.20260905-190020/extensions/todo/locales/fr.json +17 -0
- package/agent.bak.20260905-190020/extensions/todo/locales/pt-BR.json +17 -0
- package/agent.bak.20260905-190020/extensions/todo/locales/pt.json +17 -0
- package/agent.bak.20260905-190020/extensions/todo/locales/ru.json +17 -0
- package/agent.bak.20260905-190020/extensions/todo/locales/uk.json +17 -0
- package/agent.bak.20260905-190020/extensions/todo/locales/zh.json +17 -0
- package/agent.bak.20260905-190020/extensions/todo/package-lock.json +3358 -0
- package/agent.bak.20260905-190020/extensions/todo/package.json +67 -0
- package/agent.bak.20260905-190020/extensions/todo/state/i18n-bridge.ts +64 -0
- package/agent.bak.20260905-190020/extensions/todo/state/invariants.ts +20 -0
- package/agent.bak.20260905-190020/extensions/todo/state/replay.ts +38 -0
- package/agent.bak.20260905-190020/extensions/todo/state/selectors.ts +107 -0
- package/agent.bak.20260905-190020/extensions/todo/state/state-reducer.ts +326 -0
- package/agent.bak.20260905-190020/extensions/todo/state/state.ts +18 -0
- package/agent.bak.20260905-190020/extensions/todo/state/store.ts +82 -0
- package/agent.bak.20260905-190020/extensions/todo/state/task-graph.ts +57 -0
- package/agent.bak.20260905-190020/extensions/todo/todo-overlay.ts +204 -0
- package/agent.bak.20260905-190020/extensions/todo/todo.ts +155 -0
- package/agent.bak.20260905-190020/extensions/todo/tool/response-envelope.ts +109 -0
- package/agent.bak.20260905-190020/extensions/todo/tool/types.ts +206 -0
- package/agent.bak.20260905-190020/extensions/todo/view/format.ts +177 -0
- package/agent.bak.20260905-190020/extensions/todo/widget-order.ts +121 -0
- package/agent.bak.20260905-190020/extensions/trim-context/README.md +54 -0
- package/agent.bak.20260905-190020/extensions/trim-context/index.ts +487 -0
- package/agent.bak.20260905-190020/extensions/trim-context/install.ps1 +23 -0
- package/agent.bak.20260905-190020/extensions/trim-context/install.sh +21 -0
- package/agent.bak.20260905-190020/keybindings.json +7 -0
- package/bin/kiln.js +588 -358
- package/package.json +27 -21
- package/agent/install.ps1 +0 -527
- package/agent/install.sh +0 -511
- /package/{agent/extensions/status line → agent.bak.20260905-190020/extensions/ask-user}/install.ps1 +0 -0
- /package/{agent/extensions/status line → agent.bak.20260905-190020/extensions/ask-user}/install.sh +0 -0
- /package/{agent/extensions/status line → agent.bak.20260905-190020/extensions/statusline}/index.ts +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
|
+
}
|