github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 14:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add try except block to OpenAIWhisperParser (#7505)

Browse Source
This commit is contained in:
Kenny 2023-07-15 18:42:00 -04:00 committed by GitHub
parent 4cb9f1eda8
commit 0d058d4046
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
langchain/document_loaders/parsers/audio.py
View File

@ -1,3 +1,4 @@
import time
from typing import Iterator, Optional from typing import Iterator, Optional
from langchain.document_loaders.base import BaseBlobParser from langchain.document_loaders.base import BaseBlobParser
@ -55,7 +56,18 @@ class OpenAIWhisperParser(BaseBlobParser):
# Transcribe # Transcribe
print(f"Transcribing part {split_number+1}!") print(f"Transcribing part {split_number+1}!")
transcript = openai.Audio.transcribe("whisper-1", file_obj) attempts = 0
while attempts < 3:
try:
transcript = openai.Audio.transcribe("whisper-1", file_obj)
break
except Exception as e:
attempts += 1
print(f"Attempt {attempts} failed. Exception: {str(e)}")
time.sleep(5)
else:
print("Failed to transcribe after 3 attempts.")
continue
yield Document( yield Document(
page_content=transcript.text, page_content=transcript.text,