diff --git a/server/lib/readConfigFile.ts b/server/lib/readConfigFile.ts index 5774456a7..97dda31a2 100644 --- a/server/lib/readConfigFile.ts +++ b/server/lib/readConfigFile.ts @@ -79,7 +79,7 @@ export const configSchema = z .default(3001) .transform(stoi) .pipe(portSchema), - badger_override: z.string().optional(), + internal_api_override: z.string().optional(), next_port: portSchema .optional() .default(3002) diff --git a/server/routers/traefik/traefikConfigProvider.ts b/server/routers/traefik/traefikConfigProvider.ts index 898ac42ee..c2e88709b 100644 --- a/server/routers/traefik/traefikConfigProvider.ts +++ b/server/routers/traefik/traefikConfigProvider.ts @@ -20,9 +20,13 @@ export async function traefikConfigProvider( const maintenancePort = config.getRawConfig().server.next_port; const maintenanceHost = config.getRawConfig().server.internal_hostname; const pangolinUIUrl = `http://${maintenanceHost}:${maintenancePort}`; - const aiGatewayUrl = `http://${maintenanceHost}:${ - config.getRawConfig().server.internal_port - }/api/v1/ai-gateway`; + const aiGatewayUrl = + `${ + config.getRawConfig().server.internal_api_override + }/ai-gateway` || + `http://${maintenanceHost}:${ + config.getRawConfig().server.internal_port + }/api/v1/ai-gateway`; const traefikConfig = await getTraefikConfig( currentExitNodeId, @@ -41,7 +45,8 @@ export async function traefikConfigProvider( plugin: { [badgerMiddlewareName]: { apiBaseUrl: - config.getRawConfig().server.badger_override || + config.getRawConfig().server + .internal_api_override || new URL( "/api/v1", `http://${