tinybird 4.5.9.dev0__tar.gz → 4.5.10__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 4.5.9.dev0
+Version: 4.5.10
 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.10
+*******
+
+- `Fixed` `tb init` now persists `folder` in `tinybird.config.json`, so subsequent resource creation targets the configured project folder instead of the repository root.
+
+4.5.9
+*******
+
+- `Fixed` Manage correctly LLM errors when analyzing quarantine errors.
+
 4.5.8
 *******
 

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

@@ -340,10 +340,14 @@ class TinyB:
             for c in connectors
         ]
 
-    async def get_datasource(self, ds_name: str, used_by: bool = False) -> Dict[str, Any]:
+    async def get_datasource(
+        self, ds_name: str, used_by: bool = False, include_workspace_names: bool = False
+    ) -> Dict[str, Any]:
         params = {
             "attrs": "used_by" if used_by else "",
         }
+        if include_workspace_names:
+            params["include_workspace_names"] = "true"
         return await self._req(f"/v0/datasources/{ds_name}?{urlencode(params)}")
 
     async def alter_datasource(
@@ -489,12 +493,32 @@ class TinyB:
 
         return await self._req(f"/v0/dependencies?{urlencode(params)}", timeout=60)
 
-    async def datasource_share(self, datasource_id: str, current_workspace_id: str, destination_workspace_id: str):
-        params = {"origin_workspace_id": current_workspace_id, "destination_workspace_id": destination_workspace_id}
+    async def datasource_share(
+        self,
+        datasource_id: str,
+        current_workspace_id: str,
+        destination_workspace_id: Optional[str] = None,
+        destination_workspace_name: Optional[str] = None,
+    ) -> Dict[str, Any]:
+        params = {"origin_workspace_id": current_workspace_id}
+        if destination_workspace_id:
+            params["destination_workspace_id"] = destination_workspace_id
+        if destination_workspace_name:
+            params["destination_workspace_name"] = destination_workspace_name
         return await self._req(f"/v0/datasources/{datasource_id}/share", method="POST", data=params)
 
-    async def datasource_unshare(self, datasource_id: str, current_workspace_id: str, destination_workspace_id: str):
-        params = {"origin_workspace_id": current_workspace_id, "destination_workspace_id": destination_workspace_id}
+    async def datasource_unshare(
+        self,
+        datasource_id: str,
+        current_workspace_id: str,
+        destination_workspace_id: Optional[str] = None,
+        destination_workspace_name: Optional[str] = None,
+    ) -> Dict[str, Any]:
+        params = {"origin_workspace_id": current_workspace_id}
+        if destination_workspace_id:
+            params["destination_workspace_id"] = destination_workspace_id
+        if destination_workspace_name:
+            params["destination_workspace_name"] = destination_workspace_name
         return await self._req(f"/v0/datasources/{datasource_id}/share", method="DELETE", data=params)
 
     async def datasource_sync(self, datasource_id: str):
@@ -972,16 +996,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.9.dev0 → tinybird-4.5.10}/tinybird/service_datasources.py

@@ -219,6 +219,7 @@ def get_tinybird_service_datasources() -> List[Dict[str, Any]]:
                 {"name": "processed_messages", "type": "Int32"},
                 {"name": "processed_bytes", "type": "Int32"},
                 {"name": "committed_messages", "type": "Int32"},
+                {"name": "time_process", "type": "Float32"},
                 {"name": "msg", "type": "String"},
             ],
         },
@@ -879,6 +880,7 @@ def get_organization_service_datasources() -> List[Dict[str, Any]]:
                 {"name": "processed_messages", "type": "Int32"},
                 {"name": "processed_bytes", "type": "Int32"},
                 {"name": "committed_messages", "type": "Int32"},
+                {"name": "time_process", "type": "Float32"},
                 {"name": "msg", "type": "String"},
             ],
         },

{tinybird-4.5.9.dev0 → tinybird-4.5.10}/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.9.dev0'
-__revision__ = '4b9bf45'
+__version__ = '4.5.10'
+__revision__ = 'e0b2a62'

{tinybird-4.5.9.dev0 → tinybird-4.5.10}/tinybird/tb/modules/create.py

@@ -503,7 +503,6 @@ def persist_tinybird_config(root_folder: str, project_type: str, dev_mode: str,
     config_path = Path(root_folder) / "tinybird.config.json"
     config_data: Dict[str, Any] = {}
     created = not config_path.exists()
-    include = [folder]
 
     if config_path.exists():
         try:
@@ -516,17 +515,17 @@ def persist_tinybird_config(root_folder: str, project_type: str, dev_mode: str,
     if project_type == "typescript":
         updates = {
             "devMode": dev_mode,
-            "include": include,
+            "folder": folder,
         }
-        keys_to_remove = ("type", "projectType", "project_type", "dev_mode", "sdk", "folder")
-        preferred_order = ("devMode", "include")
+        keys_to_remove = ("type", "projectType", "project_type", "dev_mode", "sdk", "include")
+        preferred_order = ("devMode", "folder")
     else:
         updates = {
             "dev_mode": dev_mode,
-            "include": include,
+            "folder": folder,
         }
-        keys_to_remove = ("type", "projectType", "project_type", "devMode", "sdk", "folder")
-        preferred_order = ("dev_mode", "include")
+        keys_to_remove = ("type", "projectType", "project_type", "devMode", "sdk", "include")
+        preferred_order = ("dev_mode", "folder")
 
     changed = False
     for key in keys_to_remove:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 4.5.9.dev0
+Version: 4.5.10
 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.10
+*******
+
+- `Fixed` `tb init` now persists `folder` in `tinybird.config.json`, so subsequent resource creation targets the configured project folder instead of the repository root.
+
+4.5.9
+*******
+
+- `Fixed` Manage correctly LLM errors when analyzing quarantine errors.
+
 4.5.8
 *******