Skip to content

Commit 46c8437

Browse files
committed
Minor writerow updates
1 parent 1fb1388 commit 46c8437

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

llm/backfill_summaries.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -40,12 +40,16 @@ def make_bill_summary(bill_id, status, summary, topics):
4040

4141
# If the summary is already populated move on
4242
if summary is not None:
43-
csv_writer.write(make_bill_summary(bill_id, "previous_summary", None, None))
43+
csv_writer.writerow(
44+
make_bill_summary(bill_id, "previous_summary", None, None)
45+
)
4446
continue
4547

4648
summary = get_summary_api_function(bill_id, document_title, document_text)
4749
if summary["status"] in [-1, -2]:
48-
csv_writer.write(make_bill_summary(bill_id, "failed_summary", None, None))
50+
csv_writer.writerow(
51+
make_bill_summary(bill_id, "failed_summary", None, None)
52+
)
4953
continue
5054
# Note: `normalize_summary` does some post-processing to clean up the summaries
5155
# As of 2025-10-21 this was necessary due to the LLM prompt
@@ -55,21 +59,23 @@ def make_bill_summary(bill_id, status, summary, topics):
5559
# If the topics are already populated, just make a note of it
5660
topics = document.get("topics")
5761
if topics is not None:
58-
csv_writer.write(make_bill_summary(bill_id, "previous_topics", None, None))
62+
csv_writer.writerow(
63+
make_bill_summary(bill_id, "previous_topics", None, None)
64+
)
5965

6066
tags = get_tags_api_function_v2(bill_id, document_title, summary)
6167
# If the tags fail, make a note and at least write the summary for debugging
6268
if tags["status"] != 1:
63-
csv_writer.write(make_bill_summary(bill_id, "failed_topics", None, None))
64-
csv_writer.write(
69+
csv_writer.writerow(make_bill_summary(bill_id, "failed_topics", None, None))
70+
csv_writer.writerow(
6571
make_bill_summary(bill_id, "generated_summary", summary, None)
6672
)
6773
continue
6874
topics_and_categories = get_categories_from_topics(
6975
tags["tags"], CATEGORY_BY_TOPIC
7076
)
7177
bill.reference.update({"topics": topics_and_categories})
72-
csv_writer.write(
78+
csv_writer.writerow(
7379
make_bill_summary(
7480
bill_id, "generated_topics", summary, topics_and_categories
7581
)

0 commit comments

Comments
 (0)