fix: use "json_mode"

This commit is contained in:
Li Xin 2025-04-19 22:02:14 +08:00
parent 7497857b71
commit 19d8a59def

View File

@ -19,12 +19,12 @@ def script_writer_node(state: PodcastState):
logger.info("Generating script for podcast...")
model = get_llm_by_type(
AGENT_LLM_MAP["podcast_script_writer"]
).with_structured_output(Script)
).with_structured_output(Script, method="json_mode")
script = model.invoke(
[
SystemMessage(content=get_prompt_template("podcast_script_writer")),
HumanMessage(content=state["input"]),
],
)
logging.info(script)
print(script)
return {"script": script, "audio_chunks": []}