Upload 67 files

src/agents/agent_transcript/func.py

@@ -150,9 +150,7 @@ def script_re_outline(state: State):
     response = chain_script_re_outline.invoke(
         {
             "messages": state["messages"],
-            "prompt": state["script_re_outline_prompt"]
-            + "Output language must in: "
-            + language,
+            "prompt": state["script_re_outline_prompt"],
         }
     )
     return {
@@ -174,6 +172,9 @@ def script_writer_init(state: State):
 
 def script_writer_single(state: State):
     """Generate a single script"""
+    language = state["language"]
+    if not language:
+        language = "English"
     current_index = state.get("current_script_index", 0)
     script_count = state.get("script_count", 10)
     target_word_count = state.get("target_word_count", 8000)
@@ -198,7 +199,10 @@ def script_writer_single(state: State):
     response = chain_script_writer.invoke(
         {
             "messages": current_messages,
-            "prompt": state["script_writer_prompt"],
+            "prompt": state["script_writer_prompt"]
+            + "Output script language must in: "
+            + language
+            + "Language",
         }
     )
     script_out.append(response.content)