tinybird 0.0.1.dev291__py3-none-any.whl → 0.0.1.dev292__py3-none-any.whl

tinybird/context.py

@@ -3,10 +3,11 @@ from typing import TYPE_CHECKING
 
 # Avoid circular import error
 if TYPE_CHECKING:
-    from tinybird.user import User
+    from hfi.hfi_workspace_data import HfiWorkspaceData
+
 
 workspace_id: ContextVar[str] = ContextVar("workspace_id")
-workspace: ContextVar["User"] = ContextVar("workspace")
+hfi_workspace_data: ContextVar["HfiWorkspaceData"] = ContextVar("hfi_workspace_data")
 table_id: ContextVar[str] = ContextVar("table_id")
 hfi_frequency: ContextVar[float] = ContextVar("hfi_frequency")
 hfi_frequency_gatherer: ContextVar[float] = ContextVar("hfi_frequency_gatherer")

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__ = '0.0.1.dev291'
-__revision__ = '87da698'
+__version__ = '0.0.1.dev292'
+__revision__ = '78dba1d'

tinybird/tb/modules/agent/agent.py

@@ -34,10 +34,12 @@ from tinybird.tb.modules.agent.mock_agent import MockAgent
 from tinybird.tb.modules.agent.models import create_model
 from tinybird.tb.modules.agent.prompts import (
     agent_system_prompt,
+    fixtures_prompt,
     load_custom_project_rules,
     resources_prompt,
     secrets_prompt,
     service_datasources_prompt,
+    vendor_files_prompt,
 )
 from tinybird.tb.modules.agent.testing_agent import TestingAgent
 from tinybird.tb.modules.agent.tools.analyze import analyze_file, analyze_url
@@ -322,6 +324,14 @@ class TinybirdAgent:
         def get_project_files(ctx: RunContext[TinybirdAgentContext]) -> str:
             return resources_prompt(self.project)
 
+        @self.agent.instructions
+        def get_vendor_files(ctx: RunContext[TinybirdAgentContext]) -> str:
+            return vendor_files_prompt(self.project)
+
+        @self.agent.instructions
+        def get_fixture_files(ctx: RunContext[TinybirdAgentContext]) -> str:
+            return fixtures_prompt(self.project)
+
         @self.agent.instructions
         def get_service_datasources(ctx: RunContext[TinybirdAgentContext]) -> str:
             return service_datasources_prompt()

tinybird/tb/modules/agent/explore_agent.py

@@ -12,6 +12,7 @@ from tinybird.tb.modules.agent.prompts import (
     resources_prompt,
     service_datasources_prompt,
     tone_and_style_instructions,
+    vendor_files_prompt,
 )
 from tinybird.tb.modules.agent.tools.diff_resource import diff_resource
 from tinybird.tb.modules.agent.tools.execute_query import execute_query
@@ -76,6 +77,10 @@ Once you finish the task, return a valid response for the task to complete.
         def get_project_files(ctx: RunContext[TinybirdAgentContext]) -> str:
             return resources_prompt(self.project)
 
+        @self.agent.instructions
+        def get_vendor_files(ctx: RunContext[TinybirdAgentContext]) -> str:
+            return vendor_files_prompt(self.project)
+
         @self.agent.instructions
         def get_service_datasources(ctx: RunContext[TinybirdAgentContext]) -> str:
             return service_datasources_prompt()

tinybird/tb/modules/agent/mock_agent.py

@@ -6,7 +6,7 @@ from pydantic_ai.usage import Usage
 
 from tinybird.tb.modules.agent.animations import ThinkingAnimation
 from tinybird.tb.modules.agent.models import create_model
-from tinybird.tb.modules.agent.prompts import resources_prompt
+from tinybird.tb.modules.agent.prompts import fixtures_prompt, resources_prompt
 from tinybird.tb.modules.agent.tools.mock import generate_mock_fixture
 from tinybird.tb.modules.agent.utils import TinybirdAgentContext
 from tinybird.tb.modules.project import Project
@@ -194,6 +194,10 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
         def get_project_files(ctx: RunContext[TinybirdAgentContext]) -> str:
             return resources_prompt(self.project)
 
