langchain-ocr-lib 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

langchain_ocr_lib/di_config.py

@@ -17,6 +17,7 @@ from langfuse import Langfuse
 
 from langchain_ocr_lib.impl.chains.ocr_chain import OcrChain
 from langchain_ocr_lib.impl.settings.ollama_chat_settings import OllamaSettings
+from langchain_ocr_lib.impl.settings.vllm_chat_settings import VllmSettings
 from langchain_ocr_lib.impl.settings.openai_chat_settings import OpenAISettings
 from langchain_ocr_lib.impl.settings.llm_class_type_settings import LlmClassTypeSettings
 from langchain_ocr_lib.impl.settings.langfuse_settings import LangfuseSettings
@@ -53,6 +54,9 @@ def lib_di_config(binder: Binder):
     elif llm_class_type_settings.llm_type == "openai":
         settings = OpenAISettings()
         llm_instance = llm_provider(settings, ChatOpenAI)
+    elif llm_class_type_settings.llm_type == "vllm":
+        settings = VllmSettings()
+        llm_instance = llm_provider(settings, ChatOpenAI)
     else:
         raise NotImplementedError("Configured LLM is not implemented")
     binder.bind(LargeLanguageModelKey, llm_instance)

langchain_ocr_lib/impl/converter/pdf_converter.py

@@ -48,9 +48,11 @@ class Pdf2MarkdownConverter(File2MarkdownConverter):
                 with open(filename, "rb") as f:
                     file = f.read()
             except Exception as e:
-                raise ValueError("PDF corrupted or unsupported file type, %s" % e)
-
-        images = convert_from_bytes(file)
+                raise ValueError("PDF corrupted or unsupported file type") from e
+        try:
+            images = convert_from_bytes(file)
+        except Exception as e:
+            raise ValueError("PDF corrupted or unsupported file type") from e
 
         markdown = ""
         for image in images:
@@ -93,7 +95,10 @@ class Pdf2MarkdownConverter(File2MarkdownConverter):
             except Exception as e:
                 raise ValueError("PDF corrupted or unsupported file type") from e
 
-        images = convert_from_bytes(file)
+        try:
+            images = convert_from_bytes(file)
+        except Exception as e:
+            raise ValueError("PDF corrupted or unsupported file type") from e
 
         markdown = ""
         for image in images:

langchain_ocr_lib/impl/llms/llm_type.py

@@ -9,3 +9,4 @@ class LLMType(StrEnum):
 
     OLLAMA = "ollama"
     OPENAI = "openai"
+    VLLM = "vllm"

langchain_ocr_lib/impl/settings/openai_chat_settings.py

@@ -18,8 +18,8 @@ class OpenAISettings(BaseSettings):
         Total probability mass of tokens to consider at each step.
     temperature : float
         What sampling temperature to use.
