langchain 0.3.26__py3-none-any.whl → 0.4.0.dev0__py3-none-any.whl

langchain/chains/conversational_retrieval/base.py

@@ -22,6 +22,7 @@ from langchain_core.retrievers import BaseRetriever
 from langchain_core.runnables import RunnableConfig
 from langchain_core.vectorstores import VectorStore
 from pydantic import BaseModel, ConfigDict, Field, model_validator
+from typing_extensions import override
 
 from langchain.chains.base import Chain
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
@@ -44,18 +45,20 @@ def _get_chat_history(chat_history: list[CHAT_TURN_TYPE]) -> str:
         if isinstance(dialogue_turn, BaseMessage):
             if len(dialogue_turn.content) > 0:
                 role_prefix = _ROLE_MAP.get(
-                    dialogue_turn.type, f"{dialogue_turn.type}: "
+                    dialogue_turn.type,
+                    f"{dialogue_turn.type}: ",
                 )
                 buffer += f"\n{role_prefix}{dialogue_turn.content}"
         elif isinstance(dialogue_turn, tuple):
             human = "Human: " + dialogue_turn[0]
             ai = "Assistant: " + dialogue_turn[1]
-            buffer += "\n" + "\n".join([human, ai])
+            buffer += f"\n{human}\n{ai}"
         else:
-            raise ValueError(
+            msg = (
                 f"Unsupported chat history format: {type(dialogue_turn)}."
                 f" Full chat history: {chat_history} "
             )
+            raise ValueError(msg)  # noqa: TRY004
     return buffer
 
 
@@ -93,7 +96,7 @@ class BaseConversationalRetrievalChain(Chain):
     """An optional function to get a string of the chat history.
     If None is provided, will use a default."""
     response_if_no_docs_found: Optional[str] = None
-    """If specified, the chain will return a fixed response if no docs 
+    """If specified, the chain will return a fixed response if no docs
     are found for the question. """
 
     model_config = ConfigDict(
@@ -107,8 +110,10 @@ class BaseConversationalRetrievalChain(Chain):
         """Input keys."""
         return ["question", "chat_history"]
 
+    @override
     def get_input_schema(
-        self, config: Optional[RunnableConfig] = None
+        self,
+        config: Optional[RunnableConfig] = None,
     ) -> type[BaseModel]:
         return InputType
 
@@ -120,9 +125,9 @@ class BaseConversationalRetrievalChain(Chain):
         """
         _output_keys = [self.output_key]
         if self.return_source_documents:
-            _output_keys = _output_keys + ["source_documents"]
+            _output_keys = [*_output_keys, "source_documents"]
         if self.return_generated_question:
-            _output_keys = _output_keys + ["generated_question"]
+            _output_keys = [*_output_keys, "generated_question"]
         return _output_keys
 
     @abstractmethod
@@ -148,7 +153,9 @@ class BaseConversationalRetrievalChain(Chain):
         if chat_history_str:
             callbacks = _run_manager.get_child()
             new_question = self.question_generator.run(
-                question=question, chat_history=chat_history_str, callbacks=callbacks
+                question=question,
+                chat_history=chat_history_str,
+                callbacks=callbacks,
             )
         else:
             new_question = question
@@ -168,7 +175,9 @@ class BaseConversationalRetrievalChain(Chain):
                 new_inputs["question"] = new_question
             new_inputs["chat_history"] = chat_history_str
             answer = self.combine_docs_chain.run(
-                input_documents=docs, callbacks=_run_manager.get_child(), **new_inputs
+                input_documents=docs,
+                callbacks=_run_manager.get_child(),
+                **new_inputs,
             )
             output[self.output_key] = answer
 
@@ -200,7 +209,9 @@ class BaseConversationalRetrievalChain(Chain):
         if chat_history_str:
             callbacks = _run_manager.get_child()
             new_question = await self.question_generator.arun(
-                question=question, chat_history=chat_history_str, callbacks=callbacks
+                question=question,
+                chat_history=chat_history_str,
+                callbacks=callbacks,
             )
         else:
             new_question = question
@@ -221,7 +232,9 @@ class BaseConversationalRetrievalChain(Chain):
                 new_inputs["question"] = new_question
             new_inputs["chat_history"] = chat_history_str
             answer = await self.combine_docs_chain.arun(
-                input_documents=docs, callbacks=_run_manager.get_child(), **new_inputs
+                input_documents=docs,
+                callbacks=_run_manager.get_child(),
+                **new_inputs,
             )
             output[self.output_key] = answer
 
@@ -231,9 +244,11 @@ class BaseConversationalRetrievalChain(Chain):
             output["generated_question"] = new_question
         return output
 
+    @override
     def save(self, file_path: Union[Path, str]) -> None:
         if self.get_chat_history:
-            raise ValueError("Chain not saveable when `get_chat_history` is not None.")
+            msg = "Chain not saveable when `get_chat_history` is not None."
+            raise ValueError(msg)
         super().save(file_path)
 
 
@@ -252,68 +267,67 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
     `create_retrieval_chain`. Additional walkthroughs can be found at
     https://python.langchain.com/docs/use_cases/question_answering/chat_history
 
-        .. code-block:: python
-
-            from langchain.chains import (
-                create_history_aware_retriever,
-                create_retrieval_chain,
-            )
-            from langchain.chains.combine_documents import create_stuff_documents_chain
-            from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
-            from langchain_openai import ChatOpenAI
+    .. code-block:: python
 
+        from langchain.chains import (
+            create_history_aware_retriever,
+            create_retrieval_chain,
+        )
+        from langchain.chains.combine_documents import create_stuff_documents_chain
+        from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+        from langchain_openai import ChatOpenAI
 
-            retriever = ...  # Your retriever
+        retriever = ...  # Your retriever
 
-            llm = ChatOpenAI()
+        llm = ChatOpenAI()
 
-            # Contextualize question
-            contextualize_q_system_prompt = (
-                "Given a chat history and the latest user question "
-                "which might reference context in the chat history, "
-                "formulate a standalone question which can be understood "
-                "without the chat history. Do NOT answer the question, just "
-                "reformulate it if needed and otherwise return it as is."
-            )
-            contextualize_q_prompt = ChatPromptTemplate.from_messages(
-                [
-                    ("system", contextualize_q_system_prompt),
-                    MessagesPlaceholder("chat_history"),
-                    ("human", "{input}"),
-                ]
-            )
-            history_aware_retriever = create_history_aware_retriever(
-                llm, retriever, contextualize_q_prompt
-            )
+        # Contextualize question
+        contextualize_q_system_prompt = (
+            "Given a chat history and the latest user question "
+            "which might reference context in the chat history, "
+            "formulate a standalone question which can be understood "
+            "without the chat history. Do NOT answer the question, just "
+            "reformulate it if needed and otherwise return it as is."
+        )
+        contextualize_q_prompt = ChatPromptTemplate.from_messages(
+            [
+                ("system", contextualize_q_system_prompt),
+                MessagesPlaceholder("chat_history"),
+                ("human", "{input}"),
+            ]
+        )
+        history_aware_retriever = create_history_aware_retriever(
+            llm, retriever, contextualize_q_prompt
+        )
 
-            # Answer question
-            qa_system_prompt = (
-                "You are an assistant for question-answering tasks. Use "
-                "the following pieces of retrieved context to answer the "
-                "question. If you don't know the answer, just say that you "
-                "don't know. Use three sentences maximum and keep the answer "
-                "concise."
-                "\n\n"
-                "{context}"
-            )
-            qa_prompt = ChatPromptTemplate.from_messages(
-                [
-                    ("system", qa_system_prompt),
-                    MessagesPlaceholder("chat_history"),
-                    ("human", "{input}"),
-                ]
-            )
-            # Below we use create_stuff_documents_chain to feed all retrieved context
-            # into the LLM. Note that we can also use StuffDocumentsChain and other
-            # instances of BaseCombineDocumentsChain.
-            question_answer_chain = create_stuff_documents_chain(llm, qa_prompt)
-            rag_chain = create_retrieval_chain(
-                history_aware_retriever, question_answer_chain
-            )
+        # Answer question
+        qa_system_prompt = (
+            "You are an assistant for question-answering tasks. Use "
+            "the following pieces of retrieved context to answer the "
+            "question. If you don't know the answer, just say that you "
+            "don't know. Use three sentences maximum and keep the answer "
+            "concise."
+            "\n\n"
+            "{context}"
+        )
+        qa_prompt = ChatPromptTemplate.from_messages(
+            [
+                ("system", qa_system_prompt),
+                MessagesPlaceholder("chat_history"),
+                ("human", "{input}"),
+            ]
+        )
+        # Below we use create_stuff_documents_chain to feed all retrieved context
+        # into the LLM. Note that we can also use StuffDocumentsChain and other
+        # instances of BaseCombineDocumentsChain.
+        question_answer_chain = create_stuff_documents_chain(llm, qa_prompt)
+        rag_chain = create_retrieval_chain(
+            history_aware_retriever, question_answer_chain
+        )
 
-            # Usage:
-            chat_history = []  # Collect chat history here (a sequence of messages)
-            rag_chain.invoke({"input": query, "chat_history": chat_history})
+        # Usage:
+        chat_history = []  # Collect chat history here (a sequence of messages)
+        rag_chain.invoke({"input": query, "chat_history": chat_history})
 
     This chain takes in chat history (a list of messages) and new questions,
     and then returns an answer to that question.
@@ -360,22 +374,26 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
                 retriever=retriever,
                 question_generator=question_generator_chain,
             )
