Upload data_updating_scripts/generate_reports.py with huggingface_hub

data_updating_scripts/generate_reports.py

@@ -152,17 +152,24 @@ def create_detailed_report(
     bill_json = json.dumps(bill_copy, ensure_ascii=False, indent=2)
 
     # Thread-based 180-second timeout — signal.alarm doesn't work because
-    # the openai SDK resets SIGALRM internally
+    # the openai SDK resets SIGALRM internally.
+    # IMPORTANT: Don't use `with ThreadPoolExecutor` — the context manager's
+    # __exit__ calls shutdown(wait=True) which blocks forever if the thread
+    # is stuck in a deadlocked HTTP connection.
     def _do_invoke():
         chain = DETAILED_REPORT_PROMPT | llm
         return chain.invoke({"bill_json": bill_json})
 
-    with ThreadPoolExecutor(max_workers=1) as executor:
-        future = executor.submit(_do_invoke)
-        try:
-            result = future.result(timeout=180)
-        except FuturesTimeout:
-            raise TimeoutError("API call exceeded 180s hard timeout")
+    executor = ThreadPoolExecutor(max_workers=1)
+    future = executor.submit(_do_invoke)
+    try:
+        result = future.result(timeout=180)
+    except FuturesTimeout:
+        future.cancel()
+        executor.shutdown(wait=False)
+        raise TimeoutError("API call exceeded 180s hard timeout")
+    else:
+        executor.shutdown(wait=False)
 
     # result can be AIMessage; get text
     report_text = getattr(result, "content", str(result))