Merge pull request #51 from Portkey-AI/hotfix/debugHeader

TS Issue: debug header
Portkey-AI · Apr 20, 2024 · 64299e8 · 64299e8
2 parents a34f0c3 + 72dfa56
commit 64299e8
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 4 deletions.
diff --git a/src/_types/generalTypes.ts b/src/_types/generalTypes.ts
@@ -10,6 +10,7 @@ export interface ApiClientInterface {
     metadata?: Record<string, unknown> | null | undefined;
     Authorization?: string | null | undefined;
     cacheForceRefresh?: boolean | null | undefined;
+    debug?: boolean | null | undefined;
 }
 
 export interface APIResponseType {

diff --git a/src/apis/createHeaders.ts b/src/apis/createHeaders.ts
@@ -5,14 +5,19 @@ export const createHeaders = (config: Record<string, any>): Record<string, strin
 
 	for (let k in config) {
 		let v = config[k];
-
 		if (isEmpty(v)) continue;
 
 		// convert to snakecase
 		if (k.toLocaleLowerCase() === "authorization") {
 			headers[k.toLowerCase()] = v || ""
 			continue
 		}
+
+		// false logic (type is boolean, to handle flasy logic)
+		if (typeof v === "boolean") {
+			v = v.toString()
+		}
+
 		k = k.replace('ID', 'Id')
 			.replace(/[A-Z]/g, letter => `-${letter.toLowerCase()}`)
 		if (!isEmpty(v) && typeof v == "object") {

diff --git a/src/baseClient.ts b/src/baseClient.ts
@@ -120,10 +120,10 @@ export abstract class ApiClient {
     portkeyHeaders: Record<string, string>
 
     private fetch: Fetch;
-    constructor({ apiKey, baseURL, config, virtualKey, traceID, metadata, provider, Authorization, cacheForceRefresh }: ApiClientInterface) {
+    constructor({ apiKey, baseURL, config, virtualKey, traceID, metadata, provider, Authorization, cacheForceRefresh, debug }: ApiClientInterface) {
         this.apiKey = apiKey ?? "";
         this.baseURL = baseURL ?? "";
-        this.customHeaders = createHeaders({ apiKey, config, virtualKey, traceID, metadata, provider, Authorization, cacheForceRefresh })
+        this.customHeaders = createHeaders({ apiKey, config, virtualKey, traceID, metadata, provider, Authorization, cacheForceRefresh, debug })
         this.portkeyHeaders = this.defaultHeaders()
         this.fetch = fetch;
         this.responseHeaders = {}

diff --git a/src/client.ts b/src/client.ts
@@ -15,6 +15,7 @@ export class Portkey extends ApiClient {
 	metadata: Record<string, unknown> | null | undefined;
 	Authorization?: string;
 	cacheForceRefresh?: boolean | null | undefined;
+	debug?: boolean | null | undefined;
 	constructor({
 		apiKey = readEnv("PORTKEY_API_KEY") ?? null,
 		baseURL = readEnv("PORTKEY_BASE_URL") ?? null,
@@ -24,7 +25,8 @@ export class Portkey extends ApiClient {
 		traceID,
 		metadata,
 		Authorization,
-		cacheForceRefresh
+		cacheForceRefresh,
+		debug,
 	}: ApiClientInterface) {
 
 		super({
@@ -37,6 +39,7 @@ export class Portkey extends ApiClient {
 			metadata,
 			Authorization,
 			cacheForceRefresh,
+			debug,
 		});
 
 		this.apiKey = apiKey;
@@ -50,6 +53,7 @@ export class Portkey extends ApiClient {
 		this.traceID = traceID
 		this.metadata = metadata
 		this.cacheForceRefresh = cacheForceRefresh;
+		this.debug = debug;
 	}
 
 	completions: API.Completions = new API.Completions(this);