feat: api timeout

pdfix · Dec 9, 2024 · 8557ad9 · 8557ad9
1 parent 992fcb7
commit 8557ad9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/src/ai.py b/src/ai.py
@@ -1,5 +1,6 @@
 import base64
 
+import httpx
 from openai import OpenAI
 
 
@@ -12,8 +13,10 @@ def encode_image(image_path: str):
 def alt_description(img_path: str, api_key: str, lang: str):
     # Getting the base64 string
     base64_image = encode_image(img_path)
-    # print("Using api key: {}".format(api_key))
-    client = OpenAI(api_key=api_key)
+
+    client = OpenAI(
+        api_key=api_key, timeout=httpx.Timeout(None, connect=10, read=60, write=30)
+    )
 
     response = client.chat.completions.create(
         model="gpt-4o-mini",

diff --git a/src/process_pdf.py b/src/process_pdf.py
@@ -56,11 +56,7 @@ def render_page(doc: PdfDoc, page_num: int, bbox: PdfRect, zoom: float) -> bytea
 
 
 def update_image_alt(
-    elem: PdsStructElement,
-    doc: PdfDoc,
-    api_key: str,
-    overwrite: bool,
-    lang: str
+    elem: PdsStructElement, doc: PdfDoc, api_key: str, overwrite: bool, lang: str
 ) -> None:
     img = "image_" + str(elem.GetObject().GetId()) + ".jpg"
 
@@ -135,7 +131,7 @@ def alt_text(
     tags: str,
     license_name: str,
     license_key: str,
-    api_key: str,    
+    api_key: str,
     overwrite: bool,
     lang: str,
 ) -> None: