Haystack Integration for AssemblyAI

1
from assemblyai_haystack.transcriber
2
from haystack.document_stores.in_memory
3
from haystack
4
from haystack.components.writers
5
6
ASSEMBLYAI_API_KEY = os.environ.get("ASSEMBLYAI_API_KEY")
7
8
document_store = InMemoryDocumentStore()
9
file_url = "https://assembly.ai/wildfires.mp3"
10
11
indexing = Pipeline()
12
indexing.add_component("transcriber", AssemblyAITranscriber(api_key=ASSEMBLYAI_API_KEY))
13
indexing.add_component("writer", DocumentWriter(document_store))
14
indexing.connect("transcriber.transcription", "writer.documents")
15
indexing.run(
16
    {
17
        "transcriber": {
18
            "file_path": file_url,
19
            "summarization": None,
20
            "speaker_labels": None,
21
        }
22
    }
23
)
24
25
print("Indexed Document Count:", document_store.count_documents())

1
{
2
"transcription":
3
  [Document(
4
    id=bdf3eb20f6440cf4b15fa4fa3176eeb72bf0139a3ad4c76741724132907a5daa,
5
    content: "Smoke from hundreds of wildfires in Canada is triggering air quality alerts throughout the US. Skyli...",
6
    meta: {
7
      'transcript_id': '2335cc07-1fbf-48ba-9855-7db3eeeb80f4',
8
      'audio_url': "https://assembly.ai/wildfires.mp3"
9
      }
10
    )
11
  ],
12
13
"summarization":
14
  [Document(
15
    id=f88864d9229b30013d5248156e74d5bfd4435e73aadb0c0ce79040be10a4f308,
16
    content: "- Smoke from hundreds of wildfires in Canada is triggering air quality alerts...")],
17
18
"speaker_labels":
19
  [Document(
20
    id=a7e222bc6a965ab1032401a6fa22da2e774294ce049b9d228acbb8b100ea2ecf,
21
    content: "Smoke from hundreds of wildfires in Canada is triggering air quality...",
22
    meta: {
23
      'speaker': 'A'
24
      }
25
    ),
26
27
  Document(
28
    id=711a1888af58601e6392490a5e4ca4c10958f93a52d8f0734869c54573ea76f5,
29
    content: "Well, there's a couple of things. The season has been pretty dry already...",
30
    meta: {
31
      'speaker': 'B'
32
      }
33
    ),
34
35
  Document(
36
    id=8fc78631d420e2e6127b8bdff2830f693febb91ed1566b9a84527cf023023d9e,
37
    content: "So what is it in this haze that makes it harmful?",
38
    meta: {
39
      'speaker': 'A'
40
      }
41
    ),
42
43
    ...
44
]}