Corin1998 commited on
Commit
d79c7a6
·
verified ·
1 Parent(s): 3e95b96

Update modules/workflow.py

Browse files
Files changed (1) hide show
  1. modules/workflow.py +14 -14
modules/workflow.py CHANGED
@@ -1,8 +1,8 @@
1
  import os
2
  from modules.company_score import score_company
3
- from module.rag_retriever import retrieve_contexts
4
- from module.proposal_generator import make_proposal, suggest_next_actions
5
- from module.exporters import export_docx, export_pptx
6
  from modules.emailer import build_tracking_url
7
  from modules.utils import truncate
8
 
@@ -14,17 +14,17 @@ def _make_email(company_name: str, lead_email: str, proposal_md: str) -> dict:
14
  "redirect": os.getenv("PUBLIC_BASE_URL", "/")
15
  },
16
  )
17
- subject = f"{company_name}様向けご提案(無料PoCのご案内)"
18
  body = (
19
- f"{company_name} ご担当社様\n\n"
20
- "突然のご連絡失礼致します。貴社受けに要点を1枚にまとめた提案書を作成しました。 \n\n"
21
  f"▼提案要旨\n{truncate(proposal_md, 800)}\n\n"
22
- f"詳細はこちら(クリック計測用リンク) :\n{tracking}\n\n"
23
- "30分ほどオンラインでディスカッションでさせて戴ければ幸いです。\n"
24
  "ご検討のほど、よろしくお願いいたします。\n"
25
  "\n--\nAgent Studio"
26
  )
27
- return{"subject": subject, "body": body}
28
 
29
  def run_full_workflow(company_name: str,
30
  company_website: str,
@@ -34,7 +34,7 @@ def run_full_workflow(company_name: str,
34
  # 1) スコアリング
35
  score = score_company(company_name, company_website)
36
 
37
- # 2) コンテキスト検索(RAG)
38
  query = f"{company_name} {objective} {company_website}"
39
  top_contexts = retrieve_contexts(query, k=5)
40
 
@@ -45,19 +45,19 @@ def run_full_workflow(company_name: str,
45
  docx_path = export_docx(proposal_md, company_name)
46
  pptx_path = export_pptx(proposal_md, company_name)
47
 
48
- # 5) メール作成(送信はUIでは手動or別APIで)
49
  email = _make_email(company_name, lead_email or "", proposal_md)
50
 
51
  # 6) 次アクション提案
52
  next_actions = suggest_next_actions(proposal_md, score["overall"])
53
 
54
- return{
55
  "score": score,
56
  "top_contexts": top_contexts,
57
  "proposal_markdown": proposal_md,
58
- "exports":{
59
  "docx_path": docx_path,
60
- "pptx_path": pptx_path,
61
  },
62
  "email": email,
63
  "next_actions": next_actions
 
1
  import os
2
  from modules.company_score import score_company
3
+ from modules.rag_retriever import retrieve_contexts # ← ここを修正(modules)
4
+ from modules.proposal_generator import make_proposal, suggest_next_actions
5
+ from modules.exporters import export_docx, export_pptx
6
  from modules.emailer import build_tracking_url
7
  from modules.utils import truncate
8
 
 
14
  "redirect": os.getenv("PUBLIC_BASE_URL", "/")
15
  },
16
  )
17
+ subject = f"{company_name}様向けご提案(無料PoCのご案内)"
18
  body = (
19
+ f"{company_name} ご担当者様\n\n"
20
+ "突然のご連絡失礼いたします。貴社向けに要点を1枚にまとめた提案書を作成しました。\n\n"
21
  f"▼提案要旨\n{truncate(proposal_md, 800)}\n\n"
22
+ f"詳細はこちら(クリック計測用リンク):\n{tracking}\n\n"
23
+ "30分ほどオンラインでディスカッションできれば嬉しいです。\n"
24
  "ご検討のほど、よろしくお願いいたします。\n"
25
  "\n--\nAgent Studio"
26
  )
27
+ return {"subject": subject, "body": body}
28
 
29
  def run_full_workflow(company_name: str,
30
  company_website: str,
 
34
  # 1) スコアリング
35
  score = score_company(company_name, company_website)
36
 
37
+ # 2) コンテキスト検索(RAG
38
  query = f"{company_name} {objective} {company_website}"
39
  top_contexts = retrieve_contexts(query, k=5)
40
 
 
45
  docx_path = export_docx(proposal_md, company_name)
46
  pptx_path = export_pptx(proposal_md, company_name)
47
 
48
+ # 5) メール作成
49
  email = _make_email(company_name, lead_email or "", proposal_md)
50
 
51
  # 6) 次アクション提案
52
  next_actions = suggest_next_actions(proposal_md, score["overall"])
53
 
54
+ return {
55
  "score": score,
56
  "top_contexts": top_contexts,
57
  "proposal_markdown": proposal_md,
58
+ "exports": {
59
  "docx_path": docx_path,
60
+ "pptx_path": pptx_path
61
  },
62
  "email": email,
63
  "next_actions": next_actions