+        @self.agent.instructions
+        def get_fixture_files(ctx: RunContext[TinybirdAgentContext]) -> str:
+            return fixtures_prompt(self.project)
+
     def run(self, task: str, deps: TinybirdAgentContext, usage: Usage):
         result = self.agent.run_sync(
             task,

tinybird/tb/modules/agent/prompts.py

@@ -35,8 +35,10 @@ available_commands = [
     "`tb sink ls`: List all sinks",
     "`tb workspace current`: Show the current workspace",
     "`tb workspace clear --yes`: Delete all resources in the workspace (Only available in Tinybird Local)",
+    "`tb workspace ls`: List all workspaces",
     "`tb local start --skip-new-version`: Start Tinybird Local container in non-interactive mode",
     "`tb local restart --skip-new-version --yes`: Restart Tinybird Local container in non-interactive mode",
+    "`tb pull --only-vendored`: Pull only the vendored datasources from other workspaces",
 ]
 
 plan_instructions = """
@@ -166,7 +168,6 @@ SQL >
 
 def resources_prompt(project: Project) -> str:
     files = project.get_project_files()
-    fixture_files = project.get_fixture_files()
 
     resources_content = "# Existing resources in the project:\n"
     if files:
@@ -184,7 +185,36 @@ def resources_prompt(project: Project) -> str:
     else:
         resources_content += "No resources found"
 
+    return resources_content
+
+
+def vendor_files_prompt(project: Project) -> str:
+    files = project.get_vendored_files()
+    content = "# Datasources shared from other workspaces:\n"
+    if files:
+        resources: list[dict[str, Any]] = []
+        for filename in files:
+            file_path = Path(filename)
+            workspace_name = file_path.parent.parent.name
+            resource = {
+                "path": str(file_path.relative_to(project.folder)),
+                "type": get_resource_type(file_path),
+                "name": f"{workspace_name}.{file_path.stem}",
+                "content": file_path.read_text(),
+                "origin_workspace": workspace_name,
+            }
+            resources.append(resource)
+        content += format_as_xml(resources, root_tag="resources", item_tag="resource")
+    else:
+        content += "No datasources shared from other workspaces"
+
+    return content
+
+
+def fixtures_prompt(project: Project) -> str:
+    fixture_files = project.get_fixture_files()
     fixture_content = "# Fixture files in the project:\n"
+
     if fixture_files:
         fixtures: list[dict[str, Any]] = []
         for filename in fixture_files:
@@ -199,7 +229,7 @@ def resources_prompt(project: Project) -> str:
     else:
         fixture_content += "No fixture files found"
 
-    return resources_content + "\n" + fixture_content
+    return fixture_content
 
 
 def service_datasources_prompt() -> str:
@@ -954,6 +984,23 @@ After changes have been deployed and promoted, if you want to deploy other chang
 If after running a deployment, the error contains a recommended forward query, use it to update the .datasource file.
 </dev_notes>
 
+## Sharing datasources with other workspaces
+To share a Data Source, in the .datasource file you want to share, add the destination workspace(s). For example:
+
+```
+SHARED_WITH >
+    destination_workspace,
+    other_destination_workspace
+```
+
+## Working with shared datasources:
+
+The following limitations apply to shared datasources:
+- Shared datasources are read-only.
+- You can't share a shared datasource, only the original.
+- You can't check the quarantine of a shared datasource.
+- You can't create a Materialized View from a shared datasource.
+
 # Working with any type of pipe file:
 {pipe_instructions}
 {pipe_example}

tinybird/tb/modules/agent/tools/run_command.py

@@ -3,6 +3,7 @@ import subprocess
 import click
 from pydantic_ai import RunContext
 
+from tinybird.tb.modules.agent.prompts import available_commands
 from tinybird.tb.modules.agent.utils import (
     AgentRunCancelled,
     SubAgentRunCancelled,
@@ -20,6 +21,14 @@ def run_command(ctx: RunContext[TinybirdAgentContext], command: str):
         command (str): The command to run. Required. Examples: `tb --local sql "select 1"`, `tb --cloud datasource ls`, `tb --help`
     """
     try:
+        clean_commands = [cmd.split(":")[0].replace("`", "").split("[")[0] for cmd in available_commands]
+        available_commands_to_cloud = [f"tb --cloud {command.replace('tb ', '')}" for command in clean_commands]
+        available_commands_to_local = [f"tb --local {command.replace('tb ', '')}" for command in clean_commands]
+        all_commands = clean_commands + available_commands_to_cloud + available_commands_to_local
+
+        if not any(cmd.startswith(command) for cmd in all_commands):
+            raise SubAgentRunCancelled(f"Command {command} not found in the list of available commands")
+
         ctx.deps.thinking_animation.stop()
         force_confirmation = " deploy" in command.lower() or " truncate" in command.lower()
         confirmation = show_confirmation(

{tinybird-0.0.1.dev291.dist-info → tinybird-0.0.1.dev292.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev291
+Version: 0.0.1.dev292
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird

{tinybird-0.0.1.dev291.dist-info → tinybird-0.0.1.dev292.dist-info}/RECORD

@@ -1,5 +1,5 @@
 tinybird/connectors.py,sha256=7Gjms7b5MAaBFGi3xytsJurCylprONpFcYrzp4Fw2Rc,15241
-tinybird/context.py,sha256=FfqYfrGX_I7PKGTQo93utaKPDNVYWelg4Hsp3evX5wM,1291
+tinybird/context.py,sha256=M2jhaNBDOFLlSU2ONDPwO7Coknh2z1PNY-TlNJAkEK0,1343
 tinybird/datatypes.py,sha256=r4WCvspmrXTJHiPjjyOTiZyZl31FO3Ynkwq4LQsYm6E,11059
 tinybird/feedback_manager.py,sha256=XY8d83pRlq-LH7xHMApkaEebfXEWLjDzrGe1prpcTHE,69778
 tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
@@ -18,7 +18,7 @@ tinybird/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1w
 tinybird/datafile/parse_connection.py,sha256=tRyn2Rpr1TeWet5BXmMoQgaotbGdYep1qiTak_OqC5E,1825
 tinybird/datafile/parse_datasource.py,sha256=ssW8QeFSgglVFi3sDZj_HgkJiTJ2069v2JgqnH3CkDE,1825
 tinybird/datafile/parse_pipe.py,sha256=xf4m0Tw44QWJzHzAm7Z7FwUoUUtr7noMYjU1NiWnX0k,3880
-tinybird/tb/__cli__.py,sha256=Etfq1IZSoQiHmnt8KFW5i_w4w_P9rnn-X56chF8OH3o,247
+tinybird/tb/__cli__.py,sha256=-6WKEO5Hmmfh0zrBTi4mPaxkqDYeWDApz2022V9XsK8,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
 tinybird/tb/client.py,sha256=IQRaInDjOwr9Fzaz3_xXc3aUGqh94tM2lew7IZbB9eM,53733
@@ -70,17 +70,17 @@ tinybird/tb/modules/watch.py,sha256=No0bK1M1_3CYuMaIgylxf7vYFJ72lTJe3brz6xQ-mJo,
 tinybird/tb/modules/workspace.py,sha256=tCP1zZMwBhLRGm22TGfpSd4cHvQLAS1o_azIXv_r6uw,11172
 tinybird/tb/modules/workspace_members.py,sha256=5JdkJgfuEwbq-t6vxkBhYwgsiTDxF790wsa6Xfif9nk,8608
 tinybird/tb/modules/agent/__init__.py,sha256=i3oe3vDIWWPaicdCM0zs7D7BJ1W0k7th93ooskHAV00,54
-tinybird/tb/modules/agent/agent.py,sha256=uVZ0yGOk4uRtF74mPHIYjawIDPSCweb5HCC7eLVuf2k,37339
+tinybird/tb/modules/agent/agent.py,sha256=BodnCrX5cU8x6RYIwnBSLJ2nsHFTdqJCyB0capBKB0g,37708
 tinybird/tb/modules/agent/animations.py,sha256=4WOC5_2BracttmMCrV0H91tXfWcUzQHBUaIJc5FA7tE,3490
 tinybird/tb/modules/agent/banner.py,sha256=l6cO5Fi7lbVKp-GsBP8jf3IkjOWxg2jpAt9NBCy0WR8,4085
 tinybird/tb/modules/agent/command_agent.py,sha256=0Z08rQsir59zQAr-kkOvsKIFpIBsBSTGJJ1VgqqF5WA,3654
 tinybird/tb/modules/agent/compactor.py,sha256=BK5AxZFhrp3xWnsRnYaleiYoIWtVNc-_m650Hsopt8g,13841
-tinybird/tb/modules/agent/explore_agent.py,sha256=gyD5uV5TJwV24eeQiSwhkgfNPb4mtbeH7t2qSdoc18U,4100
+tinybird/tb/modules/agent/explore_agent.py,sha256=QVMPmX6GDW8bh65_X4yXepvAgJmNRQh1S-YwmvGHkjw,4288
 tinybird/tb/modules/agent/file_agent.py,sha256=vE8VChVbG_-uyeBtp3I6QrXeNaPaVlVoNn4c9cGTAJA,2466
 tinybird/tb/modules/agent/memory.py,sha256=vBewB_64L_wHoT4tLT6UX2uxcHwSY880QZ26F9rPqXs,3793
-tinybird/tb/modules/agent/mock_agent.py,sha256=zbAZfAqdSLUtMr2VqO0erWpzjT2F1tTcuYjvHb-gvbA,8023
+tinybird/tb/modules/agent/mock_agent.py,sha256=9_w5-MJDC3vp0U5SkM-ucqhxftfpeUM4FPLl5BAr1_0,8200
 tinybird/tb/modules/agent/models.py,sha256=eokO8XlY-kVJOsbqiVporGUAOCyKAXCO5xgTEK9SM6Y,2208
-tinybird/tb/modules/agent/prompts.py,sha256=a8ZnOZCTGRnxa_dlrTSmeaM9snOI3sMx0nwUdv7aVGI,44165
+tinybird/tb/modules/agent/prompts.py,sha256=eua-QawthfG5X6Kz1dRzOZSjXU8tj0nr4k7onmvBZnk,45792
 tinybird/tb/modules/agent/testing_agent.py,sha256=AtwtJViH7805i7djyBgDb7SSUtDyJnw0TWJu6lBFsrg,2953
 tinybird/tb/modules/agent/utils.py,sha256=alvRNtpPPRZP0sNdOXbzX3uQZHqqeh1daIAg3aquq24,32015
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -98,7 +98,7 @@ tinybird/tb/modules/agent/tools/get_openapi_definition.py,sha256=4TIMO2XzHBMhpt9
 tinybird/tb/modules/agent/tools/mock.py,sha256=Seo4WcYNLL1-SmPXutoaX94_pfOdIb47JXo8dHtUVhg,7106
 tinybird/tb/modules/agent/tools/plan.py,sha256=uAJEHZ-xXIq-EpURJYV7GUyY7IbIgactw9NWeCsIT9Y,2516
 tinybird/tb/modules/agent/tools/request_endpoint.py,sha256=bsLWrMn-ofJM3nn9vm8j_U8fdopVd3H5L0ii6ji-Kuw,4359
-tinybird/tb/modules/agent/tools/run_command.py,sha256=EOHkyK9nktlt5jVz9sQcSt4CN40LO490FjmAy3CuNmE,1793
+tinybird/tb/modules/agent/tools/run_command.py,sha256=B0jyXQW7d0vSBi7abCTlhmEsf6nvwWcVEbRpGatLFBo,2464
 tinybird/tb/modules/agent/tools/secret.py,sha256=8AGTZgHLPg1bxCA2cPMnb-zNutWEwn4emHo7kLjJC5w,4323
 tinybird/tb/modules/agent/tools/test.py,sha256=4XuEWVHLOTSO51Z9xJ08dTjk0j3IWY_JlPtSBO5aaUs,10373
 tinybird/tb/modules/datafile/build.py,sha256=NFKBrusFLU0WJNCXePAFWiEDuTaXpwc0lHlOQWEJ43s,51117
@@ -121,8 +121,8 @@ tinybird/tb_cli_modules/config.py,sha256=IsgdtFRnUrkY8-Zo32lmk6O7u3bHie1QCxLwgp4
 tinybird/tb_cli_modules/exceptions.py,sha256=pmucP4kTF4irIt7dXiG-FcnI-o3mvDusPmch1L8RCWk,3367
 tinybird/tb_cli_modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
 tinybird/tb_cli_modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09K1A,10449
-tinybird-0.0.1.dev291.dist-info/METADATA,sha256=WCSw6JMv71AZt3gonZC3XeWThiDtGWEynv8GH1jSKhE,1845
-tinybird-0.0.1.dev291.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev291.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev291.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev291.dist-info/RECORD,,
+tinybird-0.0.1.dev292.dist-info/METADATA,sha256=d2RNeqvNyqjfXrpEIlGB9lV-Kh2KUu3bD8c99JGeGJQ,1845
+tinybird-0.0.1.dev292.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev292.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev292.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev292.dist-info/RECORD,,