Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-08 07:10:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow passing api key into OpenAIWhisperParser (#7281)

Browse Source 
This just allows the user to pass in an api_key directly into
OpenAIWhisperParser. Very simple addition.
This commit is contained in:
Kenny 2023-07-07 15:07:45 -04:00 committed by GitHub
parent 4e7d0c115b
commit 34a2755a54
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,4 @@
from typing import Iterator from typing import Iterator, Optional
from langchain.document_loaders.base import BaseBlobParser from langchain.document_loaders.base import BaseBlobParser
from langchain.document_loaders.blob_loaders import Blob from langchain.document_loaders.blob_loaders import Blob
@ -9,6 +9,9 @@ class OpenAIWhisperParser(BaseBlobParser):
"""Transcribe and parse audio files. """Transcribe and parse audio files.
Audio transcription is with OpenAI Whisper model.""" Audio transcription is with OpenAI Whisper model."""
def __init__(self, api_key: Optional[str] = None):
self.api_key = api_key
def lazy_parse(self, blob: Blob) -> Iterator[Document]: def lazy_parse(self, blob: Blob) -> Iterator[Document]:
"""Lazily parse the blob.""" """Lazily parse the blob."""
@ -28,6 +31,10 @@ class OpenAIWhisperParser(BaseBlobParser):
"pydub package not found, please install it with " "`pip install pydub`" "pydub package not found, please install it with " "`pip install pydub`"
) )
# Set the API key if provided
if self.api_key:
openai.api_key = self.api_key
# Audio file from disk # Audio file from disk
audio = AudioSegment.from_file(blob.path) audio = AudioSegment.from_file(blob.path)