langchain 0.4.0.dev0__py3-none-any.whl → 1.0.0a2__py3-none-any.whl

langchain/chains/summarize/chain.py

@@ -1,174 +0,0 @@
-"""Load summarizing chains."""
-
-from collections.abc import Mapping
-from typing import Any, Optional, Protocol
-
-from langchain_core.callbacks import Callbacks
-from langchain_core.language_models import BaseLanguageModel
-from langchain_core.prompts import BasePromptTemplate
-
-from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
-from langchain.chains.combine_documents.map_reduce import MapReduceDocumentsChain
-from langchain.chains.combine_documents.reduce import ReduceDocumentsChain
-from langchain.chains.combine_documents.refine import RefineDocumentsChain
-from langchain.chains.combine_documents.stuff import StuffDocumentsChain
-from langchain.chains.llm import LLMChain
-from langchain.chains.summarize import map_reduce_prompt, refine_prompts, stuff_prompt
-
-
-class LoadingCallable(Protocol):
-    """Interface for loading the combine documents chain."""
-
-    def __call__(
-        self,
-        llm: BaseLanguageModel,
-        **kwargs: Any,
-    ) -> BaseCombineDocumentsChain:
-        """Callable to load the combine documents chain."""
-
-
-def _load_stuff_chain(
-    llm: BaseLanguageModel,
-    *,
-    prompt: BasePromptTemplate = stuff_prompt.PROMPT,
-    document_variable_name: str = "text",
-    verbose: Optional[bool] = None,
-    **kwargs: Any,
-) -> StuffDocumentsChain:
-    llm_chain = LLMChain(llm=llm, prompt=prompt, verbose=verbose)  # type: ignore[arg-type]
-    # TODO: document prompt
-    return StuffDocumentsChain(
-        llm_chain=llm_chain,
-        document_variable_name=document_variable_name,
-        verbose=verbose,  # type: ignore[arg-type]
-        **kwargs,
-    )
-
-
-def _load_map_reduce_chain(
-    llm: BaseLanguageModel,
-    *,
-    map_prompt: BasePromptTemplate = map_reduce_prompt.PROMPT,
-    combine_prompt: BasePromptTemplate = map_reduce_prompt.PROMPT,
-    combine_document_variable_name: str = "text",
-    map_reduce_document_variable_name: str = "text",
-    collapse_prompt: Optional[BasePromptTemplate] = None,
-    reduce_llm: Optional[BaseLanguageModel] = None,
-    collapse_llm: Optional[BaseLanguageModel] = None,
-    verbose: Optional[bool] = None,
-    token_max: int = 3000,
-    callbacks: Callbacks = None,
-    collapse_max_retries: Optional[int] = None,
-    **kwargs: Any,
-) -> MapReduceDocumentsChain:
-    map_chain = LLMChain(
-        llm=llm,
-        prompt=map_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
-        callbacks=callbacks,
-    )
-    _reduce_llm = reduce_llm or llm
-    reduce_chain = LLMChain(
-        llm=_reduce_llm,
-        prompt=combine_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
-        callbacks=callbacks,
-    )
-    # TODO: document prompt
-    combine_documents_chain = StuffDocumentsChain(
-        llm_chain=reduce_chain,
-        document_variable_name=combine_document_variable_name,
-        verbose=verbose,  # type: ignore[arg-type]
-        callbacks=callbacks,
-    )
-    if collapse_prompt is None:
-        collapse_chain = None
-        if collapse_llm is not None:
-            msg = (
-                "collapse_llm provided, but collapse_prompt was not: please "
-                "provide one or stop providing collapse_llm."
-            )
-            raise ValueError(msg)
-    else:
-        _collapse_llm = collapse_llm or llm
-        collapse_chain = StuffDocumentsChain(
-            llm_chain=LLMChain(
-                llm=_collapse_llm,
-                prompt=collapse_prompt,
-                verbose=verbose,  # type: ignore[arg-type]
-                callbacks=callbacks,
-            ),
-            document_variable_name=combine_document_variable_name,
-        )
-    reduce_documents_chain = ReduceDocumentsChain(
-        combine_documents_chain=combine_documents_chain,
-        collapse_documents_chain=collapse_chain,
-        token_max=token_max,
-        verbose=verbose,  # type: ignore[arg-type]
-        callbacks=callbacks,
-        collapse_max_retries=collapse_max_retries,
-    )
-    return MapReduceDocumentsChain(
-        llm_chain=map_chain,
-        reduce_documents_chain=reduce_documents_chain,
-        document_variable_name=map_reduce_document_variable_name,
-        verbose=verbose,  # type: ignore[arg-type]
-        callbacks=callbacks,
-        **kwargs,
-    )
-
-
-def _load_refine_chain(
-    llm: BaseLanguageModel,
-    *,
-    question_prompt: BasePromptTemplate = refine_prompts.PROMPT,
-    refine_prompt: BasePromptTemplate = refine_prompts.REFINE_PROMPT,
-    document_variable_name: str = "text",
-    initial_response_name: str = "existing_answer",
-    refine_llm: Optional[BaseLanguageModel] = None,
-    verbose: Optional[bool] = None,
-    **kwargs: Any,
-) -> RefineDocumentsChain:
-    initial_chain = LLMChain(llm=llm, prompt=question_prompt, verbose=verbose)  # type: ignore[arg-type]
-    _refine_llm = refine_llm or llm
-    refine_chain = LLMChain(llm=_refine_llm, prompt=refine_prompt, verbose=verbose)  # type: ignore[arg-type]
-    return RefineDocumentsChain(
-        initial_llm_chain=initial_chain,
-        refine_llm_chain=refine_chain,
-        document_variable_name=document_variable_name,
-        initial_response_name=initial_response_name,
-        verbose=verbose,  # type: ignore[arg-type]
-        **kwargs,
-    )
-
-
-def load_summarize_chain(
-    llm: BaseLanguageModel,
-    chain_type: str = "stuff",
-    verbose: Optional[bool] = None,  # noqa: FBT001
-    **kwargs: Any,
-) -> BaseCombineDocumentsChain:
-    """Load summarizing chain.
-
-    Args:
-        llm: Language Model to use in the chain.
-        chain_type: Type of document combining chain to use. Should be one of "stuff",
-            "map_reduce", and "refine".
-        verbose: Whether chains should be run in verbose mode or not. Note that this
-            applies to all chains that make up the final chain.
-
-    Returns:
-        A chain to use for summarizing.
-    """
-    loader_mapping: Mapping[str, LoadingCallable] = {
-        "stuff": _load_stuff_chain,
-        "map_reduce": _load_map_reduce_chain,
-        "refine": _load_refine_chain,
-    }
-    if chain_type not in loader_mapping:
-        msg = (
-            f"Got unsupported chain type: {chain_type}. "
-            f"Should be one of {loader_mapping.keys()}"
-        )
-        raise ValueError(msg)
-    return loader_mapping[chain_type](llm, verbose=verbose, **kwargs)

