langchain-core 0.3.68__py3-none-any.whl → 0.3.70__py3-none-any.whl

langchain_core/prompts/pipeline.py

@@ -18,16 +18,14 @@ def _get_inputs(inputs: dict, input_variables: list[str]) -> dict:
     since="0.3.22",
     removal="1.0",
     message=(
-        "This class is deprecated. Please see the docstring below or at the link"
-        " for a replacement option: "
-        "https://python.langchain.com/api_reference/core/prompts/langchain_core.prompts.pipeline.PipelinePromptTemplate.html"
+        "This class is deprecated in favor of chaining individual prompts together."
     ),
 )
 class PipelinePromptTemplate(BasePromptTemplate):
-    """[DEPRECATED] Pipeline prompt template.
+    """Pipeline prompt template.
 
     This has been deprecated in favor of chaining individual prompts together in your
-    code. E.g. using a for loop, you could do:
+    code; e.g. using a for loop, you could do:
 
     .. code-block:: python
 
@@ -81,13 +79,13 @@ class PipelinePromptTemplate(BasePromptTemplate):
             A formatted string.
         """
         for k, prompt in self.pipeline_prompts:
-            _inputs = _get_inputs(kwargs, prompt.input_variables)
+            inputs = _get_inputs(kwargs, prompt.input_variables)
             if isinstance(prompt, BaseChatPromptTemplate):
-                kwargs[k] = prompt.format_messages(**_inputs)
+                kwargs[k] = prompt.format_messages(**inputs)
             else:
-                kwargs[k] = prompt.format(**_inputs)
-        _inputs = _get_inputs(kwargs, self.final_prompt.input_variables)
-        return self.final_prompt.format_prompt(**_inputs)
+                kwargs[k] = prompt.format(**inputs)
+        inputs = _get_inputs(kwargs, self.final_prompt.input_variables)
+        return self.final_prompt.format_prompt(**inputs)
 
     async def aformat_prompt(self, **kwargs: Any) -> PromptValue:
         """Async format the prompt with the inputs.
@@ -99,13 +97,13 @@ class PipelinePromptTemplate(BasePromptTemplate):
             A formatted string.
         """
         for k, prompt in self.pipeline_prompts:
-            _inputs = _get_inputs(kwargs, prompt.input_variables)
+            inputs = _get_inputs(kwargs, prompt.input_variables)
             if isinstance(prompt, BaseChatPromptTemplate):
-                kwargs[k] = await prompt.aformat_messages(**_inputs)
+                kwargs[k] = await prompt.aformat_messages(**inputs)
             else:
-                kwargs[k] = await prompt.aformat(**_inputs)
-        _inputs = _get_inputs(kwargs, self.final_prompt.input_variables)
-        return await self.final_prompt.aformat_prompt(**_inputs)
+                kwargs[k] = await prompt.aformat(**inputs)
+        inputs = _get_inputs(kwargs, self.final_prompt.input_variables)
+        return await self.final_prompt.aformat_prompt(**inputs)
 
     def format(self, **kwargs: Any) -> str:
         """Format the prompt with the inputs.

langchain_core/prompts/prompt.py

