feat: quality fix

Author: amirai21

.husky/pre-commit

@@ -0,0 +1,4 @@
+#!/usr/bin/env sh
+. "$(dirname -- "$0")/_/husky.sh"
+
+npm run lint && npm run format && npm run quality:fix

src/examples/chatExample.ts

@@ -1,4 +1,4 @@
-import { AI21 } from "../AI21";
+import { AI21 } from '../AI21';
 
 /*
   This is a temporary example to test the API streaming/non-streaming functionality.
@@ -7,7 +7,6 @@ async function main() {
   const client = new AI21({ apiKey: process.env.AI21_API_KEY });
 
   try {
-
     console.log('-------------------------------- streaming flow --------------------------------');
 
     const streamResponse = await client.chat.completions.create({

src/examples/convRagExample.ts

@@ -1,17 +1,15 @@
-import { AI21 } from "../AI21";
+import { AI21 } from '../AI21';
 
 /*
   This is a temporary example to test the Conversational RAG functionality.
 */
 async function main() {
-
-    /* TODO - add a file upload example when library support is added and combine it with the below flow */
+  /* TODO - add a file upload example when library support is added and combine it with the below flow */
 
   const client = new AI21({ apiKey: process.env.AI21_API_KEY });
   try {
-
     /* The following example is for a question that is not in the context of files uploaded to RAG */
-    
+
     const answer_not_in_ctx_response = await client.conversationalRag.create({
       messages: [{ role: 'user', content: 'Who is the Russian president?' }],
     });

src/resources/rag/conversationalRag.ts

@@ -4,14 +4,10 @@ import { ConversationalRagRequest } from '../../types/rag/ConversationalRagReque
 import { ConversationalRagResponse } from '../../types/rag/ConversationalRagResponse';
 
 export class ConversationalRag extends APIResource {
-
-  create(body: ConversationalRagRequest, options?: Models.RequestOptions){
-    return this.client.post<ConversationalRagRequest, ConversationalRagResponse>(
-      '/conversational-rag',
-      {
-        body,
-        ...options,
-      } as Models.RequestOptions<ConversationalRagRequest>,
-    ) as Promise<ConversationalRagResponse>;
+  create(body: ConversationalRagRequest, options?: Models.RequestOptions) {
+    return this.client.post<ConversationalRagRequest, ConversationalRagResponse>('/conversational-rag', {
+      body,
+      ...options,
+    } as Models.RequestOptions<ConversationalRagRequest>) as Promise<ConversationalRagResponse>;
   }
 }

src/types/rag/ConversationalRagRequest.ts

@@ -1,15 +1,14 @@
-import { ChatMessageParam } from "../chat";
-import { RetrievalStrategy } from "./RetrievalStrategy";
+import { ChatMessageParam } from '../chat';
+import { RetrievalStrategy } from './RetrievalStrategy';
 
 export interface ConversationalRagRequest {
-    messages: ChatMessageParam[];
-    path?: string | null;
-    labels?: string[] | null;
-    file_ids?: string[] | null;
-    max_segments?: number | null;
-    retrieval_strategy?: RetrievalStrategy | string | null;
-    retrieval_similarity_threshold?: number | null;
-    max_neighbors?: number | null;
-    hybrid_search_alpha?: number | null;
-    [key: string]: any;
-  }
+  messages: ChatMessageParam[];
+  path?: string | null;
+  labels?: string[] | null;
+  file_ids?: string[] | null;
+  max_segments?: number | null;
+  retrieval_strategy?: RetrievalStrategy | string | null;
+  retrieval_similarity_threshold?: number | null;
+  max_neighbors?: number | null;
+  hybrid_search_alpha?: number | null;
+}

src/types/rag/ConversationalRagResponse.ts

@@ -1,11 +1,11 @@
-import { ChatMessage } from "../chat";
-import { ConversationalRagSource } from "./ConversationalRagSource";
+import { ChatMessage } from '../chat';
+import { ConversationalRagSource } from './ConversationalRagSource';
 
 export interface ConversationalRagResponse {
-    id: string;
-    choices: ChatMessage[];
-    search_queries?: string[] | null;
-    context_retrieved: boolean;
-    answer_in_context: boolean;
-    sources: ConversationalRagSource[];
+  id: string;
+  choices: ChatMessage[];
+  search_queries?: string[] | null;
+  context_retrieved: boolean;
+  answer_in_context: boolean;
+  sources: ConversationalRagSource[];
 }

src/types/rag/ConversationalRagSource.ts

@@ -1,9 +1,9 @@
 export interface ConversationalRagSource {
-    text: string;
-    file_id: string;
-    file_name: string;
-    score: number;
-    order?: number | null;
-    public_url?: string | null;
-    labels?: string[] | null;
+  text: string;
+  file_id: string;
+  file_name: string;
+  score: number;
+  order?: number | null;
+  public_url?: string | null;
+  labels?: string[] | null;
 }

src/types/rag/RetrievalStrategy.ts

@@ -1,6 +1,6 @@
 export enum RetrievalStrategy {
-    DEFAULT = "default",
-    SEGMENTS = "segments",
-    ADD_NEIGHBORS = "add_neighbors",
-    FULL_DOC = "full_doc"
+  DEFAULT = 'default',
+  SEGMENTS = 'segments',
+  ADD_NEIGHBORS = 'add_neighbors',
+  FULL_DOC = 'full_doc',
 }