langchain-ocr-lib 0.3.2__py3-none-any.whl → 0.4.0__py3-none-any.whl

langchain_ocr_lib/di_config.py

@@ -13,11 +13,13 @@ from langchain_ocr_lib.di_binding_keys.binding_keys import (
 )
 from langchain_ollama import ChatOllama
 from langchain_openai import ChatOpenAI
+from langchain_together import ChatTogether
 from langfuse import Langfuse
 from functools import partial
 
 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.together_ai_chat_settings import TogetherAISettings
 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
@@ -48,22 +50,29 @@ 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()
-        partial_llm_provider = partial(llm_provider,settings, ChatOllama)
+        model_name = settings.model
+        partial_llm_provider = partial(llm_provider, settings, ChatOllama)
     elif llm_class_type_settings.llm_type == "openai":
         settings = OpenAISettings()
-        partial_llm_provider = partial(llm_provider,settings, ChatOpenAI)
+        model_name = settings.model_name
+        partial_llm_provider = partial(llm_provider, settings, ChatOpenAI)
     elif llm_class_type_settings.llm_type == "vllm":
         settings = VllmSettings()
-        partial_llm_provider = partial(llm_provider,settings, ChatOpenAI)
+        model_name = settings.model_name
+        partial_llm_provider = partial(llm_provider, settings, ChatOpenAI)
+    elif llm_class_type_settings.llm_type == "together-ai":
+        settings = TogetherAISettings()
+        model_name = settings.model_name
+        partial_llm_provider = partial(llm_provider, settings, ChatTogether)
     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,

langchain_ocr_lib/impl/langfuse_manager/langfuse_manager.py