@@ -287,17 +287,17 @@ class PromptTemplate(StringPromptTemplate):
             The prompt template loaded from the template.
         """
         input_variables = get_template_variables(template, template_format)
-        _partial_variables = partial_variables or {}
+        partial_variables_ = partial_variables or {}
 
-        if _partial_variables:
+        if partial_variables_:
             input_variables = [
-                var for var in input_variables if var not in _partial_variables
+                var for var in input_variables if var not in partial_variables_
             ]
 
         return cls(
             input_variables=input_variables,
             template=template,
             template_format=template_format,
-            partial_variables=_partial_variables,
+            partial_variables=partial_variables_,
             **kwargs,
         )

langchain_core/prompts/string.py

@@ -131,12 +131,12 @@ def mustache_template_vars(
         if type_ == "end":
             section_depth -= 1
         elif (
-            type_ in ("variable", "section", "inverted section", "no escape")
+            type_ in {"variable", "section", "inverted section", "no escape"}
             and key != "."
             and section_depth == 0
         ):
             variables.add(key.split(".")[0])
-        if type_ in ("section", "inverted section"):
+        if type_ in {"section", "inverted section"}:
             section_depth += 1
     return variables
 
@@ -164,11 +164,11 @@ def mustache_schema(
         if type_ == "end":
             if section_stack:
                 prefix = section_stack.pop()
-        elif type_ in ("section", "inverted section"):
+        elif type_ in {"section", "inverted section"}:
             section_stack.append(prefix)
             prefix = prefix + tuple(key.split("."))
             fields[prefix] = False
-        elif type_ in ("variable", "no escape"):
+        elif type_ in {"variable", "no escape"}:
             fields[prefix + tuple(key.split("."))] = True
     defs: Defs = {}  # None means leaf node
     while fields:

langchain_core/rate_limiters.py

@@ -146,13 +146,12 @@ class InMemoryRateLimiter(BaseRateLimiter):
 
         Args:
             requests_per_second: The number of tokens to add per second to the bucket.
-                Must be at least 1. The tokens represent "credit" that can be used
-                to make requests.
+                The tokens represent "credit" that can be used to make requests.
             check_every_n_seconds: check whether the tokens are available
                 every this many seconds. Can be a float to represent
                 fractions of a second.
             max_bucket_size: The maximum number of tokens that can be in the bucket.
-                This is used to prevent bursts of requests.
+                Must be at least 1. Used to prevent bursts of requests.
         """
         # Number of requests that we can make per second.
         self.requests_per_second = requests_per_second

langchain_core/retrievers.py

@@ -254,13 +254,13 @@ class BaseRetriever(RunnableSerializable[RetrieverInput, RetrieverOutput], ABC):
             run_id=kwargs.pop("run_id", None),
         )
         try:
-            _kwargs = kwargs if self._expects_other_args else {}
+            kwargs_ = kwargs if self._expects_other_args else {}
             if self._new_arg_supported:
                 result = self._get_relevant_documents(
-                    input, run_manager=run_manager, **_kwargs
+                    input, run_manager=run_manager, **kwargs_
                 )
             else:
-                result = self._get_relevant_documents(input, **_kwargs)
+                result = self._get_relevant_documents(input, **kwargs_)
         except Exception as e:
             run_manager.on_retriever_error(e)
             raise
@@ -318,13 +318,13 @@ class BaseRetriever(RunnableSerializable[RetrieverInput, RetrieverOutput], ABC):
             run_id=kwargs.pop("run_id", None),
         )
         try:
-            _kwargs = kwargs if self._expects_other_args else {}
+            kwargs_ = kwargs if self._expects_other_args else {}
             if self._new_arg_supported:
                 result = await self._aget_relevant_documents(
-                    input, run_manager=run_manager, **_kwargs
+                    input, run_manager=run_manager, **kwargs_
                 )
             else:
-                result = await self._aget_relevant_documents(input, **_kwargs)
+                result = await self._aget_relevant_documents(input, **kwargs_)
         except Exception as e:
             await run_manager.on_retriever_error(e)
             raise

langchain_core/runnables/base.py

@@ -108,7 +108,7 @@ if TYPE_CHECKING:
 Other = TypeVar("Other")
 
 
