doctra 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

doctra/engines/vlm/provider.py

@@ -22,10 +22,10 @@ def make_model(
     """
     Build a callable Outlines model for VLM processing.
     
-    Creates an Outlines model instance configured for Gemini, OpenAI, Anthropic, OpenRouter, or Ollama
+    Creates an Outlines model instance configured for Gemini, OpenAI, Anthropic, OpenRouter, Qianfan, or Ollama
     providers. Only one backend is active at a time, with Gemini as the default.
 
-    :param vlm_provider: VLM provider to use ("gemini", "openai", "anthropic", "openrouter", or "ollama", default: "gemini")
+    :param vlm_provider: VLM provider to use ("gemini", "openai", "anthropic", "openrouter", "qianfan", or "ollama", default: "gemini")
     :param vlm_model: Model name to use (defaults to provider-specific defaults)
     :param api_key: API key for the VLM provider (required for all providers except Ollama)
     :return: Configured Outlines model instance
@@ -43,6 +43,8 @@ def make_model(
             vlm_model = "claude-opus-4-1"
         elif vlm_provider == "openrouter":
             vlm_model = "x-ai/grok-4"
+        elif vlm_provider == "qianfan":
+            vlm_model = "ernie-4.5-turbo-vl-32k"
         elif vlm_provider == "ollama":
             vlm_model = "llava:latest"
 
@@ -87,11 +89,24 @@ def make_model(
             vlm_model
         )
 
+    if vlm_provider == "qianfan":
+        if not api_key:
+            raise ValueError("Qianfan provider requires api_key to be passed to make_model(...).")
+        # Create the Qianfan client with OpenAI-compatible interface
+        client = openai.OpenAI(
+            base_url="https://qianfan.baidubce.com/v2",
+            api_key=api_key,
+        )
+        return outlines.from_openai(
+            client,
+            vlm_model
+        )
+
     if vlm_provider == "ollama":
         # Ollama doesn't use Outlines, so we return a custom wrapper
         return OllamaModelWrapper(vlm_model)
 
-    raise ValueError(f"Unsupported provider: {vlm_provider}. Use 'gemini', 'openai', 'anthropic', 'openrouter', or 'ollama'.")
+    raise ValueError(f"Unsupported provider: {vlm_provider}. Use 'gemini', 'openai', 'anthropic', 'openrouter', 'qianfan', or 'ollama'.")
 
 
 class OllamaModelWrapper:

doctra/engines/vlm/service.py

@@ -32,7 +32,7 @@ class VLMStructuredExtractor:
         """
         Initialize the VLMStructuredExtractor with provider configuration.
 
-        :param vlm_provider: VLM provider to use ("gemini", "openai", "anthropic", or "openrouter", default: "gemini")
+        :param vlm_provider: VLM provider to use ("gemini", "openai", "anthropic", "openrouter", "qianfan", or "ollama", default: "gemini")
         :param vlm_model: Model name to use (defaults to provider-specific defaults)
         :param api_key: API key for the VLM provider (required for all providers)
         """

doctra/version.py

@@ -1,2 +1,2 @@
 """Version information for Doctra."""
-__version__ = '0.5.0'
+__version__ = '0.5.1'

{doctra-0.5.0.dist-info → doctra-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: doctra
-Version: 0.5.0
+Version: 0.5.1
 Summary: Parse, extract, and analyze documents with ease
 Home-page: https://github.com/AdemBoukhris457/Doctra
 Author: Adem Boukhris
@@ -362,7 +362,7 @@ parser = StructuredPDFParser()
 # Parser with VLM for structured data extraction
 parser = StructuredPDFParser(
     use_vlm=True,
-    vlm_provider="openai",  # or "gemini" or "anthropic" or "openrouter"
+    vlm_provider="openai",  # or "gemini", "anthropic", "openrouter", "qianfan", "ollama"
     vlm_api_key="your_api_key_here"
 )
 
@@ -917,7 +917,7 @@ parser.display_pages_with_boxes("document.pdf")
 
 ### 🤖 VLM Integration
 - Vision Language Model support for structured data extraction
-- Multiple provider options (OpenAI, Gemini, Anthropic, OpenRouter)
+- Multiple provider options (OpenAI, Gemini, Anthropic, OpenRouter, Qianfan, Ollama)
 - Automatic conversion of charts and tables to structured formats
 
 ### 📊 Multiple Output Formats

{doctra-0.5.0.dist-info → doctra-0.5.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 doctra/__init__.py,sha256=rNLCyODOpaPb_TTP6qmQnuWZJW9JPXrxg1IfKnvb1No,773
-doctra/version.py,sha256=A8O6Kr44VM50_wm5hKwwcjBUZJFPAk64i5o7DatMRlQ,62
+doctra/version.py,sha256=b0vGLL2RHYHeqdwkHTZmk2FRb1-xEhcO1auAArG969s,62
 doctra/cli/__init__.py,sha256=4PTujjYRShOOUlZ7PwuWckShPWLC4v4CYIhJpzgyv1k,911
 doctra/cli/main.py,sha256=UhWTatY3qIeutZzVo9syLG2srbs8MZuGaLo5tk9xC_M,43108
 doctra/cli/utils.py,sha256=GKSSGi-JjNXufNekqCysSev7St1t32caYMduy0Tq96s,11971
@@ -15,8 +15,8 @@ doctra/engines/ocr/path_resolver.py,sha256=2_7Nsekt3dCDU3oVsgdr62iMrlAhbGNfYwgh4
 doctra/engines/ocr/pytesseract_engine.py,sha256=Imz2uwju6himkBiS8CH7DLxBRe-LtmMYZiOdb_6PoQw,2911
 doctra/engines/vlm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 doctra/engines/vlm/outlines_types.py,sha256=fQK6ru7XiXHaa8JPpaTTBaTk_zQ93ZyhFp4SyAnUdVU,1337
-doctra/engines/vlm/provider.py,sha256=lXQJNxDTxBHSxuEMbF37PjETEokR9o7rc2jLWEH9RnU,9943
-doctra/engines/vlm/service.py,sha256=nygxMe7uTq6Bv70ycBPL59F2a0ESp1Hix4j833p6rUM,4343
+doctra/engines/vlm/provider.py,sha256=QMr-gcbhyXgTQOHPIjIrmsLTNfkbDR69I3uR5Z2QVU0,10521
+doctra/engines/vlm/service.py,sha256=8o3JbNEkAFLNxSyu3KW7srI25PSLY-epzNZquKTxgcU,4364
 doctra/exporters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 doctra/exporters/excel_writer.py,sha256=rwyqlH73P7z413BELovQY_pS6IMkkqHEho6mbPrJ2Sk,11857
 doctra/exporters/html_writer.py,sha256=zJPoMiFF9lx9fHpdqk0y8diNNeQVC68wNvUInX918fY,46017
@@ -66,9 +66,9 @@ doctra/utils/pdf_io.py,sha256=c8EY47Z1iqVtlLFHS_n0qGuXJ5ERFaMUd84ivXV0b9E,706
 doctra/utils/progress.py,sha256=BD9YZqYLZw6yohQnyUV3w9QsQuiIrXM_EqByOSSJsDU,11912
 doctra/utils/quiet.py,sha256=5XPS-1CtJ0sVk6qgSQctdhr_wR8mP1xoJLoUbmkXROA,387
 doctra/utils/structured_utils.py,sha256=vU84dsD8wIlTyMsA9hitorGH-eroQiVuWEpBTQBUT24,1478
-doctra-0.5.0.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-doctra-0.5.0.dist-info/METADATA,sha256=tall4Spu8hFtNARaVVCNl9QedT-4VUubsV4oqrMLxoc,37168
-doctra-0.5.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-doctra-0.5.0.dist-info/entry_points.txt,sha256=4G2RHamA0llCiIXaQQm8EDkVK9JNGKbI7uDnXVFgIaY,47
-doctra-0.5.0.dist-info/top_level.txt,sha256=jI7E8jHci2gP9y0GYaWxlg9jG0O5n3FjHJJPLXDXMds,7
-doctra-0.5.0.dist-info/RECORD,,
+doctra-0.5.1.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+doctra-0.5.1.dist-info/METADATA,sha256=IInFIxxklcgLQHTvStUSTkqQXwXGly0JbZOSpBQAu0A,37202
+doctra-0.5.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+doctra-0.5.1.dist-info/entry_points.txt,sha256=4G2RHamA0llCiIXaQQm8EDkVK9JNGKbI7uDnXVFgIaY,47
+doctra-0.5.1.dist-info/top_level.txt,sha256=jI7E8jHci2gP9y0GYaWxlg9jG0O5n3FjHJJPLXDXMds,7
+doctra-0.5.1.dist-info/RECORD,,