feat: fix tests

Author: amirai21

examples/studio/conversational-rag/rag-engine.ts

@@ -1,5 +1,5 @@
 import { AI21 } from 'ai21';
-import { FileResponse, UploadFileResponse } from '../../../src/types/rag';
+import { FileResponse, UploadFileResponse } from '../../../src/types';
 import path from 'path';
 
 function sleep(ms) {

src/APIClient.ts

@@ -9,11 +9,11 @@ import {
   Headers,
   CrossPlatformResponse,
   UnifiedFormData,
+  FilePathOrFileObject,
 } from './types';
 import { AI21EnvConfig } from './EnvConfig';
 import { createFetchInstance, createFilesHandlerInstance } from './runtime';
 import { Fetch } from 'fetch';
-import { FilePathOrFileObject } from 'types/rag';
 import { BaseFilesHandler } from 'files/BaseFilesHandler';
 
 const validatePositiveInteger = (name: string, n: unknown): number => {

src/files/BaseFilesHandler.ts

@@ -1,5 +1,5 @@
+import { FilePathOrFileObject } from 'types';
 import { FormDataRequest } from 'types/API';
-import { FilePathOrFileObject } from 'types/rag';
 
 export abstract class BaseFilesHandler {
   abstract prepareFormDataRequest(file: FilePathOrFileObject): Promise<FormDataRequest>;

src/files/BrowserFilesHandler.ts

@@ -1,4 +1,4 @@
-import { FilePathOrFileObject } from 'types/rag';
+import { FilePathOrFileObject } from 'types';
 import { BaseFilesHandler } from './BaseFilesHandler';
 import { FormDataRequest } from 'types/API';
 

src/files/NodeFilesHandler.ts

@@ -1,4 +1,4 @@
-import { FilePathOrFileObject } from 'types/rag';
+import { FilePathOrFileObject } from 'types';
 import { BaseFilesHandler } from './BaseFilesHandler';
 import { FormDataRequest } from 'types/API';
 

src/resources/rag/conversationalRag.ts

@@ -1,7 +1,7 @@
 import * as Models from '../../types';
 import { APIResource } from '../../APIResource';
-import { ConversationalRagRequest } from '../../types/rag/ConversationalRagRequest';
-import { ConversationalRagResponse } from '../../types/rag/ConversationalRagResponse';
+import { ConversationalRagRequest } from '../../types';
+import { ConversationalRagResponse } from '../../types';
 
 export class ConversationalRag extends APIResource {
   create(body: ConversationalRagRequest, options?: Models.RequestOptions) {

src/resources/rag/files.ts

@@ -1,7 +1,7 @@
 import * as Models from '../../types';
 import { APIResource } from '../../APIResource';
-import { UploadFileResponse, UploadFileRequest, ListFilesFilters, UpdateFileRequest } from '../../types/rag';
-import { FileResponse } from 'types/rag/FileResponse';
+import { UploadFileResponse, UploadFileRequest, ListFilesFilters, UpdateFileRequest } from '../../types';
+import { FileResponse } from 'types/files/FileResponse';
 
 const FILES_PATH = '/library/files';