streamlit-nightly 1.45.1.dev20250507__py3-none-any.whl → 1.45.1.dev20250509__py3-none-any.whl

streamlit/connections/sql_connection.py

@@ -21,7 +21,7 @@ from __future__ import annotations
 
 from collections import ChainMap
 from copy import deepcopy
-from typing import TYPE_CHECKING, cast
+from typing import TYPE_CHECKING, Any, cast
 
 from streamlit.connections import BaseConnection
 from streamlit.connections.util import extract_from_dict
@@ -176,7 +176,7 @@ class SQLConnection(BaseConnection["Engine"]):
 
     """
 
-    def _connect(self, autocommit: bool = False, **kwargs) -> Engine:
+    def _connect(self, autocommit: bool = False, **kwargs: Any) -> Engine:
         import sqlalchemy
 
         kwargs = deepcopy(kwargs)
@@ -217,8 +217,7 @@ class SQLConnection(BaseConnection["Engine"]):
 
         if autocommit:
             return cast("Engine", eng.execution_options(isolation_level="AUTOCOMMIT"))
-        else:
-            return cast("Engine", eng)
+        return cast("Engine", eng)
 
     def query(
         self,
@@ -228,8 +227,8 @@ class SQLConnection(BaseConnection["Engine"]):
         ttl: float | int | timedelta | None = None,
         index_col: str | list[str] | None = None,
         chunksize: int | None = None,
-        params=None,
-        **kwargs,
+        params: Any | None = None,
+        **kwargs: Any,
     ) -> DataFrame:
         """Run a read-only query.
 
@@ -312,10 +311,10 @@ class SQLConnection(BaseConnection["Engine"]):
         )
         def _query(
             sql: str,
-            index_col=None,
-            chunksize=None,
-            params=None,
-            **kwargs,
+            index_col: str | list[str] | None = None,
+            chunksize: int | None = None,
+            params: Any | None = None,
+            **kwargs: Any,
         ) -> DataFrame:
             import pandas as pd
 

streamlit/cursor.py

@@ -91,7 +91,7 @@ class Cursor:
     def is_locked(self) -> bool:
         raise NotImplementedError()
 
-    def get_locked_cursor(self, **props) -> LockedCursor:
+    def get_locked_cursor(self, **props: Any) -> LockedCursor:
         raise NotImplementedError()
 
     @property
@@ -105,7 +105,7 @@ class Cursor:
 
 
 class RunningCursor(Cursor):
-    def __init__(self, root_container: int, parent_path: tuple[int, ...] = ()):
+    def __init__(self, root_container: int, parent_path: tuple[int, ...] = ()) -> None:
         """A moving pointer to a delta location in the app.
 
         RunningCursors auto-increment to the next available location when you
@@ -140,7 +140,7 @@ class RunningCursor(Cursor):
     def is_locked(self) -> bool:
         return False
 
-    def get_locked_cursor(self, **props) -> LockedCursor:
+    def get_locked_cursor(self, **props: Any) -> LockedCursor:
         locked_cursor = LockedCursor(
             root_container=self._root_container,
             parent_path=self._parent_path,
@@ -159,8 +159,8 @@ class LockedCursor(Cursor):
         root_container: int,
         parent_path: tuple[int, ...] = (),
         index: int = 0,
-        **props,
-    ):
+        **props: Any,
+    ) -> None:
         """A locked pointer to a location in the app.
 
         LockedCursors always point to the same location, even when you call
@@ -201,7 +201,7 @@ class LockedCursor(Cursor):
     def is_locked(self) -> bool:
         return True
 
-    def get_locked_cursor(self, **props) -> LockedCursor:
+    def get_locked_cursor(self, **props: Any) -> LockedCursor:
         self._props = props
         return self
 

streamlit/delta_generator.py

@@ -100,6 +100,8 @@ from streamlit.runtime.scriptrunner import enqueue_message as _enqueue_message
 from streamlit.runtime.scriptrunner import get_script_run_ctx
 
 if TYPE_CHECKING:
+    from types import TracebackType
+
     from google.protobuf.message import Message
 
     from streamlit.cursor import Cursor
