diff options
Diffstat (limited to 'packages/server/src/core')
| -rw-r--r-- | packages/server/src/core/client.ts | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/packages/server/src/core/client.ts b/packages/server/src/core/client.ts index 8823012e..341ce021 100644 --- a/packages/server/src/core/client.ts +++ b/packages/server/src/core/client.ts @@ -36,8 +36,11 @@ export class GeminiClient { constructor(private config: Config) { const userAgent = config.getUserAgent(); const apiKeyFromConfig = config.getApiKey(); + const vertexaiFlag = config.getVertexAI(); + this.client = new GoogleGenAI({ apiKey: apiKeyFromConfig === '' ? undefined : apiKeyFromConfig, + vertexai: vertexaiFlag, httpOptions: { headers: { 'User-Agent': userAgent, |