-class Runnable(Generic[Input, Output], ABC):
+class Runnable(ABC, Generic[Input, Output]):
     """A unit of work that can be invoked, batched, streamed, transformed and composed.
 
     Key Methods
@@ -1160,22 +1160,21 @@ class Runnable(Generic[Input, Output], ABC):
 
         A StreamEvent is a dictionary with the following schema:
 
-        - ``event``: **str** - Event names are of the
-            format: on_[runnable_type]_(start|stream|end).
+        - ``event``: **str** - Event names are of the format:
+          on_[runnable_type]_(start|stream|end).
         - ``name``: **str** - The name of the Runnable that generated the event.
-        - ``run_id``: **str** - randomly generated ID associated with the given execution of
-            the Runnable that emitted the event.
-            A child Runnable that gets invoked as part of the execution of a
-            parent Runnable is assigned its own unique ID.
-        - ``parent_ids``: **list[str]** - The IDs of the parent runnables that
-            generated the event. The root Runnable will have an empty list.
-            The order of the parent IDs is from the root to the immediate parent.
-            Only available for v2 version of the API. The v1 version of the API
-            will return an empty list.
+        - ``run_id``: **str** - randomly generated ID associated with the given
+          execution of the Runnable that emitted the event. A child Runnable that gets
+          invoked as part of the execution of a parent Runnable is assigned its own
+          unique ID.
+        - ``parent_ids``: **list[str]** - The IDs of the parent runnables that generated
+          the event. The root Runnable will have an empty list. The order of the parent
+          IDs is from the root to the immediate parent. Only available for v2 version of
+          the API. The v1 version of the API will return an empty list.
         - ``tags``: **Optional[list[str]]** - The tags of the Runnable that generated
-            the event.
-        - ``metadata``: **Optional[dict[str, Any]]** - The metadata of the Runnable
-            that generated the event.
+          the event.
+        - ``metadata``: **Optional[dict[str, Any]]** - The metadata of the Runnable that
+          generated the event.
         - ``data``: **dict[str, Any]**
 
 
@@ -1183,7 +1182,7 @@ class Runnable(Generic[Input, Output], ABC):
         chains. Metadata fields have been omitted from the table for brevity.
         Chain definitions have been included after the table.
 
-        **ATTENTION** This reference table is for the V2 version of the schema.
+        .. NOTE:: This reference table is for the V2 version of the schema.
 
         +----------------------+------------------+---------------------------------+-----------------------------------------------+-------------------------------------------------+
         | event                | name             | chunk                           | input                                         | output                                          |
@@ -4205,6 +4204,8 @@ class RunnableGenerator(Runnable[Input, Output]):
             return False
         return False
 
+    __hash__ = None  # type: ignore[assignment]
+
     @override
     def __repr__(self) -> str:
         return f"RunnableGenerator({self.name})"
@@ -4484,10 +4485,10 @@ class RunnableLambda(Runnable[Input, Output]):
             sig = inspect.signature(func)
             if sig.return_annotation != inspect.Signature.empty:
                 # unwrap iterator types
-                if getattr(sig.return_annotation, "__origin__", None) in (
+                if getattr(sig.return_annotation, "__origin__", None) in {
                     collections.abc.Iterator,
                     collections.abc.AsyncIterator,
-                ):
+                }:
                     return getattr(sig.return_annotation, "__args__", (Any,))[0]
                 return sig.return_annotation
         except ValueError:
@@ -4588,6 +4589,8 @@ class RunnableLambda(Runnable[Input, Output]):
             return False
         return False
 
+    __hash__ = None  # type: ignore[assignment]
+
     def __repr__(self) -> str:
         """A string representation of this Runnable."""
         if self._repr is None:

langchain_core/runnables/branch.py

@@ -111,7 +111,7 @@ class RunnableBranch(RunnableSerializable[Input, Output]):
             "Runnable[Input, Output]", coerce_to_runnable(cast("RunnableLike", default))
         )
 
-        _branches = []
+        branches_ = []
 
         for branch in branches[:-1]:
             if not isinstance(branch, (tuple, list)):
@@ -130,10 +130,10 @@ class RunnableBranch(RunnableSerializable[Input, Output]):
             condition, runnable = branch
             condition = cast("Runnable[Input, bool]", coerce_to_runnable(condition))
             runnable = coerce_to_runnable(runnable)
-            _branches.append((condition, runnable))
+            branches_.append((condition, runnable))
 
         super().__init__(
-            branches=_branches,
+            branches=branches_,
             default=default_,
         )  # type: ignore[call-arg]
 

langchain_core/runnables/graph.py

@@ -361,7 +361,7 @@ class Graph:
         """
         self.nodes.pop(node.id)
         self.edges = [
-            edge for edge in self.edges if node.id not in (edge.source, edge.target)
+            edge for edge in self.edges if node.id not in {edge.source, edge.target}
         ]
 
     def add_edge(

langchain_core/runnables/history.py

@@ -339,10 +339,10 @@ class RunnableWithMessageHistory(RunnableBindingBase):
         ).with_config(run_name="RunnableWithMessageHistory")
 
         if history_factory_config:
-            _config_specs = history_factory_config
+            config_specs = history_factory_config
         else:
             # If not provided, then we'll use the default session_id field
