tinybird 4.5.8.dev0__tar.gz → 4.5.9__tar.gz

{tinybird-4.5.8.dev0 → tinybird-4.5.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 4.5.8.dev0
+Version: 4.5.9
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -52,6 +52,16 @@ The Tinybird command-line tool allows you to use all the Tinybird functionality
 Changelog
 ----------
 
+4.5.9
+*******
+
+- `Fixed` Manage correctly LLM errors when analyzing quarantine errors.
+
+4.5.8
+*******
+
+- `Changed` `tb migrate-to-forward` removes the need to add a dummy pipe on first deployment
+
 4.5.7
 *******
 

{tinybird-4.5.8.dev0 → tinybird-4.5.9}/tinybird/client.py

@@ -972,16 +972,28 @@ class TinyB:
         kafka_sasl_mechanism="PLAIN",
         kafka_security_protocol="SASL_SSL",
         kafka_ssl_ca_pem=None,
+        kafka_sasl_oauthbearer_method=None,
+        kafka_sasl_oauthbearer_aws_region=None,
+        kafka_sasl_oauthbearer_aws_role_arn=None,
+        kafka_sasl_oauthbearer_aws_external_id=None,
     ):
+        is_oauthbearer = kafka_sasl_mechanism == "OAUTHBEARER"
         params = {
             "service": "kafka",
             "kafka_security_protocol": kafka_security_protocol,
             "kafka_sasl_mechanism": kafka_sasl_mechanism,
             "kafka_bootstrap_servers": kafka_bootstrap_servers,
-            "kafka_sasl_plain_username": kafka_key,
-            "kafka_sasl_plain_password": kafka_secret,
             "name": kafka_connection_name,
         }
+        if is_oauthbearer:
+            params["kafka_sasl_oauthbearer_method"] = kafka_sasl_oauthbearer_method
+            params["kafka_sasl_oauthbearer_aws_region"] = kafka_sasl_oauthbearer_aws_region
+            params["kafka_sasl_oauthbearer_aws_role_arn"] = kafka_sasl_oauthbearer_aws_role_arn
+            if kafka_sasl_oauthbearer_aws_external_id:
+                params["kafka_sasl_oauthbearer_aws_external_id"] = kafka_sasl_oauthbearer_aws_external_id
+        else:
+            params["kafka_sasl_plain_username"] = kafka_key
+            params["kafka_sasl_plain_password"] = kafka_secret
 
         if kafka_schema_registry_url:
             params["kafka_schema_registry_url"] = kafka_schema_registry_url

{tinybird-4.5.8.dev0 → tinybird-4.5.9}/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.5.8.dev0'
-__revision__ = '02e83cb'
+__version__ = '4.5.9'
+__revision__ = 'f7f4fc9'

{tinybird-4.5.8.dev0 → tinybird-4.5.9}/tinybird/tb/modules/datasource.py

@@ -1476,19 +1476,22 @@ def analyze_quarantine(datasource_name: str, project: Project, client: TinyB):
     user_token = config.get_user_token()
     click.echo(FeedbackManager.gray(message=f"\n» Analyzing errors in {datasource_name}_quarantine..."))
     if user_token:
-        llm = LLM(user_token=user_token, host=config.get_client().host)
-        ds_filenames = project.get_datasource_files()
-        datasource_definition = next(
-            (Path(f).read_text() for f in ds_filenames if f.endswith(f"{datasource_name}.datasource")), ""
-        )
-        response_llm = llm.ask(
-            system_prompt=quarantine_prompt(datasource_definition),
-            prompt=f"The quarantine errors are:\n{json.dumps(quarantine_data)}",
-            feature="tb_datasource_append_analyze_quarantine",
-        )
-        response = extract_xml(response_llm, "quarantine_errors")
-        error_message += "\n" + response
-        click.echo(response)
+        try:
+            llm = LLM(user_token=user_token, host=config.get_client().host)
+            ds_filenames = project.get_datasource_files()
+            datasource_definition = next(
+                (Path(f).read_text() for f in ds_filenames if f.endswith(f"{datasource_name}.datasource")), ""
+            )
+            response_llm = llm.ask(
+                system_prompt=quarantine_prompt(datasource_definition),
+                prompt=f"The quarantine errors are:\n{json.dumps(quarantine_data)}",
+                feature="tb_datasource_append_analyze_quarantine",
+            )
+            response = extract_xml(response_llm, "quarantine_errors")
+            error_message += "\n" + response
+            click.echo(response)
+        except Exception:
+            click.echo(FeedbackManager.error(message="There was an error analyzing the quarantine errors"))
     else:
         echo_safe_humanfriendly_tables_format_smart_table(
             data=[d.values() for d in res["data"]], column_names=res["data"][0].keys()

{tinybird-4.5.8.dev0 → tinybird-4.5.9}/tinybird/tb/modules/deployment.py

@@ -474,24 +474,13 @@ def migrate_to_forward(ctx: click.Context, allow_destructive_operations: bool) -
         show_migrate_to_forward_hint=False,
         return_check_result=True,
         validate_forward_workspace=False,
+        is_classic_migration=True,
     )
     if not check_result:
         message = "Deployment check did not complete. Migration cancelled."
         click.echo(FeedbackManager.error(message=message))
         sys_exit("migration_error", message)
 
-    if check_result and check_result.get("status") == "no_changes":
-        click.echo(
-            FeedbackManager.warning(
-                message=(
-                    "No deployment changes were detected. Add this dummy pipe to your workspace and run "
-                    "`tb migrate-to-forward` again:"
-                )
-            )
-        )
-        click.echo("NODE n\nSQL >\n    select 'Forward'")
-        return
-
     if not click.confirm(
         "Do you want to continue with the migration? This will also delete your branches, releases and switch your workspace from Classic to Forward.",
         default=False,
@@ -514,6 +503,7 @@ def migrate_to_forward(ctx: click.Context, allow_destructive_operations: bool) -
         output=output,
         env=env,
         validate_forward_workspace=False,
+        is_classic_migration=True,
     )
 
 

{tinybird-4.5.8.dev0 → tinybird-4.5.9}/tinybird/tb/modules/deployment_common.py

@@ -347,6 +347,7 @@ def create_deployment(
     env: Optional[str] = "cloud",
     return_check_result: bool = False,
     validate_forward_workspace: bool = True,
+    is_classic_migration: bool = False,
 ) -> Optional[Dict[str, Any]]:
     # TODO: This code is duplicated in build_server.py
     # Should be refactored to be shared
@@ -404,6 +405,8 @@ def create_deployment(
             params["allow_destructive_operations"] = "true"
         if not validate_forward_workspace:
             params["validate_forward_workspace"] = "false"
+        if is_classic_migration:
+            params["is_classic_migration"] = "true"
 
         deployment_request_sent = True
         result = api_post(

{tinybird-4.5.8.dev0 → tinybird-4.5.9}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 4.5.8.dev0
+Version: 4.5.9
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -52,6 +52,16 @@ The Tinybird command-line tool allows you to use all the Tinybird functionality
 Changelog
 ----------
 
+4.5.9
+*******
+
+- `Fixed` Manage correctly LLM errors when analyzing quarantine errors.
+
+4.5.8
+*******
+
+- `Changed` `tb migrate-to-forward` removes the need to add a dummy pipe on first deployment
+
 4.5.7
 *******