langchain 0.3.26__py3-none-any.whl → 0.3.27__py3-none-any.whl

langchain/globals.py

@@ -16,7 +16,9 @@ _debug: bool = False
 _llm_cache: Optional["BaseCache"] = None
 
 
-def set_verbose(value: bool) -> None:
+def set_verbose(
+    value: bool,  # noqa: FBT001
+) -> None:
     """Set a new value for the `verbose` global setting."""
     import langchain
 
@@ -37,7 +39,7 @@ def set_verbose(value: bool) -> None:
         # have migrated to using `set_verbose()` here.
         langchain.verbose = value
 
-    global _verbose
+    global _verbose  # noqa: PLW0603
     _verbose = value
 
 
@@ -64,14 +66,15 @@ def get_verbose() -> bool:
         # In the meantime, the `verbose` setting is considered True if either the old
         # or the new value are True. This accommodates users who haven't migrated
         # to using `set_verbose()` yet. Those users are getting deprecation warnings
-        # directing them to use `set_verbose()` when they import `langhchain.verbose`.
+        # directing them to use `set_verbose()` when they import `langchain.verbose`.
         old_verbose = langchain.verbose
 
-    global _verbose
     return _verbose or old_verbose
 
 
-def set_debug(value: bool) -> None:
+def set_debug(
+    value: bool,  # noqa: FBT001
+) -> None:
     """Set a new value for the `debug` global setting."""
     import langchain
 
@@ -90,7 +93,7 @@ def set_debug(value: bool) -> None:
         # have migrated to using `set_debug()` here.
         langchain.debug = value
 
-    global _debug
+    global _debug  # noqa: PLW0603
     _debug = value
 
 
@@ -115,10 +118,9 @@ def get_debug() -> bool:
         # In the meantime, the `debug` setting is considered True if either the old
         # or the new value are True. This accommodates users who haven't migrated
         # to using `set_debug()` yet. Those users are getting deprecation warnings
-        # directing them to use `set_debug()` when they import `langhchain.debug`.
+        # directing them to use `set_debug()` when they import `langchain.debug`.
         old_debug = langchain.debug
 
-    global _debug
     return _debug or old_debug
 
 
@@ -143,7 +145,7 @@ def set_llm_cache(value: Optional["BaseCache"]) -> None:
         # once all users have migrated to using `set_llm_cache()` here.
         langchain.llm_cache = value
 
-    global _llm_cache
+    global _llm_cache  # noqa: PLW0603
     _llm_cache = value
 
 
@@ -172,8 +174,7 @@ def get_llm_cache() -> "BaseCache":
         # or the old value if both are falsy. This accommodates users
         # who haven't migrated to using `set_llm_cache()` yet.
         # Those users are getting deprecation warnings directing them
-        # to use `set_llm_cache()` when they import `langhchain.llm_cache`.
+        # to use `set_llm_cache()` when they import `langchain.llm_cache`.
         old_llm_cache = langchain.llm_cache
 
-    global _llm_cache
     return _llm_cache or old_llm_cache

langchain/graphs/__init__.py

@@ -44,14 +44,14 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
+    "ArangoGraph",
+    "FalkorDBGraph",
+    "HugeGraph",
+    "KuzuGraph",
     "MemgraphGraph",
-    "NetworkxEntityGraph",
-    "Neo4jGraph",
     "NebulaGraph",
+    "Neo4jGraph",
     "NeptuneGraph",
-    "KuzuGraph",
-    "HugeGraph",
+    "NetworkxEntityGraph",
     "RdfGraph",
-    "ArangoGraph",
-    "FalkorDBGraph",
 ]

langchain/graphs/graph_document.py

@@ -27,7 +27,7 @@ def __getattr__(name: str) -> Any:
 
 
 __all__ = [
+    "GraphDocument",
     "Node",
     "Relationship",
-    "GraphDocument",
 ]

langchain/graphs/networkx_graph.py

@@ -30,7 +30,7 @@ def __getattr__(name: str) -> Any:
 
 __all__ = [
     "KnowledgeTriple",
-    "parse_triples",
-    "get_entities",
     "NetworkxEntityGraph",
+    "get_entities",
+    "parse_triples",
 ]

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}"