-            _config_specs = [
+            config_specs = [
                 ConfigurableFieldSpec(
                     id="session_id",
                     annotation=str,
@@ -359,7 +359,7 @@ class RunnableWithMessageHistory(RunnableBindingBase):
             output_messages_key=output_messages_key,
             bound=bound,
             history_messages_key=history_messages_key,
-            history_factory_config=_config_specs,
+            history_factory_config=config_specs,
             **kwargs,
         )
         self._history_chain = history_chain

langchain_core/runnables/router.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 from collections.abc import Mapping
-from itertools import starmap
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -206,7 +205,7 @@ class RouterRunnable(RunnableSerializable[RouterInput, Output]):
         configs = get_config_list(config, len(inputs))
         return await gather_with_concurrency(
             configs[0].get("max_concurrency"),
-            *starmap(ainvoke, zip(runnables, actual_inputs, configs)),
+            *map(ainvoke, runnables, actual_inputs, configs),
         )
 
     @override

langchain_core/runnables/utils.py

@@ -194,7 +194,7 @@ class IsLocalDict(ast.NodeVisitor):
             and isinstance(node.func.value, ast.Name)
             and node.func.value.id == self.name
             and node.func.attr == "get"
-            and len(node.args) in (1, 2)
+            and len(node.args) in {1, 2}
             and isinstance(node.args[0], ast.Constant)
             and isinstance(node.args[0].value, str)
         ):

langchain_core/stores.py

@@ -23,7 +23,7 @@ K = TypeVar("K")
 V = TypeVar("V")
 
 
-class BaseStore(Generic[K, V], ABC):
+class BaseStore(ABC, Generic[K, V]):
     """Abstract interface for a key-value store.
 
     This is an interface that's meant to abstract away the details of

langchain_core/sys_info.py

@@ -8,7 +8,7 @@ def _get_sub_deps(packages: Sequence[str]) -> list[str]:
     from importlib import metadata
 
     sub_deps = set()
-    _underscored_packages = {pkg.replace("-", "_") for pkg in packages}
+    underscored_packages = {pkg.replace("-", "_") for pkg in packages}
 
     for pkg in packages:
         try:
@@ -21,7 +21,7 @@ def _get_sub_deps(packages: Sequence[str]) -> list[str]:
 
         for req in required:
             cleaned_req = req.split(" ")[0]
-            if cleaned_req.replace("-", "_") not in _underscored_packages:
+            if cleaned_req.replace("-", "_") not in underscored_packages:
                 sub_deps.add(cleaned_req)
 
     return sorted(sub_deps, key=lambda x: x.lower())

langchain_core/tools/base.py

@@ -134,7 +134,7 @@ def _get_filtered_args(
         k: schema[k]
         for i, (k, param) in enumerate(valid_keys.items())
         if k not in filter_args
-        and (i > 0 or param.name not in ("self", "cls"))
+        and (i > 0 or param.name not in {"self", "cls"})
         and (include_injected or not _is_injected_arg_type(param.annotation))
     }
 
@@ -336,7 +336,7 @@ def create_schema_from_function(
     else:
         # Handle classmethods and instance methods
         existing_params: list[str] = list(sig.parameters.keys())
-        if existing_params and existing_params[0] in ("self", "cls") and in_class:
+        if existing_params and existing_params[0] in {"self", "cls"} and in_class:
             filter_args_ = [existing_params[0], *list(FILTERED_ARGS)]
         else:
             filter_args_ = list(FILTERED_ARGS)
@@ -776,7 +776,7 @@ class ChildTool(BaseTool):
     def run(
         self,
         tool_input: Union[str, dict[str, Any]],
-        verbose: Optional[bool] = None,
+        verbose: Optional[bool] = None,  # noqa: FBT001
         start_color: Optional[str] = "green",
         color: Optional[str] = "green",
         callbacks: Callbacks = None,
@@ -803,7 +803,7 @@ class ChildTool(BaseTool):
             run_id: The id of the run. Defaults to None.
             config: The configuration for the tool. Defaults to None.
             tool_call_id: The id of the tool call. Defaults to None.
-            kwargs: Keyword arguments to be passed to tool callbacks
+            kwargs: Keyword arguments to be passed to tool callbacks (event handler)
 
         Returns:
             The output of the tool.
@@ -846,9 +846,9 @@ class ChildTool(BaseTool):
                     tool_input, tool_call_id
                 )
                 if signature(self._run).parameters.get("run_manager"):
-                    tool_kwargs = tool_kwargs | {"run_manager": run_manager}
+                    tool_kwargs |= {"run_manager": run_manager}
                 if config_param := _get_runnable_config_param(self._run):
-                    tool_kwargs = tool_kwargs | {config_param: config}
+                    tool_kwargs |= {config_param: config}
                 response = context.run(self._run, *tool_args, **tool_kwargs)
             if self.response_format == "content_and_artifact":
                 if not isinstance(response, tuple) or len(response) != 2:
@@ -888,7 +888,7 @@ class ChildTool(BaseTool):
     async def arun(
         self,
         tool_input: Union[str, dict],
-        verbose: Optional[bool] = None,
+        verbose: Optional[bool] = None,  # noqa: FBT001
         start_color: Optional[str] = "green",
         color: Optional[str] = "green",
         callbacks: Callbacks = None,

langchain_core/tools/structured.py

@@ -197,7 +197,14 @@ class StructuredTool(BaseTool):
             description_ = source_function.__doc__ or None
         if description_ is None and args_schema:
             if isinstance(args_schema, type) and is_basemodel_subclass(args_schema):
-                description_ = args_schema.__doc__ or None
+                description_ = args_schema.__doc__
+                if (
+                    description_
+                    and "A base class for creating Pydantic models" in description_
+                ):
+                    description_ = ""
+                elif not description_:
+                    description_ = None
             elif isinstance(args_schema, dict):
                 description_ = args_schema.get("description")
             else:

langchain_core/tracers/core.py

@@ -165,7 +165,7 @@ class _TracerCore(ABC):
         **kwargs: Any,
     ) -> Run:
         """Create a chat model run."""
-        if self._schema_format not in ("streaming_events", "original+chat"):
+        if self._schema_format not in {"streaming_events", "original+chat"}:
             # Please keep this un-implemented for backwards compatibility.
             # When it's unimplemented old tracers that use the "original" format
             # fallback on the on_llm_start method implementation if they
@@ -352,7 +352,7 @@ class _TracerCore(ABC):
 
     def _get_chain_inputs(self, inputs: Any) -> Any:
         """Get the inputs for a chain run."""
-        if self._schema_format in ("original", "original+chat"):
+        if self._schema_format in {"original", "original+chat"}:
             return inputs if isinstance(inputs, dict) else {"input": inputs}
         if self._schema_format == "streaming_events":
             return {
@@ -363,7 +363,7 @@ class _TracerCore(ABC):
 
     def _get_chain_outputs(self, outputs: Any) -> Any:
         """Get the outputs for a chain run."""
-        if self._schema_format in ("original", "original+chat"):
+        if self._schema_format in {"original", "original+chat"}:
             return outputs if isinstance(outputs, dict) else {"output": outputs}
         if self._schema_format == "streaming_events":
             return {
@@ -423,7 +423,7 @@ class _TracerCore(ABC):
         if metadata:
             kwargs.update({"metadata": metadata})
 
-        if self._schema_format in ("original", "original+chat"):
+        if self._schema_format in {"original", "original+chat"}:
             inputs = {"input": input_str}
         elif self._schema_format == "streaming_events":
             inputs = {"input": inputs}

langchain_core/tracers/event_stream.py

@@ -773,7 +773,7 @@ async def _astream_events_implementation_v1(
     run_log = RunLog(state=None)  # type: ignore[arg-type]
     encountered_start_event = False
 
-    _root_event_filter = _RootEventFilter(
+    root_event_filter = _RootEventFilter(
         include_names=include_names,
         include_types=include_types,
         include_tags=include_tags,
@@ -796,7 +796,7 @@ async def _astream_events_implementation_v1(
         with_streamed_output_list=True,
         **kwargs,
     ):
-        run_log = run_log + log
+        run_log += log
 
         if not encountered_start_event:
             # Yield the start event for the root runnable.
@@ -815,7 +815,7 @@ async def _astream_events_implementation_v1(
                 parent_ids=[],  # Not supported in v1
             )
 
-            if _root_event_filter.include_event(event, state["type"]):
+            if root_event_filter.include_event(event, state["type"]):
                 yield event
 
         paths = {
@@ -901,7 +901,7 @@ async def _astream_events_implementation_v1(
                 data=data,
                 parent_ids=[],  # Not supported in v1
             )
-            if _root_event_filter.include_event(event, state["type"]):
+            if root_event_filter.include_event(event, state["type"]):
                 yield event
 
     state = run_log.state
@@ -918,7 +918,7 @@ async def _astream_events_implementation_v1(
         },
         parent_ids=[],  # Not supported in v1
     )
-    if _root_event_filter.include_event(event, state["type"]):
+    if root_event_filter.include_event(event, state["type"]):
         yield event
 
 

langchain_core/tracers/log_stream.py

@@ -130,6 +130,8 @@ class RunLogPatch:
     def __eq__(self, other: object) -> bool:
         return isinstance(other, RunLogPatch) and self.ops == other.ops
 
+    __hash__ = None  # type: ignore[assignment]
+
 
 class RunLog(RunLogPatch):
     """Run log."""
@@ -174,6 +176,8 @@ class RunLog(RunLogPatch):
         # Then compare that the ops are the same
         return super().__eq__(other)
 
+    __hash__ = None  # type: ignore[assignment]
+
 
 T = TypeVar("T")
 
@@ -694,7 +698,7 @@ async def _astream_log_implementation(
         else:
             state = RunLog(state=None)  # type: ignore[arg-type]
             async for log in stream:
-                state = state + log
+                state += log
                 yield state
     finally:
         # Wait for the runnable to finish, if not cancelled (eg. by break)

langchain_core/utils/_merge.py

@@ -64,6 +64,8 @@ def merge_dicts(left: dict[str, Any], *others: dict[str, Any]) -> dict[str, Any]
                 merged[right_k] = merge_lists(merged[right_k], right_v)
             elif merged[right_k] == right_v:
                 continue
+            elif isinstance(merged[right_k], int):
+                merged[right_k] += right_v
             else:
                 msg = (
                     f"Additional kwargs key {right_k} already exists in left dict and "

langchain_core/utils/env.py

@@ -15,12 +15,12 @@ def env_var_is_set(env_var: str) -> bool:
     Returns:
         bool: True if the environment variable is set, False otherwise.
     """
-    return env_var in os.environ and os.environ[env_var] not in (
+    return env_var in os.environ and os.environ[env_var] not in {
         "",
         "0",
         "false",
         "False",
-    )
+    }
 
 
 def get_from_dict_or_env(

langchain_core/utils/function_calling.py

@@ -294,8 +294,6 @@ def _convert_any_typed_dicts_to_pydantic(
                     raise ValueError(msg)
                 if arg_desc := arg_descriptions.get(arg):
                     field_kwargs["description"] = arg_desc
-                else:
-                    pass
                 fields[arg] = (new_arg_type, Field_v1(**field_kwargs))
             else:
                 new_arg_type = _convert_any_typed_dicts_to_pydantic(
@@ -456,7 +454,7 @@ def convert_to_openai_function(
         oai_function = {
             k: v
             for k, v in function.items()
-            if k in ("name", "description", "parameters", "strict")
+            if k in {"name", "description", "parameters", "strict"}
         }
     # a JSON schema with title and description
     elif isinstance(function, dict) and "title" in function:
@@ -734,7 +732,7 @@ def _parse_google_docstring(
         docstring_blocks = docstring.split("\n\n")
         if error_on_invalid_docstring:
             filtered_annotations = {
-                arg for arg in args if arg not in ("run_manager", "callbacks", "return")
+                arg for arg in args if arg not in {"run_manager", "callbacks", "return"}
             }
             if filtered_annotations and (
                 len(docstring_blocks) < 2
@@ -770,8 +768,8 @@ def _parse_google_docstring(
             if ":" in line:
                 arg, desc = line.split(":", maxsplit=1)
                 arg = arg.strip()
-                arg_name, _, _annotations = arg.partition(" ")
-                if _annotations.startswith("(") and _annotations.endswith(")"):
+                arg_name, _, annotations_ = arg.partition(" ")
+                if annotations_.startswith("(") and annotations_.endswith(")"):
                     arg = arg_name
                 arg_descriptions[arg] = desc.strip()
             elif arg:

langchain_core/utils/image.py

@@ -4,7 +4,7 @@ from typing import Any
 
 
 def __getattr__(name: str) -> Any:
-    if name in ("encode_image", "image_to_data_url"):
+    if name in {"encode_image", "image_to_data_url"}:
         msg = (
             f"'{name}' has been removed for security reasons.\n\n"
             f"Usage of this utility in environments with user-input paths is a "