adds showSpeakerLabel config parameter (#17)

qasim9872 · Oct 28, 2020 · 1dc6b4e · 1dc6b4e
1 parent ed630a0
commit 1dc6b4e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/AwsTranscribe.ts b/src/AwsTranscribe.ts
@@ -8,17 +8,23 @@ export class AwsTranscribe {
     private accessKeyId!: string
     private secretAccessKey!: string
     private sessionToken: string | undefined
+    private showSpeakerLabel?: boolean
 
     constructor(config?: ClientConfig) {
         // get from environment if config not provided
         this.setAccessKeyId(config?.accessKeyId || process.env.AWS_ACCESS_KEY_ID)
         this.setSecretAccessKey(config?.secretAccessKey || process.env.AWS_SECRET_ACCESS_KEY)
         this.setSessionToken(config?.sessionToken || process.env.AWS_SESSION_TOKEN)
+        this.setShowSpeakerLabel(config?.showSpeakerLabel || false)
     }
 
     private createPreSignedUrl(config: TranscribeStreamConfig) {
-        const { region, languageCode, sampleRate } = config
+        const { region, languageCode, sampleRate, showSpeakerLabel } = config
         const endpoint = "transcribestreaming." + region + ".amazonaws.com:8443"
+	let query = "language-code=" + languageCode + "&media-encoding=pcm&sample-rate=" + sampleRate
+	if (showSpeakerLabel) {
+	    query += '&show-speaker-label=true'
+	}
 
         return createPresignedURL(
             "GET",
@@ -33,7 +39,7 @@ export class AwsTranscribe {
                 protocol: "wss",
                 expires: 15,
                 region: region,
-                query: "language-code=" + languageCode + "&media-encoding=pcm&sample-rate=" + sampleRate,
+                query: query
             }
         )
     }
@@ -58,4 +64,8 @@ export class AwsTranscribe {
     setSessionToken(sessionToken: string | undefined) {
       this.sessionToken = sessionToken
     }
+
+    setShowSpeakerLabel(showSpeakerLabel: boolean | false) {
+        this.showSpeakerLabel = showSpeakerLabel
+    }
 }
diff --git a/src/types.ts b/src/types.ts
@@ -22,12 +22,14 @@ export interface ClientConfig {
     accessKeyId?: string
     secretAccessKey?: string
     sessionToken?: string
+    showSpeakerLabel?: boolean
 }
 
 export interface TranscribeStreamConfig {
     region: AVAILABLE_REGIONS
     languageCode: LANGUAGES
     sampleRate: number
+    showSpeakerLabel?: boolean
 }
 
 export interface PresignedUrlHeaders {