feat: wip

Author: amirai21

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

@@ -8,11 +8,10 @@ function sleep(ms) {
 async function uploadGetUpdateDelete(fileInput, path) {
   const client = new AI21({ apiKey: process.env.AI21_API_KEY });
   try {
-    const uploadFileResponse: UploadFileResponse = await client.ragEngine.create(
-      {
-        file: fileInput,
-        path: path,
-      });
+    const uploadFileResponse: UploadFileResponse = await client.ragEngine.create({
+      file: fileInput,
+      path: path,
+    });
     console.log(uploadFileResponse);
     let file: FileResponse = await client.ragEngine.get(uploadFileResponse.fileId);
     console.log(file);

src/resources/rag/ragEngine.ts

@@ -1,22 +1,13 @@
 import * as Models from '../../types';
 import { APIResource } from '../../APIResource';
-import {
-  UploadFileResponse,
-  UploadFileRequest,
-  ListFilesFilters,
-  UpdateFileRequest,
-} from '../../types/rag';
+import { UploadFileResponse, UploadFileRequest, ListFilesFilters, UpdateFileRequest } from '../../types/rag';
 import { FileResponse } from 'types/rag/FileResponse';
 
 const RAG_ENGINE_PATH = '/library/files';
 
 export class RAGEngine extends APIResource {
-
-  async create(
-    body: UploadFileRequest,
-    options?: Models.RequestOptions,
-  ): Promise<UploadFileResponse> {
-    const {file, ...bodyWithoutFile} = body
+  async create(body: UploadFileRequest, options?: Models.RequestOptions): Promise<UploadFileResponse> {
+    const { file, ...bodyWithoutFile } = body;
     return this.client.upload<Models.UnifiedFormData, UploadFileResponse>(RAG_ENGINE_PATH, file, {
       body: bodyWithoutFile,
       ...options,

src/types/rag/UploadFileRequest.ts

@@ -1,7 +1,7 @@
-import { FilePathOrFileObject } from "./FilePathOrFileObject";
+import { FilePathOrFileObject } from './FilePathOrFileObject';
 
 export interface UploadFileRequest {
-  file: FilePathOrFileObject
+  file: FilePathOrFileObject;
   path?: string | null;
   labels?: string[] | null;
   publicUrl?: string | null;