tinybird 0.0.1.dev250__py3-none-any.whl → 0.0.1.dev251__py3-none-any.whl

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.dev250'
-__revision__ = '1c47b89'
+__version__ = '0.0.1.dev251'
+__revision__ = 'c11760c'

tinybird/tb/modules/agent/agent.py

@@ -41,6 +41,7 @@ from tinybird.tb.modules.agent.tools.deploy import deploy
 from tinybird.tb.modules.agent.tools.deploy_check import deploy_check
 from tinybird.tb.modules.agent.tools.explore import explore_data
 from tinybird.tb.modules.agent.tools.get_endpoint_stats import get_endpoint_stats
+from tinybird.tb.modules.agent.tools.get_openapi_definition import get_openapi_definition
 from tinybird.tb.modules.agent.tools.mock import mock
 from tinybird.tb.modules.agent.tools.plan import plan
 from tinybird.tb.modules.agent.tools.preview_datafile import preview_datafile
@@ -114,6 +115,7 @@ You have access to the following tools:
 9. `read_fixture_data` - Read a fixture data file present in the project folder.
 10. `append` - Append existing fixture to a datasource.
 11. `get_endpoint_stats` - Get metrics of the requests to an endpoint.
+12. `get_openapi_definition` - Get the OpenAPI definition for all endpoints that are built/deployed to Tinybird Cloud or Local.
 
 # When creating or updating datafiles:
 1. Use `plan` tool to plan the creation or update of resources.
@@ -195,6 +197,12 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
                 Tool(
                     get_endpoint_stats, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True
                 ),
+                Tool(
+                    get_openapi_definition,
+                    docstring_format="google",
+                    require_parameter_descriptions=True,
+                    takes_ctx=True,
+                ),
             ],
         )
 
@@ -227,6 +235,9 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
                 thinking_animation=thinking_animation,
                 workspace_name=self.project.workspace_name,
                 dangerously_skip_permissions=self.dangerously_skip_permissions,
+                token=self.token,
+                user_token=self.user_token,
+                host=self.host,
             ),
             message_history=self.messages,
         )
@@ -243,11 +254,13 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
         )
         click.echo(result.output)
         click.echo("\n")
-        click.echo(f"Input tokens: {request_tokens}")
-        click.echo(f"Output tokens: {response_tokens}")
-        click.echo(f"Total tokens: {total_tokens}")
-        click.echo(f"Cost: ${cost:.6f}")
-        click.echo("\n")
+
+        if "@tinybird.co" in config.get("user_email", ""):
+            click.echo(f"Input tokens: {request_tokens}")
+            click.echo(f"Output tokens: {response_tokens}")
+            click.echo(f"Total tokens: {total_tokens}")
+            click.echo(f"Cost: ${cost:.6f}")
+            click.echo("\n")
 
 
 def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permissions: bool):
