langchain 0.2.11__py3-none-any.whl → 0.2.13__py3-none-any.whl

langchain/chains/combine_documents/stuff.py

@@ -2,12 +2,13 @@
 
 from typing import Any, Dict, List, Optional, Tuple
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from langchain_core.language_models import LanguageModelLike
 from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.prompts import BasePromptTemplate, format_document
-from langchain_core.pydantic_v1 import Extra, Field, root_validator
+from langchain_core.pydantic_v1 import Field, root_validator
 from langchain_core.runnables import Runnable, RunnablePassthrough
 
 from langchain.chains.combine_documents.base import (
@@ -95,6 +96,15 @@ def create_stuff_documents_chain(
     ).with_config(run_name="stuff_documents_chain")
 
 
+@deprecated(
+    since="0.2.13",
+    removal="1.0",
+    message=(
+        "This class is deprecated. Use the `create_stuff_documents_chain` constructor "
+        "instead. See migration guide here: "
+        "https://python.langchain.com/v0.2/docs/versions/migrating_chains/stuff_docs_chain/"  # noqa: E501
+    ),
+)
 class StuffDocumentsChain(BaseCombineDocumentsChain):
     """Chain that combines documents by stuffing into context.
 
@@ -147,10 +157,8 @@ class StuffDocumentsChain(BaseCombineDocumentsChain):
     """The string with which to join the formatted documents"""
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @root_validator(pre=True)
     def get_default_document_variable_name(cls, values: Dict) -> Dict:

langchain/chains/conversation/base.py

@@ -5,7 +5,7 @@ from typing import Dict, List
 from langchain_core._api import deprecated
 from langchain_core.memory import BaseMemory
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Extra, Field, root_validator
+from langchain_core.pydantic_v1 import Field, root_validator
 
 from langchain.chains.conversation.prompt import PROMPT
 from langchain.chains.llm import LLMChain
@@ -111,10 +111,8 @@ class ConversationChain(LLMChain):
     output_key: str = "response"  #: :meta private:
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @classmethod
     def is_lc_serializable(cls) -> bool:

langchain/chains/conversational_retrieval/base.py

@@ -18,7 +18,7 @@ from langchain_core.documents import Document
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.messages import BaseMessage
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, Extra, Field, root_validator
+from langchain_core.pydantic_v1 import BaseModel, Field, root_validator
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.runnables import RunnableConfig
 from langchain_core.vectorstores import VectorStore
@@ -97,11 +97,9 @@ class BaseConversationalRetrievalChain(Chain):
     are found for the question. """
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
-        arbitrary_types_allowed = True
         allow_population_by_field_name = True
+        arbitrary_types_allowed = True
+        extra = "forbid"
 
     @property
     def input_keys(self) -> List[str]:
@@ -445,7 +443,7 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
             combine_docs_chain_kwargs: Parameters to pass as kwargs to `load_qa_chain`
                 when constructing the combine_docs_chain.
             callbacks: Callbacks to pass to all subchains.
-            **kwargs: Additional parameters to pass when initializing
+            kwargs: Additional parameters to pass when initializing
                 ConversationalRetrievalChain
         """
         combine_docs_chain_kwargs = combine_docs_chain_kwargs or {}

langchain/chains/elasticsearch_database/base.py

@@ -6,14 +6,14 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
-from langchain_core.output_parsers import BaseLLMOutputParser
+from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.output_parsers.json import SimpleJsonOutputParser
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Extra, root_validator
+from langchain_core.pydantic_v1 import root_validator
+from langchain_core.runnables import Runnable
 
 from langchain.chains.base import Chain
 from langchain.chains.elasticsearch_database.prompts import ANSWER_PROMPT, DSL_PROMPT
-from langchain.chains.llm import LLMChain
 
 if TYPE_CHECKING:
     from elasticsearch import Elasticsearch
@@ -35,9 +35,9 @@ class ElasticsearchDatabaseChain(Chain):
             db_chain = ElasticsearchDatabaseChain.from_llm(OpenAI(), database)
     """
 
-    query_chain: LLMChain
+    query_chain: Runnable
     """Chain for creating the ES query."""
-    answer_chain: LLMChain
+    answer_chain: Runnable
     """Chain for answering the user question."""
     database: Any
     """Elasticsearch database to connect to of type elasticsearch.Elasticsearch."""
@@ -52,10 +52,8 @@ class ElasticsearchDatabaseChain(Chain):
     """Whether or not to return the intermediate steps along with the final answer."""
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @root_validator(pre=False, skip_on_failure=True)
     def validate_indices(cls, values: dict) -> dict:
@@ -137,9 +135,9 @@ class ElasticsearchDatabaseChain(Chain):
         intermediate_steps: List = []
         try:
             intermediate_steps.append(query_inputs)  # input: es generation
-            es_cmd = self.query_chain.run(
-                callbacks=_run_manager.get_child(),
-                **query_inputs,
+            es_cmd = self.query_chain.invoke(
+                query_inputs,
+                config={"callbacks": _run_manager.get_child()},
             )
 
             _run_manager.on_text(es_cmd, color="green", verbose=self.verbose)
@@ -156,9 +154,9 @@ class ElasticsearchDatabaseChain(Chain):
             _run_manager.on_text("\nAnswer:", verbose=self.verbose)
             answer_inputs: dict = {"data": result, "input": input_text}
             intermediate_steps.append(answer_inputs)  # input: final answer
-            final_result = self.answer_chain.run(
-                callbacks=_run_manager.get_child(),
-                **answer_inputs,
+            final_result = self.answer_chain.invoke(
+                answer_inputs,
+                config={"callbacks": _run_manager.get_child()},
             )
 
             intermediate_steps.append(final_result)  # output: final answer
@@ -185,7 +183,7 @@ class ElasticsearchDatabaseChain(Chain):
         *,
         query_prompt: Optional[BasePromptTemplate] = None,
         answer_prompt: Optional[BasePromptTemplate] = None,
-        query_output_parser: Optional[BaseLLMOutputParser] = None,
+        query_output_parser: Optional[BaseOutputParser] = None,
         **kwargs: Any,
     ) -> ElasticsearchDatabaseChain:
         """Convenience method to construct ElasticsearchDatabaseChain from an LLM.
@@ -197,15 +195,13 @@ class ElasticsearchDatabaseChain(Chain):
             answer_prompt: The prompt to use for answering user question given data.
             query_output_parser: The output parser to use for parsing model-generated
                 ES query. Defaults to SimpleJsonOutputParser.
-            **kwargs: Additional arguments to pass to the constructor.
+            kwargs: Additional arguments to pass to the constructor.
         """
         query_prompt = query_prompt or DSL_PROMPT
         query_output_parser = query_output_parser or SimpleJsonOutputParser()
-        query_chain = LLMChain(
-            llm=llm, prompt=query_prompt, output_parser=query_output_parser
-        )
+        query_chain = query_prompt | llm | query_output_parser
         answer_prompt = answer_prompt or ANSWER_PROMPT
-        answer_chain = LLMChain(llm=llm, prompt=answer_prompt)
+        answer_chain = answer_prompt | llm | StrOutputParser()
         return cls(
             query_chain=query_chain,
             answer_chain=answer_chain,

langchain/chains/example_generator.py

@@ -1,11 +1,10 @@
 from typing import List
 
 from langchain_core.language_models import BaseLanguageModel
+from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts.few_shot import FewShotPromptTemplate
 from langchain_core.prompts.prompt import PromptTemplate
 
-from langchain.chains.llm import LLMChain
-
 TEST_GEN_TEMPLATE_SUFFIX = "Add another example."
 
 
@@ -19,5 +18,5 @@ def generate_example(
         input_variables=[],
         example_prompt=prompt_template,
     )
-    chain = LLMChain(llm=llm, prompt=prompt)
-    return chain.predict()
+    chain = prompt | llm | StrOutputParser()
+    return chain.invoke({})

langchain/chains/flare/base.py

@@ -245,7 +245,7 @@ class FlareChain(Chain):
         Args:
             llm: Language model to use.
             max_generation_len: Maximum length of the generated response.
-            **kwargs: Additional arguments to pass to the constructor.
+            kwargs: Additional arguments to pass to the constructor.
 
         Returns:
             FlareChain class with the given language model.

langchain/chains/hyde/base.py

@@ -12,7 +12,6 @@ from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.embeddings import Embeddings
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Extra
 
 from langchain.chains.base import Chain
 from langchain.chains.hyde.prompts import PROMPT_MAP
@@ -29,10 +28,8 @@ class HypotheticalDocumentEmbedder(Chain, Embeddings):
     llm_chain: LLMChain
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @property
     def input_keys(self) -> List[str]:

langchain/chains/llm.py

@@ -23,7 +23,7 @@ from langchain_core.output_parsers import BaseLLMOutputParser, StrOutputParser
 from langchain_core.outputs import ChatGeneration, Generation, LLMResult
 from langchain_core.prompt_values import PromptValue
 from langchain_core.prompts import BasePromptTemplate, PromptTemplate
-from langchain_core.pydantic_v1 import Extra, Field
+from langchain_core.pydantic_v1 import Field
 from langchain_core.runnables import (
     Runnable,
     RunnableBinding,
@@ -96,10 +96,8 @@ class LLMChain(Chain):
     llm_kwargs: dict = Field(default_factory=dict)
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @property
     def input_keys(self) -> List[str]:

langchain/chains/llm_checker/base.py

@@ -5,10 +5,11 @@ from __future__ import annotations
 import warnings
 from typing import Any, Dict, List, Optional
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import PromptTemplate
-from langchain_core.pydantic_v1 import Extra, root_validator
+from langchain_core.pydantic_v1 import root_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -63,6 +64,15 @@ def _load_question_to_checked_assertions_chain(
     return question_to_checked_assertions_chain
 
 
+@deprecated(
+    since="0.2.13",
+    message=(
+        "See LangGraph guides for a variety of self-reflection and corrective "
+        "strategies for question-answering and other tasks: "
+        "https://langchain-ai.github.io/langgraph/tutorials/rag/langgraph_self_rag/"
+    ),
+    removal="1.0",
+)
 class LLMCheckerChain(Chain):
     """Chain for question-answering with self-verification.
 
@@ -91,10 +101,8 @@ class LLMCheckerChain(Chain):
     output_key: str = "result"  #: :meta private:
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @root_validator(pre=True)
     def raise_deprecation(cls, values: Dict) -> Dict:

langchain/chains/llm_math/base.py

@@ -13,7 +13,7 @@ from langchain_core.callbacks import (
 )
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Extra, root_validator
+from langchain_core.pydantic_v1 import root_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -40,10 +40,8 @@ class LLMMathChain(Chain):
     output_key: str = "answer"  #: :meta private:
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @root_validator(pre=True)
     def raise_deprecation(cls, values: Dict) -> Dict:

langchain/chains/llm_summarization_checker/base.py

@@ -6,10 +6,11 @@ import warnings
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts.prompt import PromptTemplate
-from langchain_core.pydantic_v1 import Extra, root_validator
+from langchain_core.pydantic_v1 import root_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -65,6 +66,15 @@ def _load_sequential_chain(
     return chain
 
 
+@deprecated(
+    since="0.2.13",
+    message=(
+        "See LangGraph guides for a variety of self-reflection and corrective "
+        "strategies for question-answering and other tasks: "
+        "https://langchain-ai.github.io/langgraph/tutorials/rag/langgraph_self_rag/"
+    ),
+    removal="1.0",
+)
 class LLMSummarizationCheckerChain(Chain):
     """Chain for question-answering with self-verification.
 
@@ -96,10 +106,8 @@ class LLMSummarizationCheckerChain(Chain):
     """Maximum number of times to check the assertions. Default to double-checking."""
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @root_validator(pre=True)
     def raise_deprecation(cls, values: Dict) -> Dict:

langchain/chains/loading.py

@@ -7,6 +7,7 @@ from pathlib import Path
 from typing import TYPE_CHECKING, Any, Union
 
 import yaml
+from langchain_core._api import deprecated
 from langchain_core.prompts.loading import (
     _load_output_parser,
     load_prompt,
@@ -649,6 +650,14 @@ type_to_loader_dict = {
 }
 
 
+@deprecated(
+    since="0.2.13",
+    message=(
+        "This function is deprecated and will be removed in langchain 1.0. "
+        "At that point chains must be imported from their respective modules."
+    ),
+    removal="1.0",
+)
 def load_chain_from_config(config: dict, **kwargs: Any) -> Chain:
     """Load chain from Config Dict."""
     if "_type" not in config:
@@ -662,6 +671,14 @@ def load_chain_from_config(config: dict, **kwargs: Any) -> Chain:
     return chain_loader(config, **kwargs)
 
 
+@deprecated(
+    since="0.2.13",
+    message=(
+        "This function is deprecated and will be removed in langchain 1.0. "
+        "At that point chains must be imported from their respective modules."
+    ),
+    removal="1.0",
+)
 def load_chain(path: Union[str, Path], **kwargs: Any) -> Chain:
     """Unified method for loading a chain from LangChainHub or local fs."""
     if isinstance(path, str) and path.startswith("lc://"):

langchain/chains/mapreduce.py

@@ -8,11 +8,11 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Mapping, Optional
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun, Callbacks
 from langchain_core.documents import Document
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Extra
 from langchain_text_splitters import TextSplitter
 
 from langchain.chains import ReduceDocumentsChain
@@ -23,6 +23,16 @@ from langchain.chains.combine_documents.stuff import StuffDocumentsChain
 from langchain.chains.llm import LLMChain
 
 
+@deprecated(
+    since="0.2.13",
+    removal="1.0",
+    message=(
+        "Refer here for a recommended map-reduce implementation using langgraph: "
+        "https://langchain-ai.github.io/langgraph/how-tos/map-reduce/. See also "
+        "migration guide: "
+        "https://python.langchain.com/v0.2/docs/versions/migrating_chains/map_reduce_chain/"  # noqa: E501
+    ),
+)
 class MapReduceChain(Chain):
     """Map-reduce chain."""
 
@@ -68,10 +78,8 @@ class MapReduceChain(Chain):
         )
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @property
     def input_keys(self) -> List[str]:

langchain/chains/natbot/base.py

@@ -7,7 +7,7 @@ from typing import Any, Dict, List, Optional
 
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
-from langchain_core.pydantic_v1 import Extra, root_validator
+from langchain_core.pydantic_v1 import root_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -48,10 +48,8 @@ class NatBotChain(Chain):
     output_key: str = "command"  #: :meta private:
 
     class Config:
-        """Configuration for this pydantic object."""
-
-        extra = Extra.forbid
         arbitrary_types_allowed = True
+        extra = "forbid"
 
     @root_validator(pre=True)
     def raise_deprecation(cls, values: Dict) -> Dict:

langchain/chains/openai_functions/__init__.py

@@ -6,6 +6,7 @@ from langchain.chains.openai_functions.base import (
 )
 from langchain.chains.openai_functions.citation_fuzzy_match import (
     create_citation_fuzzy_match_chain,
+    create_citation_fuzzy_match_runnable,
 )
 from langchain.chains.openai_functions.extraction import (
     create_extraction_chain,
@@ -32,6 +33,7 @@ __all__ = [
     "create_extraction_chain_pydantic",
     "create_extraction_chain",
     "create_citation_fuzzy_match_chain",
+    "create_citation_fuzzy_match_runnable",
     "create_qa_with_structure_chain",
     "create_qa_with_sources_chain",
     "create_structured_output_chain",

langchain/chains/openai_functions/citation_fuzzy_match.py

@@ -1,10 +1,12 @@
 from typing import Iterator, List
 
-from langchain_core.language_models import BaseLanguageModel
+from langchain_core._api import deprecated
+from langchain_core.language_models import BaseChatModel, BaseLanguageModel
 from langchain_core.messages import HumanMessage, SystemMessage
 from langchain_core.output_parsers.openai_functions import PydanticOutputFunctionsParser
 from langchain_core.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate
 from langchain_core.pydantic_v1 import BaseModel, Field
+from langchain_core.runnables import Runnable
 
 from langchain.chains.llm import LLMChain
 from langchain.chains.openai_functions.utils import get_llm_kwargs
@@ -61,6 +63,57 @@ class QuestionAnswer(BaseModel):
     )
 
 
+def create_citation_fuzzy_match_runnable(llm: BaseChatModel) -> Runnable:
+    """Create a citation fuzzy match Runnable.
+
+    Example usage:
+
+        .. code-block:: python
+
+            from langchain.chains import create_citation_fuzzy_match_runnable
+            from langchain_openai import ChatOpenAI
+
+            llm = ChatOpenAI(model="gpt-4o-mini")
+
+            context = "Alice has blue eyes. Bob has brown eyes. Charlie has green eyes."
+            question = "What color are Bob's eyes?"
+
+            chain = create_citation_fuzzy_match_runnable(llm)
+            chain.invoke({"question": question, "context": context})
+
+    Args:
+        llm: Language model to use for the chain. Must implement bind_tools.
+
+    Returns:
+        Runnable that can be used to answer questions with citations.
+    """
+    if llm.bind_tools is BaseChatModel.bind_tools:
+        raise ValueError(
+            "Language model must implement bind_tools to use this function."
+        )
+    prompt = ChatPromptTemplate(
+        [
+            SystemMessage(
+                "You are a world class algorithm to answer "
+                "questions with correct and exact citations."
+            ),
+            HumanMessagePromptTemplate.from_template(
+                "Answer question using the following context."
+                "\n\n{context}"
+                "\n\nQuestion: {question}"
+                "\n\nTips: Make sure to cite your sources, "
+                "and use the exact words from the context."
+            ),
+        ]
+    )
+    return prompt | llm.with_structured_output(QuestionAnswer)
+
+
+@deprecated(
+    since="0.2.13",
+    removal="1.0",
+    alternative="create_citation_fuzzy_match_runnable",
+)
 def create_citation_fuzzy_match_chain(llm: BaseLanguageModel) -> LLMChain:
     """Create a citation fuzzy match chain.
 

langchain/chains/openai_functions/openapi.py

@@ -6,6 +6,7 @@ from collections import defaultdict
 from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Tuple, Union
 
 import requests
+from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers.openai_functions import JsonOutputFunctionsParser
@@ -242,6 +243,15 @@ class SimpleRequestChain(Chain):
         return {self.output_key: response}
 
 
+@deprecated(
+    since="0.2.13",
+    message=(
+        "This function is deprecated and will be removed in langchain 1.0. "
+        "See API reference for replacement: "
+        "https://api.python.langchain.com/en/latest/chains/langchain.chains.openai_functions.openapi.get_openapi_chain.html"  # noqa: E501
+    ),
+    removal="1.0",
+)
 def get_openapi_chain(
     spec: Union[OpenAPISpec, str],
     llm: Optional[BaseLanguageModel] = None,
@@ -255,13 +265,90 @@ def get_openapi_chain(
 ) -> SequentialChain:
     """Create a chain for querying an API from a OpenAPI spec.
 
+    Note: this class is deprecated. See below for a replacement implementation.
+        The benefits of this implementation are:
+
+        - Uses LLM tool calling features to encourage properly-formatted API requests;
+        - Includes async support.
+
+        .. code-block:: python
+
+            from typing import Any
+
+            from langchain.chains.openai_functions.openapi import openapi_spec_to_openai_fn
+            from langchain_community.utilities.openapi import OpenAPISpec
+            from langchain_core.prompts import ChatPromptTemplate
+            from langchain_openai import ChatOpenAI
+
+            # Define API spec. Can be JSON or YAML
+            api_spec = \"\"\"
+            {
+            "openapi": "3.1.0",
+            "info": {
+                "title": "JSONPlaceholder API",
+                "version": "1.0.0"
+            },
+            "servers": [
+                {
+                "url": "https://jsonplaceholder.typicode.com"
+                }
+            ],
+            "paths": {
+                "/posts": {
+                "get": {
+                    "summary": "Get posts",
+                    "parameters": [
+                    {
+                        "name": "_limit",
+                        "in": "query",
+                        "required": false,
+                        "schema": {
+                        "type": "integer",
+                        "example": 2
+                        },
+                        "description": "Limit the number of results"
+                    }
+                    ]
+                }
+                }
+            }
+            }
+            \"\"\"
+
+            parsed_spec = OpenAPISpec.from_text(api_spec)
+            openai_fns, call_api_fn = openapi_spec_to_openai_fn(parsed_spec)
+            tools = [
+                {"type": "function", "function": fn}
+                for fn in openai_fns
+            ]
+
+            prompt = ChatPromptTemplate.from_template(
+                "Use the provided APIs to respond to this user query:\\n\\n{query}"
+            )
+            llm = ChatOpenAI(model="gpt-4o-mini", temperature=0).bind_tools(tools)
+
+            def _execute_tool(message) -> Any:
+                if tool_calls := message.tool_calls:
+                    tool_call = message.tool_calls[0]
+                    response = call_api_fn(name=tool_call["name"], fn_args=tool_call["args"])
+                    response.raise_for_status()
+                    return response.json()
+                else:
+                    return message.content
+
+            chain = prompt | llm | _execute_tool
+
+        .. code-block:: python
+
+            response = chain.invoke({"query": "Get me top two posts."})
+
     Args:
         spec: OpenAPISpec or url/file/text string corresponding to one.
         llm: language model, should be an OpenAI function-calling model, e.g.
             `ChatOpenAI(model="gpt-3.5-turbo-0613")`.
         prompt: Main prompt template to use.
         request_chain: Chain for taking the functions output and executing the request.
-    """
+    """  # noqa: E501
     try:
         from langchain_community.utilities.openapi import OpenAPISpec
     except ImportError as e: