tinybird 4.5.8__tar.gz → 4.5.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 4.5.8
+Version: 4.5.9
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -52,6 +52,11 @@ 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
 *******
 

{tinybird-4.5.8 → 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 → 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'
-__revision__ = '4b9bf45'
+__version__ = '4.5.9'
+__revision__ = 'f7f4fc9'

{tinybird-4.5.8 → 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 → tinybird-4.5.9}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 4.5.8
+Version: 4.5.9
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -52,6 +52,11 @@ 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
 *******