core: Add N(naming) ruff rules

libs/core/langchain_core/_api/beta_decorator.py

@@ -154,7 +154,7 @@ def warn_if_direct_instance(
             _name = _name or obj.fget.__qualname__
             old_doc = obj.__doc__
 
-            class _beta_property(property):
+            class _BetaProperty(property):
                 """A beta property."""
 
                 def __init__(self, fget=None, fset=None, fdel=None, doc=None):
@@ -185,7 +185,7 @@ def __set_name__(self, owner, set_name):
 
             def finalize(wrapper: Callable[..., Any], new_doc: str) -> Any:
                 """Finalize the property."""
-                return _beta_property(
+                return _BetaProperty(
                     fget=obj.fget, fset=obj.fset, fdel=obj.fdel, doc=new_doc
                 )
 

libs/core/langchain_core/_api/deprecation.py

@@ -265,7 +265,7 @@ def finalize(wrapper: Callable[..., Any], new_doc: str) -> T:
             _name = _name or cast(Union[Type, Callable], obj.fget).__qualname__
             old_doc = obj.__doc__
 
-            class _deprecated_property(property):
+            class _DeprecatedProperty(property):
                 """A deprecated property."""
 
                 def __init__(self, fget=None, fset=None, fdel=None, doc=None):  # type: ignore[no-untyped-def]
@@ -298,7 +298,7 @@ def finalize(wrapper: Callable[..., Any], new_doc: str) -> T:
                 """Finalize the property."""
                 return cast(
                     T,
-                    _deprecated_property(
+                    _DeprecatedProperty(
                         fget=obj.fget, fset=obj.fset, fdel=obj.fdel, doc=new_doc
                     ),
                 )

libs/core/langchain_core/exceptions.py

@@ -3,15 +3,15 @@
 from typing import Any, Optional
 
 
-class LangChainException(Exception):
+class LangChainException(Exception):  # noqa: N818
     """General LangChain exception."""
 
 
 class TracerException(LangChainException):
     """Base class for exceptions in tracers module."""
 
 
-class OutputParserException(ValueError, LangChainException):
+class OutputParserException(ValueError, LangChainException):  # noqa: N818
     """Exception that output parsers should raise to signify a parsing error.
 
     This exists to differentiate parsing errors from other code or execution errors

libs/core/langchain_core/language_models/base.py

@@ -19,7 +19,7 @@
 )
 
 from pydantic import BaseModel, ConfigDict, Field, field_validator
-from typing_extensions import TypeAlias, TypedDict
+from typing_extensions import TypeAlias, TypedDict, override
 
 from langchain_core._api import deprecated
 from langchain_core.messages import (
@@ -148,6 +148,7 @@ def set_verbose(cls, verbose: Optional[bool]) -> bool:
             return verbose
 
     @property
+    @override
     def InputType(self) -> TypeAlias:
         """Get the input type for this runnable."""
         from langchain_core.prompt_values import (

libs/core/langchain_core/language_models/chat_models.py

@@ -30,6 +30,7 @@
     Field,
     model_validator,
 )
+from typing_extensions import override
 
 from langchain_core._api import deprecated
 from langchain_core.caches import BaseCache
@@ -255,6 +256,7 @@ def _serialized(self) -> dict[str, Any]:
     # --- Runnable methods ---
 
     @property
+    @override
     def OutputType(self) -> Any:
         """Get the output type for this runnable."""
         return AnyMessage

libs/core/langchain_core/language_models/llms.py

@@ -37,6 +37,7 @@
     stop_after_attempt,
     wait_exponential,
 )
+from typing_extensions import override
 
 from langchain_core._api import deprecated
 from langchain_core.caches import BaseCache
@@ -324,6 +325,7 @@ def _serialized(self) -> dict[str, Any]:
     # --- Runnable methods ---
 
     @property
+    @override
     def OutputType(self) -> Type[str]:
         """Get the input type for this runnable."""
         return str

libs/core/langchain_core/output_parsers/base.py

@@ -13,6 +13,8 @@
     Union,
 )
 
+from typing_extensions import override
+
 from langchain_core.language_models import LanguageModelOutput
 from langchain_core.messages import AnyMessage, BaseMessage
 from langchain_core.outputs import ChatGeneration, Generation
@@ -66,11 +68,13 @@ class BaseGenerationOutputParser(
     """Base class to parse the output of an LLM call."""
 
     @property
+    @override
     def InputType(self) -> Any:
         """Return the input type for the parser."""
         return Union[str, AnyMessage]
 
     @property
+    @override
     def OutputType(self) -> Type[T]:
         """Return the output type for the parser."""
         # even though mypy complains this isn't valid,
@@ -151,11 +155,13 @@ def _type(self) -> str:
     """  # noqa: E501
 
     @property
+    @override
     def InputType(self) -> Any:
         """Return the input type for the parser."""
         return Union[str, AnyMessage]
 
     @property
+    @override
     def OutputType(self) -> Type[T]:
         """Return the output type for the parser.
 

libs/core/langchain_core/output_parsers/pydantic.py

@@ -3,7 +3,7 @@
 
 import pydantic
 from pydantic import SkipValidation
-from typing_extensions import Annotated
+from typing_extensions import Annotated, override
 
 from langchain_core.exceptions import OutputParserException
 from langchain_core.output_parsers import JsonOutputParser
@@ -108,6 +108,7 @@ def _type(self) -> str:
         return "pydantic"
 
     @property
+    @override
     def OutputType(self) -> Type[TBaseModel]:
         """Return the pydantic model."""
         return self.pydantic_object

libs/core/langchain_core/output_parsers/xml.py

@@ -1,6 +1,6 @@
 import re
 import xml
-import xml.etree.ElementTree as ET
+import xml.etree.ElementTree as ET  # noqa: N817
 from typing import Any, AsyncIterator, Dict, Iterator, List, Literal, Optional, Union
 from xml.etree.ElementTree import TreeBuilder
 
@@ -45,14 +45,14 @@ def __init__(self, parser: Literal["defusedxml", "xml"]) -> None:
         """
         if parser == "defusedxml":
             try:
-                from defusedxml import ElementTree as DET  # type: ignore
+                import defusedxml  # type: ignore
             except ImportError as e:
                 raise ImportError(
                     "defusedxml is not installed. "
                     "Please install it to use the defusedxml parser."
                     "You can install it with `pip install defusedxml` "
                 ) from e
-            _parser = DET.DefusedXMLParser(target=TreeBuilder())
+            _parser = defusedxml.ElementTree.DefusedXMLParser(target=TreeBuilder())
         else:
             _parser = None
         self.pull_parser = ET.XMLPullParser(["start", "end"], _parser=_parser)
@@ -188,17 +188,17 @@ def parse(self, text: str) -> Dict[str, Union[str, List[Any]]]:
         # likely if you're reading this you can move them to the top of the file
         if self.parser == "defusedxml":
             try:
-                from defusedxml import ElementTree as DET  # type: ignore
+                import defusedxml  # type: ignore
             except ImportError as e:
                 raise ImportError(
                     "defusedxml is not installed. "
                     "Please install it to use the defusedxml parser."
                     "You can install it with `pip install defusedxml`"
                     "See https://github.com/tiran/defusedxml for more details"
                 ) from e
-            _ET = DET  # Use the defusedxml parser
+            _et = defusedxml.ElementTree  # Use the defusedxml parser
         else:
-            _ET = ET  # Use the standard library parser
+            _et = ET  # Use the standard library parser
 
         match = re.search(r"```(xml)?(.*)```", text, re.DOTALL)
         if match is not None:

libs/core/langchain_core/prompts/base.py

@@ -20,7 +20,7 @@
 
 import yaml
 from pydantic import BaseModel, ConfigDict, Field, model_validator
-from typing_extensions import Self
+from typing_extensions import Self, override
 
 from langchain_core.load import dumpd
 from langchain_core.output_parsers.base import BaseOutputParser
@@ -109,6 +109,7 @@ def _serialized(self) -> dict[str, Any]:
         return dumpd(self)
 
     @property
+    @override
     def OutputType(self) -> Any:
         """Return the output type of the prompt."""
         return Union[StringPromptValue, ChatPromptValueConcrete]

libs/core/langchain_core/runnables/base.py

@@ -37,7 +37,7 @@
 )
 
 from pydantic import BaseModel, ConfigDict, Field, RootModel
-from typing_extensions import Literal, get_args, get_type_hints
+from typing_extensions import Literal, get_args, get_type_hints, override
 
 from langchain_core._api import beta_decorator
 from langchain_core.load.serializable import (
@@ -273,7 +273,7 @@ def get_name(
             return name_
 
     @property
-    def InputType(self) -> Type[Input]:
+    def InputType(self) -> Type[Input]:  # noqa: N802
         """The type of input this Runnable accepts specified as a type annotation."""
         # First loop through all parent classes and if any of them is
         # a pydantic model, we will pick up the generic parameterization
@@ -298,7 +298,7 @@ def InputType(self) -> Type[Input]:
         )
 
     @property
-    def OutputType(self) -> Type[Output]:
+    def OutputType(self) -> Type[Output]:  # noqa: N802
         """The type of output this Runnable produces specified as a type annotation."""
         # First loop through bases -- this will help generic
         # any pydantic models.
@@ -2804,11 +2804,13 @@ def is_lc_serializable(cls) -> bool:
     )
 
     @property
+    @override
     def InputType(self) -> Type[Input]:
         """The type of the input to the Runnable."""
         return self.first.InputType
 
     @property
+    @override
     def OutputType(self) -> Type[Output]:
         """The type of the output of the Runnable."""
         return self.last.OutputType
@@ -3557,6 +3559,7 @@ def get_name(
         return super().get_name(suffix, name=name)
 
     @property
+    @override
     def InputType(self) -> Any:
         """The type of the input to the Runnable."""
         for step in self.steps__.values():
@@ -4050,6 +4053,7 @@ def __init__(
             self.name = "RunnableGenerator"
 
     @property
+    @override
     def InputType(self) -> Any:
         func = getattr(self, "_transform", None) or self._atransform
         try:
@@ -4086,6 +4090,7 @@ def get_input_schema(
         )
 
     @property
+    @override
     def OutputType(self) -> Any:
         func = getattr(self, "_transform", None) or self._atransform
         try:
@@ -4331,6 +4336,7 @@ def __init__(
             pass
 
     @property
+    @override
     def InputType(self) -> Any:
         """The type of the input to this Runnable."""
         func = getattr(self, "func", None) or self.afunc
@@ -4390,6 +4396,7 @@ def get_input_schema(
         return super().get_input_schema(config)
 
     @property
+    @override
     def OutputType(self) -> Any:
         """The type of the output of this Runnable as a type annotation.
 
@@ -4939,6 +4946,7 @@ class RunnableEachBase(RunnableSerializable[List[Input], List[Output]]):
     )
 
     @property
+    @override
     def InputType(self) -> Any:
         return List[self.bound.InputType]  # type: ignore[name-defined]
 
@@ -4962,6 +4970,7 @@ def get_input_schema(
         )
 
     @property
+    @override
     def OutputType(self) -> Type[List[Output]]:
         return List[self.bound.OutputType]  # type: ignore[name-defined]
 
@@ -5255,6 +5264,7 @@ def get_name(
         return self.bound.get_name(suffix, name=name)
 
     @property
+    @override
     def InputType(self) -> Type[Input]:
         return (
             cast(Type[Input], self.custom_input_type)
@@ -5263,6 +5273,7 @@ def InputType(self) -> Type[Input]:
         )
 
     @property
+    @override
     def OutputType(self) -> Type[Output]:
         return (
             cast(Type[Output], self.custom_output_type)

libs/core/langchain_core/runnables/configurable.py

@@ -22,6 +22,7 @@
 from weakref import WeakValueDictionary
 
 from pydantic import BaseModel, ConfigDict
+from typing_extensions import override
 
 from langchain_core.runnables.base import Runnable, RunnableSerializable
 from langchain_core.runnables.config import (
@@ -74,10 +75,12 @@ def get_lc_namespace(cls) -> List[str]:
         return ["langchain", "schema", "runnable"]
 
     @property
+    @override
     def InputType(self) -> Type[Input]:
         return self.default.InputType
 
     @property
+    @override
     def OutputType(self) -> Type[Output]:
         return self.default.OutputType
 

libs/core/langchain_core/runnables/fallbacks.py

@@ -19,6 +19,7 @@
 )
 
 from pydantic import BaseModel, ConfigDict
+from typing_extensions import override
 
 from langchain_core.runnables.base import Runnable, RunnableSerializable
 from langchain_core.runnables.config import (
@@ -112,10 +113,12 @@ def when_all_is_lost(inputs):
     )
 
     @property
+    @override
     def InputType(self) -> Type[Input]:
         return self.runnable.InputType
 
     @property
+    @override
     def OutputType(self) -> Type[Output]:
         return self.runnable.OutputType