langchain/chains/summarize/map_reduce_prompt.py

@@ -1,10 +0,0 @@
-from langchain_core.prompts import PromptTemplate
-
-prompt_template = """Write a concise summary of the following:
-
-
-"{text}"
-
-
-CONCISE SUMMARY:"""
-PROMPT = PromptTemplate(template=prompt_template, input_variables=["text"])

langchain/chains/summarize/refine_prompts.py

@@ -1,23 +0,0 @@
-from langchain_core.prompts import PromptTemplate
-
-REFINE_PROMPT_TMPL = """\
-Your job is to produce a final summary.
-We have provided an existing summary up to a certain point: {existing_answer}
-We have the opportunity to refine the existing summary (only if needed) with some more context below.
-------------
-{text}
-------------
-Given the new context, refine the original summary.
-If the context isn't useful, return the original summary.\
-"""  # noqa: E501
-REFINE_PROMPT = PromptTemplate.from_template(REFINE_PROMPT_TMPL)
-
-
-prompt_template = """Write a concise summary of the following:
-
-
-"{text}"
-
-
-CONCISE SUMMARY:"""
-PROMPT = PromptTemplate.from_template(prompt_template)

langchain/chains/summarize/stuff_prompt.py

@@ -1,10 +0,0 @@
-from langchain_core.prompts import PromptTemplate
-
-prompt_template = """Write a concise summary of the following:
-
-
-"{text}"
-
-
-CONCISE SUMMARY:"""
-PROMPT = PromptTemplate(template=prompt_template, input_variables=["text"])

langchain/chains/transform.py

