kostasang commited on
Commit
a094267
·
verified ·
1 Parent(s): 7a55fbb

Upload tools.py

Browse files
Files changed (1) hide show
  1. src/tools.py +2 -1
src/tools.py CHANGED
@@ -1,5 +1,5 @@
1
  import json
2
-
3
 
4
  from langchain_community.document_loaders import \
5
  AssemblyAIAudioTranscriptLoader, WikipediaLoader
@@ -102,6 +102,7 @@ def audio_transcript(
102
  """
103
  loader = AssemblyAIAudioTranscriptLoader(
104
  file_path=audiofile_path,
 
105
  )
106
  docs = loader.load(file_path=audiofile_path)
107
  return docs[0].page_content if docs else "No transcription available."
 
1
  import json
2
+ import os
3
 
4
  from langchain_community.document_loaders import \
5
  AssemblyAIAudioTranscriptLoader, WikipediaLoader
 
102
  """
103
  loader = AssemblyAIAudioTranscriptLoader(
104
  file_path=audiofile_path,
105
+ api_key=os.environ.get("ASSEMBLY_API_KEY"),
106
  )
107
  docs = loader.load(file_path=audiofile_path)
108
  return docs[0].page_content if docs else "No transcription available."