1, Add language settings for llm outputs

2, Optimize llm prompts 3, Add timeout handling for material downloads
2024-03-26 16:48:14 +08:00
parent fdbd81d164
commit c5dad43c2c
6 changed files with 57 additions and 33 deletions
--- a/app/services/task.py
+++ b/app/services/task.py
@@ -48,7 +48,7 @@ def start(task_id, params: VideoParams):
    logger.info("\n\n## generating video script")
    video_script = params.video_script.strip()
    if not video_script:
-        video_script = llm.generate_script(video_subject=video_subject, language=language,
+        video_script = llm.generate_script(video_subject=video_subject, language=params.video_language,
                                           paragraph_number=paragraph_number)
    else:
        logger.debug(f"video script: \n{video_script}")