@@ -1,88 +0,0 @@
-"""Chain that runs an arbitrary python function."""
-
-import functools
-import logging
-from collections.abc import Awaitable
-from typing import Any, Callable, Optional
-
-from langchain_core.callbacks import (
-    AsyncCallbackManagerForChainRun,
-    CallbackManagerForChainRun,
-)
-from pydantic import Field
-from typing_extensions import override
-
-from langchain.chains.base import Chain
-
-logger = logging.getLogger(__name__)
-
-
-class TransformChain(Chain):
-    """Chain that transforms the chain output.
-
-    Example:
-        .. code-block:: python
-
-            from langchain.chains import TransformChain
-            transform_chain = TransformChain(input_variables=["text"],
-             output_variables["entities"], transform=func())
-
-    """
-
-    input_variables: list[str]
-    """The keys expected by the transform's input dictionary."""
-    output_variables: list[str]
-    """The keys returned by the transform's output dictionary."""
-    transform_cb: Callable[[dict[str, str]], dict[str, str]] = Field(alias="transform")
-    """The transform function."""
-    atransform_cb: Optional[Callable[[dict[str, Any]], Awaitable[dict[str, Any]]]] = (
-        Field(None, alias="atransform")
-    )
-    """The async coroutine transform function."""
-
-    @staticmethod
-    @functools.lru_cache
-    def _log_once(msg: str) -> None:
-        """Log a message once.
-
-        :meta private:
-        """
-        logger.warning(msg)
-
-    @property
-    def input_keys(self) -> list[str]:
-        """Expect input keys.
-
-        :meta private:
-        """
-        return self.input_variables
-
-    @property
-    def output_keys(self) -> list[str]:
-        """Return output keys.
-
-        :meta private:
-        """
-        return self.output_variables
-
-    @override
-    def _call(
-        self,
-        inputs: dict[str, str],
-        run_manager: Optional[CallbackManagerForChainRun] = None,
-    ) -> dict[str, str]:
-        return self.transform_cb(inputs)
-
-    @override
-    async def _acall(
-        self,
-        inputs: dict[str, Any],
-        run_manager: Optional[AsyncCallbackManagerForChainRun] = None,
-    ) -> dict[str, Any]:
-        if self.atransform_cb is not None:
-            return await self.atransform_cb(inputs)
-        self._log_once(
-            "TransformChain's atransform is not provided, falling"
-            " back to synchronous transform",
-        )
-        return self.transform_cb(inputs)

langchain/chat_loaders/__init__.py

@@ -1,19 +0,0 @@
-"""**Chat Loaders** load chat messages from common communications platforms.
-
-Load chat messages from various
-communications platforms such as Facebook Messenger, Telegram, and
-WhatsApp. The loaded chat messages can be used for fine-tuning models.
-
-**Class hierarchy:**
-
-.. code-block::
-
-    BaseChatLoader --> <name>ChatLoader  # Examples: WhatsAppChatLoader, IMessageChatLoader
-
-**Main helpers:**
-
-.. code-block::
-
-    ChatSession
-
-"""  # noqa: E501

langchain/chat_loaders/base.py

@@ -1,3 +0,0 @@
-from langchain_core.chat_loaders import BaseChatLoader
-
-__all__ = ["BaseChatLoader"]

langchain/chat_loaders/facebook_messenger.py

@@ -1,32 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api.module_import import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_loaders.facebook_messenger import (
-        FolderFacebookMessengerChatLoader,
-        SingleFileFacebookMessengerChatLoader,
-    )
-
-module_lookup = {
-    "SingleFileFacebookMessengerChatLoader": (
-        "langchain_community.chat_loaders.facebook_messenger"
-    ),
-    "FolderFacebookMessengerChatLoader": (
-        "langchain_community.chat_loaders.facebook_messenger"
-    ),
-}
-
-# Temporary code for backwards compatibility for deprecated imports.
-# This will eventually be removed.
-import_lookup = create_importer(
-    __package__,
-    deprecated_lookups=module_lookup,
-)
-
-
-def __getattr__(name: str) -> Any:
-    return import_lookup(name)
-
-
-__all__ = ["FolderFacebookMessengerChatLoader", "SingleFileFacebookMessengerChatLoader"]

langchain/chat_loaders/gmail.py