@@ -301,9 +303,9 @@ class DeltaGenerator(
 
     def __exit__(
         self,
-        type: Any,
-        value: Any,
-        traceback: Any,
+        typ: type[BaseException] | None,
+        exc: BaseException | None,
+        tb: TracebackType | None,
     ) -> Literal[False]:
         # with block ended
 
@@ -365,7 +367,7 @@ class DeltaGenerator(
 
         return wrapper
 
-    def __deepcopy__(self, _memo):
+    def __deepcopy__(self, _memo: Any) -> DeltaGenerator:
         dg = DeltaGenerator(
             root_container=self._root_container,
             cursor=deepcopy(self._cursor),
@@ -406,8 +408,7 @@ class DeltaGenerator(
         """
         if self._provided_cursor is None:
             return cursor.get_container_cursor(self._root_container)
-        else:
-            return self._provided_cursor
+        return self._provided_cursor
 
     @property
     def _is_top_level(self) -> bool:

streamlit/delta_generator_singletons.py

@@ -52,7 +52,7 @@ class DeltaGeneratorSingleton:
         delta_generator_cls: type[DeltaGenerator],
         status_container_cls: type[StatusContainer],
         dialog_container_cls: type[Dialog],
-    ):
+    ) -> None:
         """Registers and initializes all delta-generator classes.
 
         Parameters
@@ -138,7 +138,7 @@ class ContextVarWithLazyDefault(Generic[_T]):
     the default dg (main_dg) might not exist yet when this module is imported.
     """
 
-    def __init__(self, name: str, *, default: Callable[[], _T]):
+    def __init__(self, name: str, *, default: Callable[[], _T]) -> None:
         self._name = name
         self._default = default
         self._context_var: ContextVar[_T] | None = None

streamlit/deprecation_util.py

@@ -188,7 +188,7 @@ def _create_deprecated_obj_wrapper(obj: TObj, show_warning: Callable[[], Any]) -
             return getattr(obj, attr)
 
         @staticmethod
-        def _get_magic_functions(self_cls) -> list[str]:
+        def _get_magic_functions(self_cls: type[object]) -> list[str]:
             # ignore the handful of magic functions we cannot override without
             # breaking the Wrapper.
             ignore = ("__class__", "__dict__", "__getattribute__", "__getattr__")

streamlit/elements/arrow.py

@@ -645,8 +645,7 @@ class ArrowMixin:
             )
             self.dg._enqueue("arrow_data_frame", proto)
             return cast("DataframeState", widget_state.value)
-        else:
-            return self.dg._enqueue("arrow_data_frame", proto)
+        return self.dg._enqueue("arrow_data_frame", proto)
 
     @gather_metrics("table")
     def table(self, data: Data = None) -> DeltaGenerator:
@@ -733,7 +732,7 @@ class ArrowMixin:
         return self.dg._enqueue("arrow_table", proto)
 
     @gather_metrics("add_rows")
-    def add_rows(self, data: Data = None, **kwargs) -> DeltaGenerator | None:
+    def add_rows(self, data: Data = None, **kwargs: Any) -> DeltaGenerator | None:
         """Concatenate a dataframe to the bottom of the current one.
 
         Parameters

streamlit/elements/deck_gl_json_chart.py

@@ -269,7 +269,8 @@ class PydeckMixin:
         selection_mode: Literal[
             "single-object"
         ],  # Selection mode will only be activated by on_select param; default value here to make it work with mypy
-        on_select: Literal["ignore"],  # No default value here to make it work with mypy
+        # No default value here to make it work with mypy
+        on_select: Literal["ignore"],
         key: Key | None = None,
     ) -> DeltaGenerator: ...
 
@@ -308,21 +309,26 @@ class PydeckMixin:
         - DeckGL docs: https://github.com/uber/deck.gl/tree/master/docs
         - DeckGL JSON docs: https://github.com/uber/deck.gl/tree/master/modules/json
 
-        When using this command, Mapbox provides the map tiles to render map
-        content. Note that Mapbox is a third-party product and Streamlit accepts
-        no responsibility or liability of any kind for Mapbox or for any content
-        or information made available by Mapbox.
+        When using this command, a service called Carto_ provides the map tiles to render
+        map content. If you're using advanced PyDeck features you may need to obtain
+        an API key from Carto first. You can do that as
+        ``pydeck.Deck(api_keys={"carto": YOUR_KEY})`` or by setting the CARTO_API_KEY
+        environment variable. See `PyDeck's documentation`_ for more information.
 
-        Mapbox requires users to register and provide a token before users can
-        request map tiles. Currently, Streamlit provides this token for you, but
-        this could change at any time. We strongly recommend all users create and
-        use their own personal Mapbox token to avoid any disruptions to their
-        experience. You can do this with the ``mapbox.token`` config option. The
-        use of Mapbox is governed by Mapbox's Terms of Use.
+        Another common provider for map tiles is Mapbox_. If you prefer to use that,
+        you'll need to create an account at https://mapbox.com and specify your Mapbox
+        key when creating the ``pydeck.Deck`` object. You can do that as
+        ``pydeck.Deck(api_keys={"mapbox": YOUR_KEY})`` or by setting the MAPBOX_API_KEY
+        environment variable.
 
-        To get a token for yourself, create an account at https://mapbox.com.
-        For more info on how to set config options, see
-        https://docs.streamlit.io/develop/api-reference/configuration/config.toml.
+        .. _Carto: https://carto.com
+        .. _Mapbox: https://mapbox.com
+        .. _PyDeck's documentation: https://deckgl.readthedocs.io/en/latest/deck.html
+
+        Carto and Mapbox are third-party products and Streamlit accepts no responsibility
+        or liability of any kind for Carto or Mapbox, or for any content or information
+        made available by Carto or Mapbox. The use of Carto or Mapbox is governed by
+        their respective Terms of Use.
 
         Parameters
         ----------
@@ -408,7 +414,7 @@ class PydeckMixin:
         >>>
         >>> st.pydeck_chart(
         ...     pdk.Deck(
-        ...         map_style=None,
+        ...         map_style=None,  # Use Streamlit theme to pick map style
         ...         initial_view_state=pdk.ViewState(
         ...             latitude=37.76,
         ...             longitude=-122.4,
@@ -467,7 +473,13 @@ class PydeckMixin:
         if tooltip:
             pydeck_proto.tooltip = json.dumps(tooltip)
 
-        mapbox_token = config.get_option("mapbox.token")
+        # Get the Mapbox key from the PyDeck object first, and then fallback to the
+        # old mapbox.token config option.
+
+        mapbox_token = getattr(pydeck_obj, "mapbox_key", None)
+        if mapbox_token is None or mapbox_token == "":
+            mapbox_token = config.get_option("mapbox.token")
+
         if mapbox_token:
             pydeck_proto.mapbox_token = mapbox_token
 
@@ -476,7 +488,8 @@ class PydeckMixin:
 
         if on_select not in ["ignore", "rerun"] and not callable(on_select):
             raise StreamlitAPIException(
-                f"You have passed {on_select} to `on_select`. But only 'ignore', 'rerun', or a callable is supported."
+                f"You have passed {on_select} to `on_select`. "
+                "But only 'ignore', 'rerun', or a callable is supported."
             )
 
         if is_selection_activated:

streamlit/elements/dialog_decorator.py

@@ -15,7 +15,7 @@
 from __future__ import annotations
 
 from functools import wraps
-from typing import TYPE_CHECKING, Callable, TypeVar, cast, overload
+from typing import TYPE_CHECKING, Any, Callable, TypeVar, cast, overload
 
 from streamlit.delta_generator_singletons import (
     get_dg_singleton_instance,
@@ -74,7 +74,7 @@ def _dialog_decorator(
         )
 
     @wraps(non_optional_func)
-    def wrap(*args, **kwargs) -> None:
+    def wrap(*args: Any, **kwargs: Any) -> None:
         _assert_no_nested_dialogs()
         # Call the Dialog on the event_dg because it lives outside of the normal
         # Streamlit UI flow. For example, if it is called from the sidebar, it should
@@ -109,7 +109,7 @@ def _dialog_decorator(
 
         with dialog:
             fragmented_dialog_content()
-            return None
+            return
 
     return cast("F", wrap)
 

streamlit/elements/doc_string.py

@@ -168,25 +168,25 @@ def _marshall(
     doc_string_proto.width_config.CopyFrom(width_config)
 
 
-def _get_name(obj):
+def _get_name(obj: object) -> str | None:
     # Try to get the fully-qualified name of the object.
     # For example: st.help(bar.Baz(123))
     #    The name is bar.Baz
     name = getattr(obj, "__qualname__", None)
     if name:
-        return name
+        return cast("str", name)
 
     # Try to get the name of the object.
     # For example: st.help(bar.Baz(123))
     #   The name is Baz
-    return getattr(obj, "__name__", None)
+    return cast("str | None", getattr(obj, "__name__", None))
 
 
-def _get_module(obj):
+def _get_module(obj: object) -> str | None:
     return getattr(obj, "__module__", None)
 
 
-def _get_signature(obj):
+def _get_signature(obj: object) -> str | None:
     if not inspect.isclass(obj) and not callable(obj):
         return None
 
@@ -215,7 +215,7 @@ def _get_signature(obj):
     return sig
 
 
-def _get_docstring(obj):
+def _get_docstring(obj: object) -> str | None:
     doc_string = inspect.getdoc(obj)
 
     # Sometimes an object has no docstring, but the object's type does.
@@ -239,7 +239,7 @@ def _get_docstring(obj):
     return None
 
 
-def _get_variable_name():
+def _get_variable_name() -> str | None:
     """Try to get the name of the variable in the current line, as set by the user.
 
     For example:
@@ -256,7 +256,7 @@ def _get_variable_name():
     return _get_variable_name_from_code_str(code)
 
 
-def _get_variable_name_from_code_str(code):
+def _get_variable_name_from_code_str(code: str) -> str | None:
     tree = ast.parse(code)
 
     # Example:
@@ -295,7 +295,7 @@ def _get_variable_name_from_code_str(code):
 
     # If walrus, get name.
     # E.g. st.help(foo := 123) should give you "foo".
-    elif type(arg_node) is ast.NamedExpr:
+    if type(arg_node) is ast.NamedExpr:
         # This next "if" will always be true, but need to add this for the type-checking test to
         # pass.
         if type(arg_node.target) is ast.Name:
@@ -329,7 +329,7 @@ def _get_variable_name_from_code_str(code):
 _NEWLINES = re.compile(r"[\n\r]+")
 
 
-def _get_current_line_of_code_as_str():
+def _get_current_line_of_code_as_str() -> str | None:
     scriptrunner_frame = _get_scriptrunner_frame()
 
     if scriptrunner_frame is None:
@@ -352,7 +352,7 @@ def _get_current_line_of_code_as_str():
     return re.sub(_NEWLINES, "", code_as_string.strip())
 
 
-def _get_scriptrunner_frame():
+def _get_scriptrunner_frame() -> inspect.FrameInfo | None:
     prev_frame = None
     scriptrunner_frame = None
 
@@ -375,7 +375,7 @@ def _get_scriptrunner_frame():
     return scriptrunner_frame
 
 
-def _is_stcommand(tree, command_name):
+def _is_stcommand(tree: Any, command_name: str) -> bool:
     """Checks whether the AST in tree is a call for command_name."""
     root_node = tree.body[0].value
 
@@ -391,25 +391,25 @@ def _is_stcommand(tree, command_name):
     )
 
 
-def _get_stcommand_arg(tree):
+def _get_stcommand_arg(tree: ast.Module) -> ast.expr | None:
     """Gets the argument node for the st command in tree (AST)."""
 
-    root_node = tree.body[0].value
+    root_node = tree.body[0].value  # type: ignore
 
     if root_node.args:
-        return root_node.args[0]
+        return cast("ast.expr", root_node.args[0])
 
     return None
 
 
-def _get_type_as_str(obj):
+def _get_type_as_str(obj: object) -> str:
     if inspect.isclass(obj):
         return "class"
 
     return str(type(obj).__name__)
 
 
-def _get_first_line(text):
+def _get_first_line(text: str) -> str:
     if not text:
         return ""
 
@@ -417,7 +417,7 @@ def _get_first_line(text):
     return left
 
 
-def _get_weight(value):
+def _get_weight(value: Any) -> int:
     if inspect.ismodule(value):
         return 3
     if inspect.isclass(value):
@@ -485,12 +485,12 @@ def _get_human_readable_value(value):
     return _shorten(value_str)
 
 
-def _shorten(s, length=300):
+def _shorten(s: str, length: int = 300) -> str:
     s = s.strip()
     return s[:length] + "..." if len(s) > length else s
 
 
-def _is_computed_property(obj, attr_name):
+def _is_computed_property(obj: object, attr_name: str) -> bool:
     obj_class = getattr(obj, "__class__", None)
 
     if not obj_class:

streamlit/elements/layouts.py

@@ -28,6 +28,8 @@ from streamlit.errors import (
     StreamlitInvalidVerticalAlignmentError,
 )
 from streamlit.proto.Block_pb2 import Block as BlockProto
+from streamlit.proto.GapSize_pb2 import GapConfig, GapSize
+from streamlit.proto.HeightConfig_pb2 import HeightConfig
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.string_util import validate_icon_or_emoji
 
@@ -149,17 +151,21 @@ class LayoutsMixin:
         key = to_key(key)
         block_proto = BlockProto()
         block_proto.allow_empty = False
-        block_proto.vertical.border = border or False
+        block_proto.flex_container.border = border or False
+        block_proto.flex_container.wrap = False
 
         if height:
             # Activate scrolling container behavior:
             block_proto.allow_empty = True
-            block_proto.vertical.height = height
+
+            height_config = HeightConfig()
+            height_config.pixel_height = height
+            block_proto.flex_container.height_config.CopyFrom(height_config)
             if border is None:
                 # If border is None, we activated the
                 # border as default setting for scrolling
                 # containers.
-                block_proto.vertical.border = True
+                block_proto.flex_container.border = True
 
         if key:
             # At the moment, the ID is only used for extracting the
@@ -348,22 +354,30 @@ class LayoutsMixin:
                 vertical_alignment=vertical_alignment
             )
 
-        def column_gap(gap):
+        def column_gap(gap: str) -> GapSize.ValueType:
+            gap_mapping = {
+                "small": GapSize.SMALL,
+                "medium": GapSize.MEDIUM,
+                "large": GapSize.LARGE,
+            }
+
             if isinstance(gap, str):
                 gap_size = gap.lower()
                 valid_sizes = ["small", "medium", "large"]
 
                 if gap_size in valid_sizes:
-                    return gap_size
+                    return gap_mapping[gap_size]
 
             raise StreamlitInvalidColumnGapError(gap=gap)
 
         gap_size = column_gap(gap)
+        gap_config = GapConfig()
+        gap_config.gap_size = gap_size
 
         def column_proto(normalized_weight: float) -> BlockProto:
             col_proto = BlockProto()
             col_proto.column.weight = normalized_weight
-            col_proto.column.gap = gap_size
+            col_proto.column.gap_config.CopyFrom(gap_config)
             col_proto.column.vertical_alignment = vertical_alignment_mapping[
                 vertical_alignment
             ]
@@ -372,7 +386,12 @@ class LayoutsMixin:
             return col_proto
 
         block_proto = BlockProto()
-        block_proto.horizontal.gap = gap_size
+        block_proto.flex_container.direction = (
+            BlockProto.FlexContainer.Direction.HORIZONTAL
+        )
+        block_proto.flex_container.wrap = True
+        block_proto.flex_container.gap_config.CopyFrom(gap_config)
+        block_proto.flex_container.scale = 1
         row = self.dg._block(block_proto)
         total_weight = sum(weights)
         return [row._block(column_proto(w / total_weight)) for w in weights]