diff options
Diffstat (limited to 'packages/core')
| -rw-r--r-- | packages/core/package.json | 3 | ||||
| -rw-r--r-- | packages/core/src/core/client.ts | 12 |
2 files changed, 14 insertions, 1 deletions
diff --git a/packages/core/package.json b/packages/core/package.json index 344f3a5a..033fbddc 100644 --- a/packages/core/package.json +++ b/packages/core/package.json @@ -34,7 +34,8 @@ "fast-glob": "^3.3.3", "ignore": "^7.0.0", "shell-quote": "^1.8.2", - "strip-ansi": "^7.1.0" + "strip-ansi": "^7.1.0", + "undici": "^7.10.0" }, "devDependencies": { "@types/diff": "^7.0.2", diff --git a/packages/core/src/core/client.ts b/packages/core/src/core/client.ts index 1b953d30..3046116e 100644 --- a/packages/core/src/core/client.ts +++ b/packages/core/src/core/client.ts @@ -38,6 +38,18 @@ import { createContentGenerator, } from './contentGenerator.js'; +import { ProxyAgent, setGlobalDispatcher } from 'undici'; + +const proxy = + process.env.HTTPS_PROXY || + process.env.https_proxy || + process.env.HTTP_PROXY || + process.env.http_proxy; + +if (proxy) { + setGlobalDispatcher(new ProxyAgent(proxy)); +} + export class GeminiClient { private chat: Promise<GeminiChat>; private contentGenerator: ContentGenerator; |