Skip to content

Commit 72552d4

Browse files
committed
handled bz2 spec pdf
1 parent 32972f2 commit 72552d4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

agent/agent_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import bz2
12
import git
23
import os
34
import re
@@ -250,6 +251,9 @@ def get_message(
250251
repo_info = ""
251252

252253
if agent_config.use_spec_info:
254+
with bz2.open("spec.pdf.bz2", "rb") as in_file:
255+
with open("spec.pdf", "wb") as out_file:
256+
out_file.write(in_file.read())
253257
spec_info = (
254258
f"\n{SPEC_INFO_HEADER} "
255259
+ get_specification(specification_pdf_path=Path(repo_path, "spec.pdf"))[

0 commit comments

Comments
 (0)