tinybird 0.0.1.dev291__py3-none-any.whl → 1.0.5__py3-none-any.whl

tinybird/tb/client.py

@@ -300,33 +300,33 @@ class TinyB:
         response = self._req(f"/v0/connectors?{urlencode(params)}")
         return response["connectors"]
 
-    def connections(self, connector: Optional[str] = None):
+    def connections(self, connector: Optional[str] = None, datasources: Optional[List[Dict[str, Any]]] = None):
         response = self._req("/v0/connectors")
         connectors = response["connectors"]
+        connectors_to_return = []
+        for c in connectors:
+            if connector and c["service"] != connector:
+                continue
+            if connector == "gcscheduler":
+                continue
+            if datasources and len(datasources) > 0:
+                datasource_ids = [linker["datasource_id"] for linker in c["linkers"]]
+                datasource_names = [ds["name"] for ds in datasources if ds["id"] in datasource_ids]
+                connected_datasources = ", ".join(datasource_names) if len(datasource_names) > 0 else ""
+            else:
+                connected_datasources = str(len(c["linkers"]))
 
-        if connector:
-            return [
+            connectors_to_return.append(
                 {
                     "id": c["id"],
                     "service": c["service"],
                     "name": c["name"],
-                    "connected_datasources": len(c["linkers"]),
+                    "connected_datasources": connected_datasources,
                     **c["settings"],
                 }
-                for c in connectors
-                if c["service"] == connector
-            ]
-        return [
-            {
-                "id": c["id"],
-                "service": c["service"],
-                "name": c["name"],
-                "connected_datasources": len(c["linkers"]),
-                **c["settings"],
-            }
-            for c in connectors
-            if c["service"] != "gcscheduler"
-        ]
+            )
+
+        return connectors_to_return
 
     def get_datasource(self, ds_name: str, used_by: bool = False) -> Dict[str, Any]:
         params = {
@@ -711,7 +711,7 @@ class TinyB:
         return self._req(f"/{version}/user/workspaces/?with_environments=true&only_environments=true")
 
     def branches(self):
-        return self._req("/v0/environments")
+        return self._req("/v1/environments")
 
     def releases(self, workspace_id):
         return self._req(f"/v0/workspaces/{workspace_id}/releases")
@@ -740,7 +740,7 @@ class TinyB:
         }
         if ignore_datasources:
             params["ignore_datasources"] = ",".join(ignore_datasources)
-        return self._req(f"/v0/environments?{urlencode(params)}", method="POST", data=b"")
+        return self._req(f"/v1/environments?{urlencode(params)}", method="POST", data=b"")
 
     def branch_workspace_data(
         self,
@@ -996,10 +996,92 @@ class TinyB:
             data=json.dumps(connection_params),
         )
 
-    def kafka_list_topics(self, connection_id: str, timeout=5):
-        resp = self._req(f"/v0/connectors/{connection_id}/preview?preview_activity=false", timeout=timeout)
+    def kafka_list_topics(self, connection_id: str, timeout=10, retries=3):
+        resp = self._req(
+            f"/v0/connectors/{connection_id}/preview?preview_activity=false",
+            timeout=timeout,
+            retries=retries,
+        )
         return [x["topic"] for x in resp["preview"]]
 
