admin-valentin/open-notebook
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: fix provider routing for podcasts and add a try block to catch podcast generation issues

Browse source
This commit is contained in:
LUIS NOVO 2025-06-10 11:52:41 -03:00
parent 0a5504744c
commit 61b3583a57
1 changed files with 25 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
open_notebook/plugins/podcasts.py
View file

@ -111,19 +111,22 @@ class PodcastConfig(ObjectModel):
api_key_label = "GOOGLE_API_KEY" api_key_label = "GOOGLE_API_KEY"
llm_model_name = self.transcript_model llm_model_name = self.transcript_model
if self.provider == "gemini": if self.provider == "google":
tts_model = "geminimulti" tts_model = "gemini"
elif self.provider == "openai": elif self.provider == "openai":
tts_model = "openai" tts_model = "openai"
elif self.provider == "anthropic": elif self.provider == "anthropic":
tts_model = "anthropic" tts_model = "anthropic"
elif self.provider == "vertexai":
tts_model = "geminimulti"
elif self.provider == "elevenlabs": elif self.provider == "elevenlabs":
tts_model = "elevenlabs" tts_model = "elevenlabs"
logger.debug( logger.info(
f"Generating episode {episode_name} with config {conversation_config} and using model {llm_model_name}, tts model {tts_model}" f"Generating episode {episode_name} with config {conversation_config} and using model {llm_model_name}, tts model {tts_model}"
) )
try:
audio_file = generate_podcast( audio_file = generate_podcast(
conversation_config=conversation_config, conversation_config=conversation_config,
text=text, text=text,
@ -140,6 +143,9 @@ class PodcastConfig(ObjectModel):
audio_file=audio_file, audio_file=audio_file,
) )
episode.save() episode.save()
except Exception as e:
logger.error(f"Failed to generate episode {episode_name}: {e}")
raise
@field_validator( @field_validator(
"name", "podcast_name", "podcast_tagline", "output_language", "model" "name", "podcast_name", "podcast_tagline", "output_language", "model"