@saplingai/sapling-js 1.0.8 → 1.0.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@saplingai/sapling-js",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "JavaScript library for using the Sapling.ai API.",
   "type": "module",
   "dependencies": {

package/sapling/client.d.ts

@@ -9,12 +9,29 @@ type ConstructorOptions = {
 type EditOptions = {
    variety?: string,
    autoApply?: boolean,
+   lang?: string,
+   medical?: boolean,
+}
+
+type SpellcheckOptions = {
+   minLength?: number,
+   multipleEdits?: boolean,
+   lang?: string,
+   autoApply?: boolean,
+   variety?: string,
+   userData?: any,
 }
 
 declare class Client {
-  constructor(apiKey: string, options?: ConstructorOptions);
+   constructor(apiKey: string, options?: ConstructorOptions);
+
+   edits(text: string, sessionId: string, options?: EditOptions): Promise<AxiosResponse>;
+   acceptEdit(editUUID: string, sessionId: string): Promise<AxiosResponse>;
+   rejectEdit(editUUID: string, sessionId: string): Promise<AxiosResponse>;
+
+   spellcheck(text: string, sessionId: string, options?: SpellcheckOptions): Promise<AxiosResponse>;
 
-  edits(text: string, sessionId: string, options?: EditOptions):  Promise<AxiosResponse>;
-  accept(editUUID: string, sessionId: string): void;
-  reject(editUUID: string, sessionId: string): void;
+   complete(query: string, sessionId: string): Promise<AxiosResponse>;
+   acceptComplete(completeUUID: string, query: string, completion: string, sessionId: string): Promise<AxiosResponse>;
+   tone(text: string): Promise<AxiosResponse>;
 }

package/sapling/client.js

@@ -8,12 +8,27 @@ export class Client {
     this.hostname = options.hostname || 'https://api.sapling.ai';
     this.pathname = options.pathname || '/api/v1/';
     this.urlEndpoint = `${this.hostname}${this.pathname}`;
+    this.defaultSessionId = uuidv4();
     axios.defaults.timeout = this.timeout;
   }
 
-   edits(text, sessionId, options = {}) {
+  axiosPromise(url, method, data) {
+    return new Promise((resolve) => {
+      axios({
+        url,
+        method,
+        data,
+        timeout: this.timeout,
+      })
+      .then(function (response) {
+          resolve(response);
+      });
+    });
+  }
+
+  edits(text, sessionId, options = {}) {
     const url = `${this.urlEndpoint}edits`;
-    sessionId = sessionId || uuidv4();
+    sessionId = sessionId || this.defaultSessionId;
 
     let data = {
       text,
@@ -22,54 +37,119 @@ export class Client {
     }
 
     if (options.variety != null) {
-      data.variety = variety;
+      data.variety = options.variety;
     }
 
     if (options.autoApply != null) {
-      data.auto_apply = autoApply;
+      data.auto_apply = options.autoApply;
     }
 
-    return new Promise((resolve) => {
-      axios({
-        url,
-        data,
-        method: 'post',
-        timeout: this.timeout,
-      })
-      .then(function (response) {
-          resolve(response);
-      });
-    });
+    if (options.lang != null) {
+      data.lang = options.lang;
+    }
+
+    if (options.medical != null) {
+      data.medical = options.medical;
+    }
+
+    return this.axiosPromise(url, 'post', data);
   }
 
-  accept(editUUID, sessionId) {
+  acceptEdit(editUUID, sessionId) {
     const url = `${this.urlEndpoint}edits/${editUUID}/accept`;
-    sessionId = sessionId || uuidv4();
+    sessionId = sessionId || this.defaultSessionId;
     let data = {
       key: this.apiKey,
       session_id: sessionId,
     }
-    axios({
-      url,
-      data,
-      method: 'post',
-      timeout: this.timeout,
-    })
+
+    return this.axiosPromise(url, 'post', data);
   }
 
-  reject(editUUID, sessionId) {
+  rejectEdit(editUUID, sessionId) {
     const url = `${this.urlEndpoint}edits/${editUUID}/reject`;
-    sessionId = sessionId || uuidv4();
+    sessionId = sessionId || this.defaultSessionId;
     let data = {
       key: this.apiKey,
       session_id: sessionId,
     }
-    axios({
-      url,
-      data,
-      method: 'post',
-      timeout: this.timeout,
-    })
+
+    return this.axiosPromise(url, 'post', data);
+  }
+
+  spellcheck(text, sessionId, options = {}) {
+    const url = `${this.urlEndpoint}spellcheck`;
+    sessionId = sessionId || this.defaultSessionId;
+
+    let data = {
+      key: this.apiKey,
+      session_id: sessionId,
+      text,
+    }
+
+    if (options.minLength != null) {
+      data.min_length = minLength;
+    }
+
+    if (options.multipleEdits != null) {
+      data.multiple_edits = multipleEdits;
+    }
+
+    if (options.lang != null) {
+      data.lang = lang;
+    }
+
+    if (options.autoApply != null) {
+      data.auto_apply = autoApply;
+    }
+
+    if (options.variety != null) {
+      data.variety = variety;
+    }
+
+    if (options.userData != null) {
+      data.user_data = userData;
+    }
+
+    return this.axiosPromise(url, 'post', data);
+  }
+
+
+  complete(query, sessionId) {
+    const url = `${this.urlEndpoint}complete`;
+    sessionId = sessionId || this.defaultSessionId;
+    let data = {
+      key: this.apiKey,
+      session_id: sessionId,
+      query,
+    }
+    return this.axiosPromise(url, 'post', data);
+  }
+
+
+  acceptComplete(completeUUID, query, completion, sessionId) {
+    const url = `${this.urlEndpoint}complete/${completeUUID}/accept`;
+    sessionId = sessionId || this.defaultSessionId;
+    let data = {
+      key: this.apiKey,
+      session_id: sessionId,
+      context: {
+        query,
+        completion,
+      }
+    }
+
+    return this.axiosPromise(url, 'post', data);
+  }
+
+  tone(text) {
+    const url = `${this.urlEndpoint}tone`;
+
+    let data = {
+      key: this.apiKey,
+      text,
+    }
+    return this.axiosPromise(url, 'post', data);
   }
 
 }