feat: Stable & Hella Fast Podcast Agent with auto FFMPEG handling.

surfsense_backend/app/agents/podcaster/state.py

@@ -0,0 +1,38 @@
+"""Define the state structures for the agent."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import List, Optional
+from pydantic import BaseModel, Field
+
+
+class PodcastTranscriptEntry(BaseModel):
+    """
+    Represents a single entry in a podcast transcript.
+    """
+    speaker_id: int = Field(..., description="The ID of the speaker (0 or 1)")
+    dialog: str = Field(..., description="The dialog text spoken by the speaker")
+
+
+class PodcastTranscripts(BaseModel):
+    """
+    Represents the full podcast transcript structure.
+    """
+    podcast_transcripts: List[PodcastTranscriptEntry] = Field(
+        ..., 
+        description="List of transcript entries with alternating speakers"
+    ) 
+
+@dataclass
+class State:
+    """Defines the input state for the agent, representing a narrower interface to the outside world.
+
+    This class is used to define the initial state and structure of incoming data.
+    See: https://langchain-ai.github.io/langgraph/concepts/low_level/#state
+    for more information.
+    """
+
+    source_content: str
+    podcast_transcript: Optional[List[PodcastTranscriptEntry]] = None
+    final_podcast_file_path: Optional[str] = None