fix: Linting

Author: asafgardin

src/Client.ts

@@ -45,7 +45,8 @@ export class AI21 extends Core.APIClient {
     this._options = options;
   }
 
-  protected override authHeaders(opts: Models.FinalRequestOptions): Core.Headers {
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  protected override authHeaders(_: Models.FinalRequestOptions): Core.Headers {
     return {
       Authorization: `Bearer ${this.apiKey}`,
     };

src/Core.ts

@@ -2,7 +2,7 @@ import { AI21Error } from './errors.js';
 import { VERSION } from './version.js';
 
 import fetch from 'node-fetch';
-import { HeadersInit } from 'node-fetch';
+import { HeadersInit, RequestInit } from 'node-fetch';
 import {
   RequestOptions,
   FinalRequestOptions,
@@ -94,6 +94,7 @@ export abstract class APIClient {
     };
   }
 
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   protected authHeaders(opts: FinalRequestOptions): Headers {
     return {};
   }
@@ -123,7 +124,9 @@ export abstract class APIClient {
       ...opts,
     };
 
-    return this.performRequest(options as FinalRequestOptions).then(handleAPIResponse);
+    return this.performRequest(options as FinalRequestOptions).then(
+      (response) => handleAPIResponse<Rsp>(response) as Rsp,
+    );
   }
 
   private async performRequest(options: FinalRequestOptions): Promise<APIResponseProps> {
@@ -138,7 +141,7 @@ export abstract class APIClient {
     const response = await fetch(url, {
       method: options.method,
       headers: headers as HeadersInit,
-      signal: controller.signal as any, // Type cast to avoid AbortSignal compatibility issue
+      signal: controller.signal as RequestInit['signal'],
       body: options.body ? JSON.stringify(options.body) : undefined,
     });
 

src/ResponseHandler.ts

@@ -1,13 +1,22 @@
 import { APIResponseProps } from './models';
 import { AI21Error } from './errors';
 import { Stream } from './Streaming';
+import { Response } from 'node-fetch';
 
-export async function handleAPIResponse({ response, options }: APIResponseProps): Promise<any> {
+type APIResponse<T = unknown> = {
+  data: T;
+  response: Response;
+};
+
+export async function handleAPIResponse<T>({
+  response,
+  options,
+}: APIResponseProps): Promise<Stream<T> | APIResponse<T>> {
   if (options.stream) {
     if (!response.body) {
       throw new AI21Error('Response body is null');
     }
-    return new Stream(response);
+    return new Stream<T>(response);
   }
 
   const contentType = response.headers.get('content-type');

src/Streaming/SSEDecoder.ts

@@ -24,7 +24,7 @@ export class DefaultSSEDecoder implements SSEDecoder {
       throw new Error('Response body is null');
     }
 
-    const webReadableStream = Readable.toWeb(response.body as any) as ReadableStream;
+    const webReadableStream = Readable.toWeb(response.body as Readable);
     const reader = webReadableStream.getReader();
 
     let buffer = '';

src/Streaming/Stream.ts

@@ -1,5 +1,4 @@
 import { Response as NodeResponse } from 'node-fetch';
-import { Readable } from 'stream';
 import { DefaultSSEDecoder } from './SSEDecoder';
 import { SSEDecoder } from './SSEDecoder';
 import { SSE_DONE_MSG } from './Consts';
@@ -8,7 +7,7 @@ import { StreamingDecodeError } from '../errors';
 function getStreamMessage<T>(chunk: string): T {
   try {
     return JSON.parse(chunk);
-  } catch (e) {
+  } catch {
     throw new StreamingDecodeError(chunk);
   }
 }

src/models/chat/ChatCompletionsRequest.ts

@@ -16,7 +16,7 @@ export interface ChatCompletionCreateParamsBase {
   tools?: ToolDefinition[] | null;
   response_format?: ResponseFormat | null;
   documents?: DocumentSchema[] | null;
-  [key: string]: any;
+  [key: string]: any; // eslint-disable-line @typescript-eslint/no-explicit-any
 }
 
 export interface ChatCompletionCreateParamsNonStreaming extends ChatCompletionCreateParamsBase {
@@ -30,11 +30,3 @@ export interface ChatCompletionCreateParamsStreaming extends ChatCompletionCreat
 export type ChatCompletionCreateParams =
   | ChatCompletionCreateParamsNonStreaming
   | ChatCompletionCreateParamsStreaming;
-
-export declare namespace ChatCompletions {
-  export {
-    type ChatCompletionCreateParams,
-    type ChatCompletionCreateParamsNonStreaming,
-    type ChatCompletionCreateParamsStreaming,
-  };
-}

src/models/chat/ChatCompletionsResponse.ts

@@ -42,14 +42,3 @@ export interface UsageInfo {
   completion_tokens: number;
   total_tokens: number;
 }
-
-export declare namespace ChatCompletions {
-  export {
-    type ChatCompletionResponseChoice,
-    type ChatCompletionResponse,
-    type ChoiceDelta,
-    type ChoicesChunk,
-    type ChatCompletionChunk,
-    type UsageInfo,
-  };
-}

src/models/chat/ChatMessage.ts

@@ -25,7 +25,3 @@ export interface SystemMessage extends ChatMessage {
 }
 
 export type ChatMessageParam = UserMessage | AssistantMessage | ToolMessage | SystemMessage | ChatMessage;
-
-export declare namespace ChatCompletions {
-  export { type ChatMessage, type AssistantMessage, type ToolMessage, type UserMessage, type SystemMessage };
-}

src/models/chat/ChatModel.ts

@@ -1,5 +1 @@
 export type ChatModel = 'jamba-1.5-mini' | 'jamba-1.5-large';
-
-export declare namespace ChatModel {
-  export { type ChatModel };
-}

src/models/chat/DocumentSchema.ts

@@ -3,7 +3,3 @@ export interface DocumentSchema {
   id?: string;
   metadata?: Record<string, string>;
 }
-
-export declare namespace ChatCompletions {
-  export { type DocumentSchema };
-}