langchain-ocr-lib 0.3.1__tar.gz → 0.3.3__tar.gz

{langchain_ocr_lib-0.3.1 → langchain_ocr_lib-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langchain-ocr-lib
-Version: 0.3.1
+Version: 0.3.3
 Summary: Modular, vision-LLM-powered chain to convert image and PDF documents into clean Markdown.
 License: MIT
 Author: Andreas Klos
@@ -25,9 +25,13 @@ Requires-Dist: pytest-asyncio (>=0.25.0,<0.26.0)
 Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
 Description-Content-Type: text/markdown
 
-# langchain_ocr_lib
+# langchain-ocr-lib
 
-**langchain_ocr_lib** is the OCR processing engine behind LangChain-OCR. It provides a modular, vision-LLM-powered Chain to convert image and PDF documents into clean Markdown. Designed for direct CLI usage or integration into larger applications.
+**langchain-ocr-lib** is the OCR processing engine behind LangChain-OCR. It provides a modular, vision-LLM-powered Chain to convert image and PDF documents into clean Markdown. Designed for direct CLI usage or integration into larger applications.
+
+<div align="center">
+  <img src="./images/logo.png" alt="Logo" style="width:30%;">
+</div>
 
 ## Table of Contents
 
@@ -137,7 +141,7 @@ class Converter:
         return self._converter.convert2markdown(filename=filename)
 
 converter = Converter()
-markdown = converter.convert("../docs/invoice.pdf") # Adjust the file path as needed
+markdown = converter.convert("../examples/invoice.pdf") # Adjust the file path as needed
 print(markdown)
 ```
 
@@ -174,7 +178,7 @@ class Converter:
         self._converter.convert(filename=filename)
 
 converter = Converter()
-converter.convert("../docs/invoice.pdf") # Adjust the file path as needed
+converter.convert("../examples/invoice.pdf") # Adjust the file path as needed
 ```
 
 ### 4.3 Docker
@@ -183,6 +187,6 @@ Run OCR via Docker without local Python setup:
 
 ```bash
 docker build -t ocr -f langchain_ocr_lib/Dockerfile .
-docker run --net=host -it --rm -v ./docs:/app/docs:ro ocr docs/invoice.png
+docker run --net=host -it --rm -v ./examples:/app/examples:ro ocr examples/invoice.png
 ```
 

{langchain_ocr_lib-0.3.1 → langchain_ocr_lib-0.3.3}/README.md

@@ -1,6 +1,10 @@
-# langchain_ocr_lib
+# langchain-ocr-lib
 
-**langchain_ocr_lib** is the OCR processing engine behind LangChain-OCR. It provides a modular, vision-LLM-powered Chain to convert image and PDF documents into clean Markdown. Designed for direct CLI usage or integration into larger applications.
+**langchain-ocr-lib** is the OCR processing engine behind LangChain-OCR. It provides a modular, vision-LLM-powered Chain to convert image and PDF documents into clean Markdown. Designed for direct CLI usage or integration into larger applications.
+
+<div align="center">
+  <img src="./images/logo.png" alt="Logo" style="width:30%;">
+</div>
 
 ## Table of Contents
 
@@ -110,7 +114,7 @@ class Converter:
         return self._converter.convert2markdown(filename=filename)
 
 converter = Converter()
-markdown = converter.convert("../docs/invoice.pdf") # Adjust the file path as needed
+markdown = converter.convert("../examples/invoice.pdf") # Adjust the file path as needed
 print(markdown)
 ```
 
@@ -147,7 +151,7 @@ class Converter:
         self._converter.convert(filename=filename)
 
 converter = Converter()
-converter.convert("../docs/invoice.pdf") # Adjust the file path as needed
+converter.convert("../examples/invoice.pdf") # Adjust the file path as needed
 ```
 
 ### 4.3 Docker
@@ -156,5 +160,5 @@ Run OCR via Docker without local Python setup:
 
 ```bash
 docker build -t ocr -f langchain_ocr_lib/Dockerfile .
-docker run --net=host -it --rm -v ./docs:/app/docs:ro ocr docs/invoice.png
+docker run --net=host -it --rm -v ./examples:/app/examples:ro ocr examples/invoice.png
 ```

{langchain_ocr_lib-0.3.1 → langchain_ocr_lib-0.3.3}/pyproject.toml

@@ -7,7 +7,7 @@ langchain-ocr = "langchain_ocr_lib.main:main"
 
 [tool.poetry]
 name = "langchain-ocr-lib"
-version = "0.3.1"
+version = "0.3.3"
 description = "Modular, vision-LLM-powered chain to convert image and PDF documents into clean Markdown."
 authors = ["Andreas Klos <aklos@outlook.de>"]
 readme = "README.md"

{langchain_ocr_lib-0.3.1 → langchain_ocr_lib-0.3.3}/src/langchain_ocr_lib/di_config.py

@@ -48,22 +48,25 @@ def lib_di_config(binder: Binder):
     langfuse_settings = LangfuseSettings()
     llm_class_type_settings = LlmClassTypeSettings()
     language_settings = LanguageSettings()
-
+    model_name = ""
     if llm_class_type_settings.llm_type == "ollama":
         settings = OllamaSettings()
+        model_name = settings.model
         partial_llm_provider = partial(llm_provider,settings, ChatOllama)
     elif llm_class_type_settings.llm_type == "openai":
         settings = OpenAISettings()
+        model_name = settings.model_name
         partial_llm_provider = partial(llm_provider,settings, ChatOpenAI)
     elif llm_class_type_settings.llm_type == "vllm":
         settings = VllmSettings()
+        model_name = settings.model_name
         partial_llm_provider = partial(llm_provider,settings, ChatOpenAI)
     else:
         raise NotImplementedError("Configured LLM is not implemented")
     
     binder.bind_to_provider(LargeLanguageModelKey, partial_llm_provider)
 
-    prompt = ocr_prompt_template_builder(language=language_settings.language, model_name=settings.model)
+    prompt = ocr_prompt_template_builder(language=language_settings.language, model_name=model_name)
 
     binder.bind(
         LangfuseClientKey,
@@ -80,17 +83,19 @@ def lib_di_config(binder: Binder):
             managed_prompts={
                 OcrChain.__name__: prompt,
             },
+            enabled=langfuse_settings.enabled,
         ),
     )
 