+    def kafka_preview_group(self, connection_id: str, topic: str, group_id: str, timeout=30):
+        params = {
+            "log": "previewGroup",
+            "kafka_group_id": group_id,
+            "kafka_topic": topic,
+            "preview_group": "true",
+        }
+        return self._req(f"/v0/connectors/{connection_id}/preview?{urlencode(params)}", method="GET", timeout=timeout)
+
+    def kafka_preview_topic(self, connection_id: str, topic: str, group_id: str, timeout: int = 30) -> Dict[str, Any]:
+        """Preview a Kafka topic and return structured preview data.
+
+        Args:
+            connection_id: The ID of the Kafka connection
+            topic: The Kafka topic name to preview
+            group_id: The Kafka consumer group ID
+            timeout: Request timeout in seconds
+
+        Returns:
+            A dictionary containing:
+                - analysis: Dictionary with columns information
+                - preview: Dictionary with data and meta arrays
+                - earliestTimestamp: The earliest message timestamp (if available)
+        """
+        params = {
+            "max_records": "12",
+            "preview_activity": "true",
+            "preview_earliest_timestamp": "true",
+            "kafka_topic": topic,
+            "kafka_group_id": group_id,
+            "log": "previewTopic",
+        }
+        response = self._req(
+            f"/v0/connectors/{connection_id}/preview?{urlencode(params)}", method="GET", timeout=timeout
+        )
+
+        if not response:
+            return {
+                "analysis": {"columns": []},
+                "preview": {"data": [], "meta": []},
+                "earliestTimestamp": "",
+            }
+
+        # Extract preview data (similar to TypeScript previewKafkaTopic)
+        preview_data = response.get("preview", [])
+        if not preview_data:
+            return {
+                "analysis": {"columns": []},
+                "preview": {"data": [], "meta": []},
+                "earliestTimestamp": "",
+            }
+
+        topic_preview = preview_data[0]
+        analysis = topic_preview.get("analysis", {})
+        deserialized = topic_preview.get("deserialized", {})
+
+        # Extract columns from analysis
+        columns = analysis.get("columns", []) if analysis else []
+
+        # Extract data and meta from deserialized
+        base_data = deserialized.get("data", []) if deserialized else []
+        base_meta = deserialized.get("meta", []) if deserialized else []
+
+        # Extract earliest timestamp
+        earliest = response.get("earliest", [])
+        earliest_timestamp = earliest[0].get("timestamp", "") if earliest else ""
+
+        return {
+            "analysis": {
+                "columns": columns,
+            },
+            "preview": {
+                "data": base_data,
+                "meta": base_meta,
+            },
+            "earliestTimestamp": earliest_timestamp,
+        }
+
     def get_gcp_service_account_details(self) -> Dict[str, Any]:
         return self._req("/v0/datasources-bigquery-credentials")
 
@@ -1390,7 +1472,3 @@ class TinyB:
 
     def delete_tag(self, name: str):
         self._req(f"/v0/tags/{name}", method="DELETE")
-
-    def explore_data(self, prompt: str) -> str:
-        params = urlencode({"prompt": prompt, "host": self.host, "origin": "cli"})
-        return self._req(f"/v1/agents/explore?{params}")

tinybird/tb/config.py

@@ -40,6 +40,23 @@ CLOUD_HOSTS = {
     "https://ui.europe-west2.gcp.tinybird.co": "https://cloud.tinybird.co/gcp/europe-west2",
 }
 