+
     """
 
     retriever: BaseRetriever
     """Retriever to use to fetch documents."""
     max_tokens_limit: Optional[int] = None
     """If set, enforces that the documents returned are less than this limit.
-    This is only enforced if `combine_docs_chain` is of type StuffDocumentsChain."""
+
+    This is only enforced if ``combine_docs_chain`` is of type StuffDocumentsChain.
+    """
 
     def _reduce_tokens_below_limit(self, docs: list[Document]) -> list[Document]:
         num_docs = len(docs)
 
         if self.max_tokens_limit and isinstance(
-            self.combine_docs_chain, StuffDocumentsChain
+            self.combine_docs_chain,
+            StuffDocumentsChain,
         ):
             tokens = [
-                self.combine_docs_chain.llm_chain._get_num_tokens(doc.page_content)
+                self.combine_docs_chain.llm_chain._get_num_tokens(doc.page_content)  # noqa: SLF001
                 for doc in docs
             ]
             token_count = sum(tokens[:num_docs])
@@ -385,6 +403,7 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
 
         return docs[:num_docs]
 
+    @override
     def _get_docs(
         self,
         question: str,
@@ -394,10 +413,12 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
     ) -> list[Document]:
         """Get docs."""
         docs = self.retriever.invoke(
-            question, config={"callbacks": run_manager.get_child()}
+            question,
+            config={"callbacks": run_manager.get_child()},
         )
         return self._reduce_tokens_below_limit(docs)
 
+    @override
     async def _aget_docs(
         self,
         question: str,
@@ -407,7 +428,8 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
     ) -> list[Document]:
         """Get docs."""
         docs = await self.retriever.ainvoke(
-            question, config={"callbacks": run_manager.get_child()}
+            question,
+            config={"callbacks": run_manager.get_child()},
         )
         return self._reduce_tokens_below_limit(docs)
 
@@ -418,7 +440,7 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain):
         retriever: BaseRetriever,
         condense_question_prompt: BasePromptTemplate = CONDENSE_QUESTION_PROMPT,
         chain_type: str = "stuff",
-        verbose: bool = False,
+        verbose: bool = False,  # noqa: FBT001,FBT002
         condense_question_llm: Optional[BaseLanguageModel] = None,
         combine_docs_chain_kwargs: Optional[dict] = None,
         callbacks: Callbacks = None,
@@ -485,13 +507,15 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
 
     @model_validator(mode="before")
     @classmethod
-    def raise_deprecation(cls, values: dict) -> Any:
+    def _raise_deprecation(cls, values: dict) -> Any:
         warnings.warn(
             "`ChatVectorDBChain` is deprecated - "
-            "please use `from langchain.chains import ConversationalRetrievalChain`"
+            "please use `from langchain.chains import ConversationalRetrievalChain`",
+            stacklevel=4,
         )
         return values
 