-    binder.bind(OcrChainKey, OcrChain())
+    binder.bind(OcrChainKey if langfuse_settings.enabled else LangfuseTracedChainKey, OcrChain())
 
-    binder.bind(
-        LangfuseTracedChainKey,
-        LangfuseTracedChain(
-            settings=langfuse_settings,
-        ),
-    )
+    if langfuse_settings.enabled:
+        binder.bind(
+            LangfuseTracedChainKey,
+            LangfuseTracedChain(
+                settings=langfuse_settings,
+            ),
+        )
 
     binder.bind(PdfConverterKey, Pdf2MarkdownConverter())
     binder.bind(ImageConverterKey, Image2MarkdownConverter())

{langchain_ocr_lib-0.3.1 → langchain_ocr_lib-0.3.3}/src/langchain_ocr_lib/impl/langfuse_manager/langfuse_manager.py

@@ -32,8 +32,10 @@ class LangfuseManager:
     def __init__(
         self,
         managed_prompts: dict[str, str],
+        enabled: bool = True,
     ):
         self._managed_prompts = managed_prompts
+        self._enabled = enabled
 
     def get_langfuse_prompt(self, base_prompt_name: str) -> Optional[ChatPromptClient]:
         """
@@ -56,6 +58,10 @@ class LangfuseManager:
         Exception
             If an error occurs while retrieving the prompt template from Langfuse.
         """
+        langfuse_prompt = None
+        if not self._enabled:
+            logger.info("Langfuse is not enabled. Using fallback prompt.")
+            return langfuse_prompt
         try:
             langfuse_prompt = self._langfuse.get_prompt(base_prompt_name)
         except NotFoundError:
@@ -72,11 +78,7 @@ class LangfuseManager:
             )
             langfuse_prompt = self._langfuse.get_prompt(base_prompt_name)
         except Exception as error:
-            logger.error(
-                "Error occured while getting prompt template from langfuse. Error:\n{error}",
-                extra={error: error},
-            )
-            return None
+            logger.error(f"Error occurred while getting prompt template from langfuse. Error:\n{error}")
         return langfuse_prompt
 
     def get_base_llm(self, name: str) -> LLM:
