mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-05-04 15:58:22 +00:00
1bfcf9f429
* feat: enable ppt_composer.zh_CN.md with request.locale * fix: GeneratePPTRequest miss locale field --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
import logging
|
|
import os
|
|
import uuid
|
|
|
|
from langchain.schema import HumanMessage, SystemMessage
|
|
|
|
from src.config.agents import AGENT_LLM_MAP
|
|
from src.llms.llm import get_llm_by_type
|
|
from src.prompts.template import get_prompt_template
|
|
|
|
from .state import PPTState
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def ppt_composer_node(state: PPTState):
|
|
logger.info("Generating ppt content...")
|
|
model = get_llm_by_type(AGENT_LLM_MAP["ppt_composer"])
|
|
ppt_content = model.invoke(
|
|
[
|
|
SystemMessage(content=get_prompt_template("ppt/ppt_composer", locale=state.get("locale", "en-US"))),
|
|
HumanMessage(content=state["input"]),
|
|
],
|
|
)
|
|
logger.info(f"ppt_content: {ppt_content}")
|
|
# save the ppt content in a temp file
|
|
temp_ppt_file_path = os.path.join(os.getcwd(), f"ppt_content_{uuid.uuid4()}.md")
|
|
with open(temp_ppt_file_path, "w") as f:
|
|
f.write(ppt_content.content)
|
|
return {"ppt_content": ppt_content, "ppt_file_path": temp_ppt_file_path}
|