tinybird 4.6.21.dev0__tar.gz → 4.6.22__tar.gz

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 4.6.21.dev0
+Version: 4.6.22
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -52,6 +52,21 @@ The Tinybird command-line tool allows you to use all the Tinybird functionality
 Changelog
 ----------
 
+4.6.22
+********
+
+- `Added` ``tb pull`` demotes sink and copy pipes that exceed the workspace's sink/copy pipe plan limit to plain pipes instead of failing, stripping their ``TYPE``/``EXPORT_*``/``TARGET_DATASOURCE``/``COPY_SCHEDULE`` settings and warning that the plan must be upgraded to restore them.
+
+4.6.21
+********
+
+- `Fixed` ``tb deployment discard`` now discards the deployment that is not live, instead of assuming the live deployment is listed first. Previously, if removing the old deployment failed after a promotion, the command refused to discard it.
+
+4.6.20
+********
+
+- `Added` Copy pipes accept ``ON_DEMAND_COMPUTE true`` so scheduled runs use on-demand compute. Manual ``tb copy run`` still requires ``--on-demand-compute``, and asks for confirmation when the local datafile does not enable it (skip it with ``--yes``).
+
 4.6.19
 ********
 

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/pyproject.toml

@@ -18,34 +18,35 @@ authors = [
 dependencies = [
     "GitPython==3.1.44",
     "Markdown==3.5.1",
-    "aioboto3==14.0.0",
+    "aioboto3~=15.0",
     "aiofiles==24.1.0",
-    "aiohttp==3.11.2",
-    "anthropic==0.55.0",
+    "aiohttp~=3.11",
+    "anthropic~=1.9",
+    "async-lru==2.3.0",
     "aws-msk-iam-sasl-signer-python==1.0.2",
     "cachetools==5.5.0",
     "cffi==1.17.1",
     "chardet==4.0.0",
     "ciso8601==2.2.0",
-    "click==8.1.8",
+    "click~=8.5",
     "clickhouse-cityhash==1.0.2.5",
     "clickhouse-driver==0.2.9",
     "confluent-kafka[avro,json,schemaregistry]==2.12.2",
     "contextvars==2.4",
     "cramjam==2.11.0",
     "croniter==6.2.2",
-    "cryptography==3.4.8",
+    "cryptography~=38.0",
     "datasketch==1.6.5",
     "dnspython==2.8.0",
     "docker==7.1.0",
     "fastavro==1.12.2",
     "fastmcp==2.12.2",
-    "google-cloud-aiplatform==1.148.1",
+    "google-cloud-aiplatform~=1.148",
     "google-cloud-compute==1.48.0",
     "google-cloud-scheduler==2.20.0",
     "google-cloud-secret-manager==2.29.0",
-    "google-cloud-storage==3.4.1",
-    "google_api_core==2.25.2",
+    "google-cloud-storage~=3.4",
+    "google_api_core~=2.25",
     "googleapis_common_protos==1.66.0",
     "grpcio_status==1.67.1",
     "humanfriendly==10.0",
@@ -68,7 +69,7 @@ dependencies = [
     "opentelemetry-sdk==1.38.0",
     "orb-billing==2.17.0",
     "orjson==3.11.5",
-    "packaging==23.2",
+    "packaging~=26.3",
     "passlib==1.7.1",
     "posthog==7.28.0",
     "pre-commit==4.6.2",
@@ -82,10 +83,10 @@ dependencies = [
     "py-spy==0.4.2",
     "pyarrow==23.0.1",
     "pycurl==7.45.7",
-    "pydantic-ai-slim[anthropic]==0.5.0",
-    "pydantic-ai-slim[google]==0.5.0",
-    "pydantic-ai-slim[retries]==0.5.0",
-    "pydantic==2.12.5",
+    "pydantic-ai-slim[anthropic]~=2.0",
+    "pydantic-ai-slim[google]~=2.0",
+    "pydantic-ai-slim[retries]~=2.0",
+    "pydantic~=2.12",
     "pyjwt[crypto]==2.9.0",
     "pympler==1.1",
     "pynacl==1.5.0",
@@ -93,10 +94,10 @@ dependencies = [
     "python-dateutil==2.9.0.post0",
     "python-dotenv==1.1.0",
     "python-ulid==2.7.0",
-    "pyyaml==6.0.1",
+    "pyyaml~=6.0",
     "redis[hiredis]==5.3.1",
     "requests==2.32.4",
-    "sentry-sdk==2.69.1",
+    "sentry-sdk==2.69.2",
     "setproctitle==1.3.3",
     "shandy-sqlfmt==0.11.1",
     "shandy-sqlfmt[jinjafmt]==0.11.1",
@@ -107,7 +108,7 @@ dependencies = [
     "tabulate==0.8.10",
     "thefuzz==0.22.1",
     "tinybird-python-sdk[ai]==0.3.7",
-    "tinybird-toolset==2.6.2",
+    "tinybird-toolset==2.6.6",
     "toposort==1.10",
     "tornado==6.5.5",
     "urllib3==1.26.20",
@@ -124,9 +125,9 @@ deploy = [
     "GitPython==3.1.44",
     "ansible==8.5.0",
     "ara==1.6.1",
-    "boto3==1.36.21",
+    "boto3~=1.40",
     "fabric==3.2.2",
-    "google-auth==2.47.0",
+    "google-auth~=2.47",
     "jsonnet",
     "prettytable==3.8.0",
     "requests==2.32.4",
@@ -141,7 +142,7 @@ devtools = [
 ]
 doc = [
     "Pallets-Sphinx-Themes==2.3.0",
-    "pyOpenSSL<24.0.0",
+    "pyOpenSSL~=23.2",
     "sphinx-autobuild==2024.2.4",
     "sphinx-sitemap",
     "sphinx==7.3.7",
@@ -149,13 +150,15 @@ doc = [
     "sphinxcontrib-jquery==4.1",
 ]
 lint = [
-    "botocore-stubs==1.36.23",
+    "botocore-stubs~=1.40",
+    "flake8~=7.3",
+    "flake8-async~=26.8",
     "mypy==2.3.1",
-    "ruff==0.16.8",
+    "ruff==0.16.9",
     "types-Markdown",
     "types-PyYAML",
     "types-aiofiles==24.1.0.20241221",
-    "types-boto3-lite==1.36.21",
+    "types-boto3-lite~=1.40",
     "types-cachetools",
     "types-chardet",
     "types-paramiko",
@@ -169,21 +172,28 @@ lint = [
 test = [
     "coverage==7.10.6",
     "freezegun==1.5.5",
-    "google-auth==2.47.0",
+    "google-auth~=2.47",
     "gsutil==4.58",
+    "httpx2~=2.13",
     "hypothesis==6.156.1",
     "oauth2client==3.0.0",
     "py==1.11.0",
-    "pytest-asyncio==0.23.8",
+    "pytest-asyncio~=1.4",
     "pytest-cov==4.1.0",
     "pytest-env==1.1.3",
     "pytest-flakefinder==1.1.0",
     "pytest-timeout==2.3.1",
     "pytest-xdist==3.6.1",
-    "pytest==8.1.2",
+    "pytest~=8.1",
     "rangehttpserver==1.4.0",
     "yappi==1.7.6",
 ]
+# Synced automatically by plain `uv run`/`uv sync`, no `--group` needed.
+dev = [
+    { include-group = "test" },
+    { include-group = "deploy" },
+    { include-group = "lint" },
+]
 
 [project.scripts]
 tinybird_server = "tinybird.app:run"
@@ -193,7 +203,12 @@ tb = "tinybird.tb.cli:cli"
 colibri = "tinybird.colibri.cli:cli"
 
 [tool.uv.workspace]
-members = ["data_connector"]
+members = [
+    "chbackups",
+    "data_connector",
+    "gatherer",
+    "tinybird_orchestrator",
+]
 
 [tool.setuptools]
 include-package-data = true

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/client.py

@@ -47,14 +47,6 @@ class TimeoutException(Exception):
     pass
 
 
-class ReachRetryLimit(Exception):
-    pass
-
-
-class ConnectorNothingToLoad(Exception):
-    pass
-
-
 class JobException(Exception):
     pass
 
@@ -691,6 +683,7 @@ class TinyB:
         target_datasource: Optional[str] = None,
         schedule_cron: Optional[str] = None,
         mode: Optional[str] = None,
+        on_demand_compute: Optional[bool] = None,
     ):
         data = {"schedule_cron": schedule_cron}
 
@@ -700,6 +693,9 @@ class TinyB:
         if mode:
             data["mode"] = mode
 
+        if on_demand_compute is not None:
+            data["on_demand_compute"] = "true" if on_demand_compute else "false"
+
         return await self._req(f"/v0/pipes/{pipe_name_or_id}/nodes/{node_id}/copy", method="PUT", data=data)
 
     async def pipe_remove_copy(self, pipe_name_or_id: str, node_id: str):

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/datafile/common.py

@@ -28,6 +28,7 @@ from tinybird.ch_utils.engine import VALID_ENGINE_PARAMS, EngineParam
 from tinybird.datafile.exceptions import IncludeFileNotFoundException, ParseException, ValidationException
 from tinybird.tb.modules.exceptions import CLIPipeException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.utils.bools import parse_optional_bool
 
 # Code from sql.py has been duplicated so I can change it without breaking absolutely everything in the app
 # I'll try not to make logic changes, just error reporting changes
@@ -179,9 +180,10 @@ class CopyParameters(Parameters):
     COPY_SCHEDULE = "copy_schedule"
     COPY_MODE = "copy_mode"
     COPY_MODE_ALIAS = "mode"  # we need this because bot MODE and COPY_MODE go to `mode` variable inside the node
+    ON_DEMAND_COMPUTE = "on_demand_compute"
     MANDATORY_ATTRIBUTES = PipeParameters.MANDATORY_ATTRIBUTES.union({TARGET_DATASOURCE})
     ACCEPTED_ATTRIBUTES = PipeParameters.ACCEPTED_ATTRIBUTES.union(MANDATORY_ATTRIBUTES).union(
-        {COPY_SCHEDULE, COPY_MODE_ALIAS}
+        {COPY_SCHEDULE, COPY_MODE_ALIAS, ON_DEMAND_COMPUTE}
     )
 
 
@@ -416,6 +418,11 @@ class Datafile:
                 )
             if not croniter.is_valid(copy_schedule):
                 raise DatafileValidationError("COPY node schedule must be @on-demand or a valid cron expression.")
+        if CopyParameters.ON_DEMAND_COMPUTE in node:
+            try:
+                node[CopyParameters.ON_DEMAND_COMPUTE] = parse_optional_bool(node[CopyParameters.ON_DEMAND_COMPUTE])
+            except ValueError:
+                raise DatafileValidationError("ON_DEMAND_COMPUTE must be true or false")
         for key in node.keys():
             if key not in CopyParameters.valid_params():
                 raise DatafileValidationError(
@@ -2097,6 +2104,7 @@ def parse(
             "copy_schedule": assign_node_var(CopyParameters.COPY_SCHEDULE),
             "copy_mode": assign_node_var("mode"),
             "mode": assign_node_var("mode"),
+            "on_demand_compute": assign_node_var(CopyParameters.ON_DEMAND_COMPUTE),
             "filter": assign_node_var("filter"),
             "token": add_token,
             "include": include,

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/datatypes.py

@@ -120,52 +120,74 @@ def datetime_test(x: str) -> bool:
     return any(p.match(x) for p in datetime_patterns)
 
 
+def bounded_integer(x: str, pattern: re.Pattern[str]) -> Optional[int]:
+    if pattern.match(x) is None:
+        return None
+    digits = x.lstrip("+-").lstrip("0")
+    # 256-bit unsigned integers have at most 78 decimal digits.
+    if len(digits) > 78:
+        return None
+    return int(x) if len(x) <= 78 else int(digits or "0") * (-1 if x.startswith("-") else 1)
+
+
 def int_8_test(x: str) -> bool:
-    return intx_re.match(x) is not None and -int_8_max <= int(x) < int_8_max
+    value = bounded_integer(x, intx_re)
+    return value is not None and -int_8_max <= value < int_8_max
 
 
 def int16_test(x: str) -> bool:
-    return intx_re.match(x) is not None and -int16_max <= int(x) < int16_max
+    value = bounded_integer(x, intx_re)
+    return value is not None and -int16_max <= value < int16_max
 
 
 def int32_test(x: str) -> bool:
-    return intx_re.match(x) is not None and -int32_max <= int(x) < int32_max
+    value = bounded_integer(x, intx_re)
+    return value is not None and -int32_max <= value < int32_max
 
 
 def int64_test(x: str) -> bool:
-    return intx_re.match(x) is not None and -int64_max <= int(x) < int64_max
+    value = bounded_integer(x, intx_re)
+    return value is not None and -int64_max <= value < int64_max
 
 
 def int128_test(x: str) -> bool:
-    return intx_re.match(x) is not None and -int128_max <= int(x) < int128_max
+    value = bounded_integer(x, intx_re)
+    return value is not None and -int128_max <= value < int128_max
 
 
 def int256_test(x: str) -> bool:
-    return intx_re.match(x) is not None and -int256_max <= int(x) < int256_max
+    value = bounded_integer(x, intx_re)
+    return value is not None and -int256_max <= value < int256_max
 
 
 def uint_8_test(x: str) -> bool:
-    return uintx_re.match(x) is not None and 0 <= int(x) < uint_8_max
+    value = bounded_integer(x, uintx_re)
+    return value is not None and 0 <= value < uint_8_max
 
 
 def uint16_test(x: str) -> bool:
-    return uintx_re.match(x) is not None and 0 <= int(x) < uint16_max
+    value = bounded_integer(x, uintx_re)
+    return value is not None and 0 <= value < uint16_max
 
 
 def uint32_test(x: str) -> bool:
-    return uintx_re.match(x) is not None and 0 <= int(x) < uint32_max
+    value = bounded_integer(x, uintx_re)
+    return value is not None and 0 <= value < uint32_max
 
 
 def uint64_test(x: str) -> bool:
-    return uintx_re.match(x) is not None and 0 <= int(x) < uint64_max
+    value = bounded_integer(x, uintx_re)
+    return value is not None and 0 <= value < uint64_max
 
 
 def uint128_test(x: str) -> bool:
-    return intx_re.match(x) is not None and 0 <= int(x) < uint128_max
+    value = bounded_integer(x, intx_re)
+    return value is not None and 0 <= value < uint128_max
 
 
 def uint256_test(x: str) -> bool:
-    return intx_re.match(x) is not None and 0 <= int(x) < uint256_max
+    value = bounded_integer(x, intx_re)
+    return value is not None and 0 <= value < uint256_max
 
 
 def float_test(x: str) -> bool:

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/feedback_manager.py

@@ -572,6 +572,11 @@ class FeedbackManager:
         "You can read more about the pricing at https://www.tinybird.co/docs/classic/work-with-data/process-and-copy/materialized-views#compute-compute-separation-for-populates\n\n"
         "Do you want to proceed with on-demand compute?"
     )
+    warning_confirm_on_demand_compute_differs_from_datafile = warning_message(
+        "The local datafile of pipe '{pipe}' does not set ON_DEMAND_COMPUTE true, but --on-demand-compute was passed.\n"
+        "On-demand compute uses dedicated resources that are billed separately from your regular usage.\n\n"
+        "Do you want to proceed?"
+    )
 
     warning_confirm_delete_token = prompt_message("Do you want to delete Token {token}?")
     warning_confirm_refresh_token = prompt_message("Do you want to refresh Token {token}?")

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/sql_template.py

@@ -1,5 +1,4 @@
 import ast
-import builtins
 import linecache
 import logging
 import re
@@ -18,7 +17,7 @@ from tinybird.context import (
 )
 
 from .datatypes import testers
-from .tornado_template import VALID_CUSTOM_FUNCTION_NAMES, SecurityException, Template
+from .tornado_template import SAFE_BUILTINS, VALID_CUSTOM_FUNCTION_NAMES, SecurityException, Template
 
 VALID_ACTIVATE_FEATURES = frozenset(["analyzer", "parallel_replicas", "optimize_aggregation_in_order"])
 
@@ -1336,8 +1335,10 @@ sqlescape_for_string_expression = sqlescape_generator(
 )
 
 
-def escape_single_quote_str(s):
-    return "'" + s.replace("'", "''") + "'"
+def escape_single_quote_str(s: str) -> str:
+    # ClickHouse string literals treat backslash as an escape char, so it must be doubled too:
+    # otherwise a trailing backslash escapes the closing quote and the following input becomes SQL
+    return "'" + s.replace("\\", "\\\\").replace("'", "''") + "'"
 
 
 def expression_wrapper(x, name, escape_arrays: bool = False):
@@ -1375,7 +1376,7 @@ _namespace = {
     # they raise a pretty non understandable error but if someone
     # is using them they know what they are trying to do
     # read https://anee.me/escaping-python-jails-849c65cf306e on how to escape from python jails
-    "__buildins__": {},
+    "__builtins__": SAFE_BUILTINS,
     "__import__": {},
     "__debug__": {},
     "__doc__": {},
@@ -1388,10 +1389,13 @@ _namespace = {
 }
 
 
-reserved_vars = {"_tt_tmp", "_tt_append", "isinstance", "str", "error", "custom_error", *list(vars(builtins))}
+# Built from the builtins templates can actually resolve: names missing from SAFE_BUILTINS (`type`, `dir`,
+# `property`, `object`, ...) can only be template parameters, so they must be discovered as such to get a
+# value (None or a test-mode placeholder) instead of raising NameError.
+reserved_vars = {"_tt_tmp", "_tt_append", "isinstance", "str", "error", "custom_error", *SAFE_BUILTINS}
 for p in DEFAULT_PARAM_NAMES:  # we handle these in an specific manner
     reserved_vars.discard(p)  # `format` is part of builtins
-# Allow 'id' to be used as a template parameter - https://gitlab.com/tinybird/analytics/-/issues/19119
+# Allow 'id' as a template parameter name despite being a builtin - https://gitlab.com/tinybird/analytics/-/issues/19119
 reserved_vars.discard("id")
 error_vars = ["error", "custom_error"]
 
@@ -1468,8 +1472,7 @@ def generate(self, **kwargs) -> Tuple[str, TemplateExecutionResults]:
         return Expression(f"-- disable {feature}\n")
 
     namespace.update(_namespace)
-    # This is to fix the issue https://gitlab.com/tinybird/analytics/-/issues/19119
-    # We need to do the override here because if we modify the _namespace, we would filter out parameters from the users
+    # Default it here, not in _namespace, so it stays visible to variable extraction.
     namespace.update({"id": None})
     namespace.update(kwargs)
     namespace.update(
@@ -1478,6 +1481,8 @@ def generate(self, **kwargs) -> Tuple[str, TemplateExecutionResults]:
             # the generated source code.
             "__name__": self.name.replace(".", "_"),
             "__loader__": ObjectDict(get_source=lambda name: self.code),
+            # Set again after kwargs so request parameters can't override it
+            "__builtins__": SAFE_BUILTINS,
             "max_threads": set_max_threads,
             "tb_secret": set_tb_secret,
             "tb_var": set_tb_secret,
@@ -1817,7 +1822,7 @@ def get_var_names(t: Template):
     >>> [v['name'] for v in get_var_names(Template("{% for name in secret_names %}{{tb_secret(name)}}{% end %}"))]
     ['secret_names', 'name', 'tb_secret', 'name']
 
-    Type casting with date function in elif chain (type is in reserved_vars):
+    Type casting with date function in elif chain:
     >>> t = Template("{% if kind == 'int' %}{{Int32(val, 0)}}{% elif kind == 'date' %}{{DateTime(val)}}{% elif kind == 'diff' %}{{date_diff_in_days(val, now)}}{% end %}")
     >>> [v['name'] for v in get_var_names(t)]
     ['kind', 'Int32', 'val', 'kind', 'DateTime', 'val', 'kind', 'date_diff_in_days', 'val', 'now']
@@ -2369,7 +2374,24 @@ def wrap_vars(t, escape_arrays: bool = False):
     return var
 
 
-def get_used_tables_in_template(sql):
+def _constant_table_name(node: ast.expr) -> Optional[str]:
+    """Value of a constant table() argument: a literal or a concatenation of string literals."""
+    if isinstance(node, ast.Constant) and isinstance(node.value, (int, float)):
+        return str(node.value)
+    return _string_literal(node)
+
+
+def _string_literal(node: ast.expr) -> Optional[str]:
+    if isinstance(node, ast.Constant) and isinstance(node.value, str):
+        return node.value
+    if isinstance(node, ast.BinOp) and isinstance(node.op, ast.Add):
+        left, right = _string_literal(node.left), _string_literal(node.right)
+        if left is not None and right is not None:
+            return left + right
+    return None
+
+
+def get_used_tables_in_template(sql: str) -> List[str]:
     """
     >>> get_used_tables_in_template("select * from {{table('test')}}")
     ['test']
@@ -2383,30 +2405,42 @@ def get_used_tables_in_template(sql):
     []
     >>> get_used_tables_in_template("select * from {{table('my.test')}}, {{table('another.one')}}")
     ['my.test', 'another.one']
+
+    Only constant table names can be known without rendering, so anything else is skipped
+    >>> get_used_tables_in_template("select * from {{table(tbl)}}, {{TABLE('b')}}, {{table('my.' + 'c')}}")
+    ['b', 'my.c']
     """
+    return list(_get_used_tables_in_template_cached(sql))
+
+
+# Building the Template dominates the cost and the result only depends on the SQL, which deployments and
+# /v0/sql pipelines scan repeatedly
+@lru_cache(maxsize=2**10)
+def _get_used_tables_in_template_cached(sql: str) -> Tuple[str, ...]:
     try:
         t = Template(sql)
 
         def _n(chunks, tables):
             for x in chunks:
                 if type(x).__name__ == "_Expression":
-                    c = compile(x.expression, "<string>", "exec", dont_inherit=True)
-                    v = [x.lower() for x in c.co_names if x not in _namespace and x not in reserved_vars]
-                    if "table" in v:
-
-                        def _t(*args, **kwargs):
-                            return str(args[0])
-
-                        n = {"table": _t, "TABLE": _t}
-                        e = "_tt_tmp = %s" % x.expression
-                        exec_in(e, n)
-                        tables += [n["_tt_tmp"]]
+                    # Read the names from the AST instead of executing the expression: this runs outside
+                    # generate(), without the restricted builtins, on every pipe, query and deployment.
+                    for node in ast.walk(ast.parse(x.expression)):
+                        if (
+                            isinstance(node, ast.Call)
+                            and isinstance(node.func, ast.Name)
+                            and node.func.id in ("table", "TABLE")
+                            and node.args
+                        ):
+                            name = _constant_table_name(node.args[0])
+                            if name is not None:
+                                tables.append(name)
                 elif type(x).__name__ == "_ControlBlock":
                     _n(x.body.chunks, tables)
 
-        tables = []
+        tables: List[str] = []
         _n(t.file.body.chunks, tables)
-        return tables
+        return tuple(tables)
     except SecurityException as e:
         raise SQLTemplateException(e)
 

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/sql_template_fmt.py

@@ -214,8 +214,8 @@ class TinybirdDialect(ClickHouse):
             ],
         }
 
-        for section in override_rules:
-            for rule in override_rules[section]:
+        for section, rules in override_rules.items():
+            for rule in rules:
                 for rr in self.RULES[section]:
                     if rr.name == rule.name:
                         self.RULES[section].remove(rr)

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/syncasync.py

@@ -260,49 +260,6 @@ def _iscoroutinefunction_or_partial(func: Any) -> bool:
     return asyncio.iscoroutinefunction(func)
 
 
-class ThreadSensitiveContext:
-    """Async context manager to manage context for thread sensitive mode
-    This context manager controls which thread pool executor is used when in
-    thread sensitive mode. By default, a single thread pool executor is shared
-    within a process.
-    In Python 3.7+, the ThreadSensitiveContext() context manager may be used to
-    specify a thread pool per context.
-    In Python 3.6, usage of this context manager has no effect.
-    This context manager is re-entrant, so only the outer-most call to
-    ThreadSensitiveContext will set the context.
-    """
-
-    def __init__(self):
-        self.token = None
-
-    if contextvars:
-
-        async def __aenter__(self):
-            try:
-                SyncToAsync.thread_sensitive_context.get()
-            except LookupError:
-                self.token = SyncToAsync.thread_sensitive_context.set(self)
-
-            return self
-
-        async def __aexit__(self, exc, value, tb):
-            if not self.token:
-                return
-
-            executor = SyncToAsync.context_to_thread_executor.pop(self, None)
-            if executor:
-                executor.shutdown()
-            SyncToAsync.thread_sensitive_context.reset(self.token)
-
-    else:
-
-        async def __aenter__(self):
-            return self
-
-        async def __aexit__(self, exc, value, tb):
-            pass
-
-
 class AsyncToSync:
     """
     Utility class which turns an awaitable that only works on the thread with

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/forward/commands'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '4.6.21.dev0'
-__revision__ = '965c47ce'
+__version__ = '4.6.22'
+__revision__ = '5269fe22'

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/tb/client.py

@@ -58,14 +58,6 @@ class TimeoutException(Exception):
     pass
 
 
-class ReachRetryLimit(Exception):
-    pass
-
-
-class ConnectorNothingToLoad(Exception):
-    pass
-
-
 class JobException(Exception):
     pass
 
@@ -743,6 +735,7 @@ class TinyB:
         target_datasource: Optional[str] = None,
         schedule_cron: Optional[str] = None,
         mode: Optional[str] = None,
+        on_demand_compute: Optional[bool] = None,
     ):
         data = {"schedule_cron": schedule_cron}
 
@@ -752,6 +745,9 @@ class TinyB:
         if mode:
             data["mode"] = mode
 
+        if on_demand_compute is not None:
+            data["on_demand_compute"] = "true" if on_demand_compute else "false"
+
         return self._req(f"/v0/pipes/{pipe_name_or_id}/nodes/{node_id}/copy", method="PUT", data=data)
 
     def pipe_remove_copy(self, pipe_name_or_id: str, node_id: str):

{tinybird-4.6.21.dev0 → tinybird-4.6.22}/tinybird/tb/modules/common.py

@@ -1277,11 +1277,11 @@ class ConnectionReplacements:
         """Maps prompt parameters to API parameters."""
 
         api_params = {}
-        for key in params.keys():
+        for key, value in params.items():
             try:
-                api_params[ConnectionReplacements._PARAMS_REPLACEMENTS[service][key]] = params[key]
+                api_params[ConnectionReplacements._PARAMS_REPLACEMENTS[service][key]] = value
             except KeyError:
-                api_params[key] = params[key]
+                api_params[key] = value
 
         api_params["service"] = service
         return api_params