Coool2 commited on
Commit
4bcc56d
·
verified ·
1 Parent(s): fa9908d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -26,7 +26,7 @@ from llama_index.core.workflow import Context
26
  from llama_index.callbacks.wandb import WandbCallbackHandler
27
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
28
  from llama_index.llms.huggingface import HuggingFaceLLM
29
- from llama_index.readers.audiotranscribe.openai import OpenAIAudioTranscriptReader
30
  from llama_index.readers.file import PDFReader, DocxReader, CSVReader, ImageReader, PandasExcelReader
31
  from llama_index.readers.json import JSONReader
32
  from llama_index.readers.web import TrafilaturaWebReader
@@ -110,7 +110,7 @@ def read_and_parse_content(input_path: str) -> List[Document]:
110
  '.jpg': ImageReader(),
111
  '.jpeg': ImageReader(),
112
  '.png': ImageReader(),
113
- '.mp3': OpenAIAudioTranscriptReader(),
114
  }
115
 
116
  # --- URL Handling ---
 
26
  from llama_index.callbacks.wandb import WandbCallbackHandler
27
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
28
  from llama_index.llms.huggingface import HuggingFaceLLM
29
+ from llama_index.readers.assemblyai import AssemblyAIAudioTranscriptReader
30
  from llama_index.readers.file import PDFReader, DocxReader, CSVReader, ImageReader, PandasExcelReader
31
  from llama_index.readers.json import JSONReader
32
  from llama_index.readers.web import TrafilaturaWebReader
 
110
  '.jpg': ImageReader(),
111
  '.jpeg': ImageReader(),
112
  '.png': ImageReader(),
113
+ '.mp3': AssemblyAIAudioTranscriptReader(),
114
  }
115
 
116
  # --- URL Handling ---