GitHub-Proxy/deer-flow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/bytedance/deer-flow synced 2025-08-18 02:35:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: use "json_mode"

Browse Source
This commit is contained in:
Li Xin 2025-04-19 22:02:14 +08:00
parent 7497857b71
commit 19d8a59def
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/podcast/graph/script_writer_node.py
View File

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