benchling-sdk 1.9.0a5__py3-none-any.whl → 1.10.0__py3-none-any.whl

benchling_sdk/apps/{helpers/session_helpers.py → status/framework.py}

@@ -3,15 +3,20 @@ from __future__ import annotations
 from abc import ABC, abstractmethod
 from contextlib import AbstractContextManager
 from types import TracebackType
-from typing import cast, Generic, Iterable, List, Optional, Protocol, Type, Union
+from typing import cast, Iterable, List, Optional, Protocol, Type, TYPE_CHECKING, Union
 
 from benchling_api_client.v2.stable.types import Unset, UNSET
 
-from benchling_sdk.apps.framework import AppType
+from benchling_sdk.apps.status.errors import (
+    AppUserFacingError,
+    InvalidSessionTimeoutError,
+    MissingAttachedCanvasError,
+    SessionClosedError,
+    SessionContextClosedError,
+)
 from benchling_sdk.errors import BenchlingError
-from benchling_sdk.helpers.logging_helpers import log_stability_warning, sdk_logger, StabilityLevel
+from benchling_sdk.helpers.logging_helpers import sdk_logger
 from benchling_sdk.models import (
-    AaSequence,
     AppCanvasUpdate,
     AppSession,
     AppSessionCreate,
@@ -22,131 +27,13 @@ from benchling_sdk.models import (
     AppSessionUpdateStatus,
     BadRequestError,
     BadRequestErrorBulkError,
-    Blob,
-    Box,
-    Container,
-    CustomEntity,
-    DnaOligo,
-    DnaSequence,
-    Entry,
-    Folder,
-    Location,
-    Mixture,
-    Molecule,
-    Plate,
-    Request,
-    RnaOligo,
-    RnaSequence,
-    WorkflowOutput,
-    WorkflowTask,
 )
 
-# Taken from CHIP_SUPPORTED_COLUMN_TYPES in Benchling server
-# Anything we miss, they can still embed the ID themselves in a message
-_ReferencedSessionLinkType = Union[
-    AaSequence,
-    Blob,
-    Box,
-    Container,
-    CustomEntity,
-    DnaSequence,
-    DnaOligo,
-    Entry,
-    Folder,
-    Location,
-    Mixture,
-    Plate,
-    RnaOligo,
-    Molecule,
-    RnaSequence,
-    Request,
-    WorkflowOutput,
-    WorkflowTask,
-]
+if TYPE_CHECKING:
+    from benchling_sdk.apps.framework import App
 
 _DEFAULT_APP_ERROR_MESSAGE = "An unexpected error occurred in the app"
 
-log_stability_warning(StabilityLevel.BETA)
-
-
-class SessionClosedError(Exception):
-    """
-    Session Closed Error.
-
-    A session was inoperable because its status in Benchling was terminal.
-    """
-
-    pass
-
-
-class SessionContextClosedError(Exception):
-    """
-    Session Context Closed Error.
-
-    An operation was attempted using the session context manager after it was closed.
-    """
-
-    pass
-
-
-class InvalidSessionTimeoutError(Exception):
-    """
-    Invalid Session Timeout Error.
-
-    A session's timeout value was set to an invalid value.
-    """
-
-    pass
-
-
-class MissingAttachedCanvasError(Exception):
-    """
-    Missing Attached Canvas Error.
-
-    A canvas operation was requested, but a session context has no attached canvas.
-    """
-
-    pass
-
-
-class AppUserFacingError(Exception):
-    """
-    App User Facing Error.
-
-    Extend this class with custom exceptions you want to be written back to the user as a SessionMessage.
-
-    SessionClosingContextExitHandler will invoke messages() and write them to a user. Callers choosing to
-    write their own SessionContextExitHandler may need to replicate this behavior themselves.
-
-    This is useful for control flow where an app wants to terminate with an error state that is resolvable
-    by the user.
-
-    For example:
-
-    class InvalidUserInputError(AppUserFacingError):
-        pass
-
-    raise InvalidUserInputError("Please enter a number between 1 and 10")
-
-    This would create a message shown to the user like:
-
-    AppSessionMessageCreate("Please enter a number between 1 and 10", style=AppSessionMessageStyle.ERROR)
-    """
-
-    _messages: List[str]
-
-    def __init__(self, messages: Union[str, List[str]], *args) -> None:
-        """Initialize an AppUserFacingError with one message or a list."""
-        self._messages = [messages] if isinstance(messages, str) else messages
-        super().__init__(args)
-
-    def messages(self) -> List[AppSessionMessageCreate]:
-        """Create a series of AppSessionMessageCreate to write to a Session and displayed to the user."""
-        return [
-            AppSessionMessageCreate(content=message, style=AppSessionMessageStyle.ERROR)
-            for message in self._messages
-        ]
-
 
 class SessionProvider(Protocol):
     """Provide a Benchling App Session to convey app status."""
@@ -166,7 +53,7 @@ class SessionContextErrorProcessor(ABC):
     process_error() does not assume a context's session is active. If using the session with process_error(),
     this should be checked.
 
-    For user with SessionClosingContextExitHandler or its subclasses.
+    For use with SessionClosingContextExitHandler or its subclasses.
     """
 
     @classmethod
@@ -181,7 +68,7 @@ class SessionContextErrorProcessor(ABC):
     @abstractmethod
     def process_error(
         cls,
-        context: SessionContextManager[AppType],
+        context: SessionContextManager,
         exc_type: Type[BaseException],
         exc_value: BaseException,
         exc_traceback: TracebackType,
@@ -220,7 +107,7 @@ class AppUserFacingErrorProcessor(SessionContextErrorProcessor):
     @classmethod
     def process_error(
         cls,
-        context: SessionContextManager[AppType],
+        context: SessionContextManager,
         exc_type: Type[BaseException],
         exc_value: BaseException,
         exc_traceback: TracebackType,
@@ -269,7 +156,7 @@ class BenchlingBadRequestErrorProcessor(AppUserFacingErrorProcessor):
     @classmethod
     def process_error(
         cls,
-        context: SessionContextManager[AppType],
+        context: SessionContextManager,
         exc_type: Type[BaseException],
         exc_value: BaseException,
         exc_traceback: TracebackType,
@@ -318,7 +205,7 @@ class BenchlingBadRequestErrorProcessor(AppUserFacingErrorProcessor):
         return messages
 
 
-class SessionContextEnterHandler(ABC, Generic[AppType]):
+class SessionContextEnterHandler(ABC):
     """
     Session Context Enter Handler.
 
@@ -326,12 +213,12 @@ class SessionContextEnterHandler(ABC, Generic[AppType]):
     """
 
     @abstractmethod
-    def on_enter(self, context: SessionContextManager[AppType]) -> None:
+    def on_enter(self, context: SessionContextManager) -> None:
         """Perform on session context enter after a Session has been started with Benchling."""
         pass
 
 
-class SessionContextExitHandler(ABC, Generic[AppType]):
+class SessionContextExitHandler(ABC):
     """
     Session Context Exit Handler.
 
@@ -339,7 +226,7 @@ class SessionContextExitHandler(ABC, Generic[AppType]):
     """
 
     @abstractmethod
-    def on_success(self, context: SessionContextManager[AppType]) -> bool:
+    def on_success(self, context: SessionContextManager) -> bool:
         """
         Perform on session context exit when no errors are present.
 
@@ -354,7 +241,7 @@ class SessionContextExitHandler(ABC, Generic[AppType]):
     @abstractmethod
     def on_error(
         self,
-        context: SessionContextManager[AppType],
+        context: SessionContextManager,
         exc_type: Type[BaseException],
         exc_value: BaseException,
         exc_traceback: TracebackType,
@@ -382,6 +269,7 @@ class SessionClosingContextExitHandler(SessionContextExitHandler):
 
     On error: close the Session with Benchling with SessionStatusUpdate.FAILED.
     If error_messages were specified, those will also be written to the Session when setting the status.
+    If the session has a Canvas attached and enable_attached_canvas_on_error is True, re-enable the AppCanvas.
 
     As a best practice, error messages are not written to the session to avoid leaking
     potentially sensitive information.
@@ -390,6 +278,7 @@ class SessionClosingContextExitHandler(SessionContextExitHandler):
     _success_messages: Optional[Iterable[AppSessionMessageCreate]]
     _error_messages: Optional[Iterable[AppSessionMessageCreate]]
     _error_processors: List[Type[SessionContextErrorProcessor]]
+    _enable_attached_canvas_on_error: bool
 
     def __init__(
         self,
@@ -401,6 +290,7 @@ class SessionClosingContextExitHandler(SessionContextExitHandler):
             AppUserFacingErrorProcessor,
             BenchlingBadRequestErrorProcessor,
         ],
+        enable_attached_canvas_on_error: bool = True,
     ):
         """
         Init Session Closing Context Exit Handler.
@@ -411,8 +301,9 @@ class SessionClosingContextExitHandler(SessionContextExitHandler):
         self._success_messages = success_messages
         self._error_messages = error_messages
         self._error_processors = error_processors
+        self._enable_attached_canvas_on_error = enable_attached_canvas_on_error
 
-    def on_success(self, context: SessionContextManager[AppType]) -> bool:
+    def on_success(self, context: SessionContextManager) -> bool:
         """
         Close Active Session on Success.
 
@@ -443,7 +334,7 @@ class SessionClosingContextExitHandler(SessionContextExitHandler):
 
     def on_error(
         self,
-        context: SessionContextManager[AppType],
+        context: SessionContextManager,
         exc_type: Type[BaseException],
         exc_value: BaseException,
         exc_traceback: TracebackType,
@@ -467,7 +358,9 @@ class SessionClosingContextExitHandler(SessionContextExitHandler):
             # Pluggable specific error handling
             if optional_error_processor:
                 sdk_logger.debug("Exiting session context with error, matched handler for %s", str(exc_type))
-                return optional_error_processor.process_error(context, exc_type, exc_value, exc_traceback)
+                result = optional_error_processor.process_error(context, exc_type, exc_value, exc_traceback)
+                self._check_and_enable_canvas(context)
+                return result
             # Fallback to general error handling
             sdk_logger.debug(
                 "Exiting session context with error, automatically closing session ID %s as %s",
@@ -481,10 +374,17 @@ class SessionClosingContextExitHandler(SessionContextExitHandler):
                 context.close_session(AppSessionUpdateStatus.FAILED, messages=message_creates)
             else:
                 context.close_session(AppSessionUpdateStatus.FAILED)
+            self._check_and_enable_canvas(context)
         else:
             sdk_logger.debug("Exiting session context, no active session present")
         return False
 
+    def _check_and_enable_canvas(self, context: SessionContextManager) -> None:
+        canvas_id = context.attached_canvas_id()
+        if self._enable_attached_canvas_on_error and canvas_id:
+            context.app.benchling.apps.update_canvas(canvas_id, AppCanvasUpdate(enabled=True))
+            sdk_logger.debug("Re-enabled canvas with ID %s", context.attached_canvas_id())
+
     def _error_processor_type(
         self,
         exc_type: Type[BaseException],
@@ -497,7 +397,7 @@ class SessionClosingContextExitHandler(SessionContextExitHandler):
         return None
 
 
-def create_session_provider(app: AppType, name: str, timeout_seconds: int) -> SessionProvider:
+def create_session_provider(app: App, name: str, timeout_seconds: int) -> SessionProvider:
     """
     Create Session Provider.
 
@@ -512,7 +412,7 @@ def create_session_provider(app: AppType, name: str, timeout_seconds: int) -> Se
     return _new_session
 
 
-def existing_session_provider(app: AppType, session_id: str) -> SessionProvider:
+def existing_session_provider(app: App, session_id: str) -> SessionProvider:
     """
     Existing Session Provider.
 
@@ -537,7 +437,7 @@ def _ordered_messages(messages: Iterable[AppSessionMessageCreate]) -> List[AppSe
     return list(messages)
 
 
-class SessionContextManager(AbstractContextManager, Generic[AppType]):
+class SessionContextManager(AbstractContextManager):
     """
     Manage Benchling App Session.
 
@@ -546,19 +446,19 @@ class SessionContextManager(AbstractContextManager, Generic[AppType]):
     success_exit_handler.
     """
 
-    _app: AppType
+    _app: App
     _session_provider: SessionProvider
-    _context_enter_handler: Optional[SessionContextEnterHandler[AppType]]
-    _context_exit_handler: SessionContextExitHandler[AppType]
+    _context_enter_handler: Optional[SessionContextEnterHandler]
+    _context_exit_handler: SessionContextExitHandler
     _session: Optional[AppSession]
     _attached_canvas_id: Optional[str]
 
     def __init__(
         self,
-        app: AppType,
+        app: App,
         session_provider: SessionProvider,
-        context_enter_handler: Optional[SessionContextEnterHandler[AppType]] = None,
-        context_exit_handler: Optional[SessionContextExitHandler[AppType]] = None,
+        context_enter_handler: Optional[SessionContextEnterHandler] = None,
+        context_exit_handler: Optional[SessionContextExitHandler] = None,
     ):
         """
         Initialize SessionContextManager.
@@ -595,7 +495,7 @@ class SessionContextManager(AbstractContextManager, Generic[AppType]):
         return self._context_exit_handler.on_success(self)
 
     @property
-    def app(self) -> AppType:
+    def app(self) -> App:
         """Return the app for the session."""
         return self._app
 
@@ -711,6 +611,14 @@ class SessionContextManager(AbstractContextManager, Generic[AppType]):
         """
         return self._session is not None
 
+    def attached_canvas_id(self) -> Optional[str]:
+        """
+        Attached Canvas Id.
+
+        Returns a canvas_id associated with the active session, if one has been attached via `attach_canvas`.
+        """
+        return self._attached_canvas_id
+
     def attach_canvas(self, canvas_id: str) -> None:
         """
         Attach Canvas.
@@ -759,12 +667,12 @@ class SessionContextManager(AbstractContextManager, Generic[AppType]):
 
 
 def new_session_context(
-    app: AppType,
+    app: App,
     name: str,
     timeout_seconds: int,
     context_enter_handler: Optional[SessionContextEnterHandler] = None,
     context_exit_handler: Optional[SessionContextExitHandler] = None,
-) -> SessionContextManager[AppType]:
+) -> SessionContextManager:
     """
     Create New Session Context.
 
@@ -779,11 +687,11 @@ def new_session_context(
 
 
 def continue_session_context(
-    app: AppType,
+    app: App,
     session_id: str,
     context_enter_handler: Optional[SessionContextEnterHandler] = None,
     context_exit_handler: Optional[SessionContextExitHandler] = None,
-) -> SessionContextManager[AppType]:
+) -> SessionContextManager:
     """
     Continue Session Context.
 
@@ -795,20 +703,3 @@ def continue_session_context(
         context_enter_handler,
         context_exit_handler,
     )
-
-
-def ref(reference: _ReferencedSessionLinkType) -> str:
-    """
-    Ref.
-
-    Helper method for easily serializing a referenced object into a string embeddable in AppSessionMessageCreate
-    content.
-
-    Example:
-        dna_sequence = benchling.dna_sequences.get_by_id("seq_1234")
-        AppSessionMessageCreate(f"This is my DNA sequence {ref(dna_sequence)} for analysis"
-    """
-    # Not sure {} are possible in Benchling IDs, but the spec says we're escaping
-    unescape_id = reference.id
-    escaped_id = unescape_id.replace("{", "\\{").replace("}", "\\}")
-    return f"{{id:{escaped_id}}}"

benchling_sdk/apps/status/helpers.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+
+from benchling_sdk.apps.status.types import ReferencedSessionLinkType
+
+
+def ref(reference: ReferencedSessionLinkType) -> str:
+    """
+    Ref.
+
+    Helper method for easily serializing a referenced object into a string embeddable in AppSessionMessageCreate
+    content.
+
+    Example:
+        dna_sequence = benchling.dna_sequences.get_by_id("seq_1234")
+        AppSessionMessageCreate(f"This is my DNA sequence {ref(dna_sequence)} for analysis"
+    """
+    # Not sure {} are possible in Benchling IDs, but the spec says we're escaping
+    unescape_id = reference.id
+    escaped_id = unescape_id.replace("{", "\\{").replace("}", "\\}")
+    return f"{{id:{escaped_id}}}"

benchling_sdk/apps/status/types.py

@@ -0,0 +1,45 @@
+from typing import Union
+
+from benchling_sdk.models import (
+    AaSequence,
+    Blob,
+    Box,
+    Container,
+    CustomEntity,
+    DnaOligo,
+    DnaSequence,
+    Entry,
+    Folder,
+    Location,
+    Mixture,
+    Molecule,
+    Plate,
+    Request,
+    RnaOligo,
+    RnaSequence,
+    WorkflowOutput,
+    WorkflowTask,
+)
+
+# Taken from CHIP_SUPPORTED_COLUMN_TYPES in Benchling server
+# Anything we miss, they can still embed the ID themselves in a message
+ReferencedSessionLinkType = Union[
+    AaSequence,
+    Blob,
+    Box,
+    Container,
+    CustomEntity,
+    DnaSequence,
+    DnaOligo,
+    Entry,
+    Folder,
+    Location,
+    Mixture,
+    Plate,
+    RnaOligo,
+    Molecule,
+    RnaSequence,
+    Request,
+    WorkflowOutput,
+    WorkflowTask,
+]

benchling_sdk/apps/types.py

@@ -0,0 +1,3 @@
+from typing import Any, Dict, List, Union
+
+JsonType = Union[Dict[str, Any], List[Any], str, int, float, bool]

benchling_sdk/errors.py

@@ -183,8 +183,8 @@ class AppSessionClosedError(ExtendedBenchlingErrorBase):
 
 
 @dataclass
-class InvalidDatasetError(Exception):
-    """A general error related to Benchling Datasets."""
+class InvalidDataFrameError(Exception):
+    """A general error related to Benchling DataFrames."""
 
     message: str
 
@@ -193,8 +193,8 @@ class InvalidDatasetError(Exception):
 
 
 @dataclass
-class DatasetInProgressError(Exception):
-    """An error for Benchling datasets for unavailable data operations on a Dataset in progress."""
+class DataFrameInProgressError(Exception):
+    """An error for Benchling data frames for unavailable data operations on a DataFrame in progress."""
 
     message: str
 

benchling_sdk/models/__init__.py

@@ -632,6 +632,9 @@ __all__ = [
     "MarkdownUiBlockCreate",
     "MarkdownUiBlockType",
     "MarkdownUiBlockUpdate",
+    "MatchBasesRequest",
+    "MatchBasesRequestArchiveReason",
+    "MatchBasesRequestSort",
     "Measurement",
     "Mixture",
     "MixtureBulkUpdate",
@@ -845,6 +848,9 @@ __all__ = [
     "SchemaLinkFieldDefinition",
     "SchemaLinkFieldDefinitionType",
     "SchemaSummary",
+    "SearchBasesRequest",
+    "SearchBasesRequestArchiveReason",
+    "SearchBasesRequestSort",
     "SearchInputMultiValueUiBlock",
     "SearchInputMultiValueUiBlockCreate",
     "SearchInputMultiValueUiBlockType",
@@ -888,6 +894,8 @@ __all__ = [
     "TableNotePartType",
     "TableUiBlock",
     "TableUiBlockCreate",
+    "TableUiBlockDataFrameSource",
+    "TableUiBlockDataFrameSourceType",
     "TableUiBlockDatasetSource",
     "TableUiBlockDatasetSourceType",
     "TableUiBlockSource",
@@ -1641,6 +1649,9 @@ if TYPE_CHECKING:
     import benchling_api_client.v2.stable.models.markdown_ui_block_create
     import benchling_api_client.v2.stable.models.markdown_ui_block_type
     import benchling_api_client.v2.stable.models.markdown_ui_block_update
+    import benchling_api_client.v2.stable.models.match_bases_request
+    import benchling_api_client.v2.stable.models.match_bases_request_archive_reason
+    import benchling_api_client.v2.stable.models.match_bases_request_sort
     import benchling_api_client.v2.stable.models.measurement
     import benchling_api_client.v2.stable.models.mixture
     import benchling_api_client.v2.stable.models.mixture_bulk_update
@@ -1854,6 +1865,9 @@ if TYPE_CHECKING:
     import benchling_api_client.v2.stable.models.schema_link_field_definition
     import benchling_api_client.v2.stable.models.schema_link_field_definition_type
     import benchling_api_client.v2.stable.models.schema_summary
+    import benchling_api_client.v2.stable.models.search_bases_request
+    import benchling_api_client.v2.stable.models.search_bases_request_archive_reason
+    import benchling_api_client.v2.stable.models.search_bases_request_sort
     import benchling_api_client.v2.stable.models.search_input_multi_value_ui_block
     import benchling_api_client.v2.stable.models.search_input_multi_value_ui_block_create
     import benchling_api_client.v2.stable.models.search_input_multi_value_ui_block_type
@@ -1897,6 +1911,8 @@ if TYPE_CHECKING:
     import benchling_api_client.v2.stable.models.table_note_part_type
     import benchling_api_client.v2.stable.models.table_ui_block
     import benchling_api_client.v2.stable.models.table_ui_block_create
+    import benchling_api_client.v2.stable.models.table_ui_block_data_frame_source
+    import benchling_api_client.v2.stable.models.table_ui_block_data_frame_source_type
     import benchling_api_client.v2.stable.models.table_ui_block_dataset_source
     import benchling_api_client.v2.stable.models.table_ui_block_dataset_source_type
     import benchling_api_client.v2.stable.models.table_ui_block_source
@@ -3324,6 +3340,13 @@ if TYPE_CHECKING:
     MarkdownUiBlockUpdate = (
         benchling_api_client.v2.stable.models.markdown_ui_block_update.MarkdownUiBlockUpdate
     )
+    MatchBasesRequest = benchling_api_client.v2.stable.models.match_bases_request.MatchBasesRequest
+    MatchBasesRequestArchiveReason = (
+        benchling_api_client.v2.stable.models.match_bases_request_archive_reason.MatchBasesRequestArchiveReason
+    )
+    MatchBasesRequestSort = (
+        benchling_api_client.v2.stable.models.match_bases_request_sort.MatchBasesRequestSort
+    )
     Measurement = benchling_api_client.v2.stable.models.measurement.Measurement
     Mixture = benchling_api_client.v2.stable.models.mixture.Mixture
     MixtureBulkUpdate = benchling_api_client.v2.stable.models.mixture_bulk_update.MixtureBulkUpdate
@@ -3743,6 +3766,13 @@ if TYPE_CHECKING:
         benchling_api_client.v2.stable.models.schema_link_field_definition_type.SchemaLinkFieldDefinitionType
     )
     SchemaSummary = benchling_api_client.v2.stable.models.schema_summary.SchemaSummary
+    SearchBasesRequest = benchling_api_client.v2.stable.models.search_bases_request.SearchBasesRequest
+    SearchBasesRequestArchiveReason = (
+        benchling_api_client.v2.stable.models.search_bases_request_archive_reason.SearchBasesRequestArchiveReason
+    )
+    SearchBasesRequestSort = (
+        benchling_api_client.v2.stable.models.search_bases_request_sort.SearchBasesRequestSort
+    )
     SearchInputMultiValueUiBlock = (
         benchling_api_client.v2.stable.models.search_input_multi_value_ui_block.SearchInputMultiValueUiBlock
     )
@@ -3844,6 +3874,12 @@ if TYPE_CHECKING:
     TableNotePartType = benchling_api_client.v2.stable.models.table_note_part_type.TableNotePartType
     TableUiBlock = benchling_api_client.v2.stable.models.table_ui_block.TableUiBlock
     TableUiBlockCreate = benchling_api_client.v2.stable.models.table_ui_block_create.TableUiBlockCreate
+    TableUiBlockDataFrameSource = (
+        benchling_api_client.v2.stable.models.table_ui_block_data_frame_source.TableUiBlockDataFrameSource
+    )
+    TableUiBlockDataFrameSourceType = (
+        benchling_api_client.v2.stable.models.table_ui_block_data_frame_source_type.TableUiBlockDataFrameSourceType
+    )
     TableUiBlockDatasetSource = (
         benchling_api_client.v2.stable.models.table_ui_block_dataset_source.TableUiBlockDatasetSource
     )
@@ -4749,6 +4785,9 @@ else:
         "MarkdownUiBlockCreate": "benchling_api_client.v2.stable.models.markdown_ui_block_create",
         "MarkdownUiBlockType": "benchling_api_client.v2.stable.models.markdown_ui_block_type",
         "MarkdownUiBlockUpdate": "benchling_api_client.v2.stable.models.markdown_ui_block_update",
+        "MatchBasesRequest": "benchling_api_client.v2.stable.models.match_bases_request",
+        "MatchBasesRequestArchiveReason": "benchling_api_client.v2.stable.models.match_bases_request_archive_reason",
+        "MatchBasesRequestSort": "benchling_api_client.v2.stable.models.match_bases_request_sort",
         "Measurement": "benchling_api_client.v2.stable.models.measurement",
         "Mixture": "benchling_api_client.v2.stable.models.mixture",
         "MixtureBulkUpdate": "benchling_api_client.v2.stable.models.mixture_bulk_update",
@@ -4962,6 +5001,9 @@ else:
         "SchemaLinkFieldDefinition": "benchling_api_client.v2.stable.models.schema_link_field_definition",
         "SchemaLinkFieldDefinitionType": "benchling_api_client.v2.stable.models.schema_link_field_definition_type",
         "SchemaSummary": "benchling_api_client.v2.stable.models.schema_summary",
+        "SearchBasesRequest": "benchling_api_client.v2.stable.models.search_bases_request",
+        "SearchBasesRequestArchiveReason": "benchling_api_client.v2.stable.models.search_bases_request_archive_reason",
+        "SearchBasesRequestSort": "benchling_api_client.v2.stable.models.search_bases_request_sort",
         "SearchInputMultiValueUiBlock": "benchling_api_client.v2.stable.models.search_input_multi_value_ui_block",
         "SearchInputMultiValueUiBlockCreate": "benchling_api_client.v2.stable.models.search_input_multi_value_ui_block_create",
         "SearchInputMultiValueUiBlockType": "benchling_api_client.v2.stable.models.search_input_multi_value_ui_block_type",
@@ -5005,6 +5047,8 @@ else:
         "TableNotePartType": "benchling_api_client.v2.stable.models.table_note_part_type",
         "TableUiBlock": "benchling_api_client.v2.stable.models.table_ui_block",
         "TableUiBlockCreate": "benchling_api_client.v2.stable.models.table_ui_block_create",
+        "TableUiBlockDataFrameSource": "benchling_api_client.v2.stable.models.table_ui_block_data_frame_source",
+        "TableUiBlockDataFrameSourceType": "benchling_api_client.v2.stable.models.table_ui_block_data_frame_source_type",
         "TableUiBlockDatasetSource": "benchling_api_client.v2.stable.models.table_ui_block_dataset_source",
         "TableUiBlockDatasetSourceType": "benchling_api_client.v2.stable.models.table_ui_block_dataset_source_type",
         "TableUiBlockSource": "benchling_api_client.v2.stable.models.table_ui_block_source",