+CH_HOSTS = {
+    "https://api.tinybird.co": "https://clickhouse.tinybird.co",
+    "https://api.us-east.tinybird.co": "https://clickhouse.us-east.tinybird.co",
+    "https://api.us-east.aws.tinybird.co": "https://clickhouse.us-east-1.aws.tinybird.co",
+    "https://api.us-west-2.aws.tinybird.co": "https://clickhouse.us-west-2.aws.tinybird.co",
+    "https://api.eu-central-1.aws.tinybird.co": "https://clickhouse.eu-central-1.aws.tinybird.co",
+    "https://api.eu-west-1.aws.tinybird.co": "https://clickhouse.eu-west-1.aws.tinybird.co",
+    "https://api.europe-west2.gcp.tinybird.co": "https://clickhouse.europe-west2.gcp.tinybird.co",
+    "https://api.ap-east.aws.tinybird.co": "https://clickhouse.ap-east.aws.tinybird.co",
+    "https://ui.tinybird.co": "https://clickhouse.tinybird.co",
+    "https://ui.us-east.tinybird.co": "https://clickhouse.us-east.tinybird.co",
+    "https://ui.us-east.aws.tinybird.co": "https://clickhouse.us-east.aws.tinybird.co",
+    "https://ui.us-west-2.aws.tinybird.co": "https://clickhouse.us-west-2.aws.tinybird.co",
+    "https://ui.eu-central-1.aws.tinybird.co": "https://clickhouse.eu-central-1.aws.tinybird.co",
+    "https://ui.europe-west2.gcp.tinybird.co": "https://clickhouse.europe-west2.gcp.tinybird.co",
+}
+
 
 def get_config(
     host: Optional[str], token: Optional[str], semver: Optional[str] = None, config_file: Optional[str] = None
@@ -85,6 +102,13 @@ def get_display_cloud_host(api_host: str) -> str:
     return CLOUD_HOSTS.get(api_host, api_host)
 
 
+def get_clickhouse_host(api_host: str) -> str:
+    is_local = "localhost" in api_host
+    if is_local:
+        return "http://localhost:7182"
+    return f"{CH_HOSTS.get(api_host, api_host.replace('api.', 'clickhouse.'))}:443"
+
+
 class FeatureFlags:
     @classmethod
     def ignore_sql_errors(cls) -> bool:  # Context: #1155

tinybird/tb/modules/agent/agent.py

@@ -6,8 +6,7 @@ import sys
 import urllib.parse
 import uuid
 from functools import partial
-from pathlib import Path
-from typing import Any, Optional
+from typing import Any, Callable, Optional
 
 import click
 import humanfriendly
@@ -17,7 +16,7 @@ from requests import Response
 
 from tinybird.tb.check_pypi import CheckPypi
 from tinybird.tb.client import TinyB
-from tinybird.tb.config import CURRENT_VERSION, get_display_cloud_host
+from tinybird.tb.config import CURRENT_VERSION, get_clickhouse_host, get_display_cloud_host
 from tinybird.tb.modules.agent.animations import ThinkingAnimation
 from tinybird.tb.modules.agent.banner import display_banner
 from tinybird.tb.modules.agent.command_agent import CommandAgent
@@ -27,17 +26,18 @@ from tinybird.tb.modules.agent.file_agent import FileAgent
 from tinybird.tb.modules.agent.memory import (
     clear_history,
     clear_messages,
-    get_last_messages_from_last_user_prompt,
     save_messages,
 )
 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
@@ -65,16 +65,17 @@ from tinybird.tb.modules.common import (
     echo_safe_humanfriendly_tables_format_pretty_table,
     get_region_from_host,
     get_regions,
+    sys_exit,
     update_cli,
 )
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.deployment_common import create_deployment
-from tinybird.tb.modules.exceptions import CLIBuildException, CLIDeploymentException, CLIMockException
+from tinybird.tb.modules.exceptions import CLIBuildException, CLIDeploymentException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.llm import LLM
 from tinybird.tb.modules.local_common import get_tinybird_local_client
 from tinybird.tb.modules.login_common import login
-from tinybird.tb.modules.mock_common import append_mock_data, create_mock_data
+from tinybird.tb.modules.mock_common import append_mock_data
 from tinybird.tb.modules.project import Project
 from tinybird.tb.modules.test_common import run_tests as run_tests_common
 
@@ -89,6 +90,7 @@ class TinybirdAgent:
         project: Project,
         dangerously_skip_permissions: bool,
         prompt_mode: bool,
+        feature: Optional[str] = None,
     ):
         self.token = token
         self.user_token = user_token
@@ -98,14 +100,12 @@ class TinybirdAgent:
         self.project = project
         self.thinking_animation = ThinkingAnimation()
         self.confirmed_plan_id: Optional[str] = None