@@ -58,9 +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 None
+            return langfuse_prompt
         try:
             langfuse_prompt = self._langfuse.get_prompt(base_prompt_name)
         except NotFoundError:

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/impl/llms/llm_type.py

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

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,9 +28,16 @@ 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")
+    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")
     base_url: str = Field(
         default="https://api.openai.com/v1",

langchain_ocr_lib/impl/settings/together_ai_chat_settings.py

@@ -0,0 +1,45 @@
+"""Module contains settings regarding the Together AI API."""
+
+from pydantic import Field
+from pydantic_settings import BaseSettings
+
+
+class TogetherAISettings(BaseSettings):
+    """
+    Contains settings regarding the Together AI API.
+
+    Attributes
+    ----------
+    model_name : str
+        The Together AI model identifier.
+    together_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.
+    together_api_base : str
+        The base URL for the Together AI API endpoint.
+    """
+
+    class Config:
+        """Config class for reading fields from environment variables."""
+
+        env_prefix = "TOGETHER_"
+        case_sensitive = False
+
+    model_name: str = Field(
+        default="",
+        description="The Together AI model identifier",
+        title="Together AI Model",
+    )
+    together_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")
+    together_api_base: str = Field(
+        default="https://api.together.xyz/v1/",
+        env="API_BASE",
+        description="The base URL for the Together AI API endpoint",
+    )

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,9 +28,16 @@ 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")
+    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")
     base_url: str = Field(
         default="http://localhost:8000/v1",

{langchain_ocr_lib-0.3.2.dist-info → langchain_ocr_lib-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langchain-ocr-lib
-Version: 0.3.2
+Version: 0.4.0
 Summary: Modular, vision-LLM-powered chain to convert image and PDF documents into clean Markdown.
 License: MIT
 Author: Andreas Klos
@@ -16,6 +16,7 @@ Requires-Dist: inject (>=5.2.1,<6.0.0)
 Requires-Dist: langchain-community (>=0.3.19,<0.4.0)
 Requires-Dist: langchain-ollama (>=0.2.0,<0.3.0)
 Requires-Dist: langchain-openai (>=0.3.8,<0.4.0)
+Requires-Dist: langchain-together (>=0.3.0,<0.4.0)
 Requires-Dist: langfuse (>=2.59.7,<3.0.0)
 Requires-Dist: openai (>=1.42.0,<2.0.0)
 Requires-Dist: pdf2image (>=1.17.0,<2.0.0)

{langchain_ocr_lib-0.3.2.dist-info → langchain_ocr_lib-0.4.0.dist-info}/RECORD

@@ -5,7 +5,7 @@ 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=LvRnptts1VCDa3HVP7jtBtQTyG-QEErbyplGX86uaUA,3802
+langchain_ocr_lib/di_config.py,sha256=AhSZoYeaX6-sjfnM5J3N3ECAitQV5KFBvU0vAAu6dbQ,4285
 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
@@ -13,17 +13,18 @@ langchain_ocr_lib/impl/converter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 langchain_ocr_lib/impl/converter/image_converter.py,sha256=G1rDOCbudWNL4sDvSGJ7CeeFrWUblfWPGaZf5JsnpiM,2871
 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=C2waQ1Mvqz6bECXAcaMiBsE8TLc2kLr5QUIgXqgH_cE,5311
+langchain_ocr_lib/impl/langfuse_manager/langfuse_manager.py,sha256=6kvPfYObLmbYZAvtYNx7g9xz5wYz0PxohEzyctpGRto,5353
 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=_LKtdVuTRYX6gupkxJtEtIwrbtiMvZmG8WOxfzlm42M,286
+langchain_ocr_lib/impl/llms/llm_factory.py,sha256=pYPwUowm8r42BwOn4W1j4-cEpL9LCCkZU_pJ4xQhshU,2200
+langchain_ocr_lib/impl/llms/llm_type.py,sha256=tNyvUbgIMW7t2-lMEVo-4GyB2hk5Jw9kOaUbVtAK2LA,318
 langchain_ocr_lib/impl/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_ocr_lib/impl/settings/langfuse_settings.py,sha256=QT4_VwYj0msFbgL3qIQ-oer3Lt0qny0FFAyfssGu-q0,962
 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=YQkgD7CfOjHN5wkpJakO0GfM7-D2GqoJLP1gB2932ms,1525
-langchain_ocr_lib/impl/settings/openai_chat_settings.py,sha256=NqVfkcI8OoD8TVxyv4l0G9ycUC6LIs6Qs4kQRL24doA,1331
-langchain_ocr_lib/impl/settings/vllm_chat_settings.py,sha256=Zr4L6Urp-f1JZu7Q1dwL6671EQbrIIYL0ubJSQlod3c,1281
+langchain_ocr_lib/impl/settings/openai_chat_settings.py,sha256=5OnHo8a2iRsEvohn09Lf30MNFgS1GWVvOa5kyUusHCQ,1407
+langchain_ocr_lib/impl/settings/together_ai_chat_settings.py,sha256=Jr2dHPpa1VTKwbrCLrjxkHCkWv1-iY1w0DQoklD9SB8,1470
+langchain_ocr_lib/impl/settings/vllm_chat_settings.py,sha256=ZnijdQBSxHh6cGcU5uYth9LHpYYQyIF0XPbsTMh1Fxk,1357
 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
@@ -32,7 +33,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.3.2.dist-info/METADATA,sha256=MYG6NqcxYMkVY2UQ0SbSCdNmpSJJ98NUb5KLBSDciyg,6440
-langchain_ocr_lib-0.3.2.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-langchain_ocr_lib-0.3.2.dist-info/entry_points.txt,sha256=l4mIs0tnIgbJYuVveZySQKVBnqNMHS-8ZZtLwz8ag5k,61
-langchain_ocr_lib-0.3.2.dist-info/RECORD,,
+langchain_ocr_lib-0.4.0.dist-info/METADATA,sha256=x5QI_KtLLjUwDbY1FO3olk1wT6K7uufa3sVCOcTYXRQ,6491
+langchain_ocr_lib-0.4.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+langchain_ocr_lib-0.4.0.dist-info/entry_points.txt,sha256=l4mIs0tnIgbJYuVveZySQKVBnqNMHS-8ZZtLwz8ag5k,61
+langchain_ocr_lib-0.4.0.dist-info/RECORD,,