feat: Support custom gateway URLs in Portkey (promptfoo#840)

mldangelo · May 31, 2024 · a0e076c · a0e076c
1 parent 822ddb6
commit a0e076c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/site/docs/integrations/portkey.md b/site/docs/integrations/portkey.md
@@ -53,4 +53,5 @@ providers:
  team: xxx
  portkeyConfig: xxx
  portkeyProvider: xxx
+ portkeyApiBaseUrl: xxx
 ```
diff --git a/src/providers/portkey.ts b/src/providers/portkey.ts
@@ -38,7 +38,7 @@ export class PortkeyChatCompletionProvider extends OpenAiChatCompletionProvider
  ...providerOptions,
  config: {
  ...providerOptions.config,
- apiBaseUrl: 'https://api.portkey.ai/v1',
+ apiBaseUrl: process.env.PORTKEY_API_BASE_URL || providerOptions.config?.portkeyApiBaseUrl || 'https://api.portkey.ai/v1',
  headers,
  },
  });