@@ -378,7 +391,12 @@ def append_data(config: dict[str, Any], datasource_name: str, path: str) -> None
 
 
 def mock_data(
-    config: dict[str, Any], project: Project, datasource_name: str, data_format: str, rows: int
+    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()
@@ -388,12 +406,11 @@ def mock_data(
         raise CLIMockException(f"Datasource {datasource_name} not found")
 
     datasource_content = Path(datasource_path).read_text()
-    prompt = ""
     return create_mock_data(
         datasource_name,
         datasource_content,
         rows,
-        prompt,
+        context or "",
         cli_config,
         config,
         cli_config.get_user_token() or "",

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

@@ -0,0 +1,39 @@
+import click
+import requests
+from pydantic_ai import RunContext
+
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+
+def get_openapi_definition(ctx: RunContext[TinybirdAgentContext], cloud: bool = True):
+    """Get the OpenAPI definition for all endpoints in the project:
+
+    Args:
+        cloud (bool): Whether to get the OpenAPI spec from cloud or local. Optional.
+
+    Returns:
+        str: The OpenAPI spec for all endpoints in the project.
+    """
+
+    cloud_or_local = "Cloud" if cloud else "Local"
+    ctx.deps.thinking_animation.stop()
+
+    click.echo(FeedbackManager.highlight(message=f"» Accessing Tinybird {cloud_or_local} endpoints"))
+    try:
+        url = f"{ctx.deps.host}/v0/pipes/openapi.json"
+
+        result = requests.get(url, headers={"Authorization": f"Bearer {ctx.deps.token}"})
+        if result.status_code != 200:
+            raise Exception(f"Failed to get OpenAPI spec: {result.status_code} {result.text}")
+
+        openapi_spec = result.text
+
+        click.echo(FeedbackManager.success(message="✓ Done!"))
+        click.echo()
+        ctx.deps.thinking_animation.start()
+        return f"OpenAPI spec for {cloud_or_local} endpoints: {openapi_spec}. <dev_note>Add the base url to each path so the user can copy paste if needed: {ctx.deps.host}</dev_note>"
+    except Exception as e:
+        click.echo(FeedbackManager.error(message=f"Error getting OpenAPI spec: {e}"))
+        ctx.deps.thinking_animation.start()
+        return f"Error getting OpenAPI spec: {e}"

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

@@ -1,3 +1,5 @@
+from typing import Optional
+
 import click
 from pydantic_ai import RunContext
 
@@ -6,13 +8,20 @@ from tinybird.tb.modules.datafile.fixture import persist_fixture
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
-def mock(ctx: RunContext[TinybirdAgentContext], datasource_name: str, data_format: str, rows: int) -> str:
+def mock(
+    ctx: RunContext[TinybirdAgentContext],
+    datasource_name: str,
+    data_format: str,
+    rows: int,
+    context: Optional[str] = None,
+) -> str:
     """Create mock data for a datasource
 
     Args:
         datasource_name: Name of the datasource to create mock data for
         data_format: Format of the mock data to create. Options: ndjson, csv
         rows: Number of rows to create. If not provided, the default is 10
+        context: Extra context to be used to generate the mock data. Optional.
 
     Returns:
         str: Message indicating the success or failure of the mock data generation
@@ -35,7 +44,7 @@ def mock(ctx: RunContext[TinybirdAgentContext], datasource_name: str, data_forma
             return "User cancelled mock data generation. Stop mock data generation."
 
         click.echo(FeedbackManager.highlight(message=f"\n» Generating mock data for {datasource_name}..."))
-        data = ctx.deps.mock_data(datasource_name=datasource_name, data_format=data_format, rows=rows)
+        data = ctx.deps.mock_data(datasource_name=datasource_name, data_format=data_format, rows=rows, context=context)
         fixture_path = persist_fixture(datasource_name, data, ctx.deps.folder, format=data_format)
         ctx.deps.append_data(datasource_name=datasource_name, path=str(fixture_path))
         click.echo(FeedbackManager.success(message=f"✓ Data generated for {datasource_name}"))
@@ -43,6 +52,22 @@ def mock(ctx: RunContext[TinybirdAgentContext], datasource_name: str, data_forma
         return f"Mock data generated successfully for datasource {datasource_name}"
     except Exception as e:
         ctx.deps.thinking_animation.stop()
-        click.echo(FeedbackManager.error(message=e))
+        error_message = str(e)
+        click.echo(FeedbackManager.error(message=error_message))
+        try:
+            if "in quarantine" in error_message:
+                click.echo(
+                    FeedbackManager.highlight(message=f"\n» Looking for errors in {datasource_name}_quarantine...")
+                )
+                query = f"select * from {datasource_name}_quarantine order by insertion_date desc limit 5 FORMAT CSVWithNames"
+                quarantine_data = ctx.deps.execute_local_query(query=query)
+                error_message = (
+                    error_message
+                    + f"\nThese are the first 5 rows of the quarantine table for datasource '{datasource_name}':\n{quarantine_data}. Use again `mock` tool but add this issue to the context."
+                )
+
+        except Exception as quarantine_error:
+            error_message = error_message + f"\nError accessing to {datasource_name}_quarantine: {quarantine_error}"
+
         ctx.deps.thinking_animation.start()
-        return f"Error generating mock data: {e}"
+        return f"Error generating mock data: {error_message}"

tinybird/tb/modules/agent/utils.py

@@ -45,6 +45,9 @@ class TinybirdAgentContext(BaseModel):
     execute_cloud_query: Callable[..., str]
     execute_local_query: Callable[..., str]
     dangerously_skip_permissions: bool
+    token: str
+    user_token: str
+    host: str
 
 
 default_style = PromptStyle.from_dict(

{tinybird-0.0.1.dev250.dist-info → tinybird-0.0.1.dev251.dist-info}/METADATA

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

{tinybird-0.0.1.dev250.dist-info → tinybird-0.0.1.dev251.dist-info}/RECORD

@@ -17,7 +17,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=VsfwyuLYgBIFXG2ts697b2_s6ncyZZMxVTvvxBR-C68,247
+tinybird/tb/__cli__.py,sha256=ldJiOCMSMd-ptHqIVlOItLL_kL3ehE--VamyNzl-O8E,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
 tinybird/tb/client.py,sha256=pJbdkWMXGAqKseNAvdsRRnl_c7I-DCMB0dWCQnG82nU,54146
@@ -68,13 +68,13 @@ tinybird/tb/modules/watch.py,sha256=No0bK1M1_3CYuMaIgylxf7vYFJ72lTJe3brz6xQ-mJo,
 tinybird/tb/modules/workspace.py,sha256=Q_8HcxMsNg8QG9aBlwcWS2umrDP5IkTIHqqz3sfmGuc,11341
 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=OLJFtfgwWa3rNjf8Uwu2dtKVhxnTfZD__lXlAGQTUL4,20532
+tinybird/tb/modules/agent/agent.py,sha256=Uxo2FNNu-YW-J-AbZizZe0eKYKdZRwbPJCgA9EsXlgs,21208
 tinybird/tb/modules/agent/animations.py,sha256=4WOC5_2BracttmMCrV0H91tXfWcUzQHBUaIJc5FA7tE,3490
 tinybird/tb/modules/agent/banner.py,sha256=KX_e467uiy1gWOZ4ofTZt0GCFGQqHQ_8Ob27XLQqda0,3053
 tinybird/tb/modules/agent/memory.py,sha256=H6SJK--2L5C87B7AJd_jMqsq3sCvFvZwZXmajuT0GBE,1171
 tinybird/tb/modules/agent/models.py,sha256=LW1D27gjcd_jwFmghEzteCgToDfodX2B6B5S8BYbysw,735
 tinybird/tb/modules/agent/prompts.py,sha256=wbe6vUnm-fskceWgP13R5VW1v_YF7_wLDe-wBN6rlWw,6998
-tinybird/tb/modules/agent/utils.py,sha256=bqE6Edr6giOVjuxusHsNYiMp4F0HV1aek2ZE9r1URPg,26471
+tinybird/tb/modules/agent/utils.py,sha256=dWdjtptsBRKUAtg6wz2SNf5fmmGKMG-WKNk9miNWZ-s,26520
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tinybird/tb/modules/agent/tools/append.py,sha256=Db576-ziulykJyaWl03zUp-KI-NVURh0WcFIAPNbNPw,1974
 tinybird/tb/modules/agent/tools/build.py,sha256=LhzJMx6tbxC7gogIrxhfKJc-SDgoSR-FC6IunfaCdn8,758
@@ -83,7 +83,8 @@ tinybird/tb/modules/agent/tools/deploy.py,sha256=WrsSlaufKGOBx0S13uoMQQH2DnKue5L
 tinybird/tb/modules/agent/tools/deploy_check.py,sha256=VqMYC7l3_cihmmM_pi8w1t8rJ3P0xDc7pHs_st9k-9Q,684
 tinybird/tb/modules/agent/tools/explore.py,sha256=ihALc_kBcsjrKT3hZyicqyIowB0g_K3AtNNi-5uz9-8,412
 tinybird/tb/modules/agent/tools/get_endpoint_stats.py,sha256=_3wAvDykJitIOb5BRnP7wCy6y06y1qlULHLWB-MvS2M,1705
-tinybird/tb/modules/agent/tools/mock.py,sha256=1DeHv14M1dzzKV78kKte9xWQg7RUdQIdXH8KilnQPP0,2304
+tinybird/tb/modules/agent/tools/get_openapi_definition.py,sha256=9cQ-SUeB1NVhPJN1s8aQh9KQxqI9-DEEW1Ot5r2JbOk,1575
+tinybird/tb/modules/agent/tools/mock.py,sha256=Omog_gdEdm8YuBXNrJdHwxHqjL_ji9UIr75mALF4ozI,3408
 tinybird/tb/modules/agent/tools/plan.py,sha256=pr6LnItz6vlOeCG8GE459ExsrBEG0KLx-g02SZGNjXU,1217
 tinybird/tb/modules/agent/tools/preview_datafile.py,sha256=e9q5fR0afApcrntzFrnuHmd10ex7MG_GM6T0Pwc9bRI,850
 tinybird/tb/modules/agent/tools/read_fixture_data.py,sha256=rvTdVlZsu3rQTSWqXzpFt4LEwnBcMLIT8hlI5C7MVN4,1430
@@ -107,8 +108,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.dev250.dist-info/METADATA,sha256=5urJcqt4JiSsBmETS17ZeDZ20R3QVzATLCXLNuuIiSQ,1733
-tinybird-0.0.1.dev250.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev250.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev250.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev250.dist-info/RECORD,,
+tinybird-0.0.1.dev251.dist-info/METADATA,sha256=7hx0f5-Lb5i3niyVL2AsZ9LcJzp3cTAoX0QHuATZv5A,1733
+tinybird-0.0.1.dev251.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev251.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev251.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev251.dist-info/RECORD,,