dwani 0.1.17__tar.gz → 0.1.19__tar.gz

{dwani-0.1.17 → dwani-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dwani
-Version: 0.1.17
+Version: 0.1.19
 Summary: Multimodal API for Indian + European languages (Chat, Vision, TTS, ASR, Translate, Docs)
 Author-email: sachin <python@dwani.ai>
 License: MIT License

{dwani-0.1.17 → dwani-0.1.19}/dwani/asr.py

@@ -42,7 +42,8 @@ def asr_transcribe(client, file_path, language):
         resp = requests.post(
             f"{client.api_base}/v1/transcribe/?language={api_language}",
             headers=client._headers(),
-            files=files
+            files=files,
+            timeout=90
         )
     if resp.status_code != 200:
         raise DwaniAPIError(resp)

{dwani-0.1.17 → dwani-0.1.19}/dwani/audio.py

@@ -12,7 +12,8 @@ def audio_speech(client, input, response_format="mp3", output_file=None, languag
         headers={**client._headers(), "accept": "application/json"},
         params=params,
         data='',  # Empty body, as in the curl example
-        stream=True
+        stream=True,
+        timeout=90
     )
     if resp.status_code != 200:
         raise DwaniAPIError(resp)

{dwani-0.1.17 → dwani-0.1.19}/dwani/chat.py

@@ -46,7 +46,8 @@ def chat_direct(client, prompt, model="gemma3", system_prompt=""):
     resp = requests.post(
         url,
         headers={**client._headers(), "Content-Type": "application/json"},
-        json=payload
+        json=payload,
+        timeout=90
     )
     if resp.status_code != 200:
         raise DwaniAPIError(resp)
@@ -72,7 +73,8 @@ def chat_create(client, prompt, src_lang, tgt_lang, model="gemma3"):
     resp = requests.post(
         url,
         headers={**client._headers(), "Content-Type": "application/json"},
-        json=payload
+        json=payload,
+        timeout=90
     )
     if resp.status_code != 200:
         raise DwaniAPIError(resp)

{dwani-0.1.17 → dwani-0.1.19}/dwani/docs.py

@@ -56,7 +56,7 @@ def document_ocr_all(client, file_path, model="gemma3"):
                 headers=client._headers(),
                 files=files,
                 data=data,
-                timeout=60
+                timeout=90
             )
             resp.raise_for_status()
         except requests.RequestException as e:
@@ -67,14 +67,15 @@ def document_ocr_all(client, file_path, model="gemma3"):
     return resp.json()
 
 
-def document_ocr_number(client, file_path, page_number=1, model="gemma3"):
+def document_ocr_number(client, file_path, page_number, model="gemma3"):
     """OCR a document (image/PDF) and return extracted text."""
     logger.debug(f"Calling document_ocr: file_path={file_path}, model={model}")
     validate_model(model)
     
     data = {"model": model,
-            "page_number": str(page_number)}
-
+            "page_number": page_number}
+    
+    params = {"model": data["model"], "page_number": data["page_number"]}
     with open(file_path, "rb") as f:
         mime_type = "application/pdf" if file_path.lower().endswith('.pdf') else "image/png"
         files = {"file": (file_path, f, mime_type)}
@@ -83,8 +84,8 @@ def document_ocr_number(client, file_path, page_number=1, model="gemma3"):
                 f"{client.api_base}/v1/extract-text",
                 headers=client._headers(),
                 files=files,
-                data=data,
-                timeout=60
+                params=params,
+                timeout=90
             )
             resp.raise_for_status()
         except requests.RequestException as e:
@@ -123,7 +124,7 @@ def document_summarize(client, file_path, page_number=1, src_lang="eng_Latn", tg
                 headers=headers,
                 files=files,
                 data=data,
-                timeout=60
+                timeout=90
             )
             resp.raise_for_status()
         except requests.RequestException as e:
@@ -164,7 +165,7 @@ def extract(client, file_path, page_number=1, src_lang="eng_Latn", tgt_lang="kan
                 headers=headers,
                 files=files,
                 data=data,
-                timeout=60
+                timeout=90
             )
             resp.raise_for_status()
         except requests.RequestException as e:
@@ -216,7 +217,7 @@ def doc_query(
                 headers=headers,
                 files=files,
                 data=data,
-                timeout=60
+                timeout=90
             )
             resp.raise_for_status()
         except requests.RequestException as e:
@@ -268,7 +269,7 @@ def doc_query_kannada(
                 headers=headers,
                 files=files,
                 data=data,
-                timeout=60
+                timeout=90
             )
             resp.raise_for_status()
         except requests.RequestException as e:
@@ -281,10 +282,10 @@ def doc_query_kannada(
 
 class Documents:
     @staticmethod
-    def run_ocr_number(file_path, page_number=1,model="gemma3"):
+    def run_ocr_number(file_path, page_number=2,model="gemma3"):
         from .client import DwaniClient
         client = DwaniClient()
-        return document_ocr_number(client, file_path, page_number=page_number, model=model)
+        return document_ocr_number(client, file_path, page_number, model)
     @staticmethod
     def run_ocr_all(file_path, model="gemma3"):
         from .client import DwaniClient

{dwani-0.1.17 → dwani-0.1.19}/dwani/translate.py

@@ -66,7 +66,8 @@ def run_translate(client, sentences, src_lang, tgt_lang):
     resp = requests.post(
         url,
         headers={**client._headers(), "Content-Type": "application/json", "accept": "application/json"},
-        json=payload
+        json=payload,
+        timeout=90
     )
     if resp.status_code != 200:
         raise DwaniAPIError(resp)

{dwani-0.1.17 → dwani-0.1.19}/dwani/vision.py

@@ -52,7 +52,8 @@ def vision_direct(client, file_path, query="describe this image", model="gemma3"
             url,
             headers=headers,
             files=files,
-            data=data
+            data=data,
+            timeout=90
         )
     if resp.status_code != 200:
         raise DwaniAPIError(resp)
@@ -84,7 +85,8 @@ def vision_caption(client, file_path, query="describe the image", src_lang="eng_
             url,
             headers=headers,
             files=files,
-            data=data
+            data=data,
+            timeout=90
         )
     if resp.status_code != 200:
         raise DwaniAPIError(resp)

{dwani-0.1.17 → dwani-0.1.19}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "dwani"
 
-version = "0.1.17"
+version = "0.1.19"
 description = "Multimodal API for Indian + European languages (Chat, Vision, TTS, ASR, Translate, Docs)"
 authors = [
     { name="sachin", email="python@dwani.ai" }