Update agent.py
Browse files
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.
|
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':
|
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 ---
|