mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-24 21:38:09 +02:00
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""Propose a podcast's initial brief spec."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.podcasts.persistence import PodcastRepository
|
|
from app.podcasts.schemas import PodcastSpec
|
|
from app.podcasts.service import preferences_from
|
|
|
|
from .config import DEFAULT_MAX_MINUTES, DEFAULT_MIN_MINUTES, DEFAULT_SPEAKER_COUNT
|
|
from .graph import graph as brief_graph
|
|
from .state import BriefState
|
|
|
|
|
|
async def propose_brief(
|
|
session: AsyncSession,
|
|
*,
|
|
search_space_id: int,
|
|
speaker_count: int = DEFAULT_SPEAKER_COUNT,
|
|
min_minutes: int = DEFAULT_MIN_MINUTES,
|
|
max_minutes: int = DEFAULT_MAX_MINUTES,
|
|
focus: str | None = None,
|
|
) -> PodcastSpec:
|
|
"""Reuse the last-used language and voices, else English; return the spec."""
|
|
last_language, last_voices = preferences_from(
|
|
await PodcastRepository(session).latest_with_spec(search_space_id)
|
|
)
|
|
config = {
|
|
"configurable": {
|
|
"speaker_count": speaker_count,
|
|
"min_minutes": min_minutes,
|
|
"max_minutes": max_minutes,
|
|
"focus": focus,
|
|
"last_used_language": last_language,
|
|
"last_used_voices": last_voices,
|
|
}
|
|
}
|
|
result = await brief_graph.ainvoke(BriefState(), config=config)
|
|
return result["spec"]
|