From e3dc524783a7b7e9def47d4f55bafeb4572b88cb Mon Sep 17 00:00:00 2001
From: promet99 <54811538+promet99@users.noreply.github.com>
Date: Sun, 10 Nov 2024 20:02:25 +0900
Subject: [PATCH] fix: update openapi code to match updated return type

---
 whisper_online.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/whisper_online.py b/whisper_online.py
index ef43d42..d43ac0e 100644
--- a/whisper_online.py
+++ b/whisper_online.py
@@ -192,17 +192,17 @@ class OpenaiApiASR(ASRBase):
 
         o = []
         for word in segments.words:
-            start = word.get("start")
-            end = word.get("end")
+            start = word.start
+            end = word.end
             if any(s[0] <= start <= s[1] for s in no_speech_segments):
                 # print("Skipping word", word.get("word"), "because it's in a no-speech segment")
                 continue
-            o.append((start, end, word.get("word")))
+            o.append((start, end, word.word))
         return o
 
 
     def segments_end_ts(self, res):
-        return [s["end"] for s in res.words]
+        return [s.end for s in res.words]
 
     def transcribe(self, audio_data, prompt=None, *args, **kwargs):
         # Write the audio data to a buffer