diff --git a/seed_contract_risks.py b/seed_contract_risks.py index 77a5fe1..7213cee 100644 --- a/seed_contract_risks.py +++ b/seed_contract_risks.py @@ -279,7 +279,7 @@ def main(): title=risk["title"], content=risk["content"], doc_type=risk["doc_type"], - category=risk.get("category"), + source=risk.get("source", "BossAi"), ) count += 1 diff --git a/seed_debt_templates.py b/seed_debt_templates.py index 7b4f192..cabeef4 100644 --- a/seed_debt_templates.py +++ b/seed_debt_templates.py @@ -271,7 +271,7 @@ def main(): title=tmpl["title"], content=tmpl["content"], doc_type=tmpl["doc_type"], - category=tmpl.get("category"), + source=tmpl.get("source", "BossAi"), ) count += 1