@@ -94,9 +96,12 @@ class LangfuseManager:
             The base Large Language Model. If the Langfuse prompt is not found,
             returns the LLM with a fallback configuration.
         """
+        if not self._enabled:
+            logger.info("Langfuse is not enabled. Using fallback LLM.")
+            return self._llm
         langfuse_prompt = self.get_langfuse_prompt(name)
         if not langfuse_prompt:
-            logger.error("Using fallback for llm")
+            logger.warning("Could not retrieve prompt template from langfuse. Using fallback LLM.")
             return self._llm
 
         return self._llm.with_config({"configurable": langfuse_prompt.config})
@@ -121,7 +126,8 @@ class LangfuseManager:
         """
         langfuse_prompt = self.get_langfuse_prompt(name)
         if not langfuse_prompt:
-            logger.error("Could not retrieve prompt template from langfuse. Using fallback value.")
+            if self._enabled:
+                logger.warning("Could not retrieve prompt template from langfuse. Using fallback value.")
             fallback = self._managed_prompts[name]
             if isinstance(fallback, ChatPromptTemplate):
                 return fallback

{langchain_ocr_lib-0.3.1 → langchain_ocr_lib-0.3.3}/src/langchain_ocr_lib/impl/llms/llm_factory.py

@@ -43,6 +43,7 @@ def get_configurable_fields_from(settings: BaseSettings) -> dict[str, Configurab
         settings_of_interest = settings.model_fields[field_name]
         if settings_of_interest.title is not None:
             _fields[field_name] = ConfigurableField(id=field_name, name=settings_of_interest.title)
+    
     return _fields
 
 

{langchain_ocr_lib-0.3.1 → langchain_ocr_lib-0.3.3}/src/langchain_ocr_lib/impl/settings/langfuse_settings.py

@@ -27,3 +27,7 @@ class LangfuseSettings(BaseSettings):
     secret_key: str = Field(default="", description="The secret key for Langfuse.")
     public_key: str = Field(default="", description="The public key for Langfuse.")
     host: str = Field(default="https://api.langchain.com", description="The host for Langfuse.")
+    enabled: bool = Field(
+        default=True,
+        description="Whether to enable Langfuse. If set to False, Langfuse will not be used.",
+    )

{langchain_ocr_lib-0.3.1 → langchain_ocr_lib-0.3.3}/src/langchain_ocr_lib/impl/settings/openai_chat_settings.py

@@ -10,7 +10,7 @@ class OpenAISettings(BaseSettings):
 
     Attributes
     ----------
-    model : str
+    model_name : str
         The model identifier.
     api_key : str
         The API key for authentication.
@@ -28,7 +28,12 @@ class OpenAISettings(BaseSettings):
         env_prefix = "OPENAI_"
         case_sensitive = False
 
-    model: str = Field(default="gpt-4o-mini-search-preview-2025-03-11", description="The model identifier", title="LLM Model")
+    model_name: str = Field(
+        default="gpt-4o-mini-search-preview-2025-03-11",
+        env="MODEL",
+        description="The model identifier",
+        title="LLM Model",
+    )
     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", title="Top P")
     temperature: float = Field(default=0, description="What sampling temperature to use", title="Temperature")

{langchain_ocr_lib-0.3.1 → langchain_ocr_lib-0.3.3}/src/langchain_ocr_lib/impl/settings/vllm_chat_settings.py

@@ -10,7 +10,7 @@ class VllmSettings(BaseSettings):
 
     Attributes
     ----------
-    model : str
+    model_name : str
         The model identifier.
     api_key : str
         The API key for authentication.
@@ -28,7 +28,12 @@ class VllmSettings(BaseSettings):
         env_prefix = "VLLM_"
         case_sensitive = False
 
-    model: str = Field(default="", description="The model identifier", title="LLM Model")
+    model_name: str = Field(
+        default="",
+        env="MODEL",
+        description="The model identifier",
+        title="LLM Model",
+    )
     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", title="Top P")
     temperature: float = Field(default=0, description="What sampling temperature to use", title="Temperature")