@@ -1,23 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_loaders.gmail import GMailLoader
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {"GMailLoader": "langchain_community.chat_loaders.gmail"}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "GMailLoader",
-]

langchain/chat_loaders/imessage.py

@@ -1,23 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_loaders.imessage import IMessageChatLoader
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {"IMessageChatLoader": "langchain_community.chat_loaders.imessage"}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "IMessageChatLoader",
-]

langchain/chat_loaders/langsmith.py

@@ -1,30 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_loaders.langsmith import (
-        LangSmithDatasetChatLoader,
-        LangSmithRunChatLoader,
-    )
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {
-    "LangSmithRunChatLoader": "langchain_community.chat_loaders.langsmith",
-    "LangSmithDatasetChatLoader": "langchain_community.chat_loaders.langsmith",
-}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "LangSmithDatasetChatLoader",
-    "LangSmithRunChatLoader",
-]

langchain/chat_loaders/slack.py

@@ -1,23 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_loaders.slack import SlackChatLoader
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {"SlackChatLoader": "langchain_community.chat_loaders.slack"}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "SlackChatLoader",
-]

langchain/chat_loaders/telegram.py

@@ -1,23 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_loaders.telegram import TelegramChatLoader
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {"TelegramChatLoader": "langchain_community.chat_loaders.telegram"}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "TelegramChatLoader",
-]

langchain/chat_loaders/utils.py

@@ -1,36 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_loaders.utils import (
-        map_ai_messages,
-        map_ai_messages_in_session,
-        merge_chat_runs,
-        merge_chat_runs_in_session,
-    )
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {
-    "merge_chat_runs_in_session": "langchain_community.chat_loaders.utils",
-    "merge_chat_runs": "langchain_community.chat_loaders.utils",
-    "map_ai_messages_in_session": "langchain_community.chat_loaders.utils",
-    "map_ai_messages": "langchain_community.chat_loaders.utils",
-}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "map_ai_messages",
-    "map_ai_messages_in_session",
-    "merge_chat_runs",
-    "merge_chat_runs_in_session",
-]

langchain/chat_loaders/whatsapp.py

@@ -1,23 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_loaders.whatsapp import WhatsAppChatLoader
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {"WhatsAppChatLoader": "langchain_community.chat_loaders.whatsapp"}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "WhatsAppChatLoader",
-]

langchain/chat_models/anthropic.py

@@ -1,30 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_models.anthropic import (
-        ChatAnthropic,
-        convert_messages_to_prompt_anthropic,
-    )
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {
-    "convert_messages_to_prompt_anthropic": "langchain_community.chat_models.anthropic",
-    "ChatAnthropic": "langchain_community.chat_models.anthropic",
-}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "ChatAnthropic",
-    "convert_messages_to_prompt_anthropic",
-]

langchain/chat_models/anyscale.py

@@ -1,23 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_models.anyscale import ChatAnyscale
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {"ChatAnyscale": "langchain_community.chat_models.anyscale"}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "ChatAnyscale",
-]

langchain/chat_models/azure_openai.py

@@ -1,23 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_models.azure_openai import AzureChatOpenAI
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {"AzureChatOpenAI": "langchain_community.chat_models.azure_openai"}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "AzureChatOpenAI",
-]

langchain/chat_models/azureml_endpoint.py

@@ -1,30 +0,0 @@
-from typing import TYPE_CHECKING, Any
-
-from langchain._api import create_importer
-
-if TYPE_CHECKING:
-    from langchain_community.chat_models.azureml_endpoint import (
-        AzureMLChatOnlineEndpoint,
-        LlamaContentFormatter,
-    )
-
-# Create a way to dynamically look up deprecated imports.
-# Used to consolidate logic for raising deprecation warnings and
-# handling optional imports.
-DEPRECATED_LOOKUP = {
-    "LlamaContentFormatter": "langchain_community.chat_models.azureml_endpoint",
-    "AzureMLChatOnlineEndpoint": "langchain_community.chat_models.azureml_endpoint",
-}
-
-_import_attribute = create_importer(__package__, deprecated_lookups=DEPRECATED_LOOKUP)
-
-
-def __getattr__(name: str) -> Any:
-    """Look up attributes dynamically."""
-    return _import_attribute(name)
-
-
-__all__ = [
-    "AzureMLChatOnlineEndpoint",
-    "LlamaContentFormatter",
-]