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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langchain-ocr-lib
-Version: 0.3.1
+Version: 0.3.2
 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.2}/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.2}/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.2"
 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.2}/src/langchain_ocr_lib/di_config.py

@@ -80,17 +80,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.2}/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,9 @@ class LangfuseManager:
         Exception
             If an error occurs while retrieving the prompt template from Langfuse.
         """
+        if not self._enabled:
+            logger.info("Langfuse is not enabled. Using fallback prompt.")
+            return None
         try:
             langfuse_prompt = self._langfuse.get_prompt(base_prompt_name)
         except NotFoundError:
@@ -72,11 +77,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 +95,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 +125,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.2}/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.",
+    )