-        if prompt_mode:
-            self.messages: list[ModelMessage] = get_last_messages_from_last_user_prompt()
-        else:
-            self.messages = []
+        self.feature = feature
+        self.messages: list[ModelMessage] = []
         cli_config = CLIConfig.get_project_config()
         regions = get_regions(cli_config)
         self.agent = Agent(
-            model=create_model(user_token, host, workspace_id),
+            model=create_model(user_token, host, workspace_id, feature=feature),
             deps_type=TinybirdAgentContext,
             instructions=[agent_system_prompt],
             tools=[
@@ -282,7 +282,14 @@ class TinybirdAgent:
 - API Host: {ctx.deps.local_host}
 - Token: {ctx.deps.local_token}
 - UI Dashboard URL: {get_display_cloud_host(ctx.deps.local_host)}/{ctx.deps.workspace_name}
-            """
+- ClickHouse native HTTP interface:
+    - Protocol: HTTP
+    - Host: localhost
+    - Port: 7182
+    - Full URL: http://localhost:7182
+    - Username: {ctx.deps.workspace_name}  # Optional, for identification purposes
+    - Password: __TB_CLOUD_TOKEN__          # Your Tinybird auth token
+"""
 
         @self.agent.instructions
         def get_cloud_host(ctx: RunContext[TinybirdAgentContext]) -> str:
@@ -300,6 +307,7 @@ class TinybirdAgent:
 
             region_provider = region["provider"]
             region_name = region["name"]
+            ch_host = get_clickhouse_host(ctx.deps.host)
             return f"""
 # Tinybird Cloud info (region details):
 - API Host: {ctx.deps.host}
@@ -308,6 +316,14 @@ class TinybirdAgent:
 - Region provider: {region_provider}
 - Region name: {region_name}
 - UI Dashboard URL: {get_display_cloud_host(ctx.deps.host)}/{ctx.deps.workspace_name}
+- ClickHouse native HTTP interface:
+    - Protocol: HTTPS
+    - Host: {ch_host.replace("https://", "").replace(":443", "")}
+    - Port: 443 (HTTPS)
+    - SSL/TLS: Required (enabled)
+    - Full URL: {ch_host}
+    - Username: {ctx.deps.workspace_name}  # Optional, for identification purposes
+    - Password: __TB_CLOUD_TOKEN__          # Your Tinybird auth token
 """
 
         @self.agent.instructions
@@ -322,6 +338,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()
@@ -355,7 +379,6 @@ class TinybirdAgent:
             build_project=partial(build_project, project=project, config=config),
             deploy_project=partial(deploy_project, project=project, config=config),
             deploy_check_project=partial(deploy_check_project, project=project, config=config),
-            mock_data=partial(mock_data, project=project, config=config),
             append_data_local=partial(append_data_local, config=config),
             append_data_cloud=partial(append_data_cloud, config=config),
             analyze_fixture=partial(analyze_fixture, config=config),
@@ -363,7 +386,7 @@ class TinybirdAgent:
             execute_query_local=partial(execute_query_local, config=config),
             request_endpoint_cloud=partial(request_endpoint_cloud, config=config),
             request_endpoint_local=partial(request_endpoint_local, config=config),
-            build_project_test=partial(build_project_test, project=project, client=test_client),
+            build_project_test=partial(build_project_test, project=project, client=test_client, config=config),
             get_pipe_data_test=partial(get_pipe_data_test, client=test_client),
             get_datasource_datafile_cloud=partial(get_datasource_datafile_cloud, config=config),
             get_datasource_datafile_local=partial(get_datasource_datafile_local, config=config),
@@ -372,7 +395,7 @@ class TinybirdAgent:
             get_connection_datafile_cloud=partial(get_connection_datafile_cloud, config=config),
             get_connection_datafile_local=partial(get_connection_datafile_local, config=config),
             get_project_files=project.get_project_files,
-            run_tests=partial(run_tests, project=project, client=test_client),
+            run_tests=partial(run_tests, project=project, client=test_client, config=config),
             folder=folder,
             thinking_animation=self.thinking_animation,
             workspace_id=self.workspace_id,
@@ -402,7 +425,6 @@ class TinybirdAgent:
         save_messages(new_messages)
         self.thinking_animation.stop()
         click.echo(result.output)
-        self.echo_usage(config)
 
     async def run_iter(self, user_prompt: str, config: dict[str, Any], run_id: Optional[str] = None) -> None:
         model = create_model(self.user_token, self.host, self.workspace_id, run_id=run_id)
@@ -429,34 +451,67 @@ class TinybirdAgent:
             self.messages.extend(new_messages)
             save_messages(new_messages)
             self.thinking_animation.stop()
-            self.echo_usage(config)
 
-    def echo_usage(self, config: dict[str, Any]) -> None:
+    def echo_usage(self, config: dict[str, Any], show_credits: bool = False) -> None:
         try:
             client = _get_tb_client(config["user_token"], config["host"])
             workspace_id = config.get("id", "")
             workspace = client.workspace(workspace_id, with_organization=True, version="v1")
+            is_free_plan = workspace["organization"]["plan"].get("billing") == "free_shared_infrastructure_usage"
+
+            if not is_free_plan and not show_credits:
+                return
+
             limits_data = client.organization_limits(workspace["organization"]["id"])
-            ai_credits_limits = limits_data.get("limits", {}).get("ai_credits", {})
-            current_ai_credits = ai_credits_limits.get("quantity") or 0
-            ai_credits = ai_credits_limits.get("max") or 0
-            remaining_credits = round(max(ai_credits - current_ai_credits, 0), 2)
-            current_ai_credits = round(min(ai_credits, current_ai_credits), 2)
-            if not ai_credits:
+            llm_usage_limits = limits_data.get("limits", {}).get("llm_usage", {})
+            current_llm_usage = llm_usage_limits.get("quantity") or 0
+            llm_usage = llm_usage_limits.get("max") or 0
+            remaining_credits = round(max(llm_usage - current_llm_usage, 0), 2)
+            current_llm_usage = round(min(llm_usage, current_llm_usage), 2)
+
+            if not llm_usage:
                 return
-            warning_threshold = ai_credits * 0.8
-            message_color = FeedbackManager.warning if current_ai_credits >= warning_threshold else FeedbackManager.gray
+
+            def get_message(current_llm_usage, llm_usage: int) -> tuple[Callable[..., str], str, bool]:
+                warning_threshold = llm_usage * 0.8
+                ui_host = get_display_cloud_host(config["host"])
+
+                if is_free_plan:
+                    upgrade_link = f"{ui_host}/organizations/{workspace['organization']['name']}/upgrade?from=agent"
+                    if current_llm_usage >= llm_usage:
+                        return (
+                            FeedbackManager.error,
+                            f" You have reached the maximum number of credits. Please upgrade to continue using Tinybird Code: {upgrade_link}",
+                            True,
+                        )
+                    if current_llm_usage >= warning_threshold:
+                        return (
+                            FeedbackManager.warning,
+                            f" You are reaching the maximum number of credits. Please upgrade to continue using Tinybird Code: {upgrade_link}",
+                            False,
+                        )
+                return FeedbackManager.gray, "", False
+
+            message_color, upgrade_message, should_exit = get_message(current_llm_usage, llm_usage)
             click.echo(
                 message_color(
-                    message=f"{remaining_credits} credits left ({current_ai_credits}/{ai_credits}). You can continue using Tinybird Code. Limits will be enforced soon."
+                    message=f"{remaining_credits} credits left ({current_llm_usage}/{llm_usage}).{upgrade_message}"
                 )
             )
+
+            if should_exit:
+                sys_exit("tinybird_code_error", "Maximum number of credits reached")
+
         except Exception:
             pass
 
 
 def run_agent(
-    config: dict[str, Any], project: Project, dangerously_skip_permissions: bool, prompt: Optional[str] = None
+    config: dict[str, Any],
+    project: Project,
+    dangerously_skip_permissions: bool,
+    prompt: Optional[str] = None,
+    feature: Optional[str] = None,
 ):
     if not prompt:
         latest_version = CheckPypi().get_latest_version()
@@ -471,10 +526,13 @@ def run_agent(
             )
             if yes:
                 update_cli()
-    click.echo(FeedbackManager.highlight(message="» Initializing Tinybird Code..."))
-    token = config.get("token", None)
-    host = config.get("host", None)
-    user_token = config.get("user_token", None)
+
+    if not prompt:
+        click.echo(FeedbackManager.highlight(message="» Initializing Tinybird Code..."))
+
+    token = config.get("token", "")
+    host = config.get("host", "")
+    user_token = config.get("user_token", "")
     workspace_id = config.get("id", "")
     workspace_name = config.get("name", "")
     try:
@@ -491,8 +549,8 @@ def run_agent(
                 login(host, auth_host="https://cloud.tinybird.co", workspace=None, interactive=False, method="browser")
                 cli_config = CLIConfig.get_project_config()
                 config = {**config, **cli_config.to_dict()}
-                token = cli_config.get_token()
-                user_token = cli_config.get_user_token()
+                token = cli_config.get_token() or ""
+                user_token = cli_config.get_user_token() or ""
                 host = cli_config.get_host()
                 workspace_id = cli_config.get("id", "")
                 workspace_name = cli_config.get("name", "")
@@ -530,6 +588,7 @@ def run_agent(
             project,
             dangerously_skip_permissions,
             prompt_mode,
+            feature,
         )
 
         # Print mode: run once with the provided prompt and exit
@@ -551,7 +610,6 @@ def run_agent(
             )
         )
         agent.echo_usage(config)
-        click.echo()
 
     except Exception as e:
         click.echo(FeedbackManager.error(message=f"Failed to initialize agent: {e}"))
@@ -604,6 +662,9 @@ def run_agent(
                 elif user_input.lower() == "/help":
                     subprocess.run(["tb", "--help"], check=True)
                     continue
+                elif user_input.lower() == "/usage":
+                    agent.echo_usage(config, show_credits=True)
+                    continue
                 elif user_input.strip() == "":
                     continue
                 else:
@@ -642,6 +703,7 @@ def build_project(
     build_error = build_process(
         project=project,
         tb_client=client,
+        config=config,
         watch=False,
         silent=silent,
         exit_on_error=False,
@@ -654,11 +716,13 @@ def build_project(
 def build_project_test(
     client: TinyB,
     project: Project,
+    config: dict[str, Any],
     silent: bool = False,
 ) -> None:
     build_error = build_process(
         project=project,
         tb_client=client,
+        config=config,
         watch=False,
         silent=silent,
         exit_on_error=False,
@@ -703,36 +767,6 @@ def append_data_cloud(config: dict[str, Any], datasource_name: str, path: str) -
     append_mock_data(client, datasource_name, path)
 
 
-def mock_data(
-    config: dict[str, Any],
-    project: Project,
-    datasource_name: str,
-    data_format: str,
-    rows: int,
-    context: Optional[str] = None,
-) -> list[dict[str, Any]]:
-    client = get_tinybird_local_client(config, test=False, silent=False)
-    cli_config = CLIConfig.get_project_config()
-    datasource_path = project.get_resource_path(datasource_name, "datasource")
-
-    if not datasource_path:
-        raise CLIMockException(f"Datasource {datasource_name} not found")
-
-    datasource_content = Path(datasource_path).read_text()
-    return create_mock_data(
-        datasource_name,
-        datasource_content,
-        rows,
-        context or "",
-        cli_config,
-        config,
-        cli_config.get_user_token() or "",
-        client,
-        data_format,
-        project.folder,
-    )
-
-
 def analyze_fixture(config: dict[str, Any], fixture_path: str, format: str = "json") -> dict[str, Any]:
     local_client = get_tinybird_local_client(config, test=False, silent=True)
     meta, _data = _analyze(fixture_path, local_client, format)
@@ -811,9 +845,11 @@ def get_connection_datafile_local(config: dict[str, Any], connection_name: str)
         return "Connection not found"
 
 
-def run_tests(client: TinyB, project: Project, pipe_name: Optional[str] = None) -> Optional[str]:
+def run_tests(
+    client: TinyB, project: Project, config: dict[str, Any], pipe_name: Optional[str] = None
+) -> Optional[str]:
     try:
-        return run_tests_common(name=(pipe_name,) if pipe_name else (), project=project, client=client)
+        return run_tests_common(name=(pipe_name,) if pipe_name else (), project=project, client=client, config=config)
     except SystemExit as e:
         raise Exception(e.args[0])
 

tinybird/tb/modules/agent/banner.py

@@ -54,20 +54,20 @@ def display_banner():
         """Convert RGB values to ANSI escape code"""
         if use_truecolor:
             return f"\033[38;2;{r};{g};{b}m"
-        else:
-            # Convert to 8-bit color (256 color palette)
-            # Simple approximation: map RGB to 216-color cube + grayscale
-            if r == g == b:
-                # Grayscale
-                gray = int(r / 255 * 23) + 232
-                return f"\033[38;5;{gray}m"
-            else:
-                # Color cube (6x6x6)
-                r_idx = int(r / 255 * 5)
-                g_idx = int(g / 255 * 5)
-                b_idx = int(b / 255 * 5)
-                color_idx = 16 + (36 * r_idx) + (6 * g_idx) + b_idx
-                return f"\033[38;5;{color_idx}m"
+
+        # Convert to 8-bit color (256 color palette)
+        # Simple approximation: map RGB to 216-color cube + grayscale
+        if r == g == b:
+            # Grayscale
+            gray = int(r / 255 * 23) + 232
+            return f"\033[38;5;{gray}m"
+
+        # Color cube (6x6x6)
+        r_idx = int(r / 255 * 5)
+        g_idx = int(g / 255 * 5)
+        b_idx = int(b / 255 * 5)
+        color_idx = 16 + (36 * r_idx) + (6 * g_idx) + b_idx
+        return f"\033[38;5;{color_idx}m"
 
     # Define solid color (corresponding to #27f795)
     solid_color = [39, 247, 149]  # #27f795 in RGB
@@ -84,4 +84,4 @@ def display_banner():
                 colored_line += f"{color_code}{char}"
 
         click.echo(colored_line + reset)
-    click.echo()
+    click.echo("")

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,