fix: build errors

Author: VisargD

src/apm/loki/envConfig.ts

@@ -3,7 +3,7 @@ import { Environment } from '../../utils/env';
 const requiredEnvVars = ['NODE_ENV', 'SERVICE_NAME', 'LOKI_AUTH', 'LOKI_HOST'];
 
 export const loadAndValidateEnv = () => {
-  const env = Environment({}) as Record<string, string>;
+  const env = Environment() as Record<string, string>;
   requiredEnvVars.forEach((varName) => {
     if (!env[varName]) {
       console.error(`Missing required environment variable: ${varName}`);

src/apm/loki/logger.ts

@@ -4,7 +4,7 @@ let LokiLogger: any;
 
 try {
   const { createLogger, transports, format } = await import('winston');
-  const LokiTransport = await import('winston-loki');
+  const LokiTransport = (await import('winston-loki')).default;
 
   const envVars = loadAndValidateEnv();
 

src/apm/prometheus/envConfig.ts

@@ -8,7 +8,7 @@ const requiredEnvVars = [
 ];
 
 export const loadAndValidateEnv = (): { [key: string]: string } => {
-  const env = Environment({}) as Record<string, string>;
+  const env = Environment() as Record<string, string>;
   requiredEnvVars.forEach((varName) => {
     if (!env[varName]) {
       console.error(`Missing required environment variable: ${varName}`);

src/apm/prometheus/prometheusClient.ts

@@ -23,7 +23,7 @@ client.collectDefaultMetrics({
 
 const loadMetadataKeys = () => {
   return (
-    Environment({})
+    Environment()
       .PROMETHEUS_LABELS_METADATA_ALLOWED_KEYS?.replaceAll(' ', '')
       .split(',') ?? []
   ).map((key: string) => `metadata_${key}`);
@@ -305,7 +305,7 @@ export const llmCacheProcessingDurationMilliseconds = new client.Histogram({
 export const getCustomLabels = (metadata: string | undefined) => {
   let customLabels: Record<string, any> = {};
   const allowedKeys =
-    Environment({}).PROMETHEUS_LABELS_METADATA_ALLOWED_KEYS?.split(',') ?? [];
+    Environment().PROMETHEUS_LABELS_METADATA_ALLOWED_KEYS?.split(',') ?? [];
   if (typeof metadata === 'string') {
     try {
       const parsedMetadata = JSON.parse(metadata);

src/globals.ts

@@ -279,5 +279,19 @@ export const AtomicOperations = {
 export enum RateLimiterKeyTypes {
   VIRTUAL_KEY = 'VIRTUAL_KEY',
   API_KEY = 'API_KEY',
+  WORKSPACE = 'WORKSPACE',
   INTEGRATION_WORKSPACE = 'INTEGRATION_WORKSPACE',
 }
+
+export const METRICS_KEYS = {
+  AUTH_N_MIDDLEWARE_START: 'authNMiddlewareStart',
+  AUTH_N_MIDDLEWARE_END: 'authNMiddlewareEnd',
+  API_KEY_RATE_LIMIT_CHECK_START: 'apiKeyRateLimitCheckStart',
+  API_KEY_RATE_LIMIT_CHECK_END: 'apiKeyRateLimitCheckEnd',
+  PORTKEY_MIDDLEWARE_PRE_REQUEST_START: 'portkeyMiddlewarePreRequestStart',
+  PORTKEY_MIDDLEWARE_PRE_REQUEST_END: 'portkeyMiddlewarePreRequestEnd',
+  PORTKEY_MIDDLEWARE_POST_REQUEST_START: 'portkeyMiddlewarePostRequestStart',
+  PORTKEY_MIDDLEWARE_POST_REQUEST_END: 'portkeyMiddlewarePostRequestEnd',
+  LLM_CACHE_GET_START: 'llmCacheGetStart',
+  LLM_CACHE_GET_END: 'llmCacheGetEnd',
+};

src/middlewares/requestValidator/index.ts

@@ -47,7 +47,7 @@ const BLOCKED_TLDS = [
 
 // Parse allowed custom hosts from environment variable
 // Format: comma-separated list of domains/IPs (e.g., "localhost,127.0.0.1,example.com")
-const TRUSTED_CUSTOM_HOSTS = (c: Context) => {
+const TRUSTED_CUSTOM_HOSTS = (c?: Context) => {
   const envVar = Environment(c)?.TRUSTED_CUSTOM_HOSTS;
   if (!envVar) {
     // Default allowed hosts for local development

src/providers/google-vertex-ai/chatComplete.ts

@@ -98,7 +98,7 @@ export const VertexGoogleChatCompleteConfig: ProviderConfig = {
               functionResponse: {
                 name: message.name ?? 'gateway-tool-filler-name',
                 response: {
-                  output: message.content,
+                  output: message.content ?? '',
                 },
               },
             });

src/providers/google/chatComplete.ts

@@ -9,7 +9,6 @@ import {
   SYSTEM_MESSAGE_ROLES,
   MESSAGE_ROLES,
 } from '../../types/requestBody';
-import { buildGoogleSearchRetrievalTool } from '../google-vertex-ai/chatComplete';
 import { VERTEX_MODALITY } from '../google-vertex-ai/types';
 import {
   getMimeType,
@@ -111,7 +110,7 @@ interface GoogleFunctionResponseMessagePart {
     name: string;
     response: {
       name?: string;
-      content: string | ContentType[];
+      output: string | ContentType[];
     };
   };
 }
@@ -210,7 +209,7 @@ export const GoogleChatCompleteConfig: ProviderConfig = {
               functionResponse: {
                 name: message.name ?? 'gateway-tool-filler-name',
                 response: {
-                  output: message.content,
+                  output: message.content ?? '',
                 },
               },
             });

src/providers/types.ts

@@ -53,6 +53,7 @@ export interface ProviderAPIConfig {
     transformedRequestBody: Record<string, any>;
     transformedRequestUrl: string;
     gatewayRequestBody?: Params;
+    headers?: Record<string, string>;
   }) => Promise<Record<string, any>> | Record<string, any>;
   /** A function to generate the baseURL based on parameters */
   getBaseURL: (args: {