handled bz2 spec pdf

commit-0 · Sep 28, 2024 · 72552d4 · 72552d4
1 parent 32972f2
commit 72552d4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/agent/agent_utils.py b/agent/agent_utils.py
@@ -1,3 +1,4 @@
+import bz2
 import git
 import os
 import re
@@ -250,6 +251,9 @@ def get_message(
         repo_info = ""
 
     if agent_config.use_spec_info:
+        with bz2.open("spec.pdf.bz2", "rb") as in_file:
+            with open("spec.pdf", "wb") as out_file:
+                out_file.write(in_file.read())
         spec_info = (
             f"\n{SPEC_INFO_HEADER} "
             + get_specification(specification_pdf_path=Path(repo_path, "spec.pdf"))[