+    @override
     def _get_docs(
         self,
         question: str,
@@ -503,7 +527,9 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
         vectordbkwargs = inputs.get("vectordbkwargs", {})
         full_kwargs = {**self.search_kwargs, **vectordbkwargs}
         return self.vectorstore.similarity_search(
-            question, k=self.top_k_docs_for_context, **full_kwargs
+            question,
+            k=self.top_k_docs_for_context,
+            **full_kwargs,
         )
 
     async def _aget_docs(
@@ -514,7 +540,8 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
         run_manager: AsyncCallbackManagerForChainRun,
     ) -> list[Document]:
         """Get docs."""
-        raise NotImplementedError("ChatVectorDBChain does not support async")
+        msg = "ChatVectorDBChain does not support async"
+        raise NotImplementedError(msg)
 
     @classmethod
     def from_llm(
@@ -536,7 +563,9 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
             **combine_docs_chain_kwargs,
         )
         condense_question_chain = LLMChain(
-            llm=llm, prompt=condense_question_prompt, callbacks=callbacks
+            llm=llm,
+            prompt=condense_question_prompt,
+            callbacks=callbacks,
         )
         return cls(
             vectorstore=vectorstore,

langchain/chains/conversational_retrieval/prompts.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts.prompt import PromptTemplate
 
 _template = """Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question, in its original language.
@@ -6,7 +5,7 @@ _template = """Given the following conversation and a follow up question, rephra
 Chat History:
 {chat_history}
 Follow Up Input: {question}
-Standalone question:"""
+Standalone question:"""  # noqa: E501
 CONDENSE_QUESTION_PROMPT = PromptTemplate.from_template(_template)
 
 prompt_template = """Use the following pieces of context to answer the question at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer.
@@ -14,7 +13,7 @@ prompt_template = """Use the following pieces of context to answer the question
 {context}
 
 Question: {question}
-Helpful Answer:"""
+Helpful Answer:"""  # noqa: E501
 QA_PROMPT = PromptTemplate(
     template=prompt_template, input_variables=["context", "question"]
 )

langchain/chains/elasticsearch_database/base.py

@@ -34,6 +34,7 @@ class ElasticsearchDatabaseChain(Chain):
 
             database = Elasticsearch("http://localhost:9200")
             db_chain = ElasticsearchDatabaseChain.from_llm(OpenAI(), database)
+
     """
 
     query_chain: Runnable
@@ -58,11 +59,10 @@ class ElasticsearchDatabaseChain(Chain):
     )
 
     @model_validator(mode="after")
-    def validate_indices(self) -> Self:
+    def _validate_indices(self) -> Self:
         if self.include_indices and self.ignore_indices:
-            raise ValueError(
-                "Cannot specify both 'include_indices' and 'ignore_indices'."
-            )
+            msg = "Cannot specify both 'include_indices' and 'ignore_indices'."
+            raise ValueError(msg)
         return self
 
     @property
@@ -81,8 +81,7 @@ class ElasticsearchDatabaseChain(Chain):
         """
         if not self.return_intermediate_steps:
             return [self.output_key]
-        else:
-            return [self.output_key, INTERMEDIATE_STEPS_KEY]
+        return [self.output_key, INTERMEDIATE_STEPS_KEY]
 
     def _list_indices(self) -> list[str]:
         all_indices = [
@@ -111,7 +110,7 @@ class ElasticsearchDatabaseChain(Chain):
             [
                 "Mapping for index {}:\n{}".format(index, mappings[index]["mappings"])
                 for index in mappings
-            ]
+            ],
         )
 
     def _search(self, indices: list[str], query: str) -> str:
@@ -144,7 +143,7 @@ class ElasticsearchDatabaseChain(Chain):
 
             _run_manager.on_text(es_cmd, color="green", verbose=self.verbose)
             intermediate_steps.append(
-                es_cmd
+                es_cmd,
             )  # output: elasticsearch dsl generation (no checker)
             intermediate_steps.append({"es_cmd": es_cmd})  # input: ES search
             result = self._search(indices=indices, query=es_cmd)
@@ -166,12 +165,13 @@ class ElasticsearchDatabaseChain(Chain):
             chain_result: dict[str, Any] = {self.output_key: final_result}
             if self.return_intermediate_steps:
                 chain_result[INTERMEDIATE_STEPS_KEY] = intermediate_steps
-            return chain_result
         except Exception as exc:
             # Append intermediate steps to exception, to aid in logging and later
             # improvement of few shot prompt seeds
             exc.intermediate_steps = intermediate_steps  # type: ignore[attr-defined]
-            raise exc
+            raise
+
+        return chain_result
 
     @property
     def _chain_type(self) -> str:

langchain/chains/elasticsearch_database/prompts.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts.prompt import PromptTemplate
 
 PROMPT_SUFFIX = """Only use the following Elasticsearch indices:
@@ -17,7 +16,7 @@ Use the following format:
 
 Question: Question here
 ESQuery: Elasticsearch Query formatted as json
-"""
+"""  # noqa: E501
 
 DSL_PROMPT = PromptTemplate.from_template(DEFAULT_DSL_TEMPLATE + PROMPT_SUFFIX)
 
@@ -31,6 +30,6 @@ Answer: Final answer here
 
 Question: {input}
 Data: {data}
-Answer:"""
+Answer:"""  # noqa: E501
 
 ANSWER_PROMPT = PromptTemplate.from_template(DEFAULT_ANSWER_TEMPLATE)

langchain/chains/ernie_functions/__init__.py

@@ -36,9 +36,9 @@ def __getattr__(name: str) -> Any:
 
 __all__ = [
     "convert_to_ernie_function",
-    "create_structured_output_chain",
     "create_ernie_fn_chain",
-    "create_structured_output_runnable",
     "create_ernie_fn_runnable",
+    "create_structured_output_chain",
+    "create_structured_output_runnable",
     "get_ernie_output_parser",
 ]

langchain/chains/example_generator.py

@@ -7,7 +7,9 @@ TEST_GEN_TEMPLATE_SUFFIX = "Add another example."
 
 
 def generate_example(
-    examples: list[dict], llm: BaseLanguageModel, prompt_template: PromptTemplate
+    examples: list[dict],
+    llm: BaseLanguageModel,
+    prompt_template: PromptTemplate,
 ) -> str:
     """Return another example given a list of examples for a prompt."""
     prompt = FewShotPromptTemplate(

langchain/chains/flare/base.py

@@ -15,6 +15,7 @@ from langchain_core.prompts import BasePromptTemplate
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.runnables import Runnable
 from pydantic import Field
+from typing_extensions import override
 
 from langchain.chains.base import Chain
 from langchain.chains.flare.prompts import (
@@ -44,6 +45,7 @@ class QuestionGeneratorChain(LLMChain):
     """Prompt template for the chain."""
 
     @classmethod
+    @override
     def is_lc_serializable(cls) -> bool:
         return False
 
@@ -69,7 +71,8 @@ def _low_confidence_spans(
             "NumPy not found in the current Python environment. FlareChain will use a "
             "pure Python implementation for internal calculations, which may "
             "significantly impact performance, especially for large datasets. For "
-            "optimal speed and efficiency, consider installing NumPy: pip install numpy"
+            "optimal speed and efficiency, consider installing NumPy: pip install "
+            "numpy",
         )
         import math
 
@@ -171,7 +174,8 @@ class FlareChain(Chain):
         callbacks = _run_manager.get_child()
         if isinstance(self.question_generator_chain, LLMChain):
             question_gen_outputs = self.question_generator_chain.apply(
-                question_gen_inputs, callbacks=callbacks
+                question_gen_inputs,
+                callbacks=callbacks,
             )
             questions = [
                 output[self.question_generator_chain.output_keys[0]]
@@ -179,10 +183,13 @@ class FlareChain(Chain):
             ]
         else:
             questions = self.question_generator_chain.batch(
-                question_gen_inputs, config={"callbacks": callbacks}
+                question_gen_inputs,
+                config={"callbacks": callbacks},
             )
         _run_manager.on_text(
-            f"Generated Questions: {questions}", color="yellow", end="\n"
+            f"Generated Questions: {questions}",
+            color="yellow",
+            end="\n",
         )
         return self._do_generation(questions, user_input, response, _run_manager)
 
@@ -197,15 +204,18 @@ class FlareChain(Chain):
 
         response = ""
 
-        for i in range(self.max_iter):
+        for _i in range(self.max_iter):
             _run_manager.on_text(
-                f"Current Response: {response}", color="blue", end="\n"
+                f"Current Response: {response}",
+                color="blue",
+                end="\n",
             )
             _input = {"user_input": user_input, "context": "", "response": response}
             tokens, log_probs = _extract_tokens_and_log_probs(
                 self.response_chain.invoke(
-                    _input, {"callbacks": _run_manager.get_child()}
-                )
+                    _input,
+                    {"callbacks": _run_manager.get_child()},
+                ),
             )
             low_confidence_spans = _low_confidence_spans(
                 tokens,
@@ -236,7 +246,10 @@ class FlareChain(Chain):
 
     @classmethod
     def from_llm(
-        cls, llm: BaseLanguageModel, max_generation_len: int = 32, **kwargs: Any
+        cls,
+        llm: BaseLanguageModel,
+        max_generation_len: int = 32,
+        **kwargs: Any,
     ) -> FlareChain:
         """Creates a FlareChain from a language model.
 
@@ -250,14 +263,17 @@ class FlareChain(Chain):
         """
         try:
             from langchain_openai import ChatOpenAI
-        except ImportError:
-            raise ImportError(
+        except ImportError as e:
+            msg = (
                 "OpenAI is required for FlareChain. "
                 "Please install langchain-openai."
                 "pip install langchain-openai"
             )
+            raise ImportError(msg) from e
         llm = ChatOpenAI(
-            max_completion_tokens=max_generation_len, logprobs=True, temperature=0
+            max_completion_tokens=max_generation_len,
+            logprobs=True,
+            temperature=0,
         )
         response_chain = PROMPT | llm
         question_gen_chain = QUESTION_GENERATOR_PROMPT | llm | StrOutputParser()

langchain/chains/flare/prompts.py

@@ -1,5 +1,6 @@
 from langchain_core.output_parsers import BaseOutputParser
 from langchain_core.prompts import PromptTemplate
+from typing_extensions import override
 
 
 class FinishedOutputParser(BaseOutputParser[tuple[str, bool]]):
@@ -8,6 +9,7 @@ class FinishedOutputParser(BaseOutputParser[tuple[str, bool]]):
     finished_value: str = "FINISHED"
     """Value that indicates the output is finished."""
 
+    @override
     def parse(self, text: str) -> tuple[str, bool]:
         cleaned = text.strip()
         finished = self.finished_value in cleaned

langchain/chains/graph_qa/cypher.py

@@ -31,9 +31,9 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "GraphCypherQAChain",
+    "CYPHER_GENERATION_PROMPT",
     "INTERMEDIATE_STEPS_KEY",
+    "GraphCypherQAChain",
     "construct_schema",
     "extract_cypher",
-    "CYPHER_GENERATION_PROMPT",
 ]

langchain/chains/graph_qa/falkordb.py

@@ -26,4 +26,4 @@ def __getattr__(name: str) -> Any:
     return _import_attribute(name)
 
 
-__all__ = ["FalkorDBQAChain", "INTERMEDIATE_STEPS_KEY", "extract_cypher"]
+__all__ = ["INTERMEDIATE_STEPS_KEY", "FalkorDBQAChain", "extract_cypher"]

langchain/chains/graph_qa/gremlin.py

@@ -30,7 +30,7 @@ def __getattr__(name: str) -> Any:
 
 __all__ = [
     "GRAPHDB_SPARQL_FIX_TEMPLATE",
-    "GremlinQAChain",
     "INTERMEDIATE_STEPS_KEY",
+    "GremlinQAChain",
     "extract_gremlin",
 ]

langchain/chains/graph_qa/neptune_sparql.py

@@ -30,7 +30,7 @@ def __getattr__(name: str) -> Any:
 
 __all__ = [
     "INTERMEDIATE_STEPS_KEY",
-    "NeptuneSparqlQAChain",
     "SPARQL_GENERATION_TEMPLATE",
+    "NeptuneSparqlQAChain",
     "extract_sparql",
 ]