-    vision_capable : bool
-        Flag to enable a vision capable model.
+    base_url : str
+        The base URL for the OpenAI API endpoint.
     """
 
     class Config:
@@ -31,5 +31,8 @@ class OpenAISettings(BaseSettings):
     model: str = Field(default="gpt-4o-mini-search-preview-2025-03-11", description="The model identifier")
     api_key: str = Field(default="", description="The API key for authentication")
     top_p: float = Field(default=1.0, description="Total probability mass of tokens to consider at each step")
-    temperature: float = Field(default=0.7, description="What sampling temperature to use")
-    vision_capable: bool = Field(default=False, description="Enable a vision capable model")
+    temperature: float = Field(default=0, description="What sampling temperature to use")
+    base_url: str = Field(
+        default="https://api.openai.com/v1",
+        description="The base URL for the OpenAI API endpoint",
+    )

langchain_ocr_lib/impl/settings/vllm_chat_settings.py

@@ -0,0 +1,38 @@
+"""Module contains settings regarding the Vllm API."""
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class VllmSettings(BaseSettings):
+    """
+    Contains settings regarding the Vllm API.
+
+    Attributes
+    ----------
+    model : str
+        The model identifier.
+    api_key : str
+        The API key for authentication.
+    top_p : float
+        Total probability mass of tokens to consider at each step.
+    temperature : float
+        What sampling temperature to use.
+    base_url : str
+        The base URL for the Vllm API endpoint.
+    """
+
+    class Config:
+        """Config class for reading fields from environment variables."""
+
+        env_prefix = "VLLM_"
+        case_sensitive = False
+
+    model: str = Field(default="", description="The model identifier")
+    api_key: str = Field(default="", description="The API key for authentication")
+    top_p: float = Field(default=1.0, description="Total probability mass of tokens to consider at each step")
+    temperature: float = Field(default=0, description="What sampling temperature to use")
+    base_url: str = Field(
+        default="http://localhost:8000/v1",
+        description="The base URL for the Vllm API endpoint",
+    )

{langchain_ocr_lib-0.2.0.dist-info → langchain_ocr_lib-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-ocr-lib
-Version: 0.2.0
+Version: 0.3.0
 Summary: 
 License: MIT
 Author: Andreas Klos
@@ -54,7 +54,7 @@ This package offers the core functionality to extract text from documents using
 
 ## 2. Features
 
-- **Vision-Language OCR:** Supports Ollama. Other LLM providers will be added soon.
+- **Vision-Language OCR:** Supports Ollama, vLLM and OpenAI (and other OpenAI conform providers). Other LLM providers can be easily integrated.
 - **CLI Interface:** Simple local execution via command line or container
 - **Highly Configurable:** Use environment variables to configure the OCR
 - **Dependency Injection:** Easily swap out components for custom implementations
@@ -70,7 +70,7 @@ This package offers the core functionality to extract text from documents using
 - **Python:** 3.11+
 - **Poetry:** [Install Poetry](https://python-poetry.org/docs/)
 - **Docker:** For containerized CLI usage (optional)
-- **Ollama:** Follow instructions [here](https://ollama.com)
+- **Ollama:** Follow instructions [here](https://ollama.com) (other LLM providers can be used as well, see [here](#2-features))
 - **Langfuse:** Different options for self hosting, see [here](https://langfuse.com/self-hosting) (optional, for observability)
 
 ### 3.2 Environment Setup

{langchain_ocr_lib-0.2.0.dist-info → langchain_ocr_lib-0.3.0.dist-info}/RECORD

@@ -5,24 +5,25 @@ langchain_ocr_lib/converter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 langchain_ocr_lib/converter/converter.py,sha256=oDUNzVWD743RgqIal7T4OVv-Z1RKE9uQYzAIPpgY3o8,1280
 langchain_ocr_lib/di_binding_keys/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/di_binding_keys/binding_keys.py,sha256=jE8rwNcLaI0NflIMkK0vu0LVy5o4y0pYgdjbpDNTGyk,338
-langchain_ocr_lib/di_config.py,sha256=m6ZYs5dNtAKFGVmyR8uTLOlFX3l71EI4cMkNL6asxME,3316
+langchain_ocr_lib/di_config.py,sha256=eYzDi_LJaYY_JhRnNqW3VYGd3N1QblaGFjWlL_6Vx9c,3537
 langchain_ocr_lib/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/impl/chains/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/impl/chains/ocr_chain.py,sha256=stE8RLE1ieRHf6XHreKCRfhNfXzw9fNLTake7xQBGL8,2673
 langchain_ocr_lib/impl/converter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/impl/converter/image_converter.py,sha256=G1rDOCbudWNL4sDvSGJ7CeeFrWUblfWPGaZf5JsnpiM,2871
-langchain_ocr_lib/impl/converter/pdf_converter.py,sha256=ssj8DL_9wf6kMhjUhDkw0gwSwNLrvgh8nBRspwj60Vk,3510
+langchain_ocr_lib/impl/converter/pdf_converter.py,sha256=pTHPojuNLCSWJp4FzXBHshXva2sBGyOs6Y7jnKJrnNo,3760
 langchain_ocr_lib/impl/langfuse_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/impl/langfuse_manager/langfuse_manager.py,sha256=AEF1iFYghr-62gcPcUb4Yi3DNRYfe-JsIWd3ymsIU8I,5403
 langchain_ocr_lib/impl/llms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/impl/llms/llm_factory.py,sha256=9DsUdoYNrjeWLGA9ISDdHN2cxcQ7DquNQ5it6zSxHlg,2199
-langchain_ocr_lib/impl/llms/llm_type.py,sha256=_Ap7yStlBn0tyOyfVLH1c2j2A9-ccsTCxAm7bgoRQnM,268
+langchain_ocr_lib/impl/llms/llm_type.py,sha256=_LKtdVuTRYX6gupkxJtEtIwrbtiMvZmG8WOxfzlm42M,286
 langchain_ocr_lib/impl/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/impl/settings/langfuse_settings.py,sha256=5lr3tVeiHXDUaYtWAnZPXrKxBJgM2wgaz7yyZThhCsE,812
 langchain_ocr_lib/impl/settings/language_settings.py,sha256=tdAC1t5wGu1MoH1jhjkDnxnX4Ui7giwxt7Qm8_LPkP8,627
 langchain_ocr_lib/impl/settings/llm_class_type_settings.py,sha256=4KC6zxby13wn38rB8055J8LNVTsmUfrOiyLtLuToHaM,598
 langchain_ocr_lib/impl/settings/ollama_chat_settings.py,sha256=8RWMsaK4qDrqC6Mrxekr8IEDYwcvjYwhw9xDwZemxI4,1506
-langchain_ocr_lib/impl/settings/openai_chat_settings.py,sha256=cXzxe-sea8VCK2M_u9ZIL4l8AR_YdhmA-phZa9fwf8o,1233
+langchain_ocr_lib/impl/settings/openai_chat_settings.py,sha256=gZqmFYDtF0l5lEAnuT2VzdqLWKnTPSK_lTeg7ERmJas,1276
+langchain_ocr_lib/impl/settings/vllm_chat_settings.py,sha256=y8PPNUcce1uA4kEu6p0p5vCwCOGp9uEEvHbCoS1Ohh8,1226
 langchain_ocr_lib/impl/tracers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/impl/tracers/langfuse_traced_chain.py,sha256=syjwNt8HfVmaWXZ-ElFYsc-KwpnKQz2LE3K5jV7c3GE,1599
 langchain_ocr_lib/language_mapping/language_mapping.py,sha256=VY7WkkZauoHNxkvgUYbig0rDmlKqDkz24cXMd6A7txM,700
@@ -31,7 +32,7 @@ langchain_ocr_lib/prompt_templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQe
 langchain_ocr_lib/prompt_templates/ocr_prompt.py,sha256=3Be1AL-HJkxPnAP0DNH1MqvAxFWTCeM5UOKP63xkHsY,3543
 langchain_ocr_lib/tracers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/tracers/traced_chain.py,sha256=uxRkdLNn_G6dAsti_gUuF7muhIj10xrOUL7HUga40oc,3056
-langchain_ocr_lib-0.2.0.dist-info/METADATA,sha256=ATjGYYdjA2U-TmsoUrXP_5aglABNsQybWT2_JL_zd8U,6113
-langchain_ocr_lib-0.2.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-langchain_ocr_lib-0.2.0.dist-info/entry_points.txt,sha256=l4mIs0tnIgbJYuVveZySQKVBnqNMHS-8ZZtLwz8ag5k,61
-langchain_ocr_lib-0.2.0.dist-info/RECORD,,
+langchain_ocr_lib-0.3.0.dist-info/METADATA,sha256=IaqIz9OXgu5WQXwEVpLmMNLmz2w3IowWmdZ7kt5O6VM,6240
+langchain_ocr_lib-0.3.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+langchain_ocr_lib-0.3.0.dist-info/entry_points.txt,sha256=l4mIs0tnIgbJYuVveZySQKVBnqNMHS-8ZZtLwz8ag5k,61
+langchain_ocr_lib-0.3.0.dist-info/RECORD,,