enable luis vnext train mode (#1099)

Author: feich-ms

packages/lu/src/parser/lubuild/builder.ts

@@ -201,6 +201,9 @@ export class Builder {
     // default to false
     let directVersionPublish = options.directVersionPublish || false
 
+    // set train mode
+    let trainMode = options.trainMode
+
     // settings assets like app id and version returned from luis api call
     let settingsAssets: any[] = []
 
@@ -257,7 +260,7 @@ export class Builder {
 
               if (needTrainAndPublish) {
                 // train and publish application
-                await this.trainAndPublishApplication(luBuildCore, recognizer, timeBucketOfRequests, isStaging, directVersionPublish)
+                await this.trainAndPublishApplication(luBuildCore, recognizer, timeBucketOfRequests, isStaging, directVersionPublish, trainMode)
               }
 
               // init settings asset
@@ -425,11 +428,11 @@ export class Builder {
     return true
   }
 
-  async trainAndPublishApplication(luBuildCore: LuBuildCore, recognizer: Recognizer, timeBucket: number, isStaging: boolean, directVersionPublish: boolean) {
+  async trainAndPublishApplication(luBuildCore: LuBuildCore, recognizer: Recognizer, timeBucket: number, isStaging: boolean, directVersionPublish: boolean, trainMode: string) {
     // send train application request
     this.handler(`${recognizer.getLuPath()} training version=${recognizer.versionId}\n`)
     await delay(timeBucket)
-    await luBuildCore.trainApplication(recognizer.getAppId(), recognizer.versionId)
+    await luBuildCore.trainApplication(recognizer.getAppId(), recognizer.versionId, trainMode)
     this.handler(`${recognizer.getLuPath()} waiting for training for version=${recognizer.versionId}...\n`)
     let done = true
     do {

packages/lu/src/parser/lubuild/core.ts

@@ -22,6 +22,7 @@ export class LuBuildCore {
   private readonly retryCount: number
   private readonly retryDuration: number
   private readonly headers: any
+  private readonly trainMode: string | undefined
 
   constructor(subscriptionKey: string, endpoint: string, retryCount: number, retryDuration: number) {
     this.subscriptionKey = subscriptionKey
@@ -37,6 +38,9 @@ export class LuBuildCore {
     // set user agent
     const luisUserAgent = process.env['LUIS_USER_AGENT'] || this.getUserAgent()
 
+    // set luis train mode
+    this.trainMode = process.env['LUIS_TRAIN_MODE']
+
     // set headers
     this.headers = {
       'Content-Type': 'application/json',
@@ -299,19 +303,24 @@ export class LuBuildCore {
     }
   }
 
-  public async trainApplication(appId: string, versionId: string) {
+  public async trainApplication(appId: string, versionId: string, trainMode: string) {
+    let mode = trainMode || this.trainMode
+    let url = `${this.endpoint}/luis/authoring/v3.0-preview/apps/${appId}/versions/${versionId}/train`
+    url += mode ? `?mode=${mode}` : ''
+
+    let messageData
     let retryCount = this.retryCount + 1
-    let error
+    let error: any
     while (retryCount > 0) {
-      if (error === undefined || error.statusCode === rateLimitErrorCode) {
-        try {
-          await this.client.train.trainVersion(appId, versionId)
-          break
-        } catch (e) {
-          error = e
-          retryCount--
-          if (retryCount > 0) await delay(this.retryDuration)
-        }
+      if (error === undefined || error.code === rateLimitErrorCode.toString()) {
+        let response = await fetch(url, {method: 'POST', headers: this.headers})
+        messageData = await response.json()
+
+        if (messageData.error === undefined) break
+
+        error = messageData.error
+        retryCount--
+        if (retryCount > 0) await delay(this.retryDuration)
       } else {
         throw error
       }
@@ -320,6 +329,8 @@ export class LuBuildCore {
     if (retryCount === 0) {
       throw error
     }
+
+    return messageData
   }
 
   public async getTrainingStatus(appId: string, versionId: string) {