Agent_studio / modules /proposal_generator.py
Corin1998's picture
Create proposal_generator.py
43c4b04 verified
from typing import List
from modules.llm import generate
SYSTEM_PROMPT = """あなたは営業企画のエキスパートです。以下の制約で日本語の提案文を作成します。
- トーン:簡潔・価値訴求・次アクション明確
- 構成:①現状理解 ②提案価値 ③導入プラン ④期待効果 ⑤CTA
- 出力:Markdown
"""
def make_proposal(company_name: str, objective: str, top_contexts: List[str], temperature: float=0.4) -> str:
contexts_md = "\n\n".join([f"- {c}" for c in top_contexts])
user_prompt = f"""
会社名: {company_name}
目的: {objective or "初回接触(無料相談/PoCの打診)"}
参考コンテキスト(RAG上位):
{contexts_md}
上記を踏まえ、1枚提案のMarkdownを出力してください。見出し・箇条書きを効果的に。
"""
return generate(SYSTEM_PROMPT, user_prompt, temperature=temperature, max_new_tokens=600)
def suggest_next_actions(proposal_md: str, score_overall: float, temperature: float=0.3) -> str:
system = "あなたは営業マネージャーです。短い箇条書きで次アクションを提案してください。"
user = f"提案ドラフト:\n{proposal_md}\n\n見込み度(0-100): {score_overall}\n反応:未送信または未反応想定。"
return generate(system, user, temperature=temperature, max_new_tokens=256)