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

langchain/hub.py

@@ -1,4 +1,4 @@
-"""Interface with the LangChain Hub."""
+"""Interface with the `LangChain Hub <https://smith.langchain.com/hub>`__."""
 
 from __future__ import annotations
 
@@ -21,25 +21,25 @@ def _get_client(
         ls_client = LangSmithClient(api_url, api_key=api_key)
         if hasattr(ls_client, "push_prompt") and hasattr(ls_client, "pull_prompt"):
             return ls_client
-        else:
-            from langchainhub import Client as LangChainHubClient
+        from langchainhub import Client as LangChainHubClient
 
-            return LangChainHubClient(api_url, api_key=api_key)
+        return LangChainHubClient(api_url, api_key=api_key)
     except ImportError:
         try:
             from langchainhub import Client as LangChainHubClient
 
             return LangChainHubClient(api_url, api_key=api_key)
         except ImportError as e:
-            raise ImportError(
+            msg = (
                 "Could not import langsmith or langchainhub (deprecated),"
                 "please install with `pip install langsmith`."
-            ) from e
+            )
+            raise ImportError(msg) from e
 
 
 def push(
     repo_full_name: str,
-    object: Any,
+    object: Any,  # noqa: A002
     *,
     api_url: Optional[str] = None,
     api_key: Optional[str] = None,
@@ -81,14 +81,13 @@ def push(
 
     # Then it's langchainhub
     manifest_json = dumps(object)
-    message = client.push(
+    return client.push(
         repo_full_name,
         manifest_json,
         parent_commit_hash=parent_commit_hash,
         new_repo_is_public=new_repo_is_public,
         new_repo_description=new_repo_description,
     )
-    return message
 
 
 def pull(
@@ -112,8 +111,7 @@ def pull(
 
     # Then it's langsmith
     if hasattr(client, "pull_prompt"):
-        response = client.pull_prompt(owner_repo_commit, include_model=include_model)
-        return response
+        return client.pull_prompt(owner_repo_commit, include_model=include_model)
 
     # Then it's langchainhub
     if hasattr(client, "pull_repo"):

langchain/indexes/__init__.py

@@ -40,11 +40,11 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    # Keep sorted
-    "aindex",
     "GraphIndexCreator",
-    "index",
     "IndexingResult",
     "SQLRecordManager",
     "VectorstoreIndexCreator",
+    # Keep sorted
+    "aindex",
+    "index",
 ]

langchain/indexes/_sql_record_manager.py

@@ -119,10 +119,12 @@ class SQLRecordManager(RecordManager):
         """
         super().__init__(namespace=namespace)
         if db_url is None and engine is None:
-            raise ValueError("Must specify either db_url or engine")
+            msg = "Must specify either db_url or engine"
+            raise ValueError(msg)
 
         if db_url is not None and engine is not None:
-            raise ValueError("Must specify either db_url or engine, not both")
+            msg = "Must specify either db_url or engine, not both"
+            raise ValueError(msg)
 
         _engine: Union[Engine, AsyncEngine]
         if db_url:
@@ -134,7 +136,8 @@ class SQLRecordManager(RecordManager):
             _engine = engine
 
         else:
-            raise AssertionError("Something went wrong with configuration of engine.")
+            msg = "Something went wrong with configuration of engine."
+            raise AssertionError(msg)
 
         _session_factory: Union[sessionmaker[Session], async_sessionmaker[AsyncSession]]
         if isinstance(_engine, AsyncEngine):
@@ -149,7 +152,8 @@ class SQLRecordManager(RecordManager):
     def create_schema(self) -> None:
         """Create the database schema."""
         if isinstance(self.engine, AsyncEngine):
-            raise AssertionError("This method is not supported for async engines.")
+            msg = "This method is not supported for async engines."
+            raise AssertionError(msg)  # noqa: TRY004
 
         Base.metadata.create_all(self.engine)
 
@@ -157,7 +161,8 @@ class SQLRecordManager(RecordManager):
         """Create the database schema."""
 
         if not isinstance(self.engine, AsyncEngine):
-            raise AssertionError("This method is not supported for sync engines.")
+            msg = "This method is not supported for sync engines."
+            raise AssertionError(msg)  # noqa: TRY004
 
         async with self.engine.begin() as session:
             await session.run_sync(Base.metadata.create_all)
@@ -167,7 +172,8 @@ class SQLRecordManager(RecordManager):
         """Create a session and close it after use."""
 
         if isinstance(self.session_factory, async_sessionmaker):
-            raise AssertionError("This method is not supported for async engines.")
+            msg = "This method is not supported for async engines."
+            raise AssertionError(msg)  # noqa: TRY004
 
         session = self.session_factory()
         try:
@@ -180,7 +186,8 @@ class SQLRecordManager(RecordManager):
         """Create a session and close it after use."""
 
         if not isinstance(self.session_factory, async_sessionmaker):
-            raise AssertionError("This method is not supported for sync engines.")
+            msg = "This method is not supported for sync engines."
+            raise AssertionError(msg)  # noqa: TRY004
 
         async with self.session_factory() as session:
             yield session
@@ -206,13 +213,15 @@ class SQLRecordManager(RecordManager):
             elif self.dialect == "postgresql":
                 query = text("SELECT EXTRACT (EPOCH FROM CURRENT_TIMESTAMP);")
             else:
-                raise NotImplementedError(f"Not implemented for dialect {self.dialect}")
+                msg = f"Not implemented for dialect {self.dialect}"
+                raise NotImplementedError(msg)
 
             dt = session.execute(query).scalar()
             if isinstance(dt, decimal.Decimal):
                 dt = float(dt)
             if not isinstance(dt, float):
-                raise AssertionError(f"Unexpected type for datetime: {type(dt)}")
+                msg = f"Unexpected type for datetime: {type(dt)}"
+                raise AssertionError(msg)  # noqa: TRY004
             return dt
 
     async def aget_time(self) -> float:
@@ -236,14 +245,16 @@ class SQLRecordManager(RecordManager):
             elif self.dialect == "postgresql":
                 query = text("SELECT EXTRACT (EPOCH FROM CURRENT_TIMESTAMP);")
             else:
-                raise NotImplementedError(f"Not implemented for dialect {self.dialect}")
+                msg = f"Not implemented for dialect {self.dialect}"
+                raise NotImplementedError(msg)
 
             dt = (await session.execute(query)).scalar_one_or_none()
 
             if isinstance(dt, decimal.Decimal):
                 dt = float(dt)
             if not isinstance(dt, float):
-                raise AssertionError(f"Unexpected type for datetime: {type(dt)}")
+                msg = f"Unexpected type for datetime: {type(dt)}"
+                raise AssertionError(msg)  # noqa: TRY004
             return dt
 
     def update(
@@ -258,10 +269,11 @@ class SQLRecordManager(RecordManager):
             group_ids = [None] * len(keys)
 
         if len(keys) != len(group_ids):
-            raise ValueError(
+            msg = (
                 f"Number of keys ({len(keys)}) does not match number of "
                 f"group_ids ({len(group_ids)})"
             )
+            raise ValueError(msg)
 
         # Get the current time from the server.
         # This makes an extra round trip to the server, should not be a big deal
@@ -274,7 +286,8 @@ class SQLRecordManager(RecordManager):
 
         if time_at_least and update_time < time_at_least:
             # Safeguard against time sync issues
-            raise AssertionError(f"Time sync issue: {update_time} < {time_at_least}")
+            msg = f"Time sync issue: {update_time} < {time_at_least}"
+            raise AssertionError(msg)
 
         records_to_upsert = [
             {
@@ -294,14 +307,14 @@ class SQLRecordManager(RecordManager):
                 # Note: uses SQLite insert to make on_conflict_do_update work.
                 # This code needs to be generalized a bit to work with more dialects.
                 sqlite_insert_stmt: SqliteInsertType = sqlite_insert(
-                    UpsertionRecord
+                    UpsertionRecord,
                 ).values(records_to_upsert)
                 stmt = sqlite_insert_stmt.on_conflict_do_update(
                     [UpsertionRecord.key, UpsertionRecord.namespace],
-                    set_=dict(
-                        updated_at=sqlite_insert_stmt.excluded.updated_at,
-                        group_id=sqlite_insert_stmt.excluded.group_id,
-                    ),
+                    set_={
+                        "updated_at": sqlite_insert_stmt.excluded.updated_at,
+                        "group_id": sqlite_insert_stmt.excluded.group_id,
+                    },
                 )
             elif self.dialect == "postgresql":
                 from sqlalchemy.dialects.postgresql import Insert as PgInsertType
@@ -310,17 +323,18 @@ class SQLRecordManager(RecordManager):
                 # Note: uses postgresql insert to make on_conflict_do_update work.
                 # This code needs to be generalized a bit to work with more dialects.
                 pg_insert_stmt: PgInsertType = pg_insert(UpsertionRecord).values(
-                    records_to_upsert
+                    records_to_upsert,
                 )
                 stmt = pg_insert_stmt.on_conflict_do_update(  # type: ignore[assignment]
                     "uix_key_namespace",  # Name of constraint
-                    set_=dict(
-                        updated_at=pg_insert_stmt.excluded.updated_at,
-                        group_id=pg_insert_stmt.excluded.group_id,
-                    ),
+                    set_={
+                        "updated_at": pg_insert_stmt.excluded.updated_at,
+                        "group_id": pg_insert_stmt.excluded.group_id,
+                    },
                 )
             else:
-                raise NotImplementedError(f"Unsupported dialect {self.dialect}")
+                msg = f"Unsupported dialect {self.dialect}"
+                raise NotImplementedError(msg)
 
             session.execute(stmt)
             session.commit()
@@ -337,10 +351,11 @@ class SQLRecordManager(RecordManager):
             group_ids = [None] * len(keys)
 
         if len(keys) != len(group_ids):
-            raise ValueError(
+            msg = (
                 f"Number of keys ({len(keys)}) does not match number of "
                 f"group_ids ({len(group_ids)})"
             )
+            raise ValueError(msg)
 
         # Get the current time from the server.
         # This makes an extra round trip to the server, should not be a big deal
@@ -353,7 +368,8 @@ class SQLRecordManager(RecordManager):
 
         if time_at_least and update_time < time_at_least:
             # Safeguard against time sync issues
-            raise AssertionError(f"Time sync issue: {update_time} < {time_at_least}")
+            msg = f"Time sync issue: {update_time} < {time_at_least}"
+            raise AssertionError(msg)
 
         records_to_upsert = [
             {
@@ -373,14 +389,14 @@ class SQLRecordManager(RecordManager):
                 # Note: uses SQLite insert to make on_conflict_do_update work.
                 # This code needs to be generalized a bit to work with more dialects.
                 sqlite_insert_stmt: SqliteInsertType = sqlite_insert(
-                    UpsertionRecord
+                    UpsertionRecord,
                 ).values(records_to_upsert)
                 stmt = sqlite_insert_stmt.on_conflict_do_update(
                     [UpsertionRecord.key, UpsertionRecord.namespace],
-                    set_=dict(
-                        updated_at=sqlite_insert_stmt.excluded.updated_at,
-                        group_id=sqlite_insert_stmt.excluded.group_id,
-                    ),
+                    set_={
+                        "updated_at": sqlite_insert_stmt.excluded.updated_at,
+                        "group_id": sqlite_insert_stmt.excluded.group_id,
+                    },
                 )
             elif self.dialect == "postgresql":
                 from sqlalchemy.dialects.postgresql import Insert as PgInsertType
@@ -389,17 +405,18 @@ class SQLRecordManager(RecordManager):
                 # Note: uses SQLite insert to make on_conflict_do_update work.
                 # This code needs to be generalized a bit to work with more dialects.
                 pg_insert_stmt: PgInsertType = pg_insert(UpsertionRecord).values(
-                    records_to_upsert
+                    records_to_upsert,
                 )
                 stmt = pg_insert_stmt.on_conflict_do_update(  # type: ignore[assignment]
                     "uix_key_namespace",  # Name of constraint
-                    set_=dict(
-                        updated_at=pg_insert_stmt.excluded.updated_at,
-                        group_id=pg_insert_stmt.excluded.group_id,
-                    ),
+                    set_={
+                        "updated_at": pg_insert_stmt.excluded.updated_at,
+                        "group_id": pg_insert_stmt.excluded.group_id,
+                    },
                 )
             else:
-                raise NotImplementedError(f"Unsupported dialect {self.dialect}")
+                msg = f"Unsupported dialect {self.dialect}"
+                raise NotImplementedError(msg)
 
             await session.execute(stmt)
             await session.commit()
@@ -412,10 +429,10 @@ class SQLRecordManager(RecordManager):
                 and_(
                     UpsertionRecord.key.in_(keys),
                     UpsertionRecord.namespace == self.namespace,
-                )
+                ),
             )
             records = filtered_query.all()
-        found_keys = set(r.key for r in records)
+        found_keys = {r.key for r in records}
         return [k in found_keys for k in keys]
 
     async def aexists(self, keys: Sequence[str]) -> list[bool]:
@@ -428,8 +445,8 @@ class SQLRecordManager(RecordManager):
                             and_(
                                 UpsertionRecord.key.in_(keys),
                                 UpsertionRecord.namespace == self.namespace,
-                            )
-                        )
+                            ),
+                        ),
                     )
                 )
                 .scalars()
@@ -450,7 +467,7 @@ class SQLRecordManager(RecordManager):
         session: Session
         with self._make_session() as session:
             query: Query = session.query(UpsertionRecord).filter(
-                UpsertionRecord.namespace == self.namespace
+                UpsertionRecord.namespace == self.namespace,
             )
 
             if after:
@@ -477,7 +494,7 @@ class SQLRecordManager(RecordManager):
         session: AsyncSession
         async with self._amake_session() as session:
             query: Query = select(UpsertionRecord.key).filter(  # type: ignore[assignment]
-                UpsertionRecord.namespace == self.namespace
+                UpsertionRecord.namespace == self.namespace,
             )
 
             # mypy does not recognize .all() or .filter()
@@ -501,7 +518,7 @@ class SQLRecordManager(RecordManager):
                 and_(
                     UpsertionRecord.key.in_(keys),
                     UpsertionRecord.namespace == self.namespace,
-                )
+                ),
             )
 
             filtered_query.delete()
@@ -515,8 +532,8 @@ class SQLRecordManager(RecordManager):
                     and_(
                         UpsertionRecord.key.in_(keys),
                         UpsertionRecord.namespace == self.namespace,
-                    )
-                )
+                    ),
+                ),
             )
 
             await session.commit()

langchain/indexes/prompts/entity_extraction.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts.prompt import PromptTemplate
 
 _DEFAULT_ENTITY_EXTRACTION_TEMPLATE = """You are an AI assistant reading the transcript of a conversation between an AI and a human. Extract all of the proper nouns from the last line of conversation. As a guideline, a proper noun is generally capitalized. You should definitely extract all names and places.
@@ -34,7 +33,7 @@ Conversation history (for reference only):
 Last line of conversation (for extraction):
 Human: {input}
 
-Output:"""
+Output:"""  # noqa: E501
 ENTITY_EXTRACTION_PROMPT = PromptTemplate(
     input_variables=["history", "input"], template=_DEFAULT_ENTITY_EXTRACTION_TEMPLATE
 )

langchain/indexes/prompts/entity_summarization.py

@@ -1,4 +1,3 @@
-# flake8: noqa
 from langchain_core.prompts.prompt import PromptTemplate
 
 _DEFAULT_ENTITY_SUMMARIZATION_TEMPLATE = """You are an AI assistant helping a human keep track of facts about relevant people, places, and concepts in their life. Update the summary of the provided entity in the "Entity" section based on the last line of your conversation with the human. If you are writing the summary for the first time, return a single sentence.
@@ -17,7 +16,7 @@ Existing summary of {entity}:
 
 Last line of conversation:
 Human: {input}
-Updated summary:"""
+Updated summary:"""  # noqa: E501
 
 ENTITY_SUMMARIZATION_PROMPT = PromptTemplate(
     input_variables=["entity", "summary", "history", "input"],

langchain/indexes/prompts/knowledge_triplet_extraction.py

@@ -1,5 +1,3 @@
-# flake8: noqa
-
 from langchain_core.prompts.prompt import PromptTemplate
 
 KG_TRIPLE_DELIMITER = "<|>"
@@ -25,7 +23,7 @@ _DEFAULT_KNOWLEDGE_TRIPLE_EXTRACTION_TEMPLATE = (
     "END OF EXAMPLE\n\n"
     "EXAMPLE\n"
     "Oh huh. I know Descartes likes to drive antique scooters and play the mandolin.\n"
-    f"Output: (Descartes, likes to drive, antique scooters){KG_TRIPLE_DELIMITER}(Descartes, plays, mandolin)\n"
+    f"Output: (Descartes, likes to drive, antique scooters){KG_TRIPLE_DELIMITER}(Descartes, plays, mandolin)\n"  # noqa: E501
     "END OF EXAMPLE\n\n"
     "EXAMPLE\n"
     "{text}"

langchain/indexes/vectorstore.py

@@ -48,16 +48,19 @@ class VectorStoreIndexWrapper(BaseModel):
             The result string from the RetrievalQA chain.
         """
         if llm is None:
-            raise NotImplementedError(
+            msg = (
                 "This API has been changed to require an LLM. "
                 "Please provide an llm to use for querying the vectorstore.\n"
                 "For example,\n"
                 "from langchain_openai import OpenAI\n"
                 "llm = OpenAI(temperature=0)"
             )
+            raise NotImplementedError(msg)
         retriever_kwargs = retriever_kwargs or {}
         chain = RetrievalQA.from_chain_type(
-            llm, retriever=self.vectorstore.as_retriever(**retriever_kwargs), **kwargs
+            llm,
+            retriever=self.vectorstore.as_retriever(**retriever_kwargs),
+            **kwargs,
         )
         return chain.invoke({chain.input_key: question})[chain.output_key]
 
@@ -80,16 +83,19 @@ class VectorStoreIndexWrapper(BaseModel):
             The asynchronous result string from the RetrievalQA chain.
         """
         if llm is None:
-            raise NotImplementedError(
+            msg = (
                 "This API has been changed to require an LLM. "
                 "Please provide an llm to use for querying the vectorstore.\n"
                 "For example,\n"
                 "from langchain_openai import OpenAI\n"
                 "llm = OpenAI(temperature=0)"
             )
+            raise NotImplementedError(msg)
         retriever_kwargs = retriever_kwargs or {}
         chain = RetrievalQA.from_chain_type(
-            llm, retriever=self.vectorstore.as_retriever(**retriever_kwargs), **kwargs
+            llm,
+            retriever=self.vectorstore.as_retriever(**retriever_kwargs),
+            **kwargs,
         )
         return (await chain.ainvoke({chain.input_key: question}))[chain.output_key]
 
@@ -112,16 +118,19 @@ class VectorStoreIndexWrapper(BaseModel):
             A dictionary containing the answer and source documents.
         """
         if llm is None:
-            raise NotImplementedError(
+            msg = (
                 "This API has been changed to require an LLM. "
                 "Please provide an llm to use for querying the vectorstore.\n"
                 "For example,\n"
                 "from langchain_openai import OpenAI\n"
                 "llm = OpenAI(temperature=0)"
             )
+            raise NotImplementedError(msg)
         retriever_kwargs = retriever_kwargs or {}
         chain = RetrievalQAWithSourcesChain.from_chain_type(
-            llm, retriever=self.vectorstore.as_retriever(**retriever_kwargs), **kwargs
+            llm,
+            retriever=self.vectorstore.as_retriever(**retriever_kwargs),
+            **kwargs,
         )
         return chain.invoke({chain.question_key: question})
 
@@ -144,16 +153,19 @@ class VectorStoreIndexWrapper(BaseModel):
             A dictionary containing the answer and source documents.
         """
         if llm is None:
-            raise NotImplementedError(
+            msg = (
                 "This API has been changed to require an LLM. "
                 "Please provide an llm to use for querying the vectorstore.\n"
                 "For example,\n"
                 "from langchain_openai import OpenAI\n"
                 "llm = OpenAI(temperature=0)"
             )
+            raise NotImplementedError(msg)
         retriever_kwargs = retriever_kwargs or {}
         chain = RetrievalQAWithSourcesChain.from_chain_type(
-            llm, retriever=self.vectorstore.as_retriever(**retriever_kwargs), **kwargs
+            llm,
+            retriever=self.vectorstore.as_retriever(**retriever_kwargs),
+            **kwargs,
         )
         return await chain.ainvoke({chain.question_key: question})
 
@@ -164,14 +176,14 @@ def _get_in_memory_vectorstore() -> type[VectorStore]:
 
     try:
         from langchain_community.vectorstores.inmemory import InMemoryVectorStore
-    except ImportError:
-        raise ImportError(
-            "Please install langchain-community to use the InMemoryVectorStore."
-        )
+    except ImportError as e:
+        msg = "Please install langchain-community to use the InMemoryVectorStore."
+        raise ImportError(msg) from e
     warnings.warn(
         "Using InMemoryVectorStore as the default vectorstore."
         "This memory store won't persist data. You should explicitly"
-        "specify a vectorstore when using VectorstoreIndexCreator"
+        "specify a vectorstore when using VectorstoreIndexCreator",
+        stacklevel=3,
     )
     return InMemoryVectorStore
 
@@ -180,7 +192,7 @@ class VectorstoreIndexCreator(BaseModel):
     """Logic for creating indexes."""
 
     vectorstore_cls: type[VectorStore] = Field(
-        default_factory=_get_in_memory_vectorstore
+        default_factory=_get_in_memory_vectorstore,
     )
     embedding: Embeddings
     text_splitter: TextSplitter = Field(default_factory=_get_default_text_splitter)
@@ -216,8 +228,7 @@ class VectorstoreIndexCreator(BaseModel):
         """
         docs = []
         for loader in loaders:
-            async for doc in loader.alazy_load():
-                docs.append(doc)
+            docs.extend([doc async for doc in loader.alazy_load()])
         return await self.afrom_documents(docs)
 
     def from_documents(self, documents: list[Document]) -> VectorStoreIndexWrapper:
@@ -231,12 +242,15 @@ class VectorstoreIndexCreator(BaseModel):
         """
         sub_docs = self.text_splitter.split_documents(documents)
         vectorstore = self.vectorstore_cls.from_documents(
-            sub_docs, self.embedding, **self.vectorstore_kwargs
+            sub_docs,
+            self.embedding,
+            **self.vectorstore_kwargs,
         )
         return VectorStoreIndexWrapper(vectorstore=vectorstore)
 
     async def afrom_documents(
-        self, documents: list[Document]
+        self,
+        documents: list[Document],
     ) -> VectorStoreIndexWrapper:
         """Asynchronously create a vectorstore index from a list of documents.
 
@@ -248,6 +262,8 @@ class VectorstoreIndexCreator(BaseModel):
         """
         sub_docs = self.text_splitter.split_documents(documents)
         vectorstore = await self.vectorstore_cls.afrom_documents(
-            sub_docs, self.embedding, **self.vectorstore_kwargs
+            sub_docs,
+            self.embedding,
+            **self.vectorstore_kwargs,
         )
         return VectorStoreIndexWrapper(vectorstore=vectorstore)

langchain/llms/__init__.py

@@ -552,6 +552,7 @@ def __getattr__(name: str) -> Any:
             "Please import from langchain-community instead:\n\n"
             f"`from langchain_community.llms import {name}`.\n\n"
             "To install langchain-community run `pip install -U langchain-community`.",
+            stacklevel=2,
             category=LangChainDeprecationWarning,
         )
 
@@ -561,12 +562,13 @@ def __getattr__(name: str) -> Any:
             k: v() for k, v in get_type_to_cls_dict().items()
         }
         return type_to_cls_dict
-    else:
-        return getattr(llms, name)
+    return getattr(llms, name)
 
 
 __all__ = [
     "AI21",
+    "RWKV",
+    "VLLM",
     "AlephAlpha",
     "AmazonAPIGateway",
     "Anthropic",
@@ -592,8 +594,8 @@ __all__ = [
     "FakeListLLM",
     "Fireworks",
     "ForefrontAI",
-    "GigaChat",
     "GPT4All",
+    "GigaChat",
     "GooglePalm",
     "GooseAI",
     "GradientLLM",
@@ -602,18 +604,20 @@ __all__ = [
     "HuggingFacePipeline",
     "HuggingFaceTextGenInference",
     "HumanInputLLM",
+    "JavelinAIGateway",
     "KoboldApiLLM",
     "LlamaCpp",
-    "TextGen",
     "ManifestWrapper",
     "Minimax",
     "MlflowAIGateway",
     "Modal",
     "MosaicML",
-    "Nebula",
     "NIBittensorLLM",
     "NLPCloud",
+    "Nebula",
+    "OctoAIEndpoint",
     "Ollama",
+    "OpaquePrompts",
     "OpenAI",
     "OpenAIChat",
     "OpenLLM",
@@ -625,28 +629,24 @@ __all__ = [
     "PredictionGuard",
     "PromptLayerOpenAI",
     "PromptLayerOpenAIChat",
-    "OpaquePrompts",
-    "RWKV",
+    "QianfanLLMEndpoint",
     "Replicate",
     "SagemakerEndpoint",
     "SelfHostedHuggingFaceLLM",
     "SelfHostedPipeline",
     "StochasticAI",
+    "TextGen",
     "TitanTakeoff",
     "TitanTakeoffPro",
     "Tongyi",
+    "VLLMOpenAI",
     "VertexAI",
     "VertexAIModelGarden",
-    "VLLM",
-    "VLLMOpenAI",
+    "VolcEngineMaasLLM",
     "WatsonxLLM",
     "Writer",
-    "OctoAIEndpoint",
     "Xinference",
-    "JavelinAIGateway",
-    "QianfanLLMEndpoint",
     "YandexGPT",
-    "VolcEngineMaasLLM",
 ]
 
 

langchain/llms/ai21.py

@@ -23,6 +23,6 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
-    "AI21PenaltyData",
     "AI21",
+    "AI21PenaltyData",
 ]