tinybird 0.0.1.dev249__py3-none-any.whl → 0.0.1.dev251__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of tinybird might be problematic. Click here for more details.
- tinybird/prompts.py +1 -1
- tinybird/sql_template.py +5 -1
- tinybird/tb/__cli__.py +2 -2
- tinybird/tb/modules/agent/agent.py +68 -27
- tinybird/tb/modules/agent/animations.py +26 -7
- tinybird/tb/modules/agent/prompts.py +4 -2
- tinybird/tb/modules/agent/tools/append.py +13 -22
- tinybird/tb/modules/agent/tools/create_datafile.py +29 -25
- tinybird/tb/modules/agent/tools/deploy.py +16 -25
- tinybird/tb/modules/agent/tools/get_endpoint_stats.py +51 -0
- tinybird/tb/modules/agent/tools/get_openapi_definition.py +39 -0
- tinybird/tb/modules/agent/tools/mock.py +43 -29
- tinybird/tb/modules/agent/tools/plan.py +13 -23
- tinybird/tb/modules/agent/tools/read_fixture_data.py +8 -0
- tinybird/tb/modules/agent/utils.py +48 -1
- {tinybird-0.0.1.dev249.dist-info → tinybird-0.0.1.dev251.dist-info}/METADATA +1 -1
- {tinybird-0.0.1.dev249.dist-info → tinybird-0.0.1.dev251.dist-info}/RECORD +20 -18
- {tinybird-0.0.1.dev249.dist-info → tinybird-0.0.1.dev251.dist-info}/WHEEL +0 -0
- {tinybird-0.0.1.dev249.dist-info → tinybird-0.0.1.dev251.dist-info}/entry_points.txt +0 -0
- {tinybird-0.0.1.dev249.dist-info → tinybird-0.0.1.dev251.dist-info}/top_level.txt +0 -0
tinybird/prompts.py
CHANGED
|
@@ -815,7 +815,7 @@ pipe_instructions = """
|
|
|
815
815
|
<pipe_file_instructions>
|
|
816
816
|
- The pipe names must be unique.
|
|
817
817
|
- Nodes do NOT use the same name as the Pipe they belong to. So if the pipe name is "my_pipe", the nodes must be named different like "my_pipe_node_1", "my_pipe_node_2", etc.
|
|
818
|
-
-
|
|
818
|
+
- Node names MUST be different from the resource names in the project.
|
|
819
819
|
- Avoid more than one node per pipe unless it is really necessary or requested by the user.
|
|
820
820
|
- No indentation is allowed for property names: DESCRIPTION, NODE, SQL, TYPE, etc.
|
|
821
821
|
- Allowed TYPE values are: endpoint, copy, materialized, sink.
|
tinybird/sql_template.py
CHANGED
|
@@ -1388,13 +1388,14 @@ _namespace = {
|
|
|
1388
1388
|
"close": None,
|
|
1389
1389
|
"print": None,
|
|
1390
1390
|
"input": None,
|
|
1391
|
-
"id": None,
|
|
1392
1391
|
}
|
|
1393
1392
|
|
|
1394
1393
|
|
|
1395
1394
|
reserved_vars = set(["_tt_tmp", "_tt_append", "isinstance", "str", "error", "custom_error", *list(vars(builtins))])
|
|
1396
1395
|
for p in DEFAULT_PARAM_NAMES: # we handle these in an specific manner
|
|
1397
1396
|
reserved_vars.discard(p) # `format` is part of builtins
|
|
1397
|
+
# Allow 'id' to be used as a template parameter - https://gitlab.com/tinybird/analytics/-/issues/19119
|
|
1398
|
+
reserved_vars.discard("id")
|
|
1398
1399
|
error_vars = ["error", "custom_error"]
|
|
1399
1400
|
|
|
1400
1401
|
|
|
@@ -1463,6 +1464,9 @@ def generate(self, **kwargs) -> Tuple[str, TemplateExecutionResults]:
|
|
|
1463
1464
|
return Expression(f"-- disable {feature}\n")
|
|
1464
1465
|
|
|
1465
1466
|
namespace.update(_namespace)
|
|
1467
|
+
# This is to fix the issue https://gitlab.com/tinybird/analytics/-/issues/19119
|
|
1468
|
+
# We need to do the override here because if we modify the _namespace, we would filter out parameters from the users
|
|
1469
|
+
namespace.update({"id": None})
|
|
1466
1470
|
namespace.update(kwargs)
|
|
1467
1471
|
namespace.update(
|
|
1468
1472
|
{
|
tinybird/tb/__cli__.py
CHANGED
|
@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
|
|
|
4
4
|
__url__ = 'https://www.tinybird.co/docs/forward/commands'
|
|
5
5
|
__author__ = 'Tinybird'
|
|
6
6
|
__author_email__ = 'support@tinybird.co'
|
|
7
|
-
__version__ = '0.0.1.
|
|
8
|
-
__revision__ = '
|
|
7
|
+
__version__ = '0.0.1.dev251'
|
|
8
|
+
__revision__ = 'c11760c'
|
|
@@ -4,12 +4,9 @@ import sys
|
|
|
4
4
|
from datetime import datetime
|
|
5
5
|
from functools import partial
|
|
6
6
|
from pathlib import Path
|
|
7
|
-
from typing import Any
|
|
7
|
+
from typing import Any, Optional
|
|
8
8
|
|
|
9
9
|
import click
|
|
10
|
-
from prompt_toolkit import prompt
|
|
11
|
-
from prompt_toolkit.cursor_shapes import CursorShape
|
|
12
|
-
from prompt_toolkit.styles import Style
|
|
13
10
|
from pydantic_ai import Agent, Tool
|
|
14
11
|
from pydantic_ai.messages import ModelMessage
|
|
15
12
|
|
|
@@ -29,7 +26,7 @@ from tinybird.prompts import (
|
|
|
29
26
|
from tinybird.tb.client import TinyB
|
|
30
27
|
from tinybird.tb.modules.agent.animations import ThinkingAnimation
|
|
31
28
|
from tinybird.tb.modules.agent.banner import display_banner
|
|
32
|
-
from tinybird.tb.modules.agent.memory import clear_history
|
|
29
|
+
from tinybird.tb.modules.agent.memory import clear_history
|
|
33
30
|
from tinybird.tb.modules.agent.models import create_model, model_costs
|
|
34
31
|
from tinybird.tb.modules.agent.prompts import (
|
|
35
32
|
datafile_instructions,
|
|
@@ -43,11 +40,13 @@ from tinybird.tb.modules.agent.tools.create_datafile import create_datafile
|
|
|
43
40
|
from tinybird.tb.modules.agent.tools.deploy import deploy
|
|
44
41
|
from tinybird.tb.modules.agent.tools.deploy_check import deploy_check
|
|
45
42
|
from tinybird.tb.modules.agent.tools.explore import explore_data
|
|
43
|
+
from tinybird.tb.modules.agent.tools.get_endpoint_stats import get_endpoint_stats
|
|
44
|
+
from tinybird.tb.modules.agent.tools.get_openapi_definition import get_openapi_definition
|
|
46
45
|
from tinybird.tb.modules.agent.tools.mock import mock
|
|
47
46
|
from tinybird.tb.modules.agent.tools.plan import plan
|
|
48
47
|
from tinybird.tb.modules.agent.tools.preview_datafile import preview_datafile
|
|
49
48
|
from tinybird.tb.modules.agent.tools.read_fixture_data import read_fixture_data
|
|
50
|
-
from tinybird.tb.modules.agent.utils import TinybirdAgentContext
|
|
49
|
+
from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_input
|
|
51
50
|
from tinybird.tb.modules.build_common import process as build_process
|
|
52
51
|
from tinybird.tb.modules.common import _analyze, _get_tb_client
|
|
53
52
|
from tinybird.tb.modules.config import CLIConfig
|
|
@@ -115,6 +114,8 @@ You have access to the following tools:
|
|
|
115
114
|
8. `mock` - Create mock data for a landing datasource.
|
|
116
115
|
9. `read_fixture_data` - Read a fixture data file present in the project folder.
|
|
117
116
|
10. `append` - Append existing fixture to a datasource.
|
|
117
|
+
11. `get_endpoint_stats` - Get metrics of the requests to an endpoint.
|
|
118
|
+
12. `get_openapi_definition` - Get the OpenAPI definition for all endpoints that are built/deployed to Tinybird Cloud or Local.
|
|
118
119
|
|
|
119
120
|
# When creating or updating datafiles:
|
|
120
121
|
1. Use `plan` tool to plan the creation or update of resources.
|
|
@@ -126,6 +127,24 @@ You have access to the following tools:
|
|
|
126
127
|
7. If the datafile was not created, finish the process and just wait for a new user prompt.
|
|
127
128
|
8. If the datafile was created successfully, but the built failed, try to fix the error and repeat the process.
|
|
128
129
|
|
|
130
|
+
# When creating a landing datasource given a .ndjson file:
|
|
131
|
+
- If the user does not specify anything about the desired schema, create a schema like this:
|
|
132
|
+
SCHEMA >
|
|
133
|
+
`data` String `json:$`
|
|
134
|
+
- Use always json paths with .ndjson files.
|
|
135
|
+
|
|
136
|
+
# When user wants to optimize an endpoint:
|
|
137
|
+
First check if the query is optimized. E.g is filtering by a column present in the sorting key.
|
|
138
|
+
Avoid when possible to not update the landing datasource.
|
|
139
|
+
Check endpoint stats to analyze how the endpoint is performing. Use `get_endpoint_stats` tool to get the stats.
|
|
140
|
+
When your data is in Tinybird, you can create intermediate data sources to preprocess data and make the endpoints faster. This can be done by using materialized views or copy pipes.
|
|
141
|
+
- Copy pipes capture the result of a pipe at a specific point in time and write it to a target data source. They can run on a schedule or run on demand, making them ideal for event-sourced snapshots, data experimentation, and deduplication with snapshots.
|
|
142
|
+
- Materialized views continuously re-evaluate a query as new events are inserted, maintaining an always up-to-date derived dataset. Unlike copy pipes which create point-in-time snapshots, materialized views provide real-time transformations of your data.
|
|
143
|
+
Each approach has its own strengths and use cases:
|
|
144
|
+
- Use copy pipes when you need scheduled or on-demand snapshots of your data.
|
|
145
|
+
- Use materialized views when you need continuous, real-time transformations.
|
|
146
|
+
Finally, update the existing endpoint itself, do not add a new one.
|
|
147
|
+
|
|
129
148
|
IMPORTANT: If the user cancels some of the steps or there is an error in file creation, DO NOT continue with the plan. Stop the process and wait for the user before using any other tool.
|
|
130
149
|
IMPORTANT: Every time you finish a plan and start a new resource creation or update process, create a new plan before starting with the changes.
|
|
131
150
|
|
|
@@ -175,6 +194,15 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
|
|
|
175
194
|
Tool(mock, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
|
|
176
195
|
Tool(read_fixture_data, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
|
|
177
196
|
Tool(append, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
|
|
197
|
+
Tool(
|
|
198
|
+
get_endpoint_stats, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True
|
|
199
|
+
),
|
|
200
|
+
Tool(
|
|
201
|
+
get_openapi_definition,
|
|
202
|
+
docstring_format="google",
|
|
203
|
+
require_parameter_descriptions=True,
|
|
204
|
+
takes_ctx=True,
|
|
205
|
+
),
|
|
178
206
|
],
|
|
179
207
|
)
|
|
180
208
|
|
|
@@ -186,7 +214,7 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
|
|
|
186
214
|
user_prompt = f"{user_prompt}\n\n{resources_prompt(project)}"
|
|
187
215
|
client = TinyB(token=self.token, host=self.host)
|
|
188
216
|
folder = self.project.folder
|
|
189
|
-
|
|
217
|
+
click.echo()
|
|
190
218
|
thinking_animation = ThinkingAnimation(message="Chirping", delay=0.15)
|
|
191
219
|
thinking_animation.start()
|
|
192
220
|
result = self.agent.run_sync(
|
|
@@ -200,11 +228,16 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
|
|
|
200
228
|
mock_data=partial(mock_data, project=project, config=config),
|
|
201
229
|
append_data=partial(append_data, config=config),
|
|
202
230
|
analyze_fixture=partial(analyze_fixture, config=config),
|
|
231
|
+
execute_cloud_query=partial(execute_cloud_query, config=config),
|
|
232
|
+
execute_local_query=partial(execute_local_query, config=config),
|
|
203
233
|
get_project_files=project.get_project_files,
|
|
204
234
|
folder=folder,
|
|
205
235
|
thinking_animation=thinking_animation,
|
|
206
236
|
workspace_name=self.project.workspace_name,
|
|
207
237
|
dangerously_skip_permissions=self.dangerously_skip_permissions,
|
|
238
|
+
token=self.token,
|
|
239
|
+
user_token=self.user_token,
|
|
240
|
+
host=self.host,
|
|
208
241
|
),
|
|
209
242
|
message_history=self.messages,
|
|
210
243
|
)
|
|
@@ -219,13 +252,15 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
|
|
|
219
252
|
request_tokens * model_costs["input_cost_per_token"]
|
|
220
253
|
+ response_tokens * model_costs["output_cost_per_token"]
|
|
221
254
|
)
|
|
222
|
-
click.echo()
|
|
223
255
|
click.echo(result.output)
|
|
224
256
|
click.echo("\n")
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
257
|
+
|
|
258
|
+
if "@tinybird.co" in config.get("user_email", ""):
|
|
259
|
+
click.echo(f"Input tokens: {request_tokens}")
|
|
260
|
+
click.echo(f"Output tokens: {response_tokens}")
|
|
261
|
+
click.echo(f"Total tokens: {total_tokens}")
|
|
262
|
+
click.echo(f"Cost: ${cost:.6f}")
|
|
263
|
+
click.echo("\n")
|
|
229
264
|
|
|
230
265
|
|
|
231
266
|
def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permissions: bool):
|
|
@@ -233,6 +268,7 @@ def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permiss
|
|
|
233
268
|
host = config.get("host", None)
|
|
234
269
|
user_token = config.get("user_token", None)
|
|
235
270
|
workspace_id = config.get("id", None)
|
|
271
|
+
workspace_name = config.get("name", None)
|
|
236
272
|
try:
|
|
237
273
|
if not token or not host or not workspace_id or not user_token:
|
|
238
274
|
yes = click.confirm(
|
|
@@ -274,20 +310,11 @@ def run_agent(config: dict[str, Any], project: Project, dangerously_skip_permiss
|
|
|
274
310
|
try:
|
|
275
311
|
while True:
|
|
276
312
|
try:
|
|
277
|
-
user_input =
|
|
278
|
-
[("class:prompt", f"tb ({project.workspace_name}) » ")],
|
|
279
|
-
history=load_history(),
|
|
280
|
-
cursor=CursorShape.BLOCK,
|
|
281
|
-
style=Style.from_dict(
|
|
282
|
-
{
|
|
283
|
-
"prompt": "#40a8a8 bold",
|
|
284
|
-
"": "", # Normal color for user input
|
|
285
|
-
}
|
|
286
|
-
),
|
|
287
|
-
)
|
|
313
|
+
user_input = show_input(workspace_name)
|
|
288
314
|
if user_input.startswith("tb "):
|
|
289
315
|
cmd_parts = shlex.split(user_input)
|
|
290
|
-
subprocess.run(cmd_parts
|
|
316
|
+
subprocess.run(cmd_parts)
|
|
317
|
+
click.echo()
|
|
291
318
|
continue
|
|
292
319
|
if user_input.lower() in ["/exit", "/quit"]:
|
|
293
320
|
click.echo(FeedbackManager.info(message="Goodbye!"))
|
|
@@ -364,7 +391,12 @@ def append_data(config: dict[str, Any], datasource_name: str, path: str) -> None
|
|
|
364
391
|
|
|
365
392
|
|
|
366
393
|
def mock_data(
|
|
367
|
-
config: dict[str, Any],
|
|
394
|
+
config: dict[str, Any],
|
|
395
|
+
project: Project,
|
|
396
|
+
datasource_name: str,
|
|
397
|
+
data_format: str,
|
|
398
|
+
rows: int,
|
|
399
|
+
context: Optional[str] = None,
|
|
368
400
|
) -> list[dict[str, Any]]:
|
|
369
401
|
client = get_tinybird_local_client(config, test=False, silent=False)
|
|
370
402
|
cli_config = CLIConfig.get_project_config()
|
|
@@ -374,12 +406,11 @@ def mock_data(
|
|
|
374
406
|
raise CLIMockException(f"Datasource {datasource_name} not found")
|
|
375
407
|
|
|
376
408
|
datasource_content = Path(datasource_path).read_text()
|
|
377
|
-
prompt = ""
|
|
378
409
|
return create_mock_data(
|
|
379
410
|
datasource_name,
|
|
380
411
|
datasource_content,
|
|
381
412
|
rows,
|
|
382
|
-
|
|
413
|
+
context or "",
|
|
383
414
|
cli_config,
|
|
384
415
|
config,
|
|
385
416
|
cli_config.get_user_token() or "",
|
|
@@ -393,3 +424,13 @@ def analyze_fixture(config: dict[str, Any], fixture_path: str) -> dict[str, Any]
|
|
|
393
424
|
local_client = get_tinybird_local_client(config, test=False, silent=True)
|
|
394
425
|
meta, _data = _analyze(fixture_path, local_client, Path(fixture_path).suffix.lstrip("."))
|
|
395
426
|
return meta
|
|
427
|
+
|
|
428
|
+
|
|
429
|
+
def execute_cloud_query(config: dict[str, Any], query: str, pipe_name: Optional[str] = None) -> str:
|
|
430
|
+
client = _get_tb_client(config["token"], config["host"])
|
|
431
|
+
return client.query(sql=query, pipeline=pipe_name)
|
|
432
|
+
|
|
433
|
+
|
|
434
|
+
def execute_local_query(config: dict[str, Any], query: str, pipe_name: Optional[str] = None) -> str:
|
|
435
|
+
local_client = get_tinybird_local_client(config, test=False, silent=True)
|
|
436
|
+
return local_client.query(sql=query, pipeline=pipe_name)
|
|
@@ -4,15 +4,13 @@ import threading
|
|
|
4
4
|
from time import sleep
|
|
5
5
|
from typing import Optional
|
|
6
6
|
|
|
7
|
-
import click
|
|
8
|
-
|
|
9
7
|
|
|
10
8
|
class ThinkingAnimation:
|
|
11
9
|
"""Thinking animation that shows changing sparkles as a prefix."""
|
|
12
10
|
|
|
13
11
|
def __init__(
|
|
14
12
|
self,
|
|
15
|
-
message: str = "
|
|
13
|
+
message: str = "Chirping",
|
|
16
14
|
delay: float = 0.15,
|
|
17
15
|
colors: bool = True,
|
|
18
16
|
dots: bool = True,
|
|
@@ -37,17 +35,35 @@ class ThinkingAnimation:
|
|
|
37
35
|
]
|
|
38
36
|
self.reset_color = "\033[0m"
|
|
39
37
|
|
|
40
|
-
def start(self):
|
|
38
|
+
def start(self, message: Optional[str] = None):
|
|
41
39
|
"""Start the animation in a separate thread."""
|
|
40
|
+
# Stop any existing animation first
|
|
41
|
+
if self.running:
|
|
42
|
+
self._stop_without_reset()
|
|
43
|
+
|
|
44
|
+
if message:
|
|
45
|
+
self.message = message
|
|
42
46
|
self.running = True
|
|
43
47
|
self.thread = threading.Thread(target=self._run_animation)
|
|
44
48
|
self.thread.daemon = True
|
|
45
49
|
|
|
46
|
-
|
|
50
|
+
# Clear the current line before starting new animation
|
|
51
|
+
sys.stdout.write("\r" + " " * (len(self.message) + 10) + "\r")
|
|
52
|
+
sys.stdout.flush()
|
|
47
53
|
self.thread.start()
|
|
48
54
|
|
|
49
55
|
def stop(self):
|
|
50
56
|
"""Stop the animation."""
|
|
57
|
+
self.message = "Chirping"
|
|
58
|
+
self.running = False
|
|
59
|
+
if self.thread:
|
|
60
|
+
self.thread.join()
|
|
61
|
+
# Clear the line and reset cursor position
|
|
62
|
+
sys.stdout.write("\r" + " " * (len(self.message) + 10) + "\r")
|
|
63
|
+
sys.stdout.flush()
|
|
64
|
+
|
|
65
|
+
def _stop_without_reset(self):
|
|
66
|
+
"""Stop the animation without resetting the message."""
|
|
51
67
|
self.running = False
|
|
52
68
|
if self.thread:
|
|
53
69
|
self.thread.join()
|
|
@@ -78,8 +94,11 @@ class ThinkingAnimation:
|
|
|
78
94
|
else:
|
|
79
95
|
display_message = self.message
|
|
80
96
|
|
|
81
|
-
# Print the message with the prefix sparkle
|
|
82
|
-
|
|
97
|
+
# Print the message with the prefix sparkle, padding to clear any leftover characters
|
|
98
|
+
line_content = f"{colored_sparkle} {display_message}"
|
|
99
|
+
# Pad with spaces to clear any leftover characters from longer previous messages
|
|
100
|
+
padded_line = line_content.ljust(len(self.message) + 10)
|
|
101
|
+
sys.stdout.write(f"\r{padded_line}")
|
|
83
102
|
sys.stdout.flush()
|
|
84
103
|
|
|
85
104
|
sleep(self.delay)
|
|
@@ -22,8 +22,10 @@ Steps:
|
|
|
22
22
|
<dev_notes>
|
|
23
23
|
You can skip steps where resources will not be created or updated.
|
|
24
24
|
Always add 'Build project' step after generating resources.
|
|
25
|
-
Always add 'Generate mock data' step after building project if a landing datasource was created.
|
|
26
|
-
Always add 'Append existing fixture' step after building project if a
|
|
25
|
+
Always add 'Generate mock data' step after building project if a landing datasource was created without providing a fixture file.
|
|
26
|
+
Always add 'Append existing fixture' step after building project if a landing datasource was created after providing a fixture file.
|
|
27
|
+
Solve the specific user request, do not add extra steps that are not related to the user request.
|
|
28
|
+
Reuse the existing resources if possible.
|
|
27
29
|
</dev_notes>
|
|
28
30
|
|
|
29
31
|
Resource dependencies:
|
|
@@ -1,29 +1,10 @@
|
|
|
1
1
|
import click
|
|
2
2
|
from pydantic_ai import RunContext
|
|
3
3
|
|
|
4
|
-
from tinybird.tb.modules.agent.utils import TinybirdAgentContext,
|
|
4
|
+
from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_confirmation, show_input
|
|
5
5
|
from tinybird.tb.modules.feedback_manager import FeedbackManager
|
|
6
6
|
|
|
7
7
|
|
|
8
|
-
def get_append_confirmation(datasource_name: str) -> bool:
|
|
9
|
-
"""Get user confirmation for appending existing fixture"""
|
|
10
|
-
while True:
|
|
11
|
-
result = show_options(
|
|
12
|
-
options=["Yes, append existing fixture", "No, and tell Tinybird Code what to do"],
|
|
13
|
-
title=f"Do you want to append existing fixture for datasource {datasource_name}?",
|
|
14
|
-
)
|
|
15
|
-
|
|
16
|
-
if result is None: # Cancelled
|
|
17
|
-
return False
|
|
18
|
-
|
|
19
|
-
if result.startswith("Yes"):
|
|
20
|
-
return True
|
|
21
|
-
elif result.startswith("No"):
|
|
22
|
-
return False
|
|
23
|
-
|
|
24
|
-
return False
|
|
25
|
-
|
|
26
|
-
|
|
27
8
|
def append(ctx: RunContext[TinybirdAgentContext], datasource_name: str, fixture_pathname: str) -> str:
|
|
28
9
|
"""Append existing fixture to a datasource
|
|
29
10
|
|
|
@@ -36,10 +17,20 @@ def append(ctx: RunContext[TinybirdAgentContext], datasource_name: str, fixture_
|
|
|
36
17
|
"""
|
|
37
18
|
try:
|
|
38
19
|
ctx.deps.thinking_animation.stop()
|
|
39
|
-
confirmation =
|
|
20
|
+
confirmation = show_confirmation(
|
|
21
|
+
title=f"Append existing fixture for datasource {datasource_name}?",
|
|
22
|
+
skip_confirmation=ctx.deps.dangerously_skip_permissions,
|
|
23
|
+
)
|
|
24
|
+
|
|
40
25
|
ctx.deps.thinking_animation.start()
|
|
41
26
|
|
|
42
|
-
if
|
|
27
|
+
if confirmation == "review":
|
|
28
|
+
click.echo()
|
|
29
|
+
feedback = show_input(ctx.deps.workspace_name)
|
|
30
|
+
ctx.deps.thinking_animation.start()
|
|
31
|
+
return f"User did not confirm the proposed plan and gave the following feedback: {feedback}"
|
|
32
|
+
|
|
33
|
+
if confirmation == "cancel":
|
|
43
34
|
return "User rejected appending existing fixture. Skipping..."
|
|
44
35
|
|
|
45
36
|
ctx.deps.thinking_animation.stop()
|
|
@@ -3,31 +3,17 @@ from pathlib import Path
|
|
|
3
3
|
import click
|
|
4
4
|
from pydantic_ai import RunContext
|
|
5
5
|
|
|
6
|
-
from tinybird.tb.modules.agent.utils import
|
|
6
|
+
from tinybird.tb.modules.agent.utils import (
|
|
7
|
+
Datafile,
|
|
8
|
+
TinybirdAgentContext,
|
|
9
|
+
create_terminal_box,
|
|
10
|
+
show_confirmation,
|
|
11
|
+
show_input,
|
|
12
|
+
)
|
|
7
13
|
from tinybird.tb.modules.exceptions import CLIBuildException
|
|
8
14
|
from tinybird.tb.modules.feedback_manager import FeedbackManager
|
|
9
15
|
|
|
10
16
|
|
|
11
|
-
def get_resource_confirmation(resource: Datafile, exists: bool) -> bool:
|
|
12
|
-
"""Get user confirmation for creating a resource"""
|
|
13
|
-
while True:
|
|
14
|
-
action = "create" if not exists else "update"
|
|
15
|
-
result = show_options(
|
|
16
|
-
options=[f"Yes, {action} {resource.type} '{resource.name}'", "No, and tell Tinybird Code what to do"],
|
|
17
|
-
title=f"What would you like to do with {resource.type} '{resource.name}'?",
|
|
18
|
-
)
|
|
19
|
-
|
|
20
|
-
if result is None: # Cancelled
|
|
21
|
-
return False
|
|
22
|
-
|
|
23
|
-
if result.startswith("Yes"):
|
|
24
|
-
return True
|
|
25
|
-
elif result.startswith("No"):
|
|
26
|
-
return False
|
|
27
|
-
|
|
28
|
-
return False
|
|
29
|
-
|
|
30
|
-
|
|
31
17
|
def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -> str:
|
|
32
18
|
"""Given a resource representation, create a file in the project folder
|
|
33
19
|
|
|
@@ -48,19 +34,34 @@ def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -
|
|
|
48
34
|
else:
|
|
49
35
|
content = create_terminal_box(resource.content, title=resource.pathname)
|
|
50
36
|
click.echo(content)
|
|
51
|
-
confirmation = ctx.deps.dangerously_skip_permissions or get_resource_confirmation(resource, exists)
|
|
52
37
|
|
|
53
|
-
if not
|
|
38
|
+
action = "Create" if not exists else "Update"
|
|
39
|
+
confirmation = show_confirmation(
|
|
40
|
+
title=f"{action} '{resource.pathname}'?",
|
|
41
|
+
skip_confirmation=ctx.deps.dangerously_skip_permissions,
|
|
42
|
+
)
|
|
43
|
+
|
|
44
|
+
if confirmation == "review":
|
|
45
|
+
click.echo()
|
|
46
|
+
feedback = show_input(ctx.deps.workspace_name)
|
|
47
|
+
ctx.deps.thinking_animation.start()
|
|
48
|
+
return f"User did not confirm the proposed changes and gave the following feedback: {feedback}"
|
|
49
|
+
|
|
50
|
+
if confirmation == "cancel":
|
|
54
51
|
ctx.deps.thinking_animation.start()
|
|
55
|
-
return f"
|
|
52
|
+
return f"User cancelled {action} of {resource.pathname}. Stop resource creation."
|
|
56
53
|
|
|
54
|
+
action_text = "Creating" if not exists else "Updating"
|
|
55
|
+
click.echo(FeedbackManager.highlight(message=f"\n» {action_text} {resource.pathname}..."))
|
|
57
56
|
folder_path = path.parent
|
|
58
57
|
folder_path.mkdir(parents=True, exist_ok=True)
|
|
59
58
|
path.touch(exist_ok=True)
|
|
60
59
|
path.write_text(resource.content)
|
|
61
60
|
ctx.deps.build_project(test=True, silent=True)
|
|
61
|
+
action_text = "created" if not exists else "updated"
|
|
62
|
+
click.echo(FeedbackManager.success(message=f"✓ {resource.pathname} {action_text} successfully"))
|
|
62
63
|
ctx.deps.thinking_animation.start()
|
|
63
|
-
return f"
|
|
64
|
+
return f"{action_text} {resource.pathname}"
|
|
64
65
|
|
|
65
66
|
except CLIBuildException as e:
|
|
66
67
|
ctx.deps.thinking_animation.stop()
|
|
@@ -68,4 +69,7 @@ def create_datafile(ctx: RunContext[TinybirdAgentContext], resource: Datafile) -
|
|
|
68
69
|
ctx.deps.thinking_animation.start()
|
|
69
70
|
return f"Error building project: {e}"
|
|
70
71
|
except Exception as e:
|
|
72
|
+
ctx.deps.thinking_animation.stop()
|
|
73
|
+
click.echo(FeedbackManager.error(message=e))
|
|
74
|
+
ctx.deps.thinking_animation.start()
|
|
71
75
|
return f"Error creating {resource.pathname}: {e}"
|
|
@@ -1,41 +1,32 @@
|
|
|
1
1
|
import click
|
|
2
2
|
from pydantic_ai import RunContext
|
|
3
3
|
|
|
4
|
-
from tinybird.tb.modules.agent.utils import TinybirdAgentContext,
|
|
4
|
+
from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_confirmation, show_input
|
|
5
5
|
from tinybird.tb.modules.feedback_manager import FeedbackManager
|
|
6
6
|
|
|
7
7
|
|
|
8
|
-
def get_deploy_confirmation() -> bool:
|
|
9
|
-
"""Get user confirmation for deploying the project"""
|
|
10
|
-
while True:
|
|
11
|
-
result = show_options(
|
|
12
|
-
options=["Yes, deploy the project", "No, and tell Tinybird Code what to do"],
|
|
13
|
-
title="Do you want to deploy the project?",
|
|
14
|
-
)
|
|
15
|
-
|
|
16
|
-
if result is None: # Cancelled
|
|
17
|
-
return False
|
|
18
|
-
|
|
19
|
-
if result.startswith("Yes"):
|
|
20
|
-
return True
|
|
21
|
-
elif result.startswith("No"):
|
|
22
|
-
return False
|
|
23
|
-
|
|
24
|
-
return False
|
|
25
|
-
|
|
26
|
-
|
|
27
8
|
def deploy(ctx: RunContext[TinybirdAgentContext]) -> str:
|
|
28
9
|
"""Deploy the project"""
|
|
29
10
|
try:
|
|
30
11
|
ctx.deps.thinking_animation.stop()
|
|
31
|
-
confirmation =
|
|
32
|
-
|
|
12
|
+
confirmation = show_confirmation(
|
|
13
|
+
title="Deploy the project?",
|
|
14
|
+
skip_confirmation=ctx.deps.dangerously_skip_permissions,
|
|
15
|
+
)
|
|
33
16
|
|
|
34
|
-
if
|
|
35
|
-
|
|
17
|
+
if confirmation == "review":
|
|
18
|
+
click.echo()
|
|
19
|
+
feedback = show_input(ctx.deps.workspace_name)
|
|
20
|
+
ctx.deps.thinking_animation.start()
|
|
21
|
+
return f"User did not confirm deployment and gave the following feedback: {feedback}"
|
|
36
22
|
|
|
37
|
-
|
|
23
|
+
if confirmation == "cancel":
|
|
24
|
+
ctx.deps.thinking_animation.start()
|
|
25
|
+
return "User cancelled deployment. Stop deployment."
|
|
26
|
+
|
|
27
|
+
click.echo(FeedbackManager.highlight(message="\n» Deploying project..."))
|
|
38
28
|
ctx.deps.deploy_project()
|
|
29
|
+
click.echo(FeedbackManager.success(message="✓ Project deployed successfully"))
|
|
39
30
|
ctx.deps.thinking_animation.start()
|
|
40
31
|
return "Project deployed successfully"
|
|
41
32
|
except Exception as e:
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
import click
|
|
2
|
+
from pydantic_ai import RunContext
|
|
3
|
+
|
|
4
|
+
from tinybird.tb.modules.agent.utils import TinybirdAgentContext
|
|
5
|
+
from tinybird.tb.modules.feedback_manager import FeedbackManager
|
|
6
|
+
|
|
7
|
+
|
|
8
|
+
def get_endpoint_stats(
|
|
9
|
+
ctx: RunContext[TinybirdAgentContext], endpoint_name: str, interval_days: int = 1, cloud: bool = True
|
|
10
|
+
):
|
|
11
|
+
"""Get stats for an endpoint:
|
|
12
|
+
|
|
13
|
+
Args:
|
|
14
|
+
endpoint_name (str): The name of the endpoint to get stats for. Required.
|
|
15
|
+
interval_days (int): The number of days to get stats for. Optional.
|
|
16
|
+
cloud (bool): Whether to get stats from cloud or local. Optional.
|
|
17
|
+
|
|
18
|
+
Returns:
|
|
19
|
+
str: The result of the stats.
|
|
20
|
+
"""
|
|
21
|
+
if interval_days == 1:
|
|
22
|
+
pipe_stats = "tinybird.pipe_stats_rt"
|
|
23
|
+
date_column = "start_datetime"
|
|
24
|
+
else:
|
|
25
|
+
pipe_stats = "tinybird.pipe_stats"
|
|
26
|
+
date_column = "date"
|
|
27
|
+
|
|
28
|
+
days = "day" if interval_days == 1 else "days"
|
|
29
|
+
cloud_or_local = "cloud" if cloud else "local"
|
|
30
|
+
ctx.deps.thinking_animation.stop()
|
|
31
|
+
|
|
32
|
+
click.echo(
|
|
33
|
+
FeedbackManager.highlight(
|
|
34
|
+
message=f"» Analyzing {cloud_or_local} requests in the last {interval_days} {days} for '{endpoint_name}' endpoint"
|
|
35
|
+
)
|
|
36
|
+
)
|
|
37
|
+
|
|
38
|
+
query = f"""SELECT * FROM {pipe_stats}
|
|
39
|
+
WHERE {date_column} > NOW() - INTERVAL {interval_days} DAY
|
|
40
|
+
AND pipe_name = '{endpoint_name}'
|
|
41
|
+
LIMIT 100
|
|
42
|
+
FORMAT JSON
|
|
43
|
+
"""
|
|
44
|
+
|
|
45
|
+
execute_query = ctx.deps.execute_cloud_query if cloud else ctx.deps.execute_local_query
|
|
46
|
+
|
|
47
|
+
result = execute_query(query=query)
|
|
48
|
+
click.echo(FeedbackManager.success(message="✓ Done!"))
|
|
49
|
+
click.echo()
|
|
50
|
+
ctx.deps.thinking_animation.start()
|
|
51
|
+
return f"Result for {endpoint_name} in the last {interval_days} {days}: {result}"
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
import click
|
|
2
|
+
import requests
|
|
3
|
+
from pydantic_ai import RunContext
|
|
4
|
+
|
|
5
|
+
from tinybird.tb.modules.agent.utils import TinybirdAgentContext
|
|
6
|
+
from tinybird.tb.modules.feedback_manager import FeedbackManager
|
|
7
|
+
|
|
8
|
+
|
|
9
|
+
def get_openapi_definition(ctx: RunContext[TinybirdAgentContext], cloud: bool = True):
|
|
10
|
+
"""Get the OpenAPI definition for all endpoints in the project:
|
|
11
|
+
|
|
12
|
+
Args:
|
|
13
|
+
cloud (bool): Whether to get the OpenAPI spec from cloud or local. Optional.
|
|
14
|
+
|
|
15
|
+
Returns:
|
|
16
|
+
str: The OpenAPI spec for all endpoints in the project.
|
|
17
|
+
"""
|
|
18
|
+
|
|
19
|
+
cloud_or_local = "Cloud" if cloud else "Local"
|
|
20
|
+
ctx.deps.thinking_animation.stop()
|
|
21
|
+
|
|
22
|
+
click.echo(FeedbackManager.highlight(message=f"» Accessing Tinybird {cloud_or_local} endpoints"))
|
|
23
|
+
try:
|
|
24
|
+
url = f"{ctx.deps.host}/v0/pipes/openapi.json"
|
|
25
|
+
|
|
26
|
+
result = requests.get(url, headers={"Authorization": f"Bearer {ctx.deps.token}"})
|
|
27
|
+
if result.status_code != 200:
|
|
28
|
+
raise Exception(f"Failed to get OpenAPI spec: {result.status_code} {result.text}")
|
|
29
|
+
|
|
30
|
+
openapi_spec = result.text
|
|
31
|
+
|
|
32
|
+
click.echo(FeedbackManager.success(message="✓ Done!"))
|
|
33
|
+
click.echo()
|
|
34
|
+
ctx.deps.thinking_animation.start()
|
|
35
|
+
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>"
|
|
36
|
+
except Exception as e:
|
|
37
|
+
click.echo(FeedbackManager.error(message=f"Error getting OpenAPI spec: {e}"))
|
|
38
|
+
ctx.deps.thinking_animation.start()
|
|
39
|
+
return f"Error getting OpenAPI spec: {e}"
|
|
@@ -1,52 +1,50 @@
|
|
|
1
|
+
from typing import Optional
|
|
2
|
+
|
|
1
3
|
import click
|
|
2
4
|
from pydantic_ai import RunContext
|
|
3
5
|
|
|
4
|
-
from tinybird.tb.modules.agent.utils import TinybirdAgentContext,
|
|
6
|
+
from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_confirmation, show_input
|
|
5
7
|
from tinybird.tb.modules.datafile.fixture import persist_fixture
|
|
6
8
|
from tinybird.tb.modules.feedback_manager import FeedbackManager
|
|
7
9
|
|
|
8
10
|
|
|
9
|
-
def
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
if result is None: # Cancelled
|
|
18
|
-
return False
|
|
19
|
-
|
|
20
|
-
if result.startswith("Yes"):
|
|
21
|
-
return True
|
|
22
|
-
elif result.startswith("No"):
|
|
23
|
-
return False
|
|
24
|
-
|
|
25
|
-
return False
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
def mock(ctx: RunContext[TinybirdAgentContext], datasource_name: str, data_format: str, rows: int) -> str:
|
|
11
|
+
def mock(
|
|
12
|
+
ctx: RunContext[TinybirdAgentContext],
|
|
13
|
+
datasource_name: str,
|
|
14
|
+
data_format: str,
|
|
15
|
+
rows: int,
|
|
16
|
+
context: Optional[str] = None,
|
|
17
|
+
) -> str:
|
|
29
18
|
"""Create mock data for a datasource
|
|
30
19
|
|
|
31
20
|
Args:
|
|
32
21
|
datasource_name: Name of the datasource to create mock data for
|
|
33
22
|
data_format: Format of the mock data to create. Options: ndjson, csv
|
|
34
23
|
rows: Number of rows to create. If not provided, the default is 10
|
|
24
|
+
context: Extra context to be used to generate the mock data. Optional.
|
|
35
25
|
|
|
36
26
|
Returns:
|
|
37
27
|
str: Message indicating the success or failure of the mock data generation
|
|
38
28
|
"""
|
|
39
29
|
try:
|
|
40
30
|
ctx.deps.thinking_animation.stop()
|
|
41
|
-
confirmation =
|
|
42
|
-
|
|
31
|
+
confirmation = show_confirmation(
|
|
32
|
+
title=f"Generate mock data for datasource '{datasource_name}'?",
|
|
33
|
+
skip_confirmation=ctx.deps.dangerously_skip_permissions,
|
|
34
|
+
)
|
|
43
35
|
|
|
44
|
-
if
|
|
45
|
-
|
|
36
|
+
if confirmation == "review":
|
|
37
|
+
click.echo()
|
|
38
|
+
feedback = show_input(ctx.deps.workspace_name)
|
|
39
|
+
ctx.deps.thinking_animation.start()
|
|
40
|
+
return f"User did not confirm mock data generation and gave the following feedback: {feedback}"
|
|
41
|
+
|
|
42
|
+
if confirmation == "cancel":
|
|
43
|
+
ctx.deps.thinking_animation.start()
|
|
44
|
+
return "User cancelled mock data generation. Stop mock data generation."
|
|
46
45
|
|
|
47
|
-
ctx.deps.thinking_animation.stop()
|
|
48
46
|
click.echo(FeedbackManager.highlight(message=f"\n» Generating mock data for {datasource_name}..."))
|
|
49
|
-
data = ctx.deps.mock_data(datasource_name=datasource_name, data_format=data_format, rows=rows)
|
|
47
|
+
data = ctx.deps.mock_data(datasource_name=datasource_name, data_format=data_format, rows=rows, context=context)
|
|
50
48
|
fixture_path = persist_fixture(datasource_name, data, ctx.deps.folder, format=data_format)
|
|
51
49
|
ctx.deps.append_data(datasource_name=datasource_name, path=str(fixture_path))
|
|
52
50
|
click.echo(FeedbackManager.success(message=f"✓ Data generated for {datasource_name}"))
|
|
@@ -54,6 +52,22 @@ def mock(ctx: RunContext[TinybirdAgentContext], datasource_name: str, data_forma
|
|
|
54
52
|
return f"Mock data generated successfully for datasource {datasource_name}"
|
|
55
53
|
except Exception as e:
|
|
56
54
|
ctx.deps.thinking_animation.stop()
|
|
57
|
-
|
|
55
|
+
error_message = str(e)
|
|
56
|
+
click.echo(FeedbackManager.error(message=error_message))
|
|
57
|
+
try:
|
|
58
|
+
if "in quarantine" in error_message:
|
|
59
|
+
click.echo(
|
|
60
|
+
FeedbackManager.highlight(message=f"\n» Looking for errors in {datasource_name}_quarantine...")
|
|
61
|
+
)
|
|
62
|
+
query = f"select * from {datasource_name}_quarantine order by insertion_date desc limit 5 FORMAT CSVWithNames"
|
|
63
|
+
quarantine_data = ctx.deps.execute_local_query(query=query)
|
|
64
|
+
error_message = (
|
|
65
|
+
error_message
|
|
66
|
+
+ 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."
|
|
67
|
+
)
|
|
68
|
+
|
|
69
|
+
except Exception as quarantine_error:
|
|
70
|
+
error_message = error_message + f"\nError accessing to {datasource_name}_quarantine: {quarantine_error}"
|
|
71
|
+
|
|
58
72
|
ctx.deps.thinking_animation.start()
|
|
59
|
-
return f"Error generating mock data: {
|
|
73
|
+
return f"Error generating mock data: {error_message}"
|
|
@@ -1,26 +1,7 @@
|
|
|
1
1
|
import click
|
|
2
2
|
from pydantic_ai import RunContext
|
|
3
3
|
|
|
4
|
-
from tinybird.tb.modules.agent.utils import TinybirdAgentContext,
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
def get_plan_confirmation() -> bool:
|
|
8
|
-
"""Get user confirmation for implementing a plan"""
|
|
9
|
-
while True:
|
|
10
|
-
result = show_options(
|
|
11
|
-
options=["Yes, implement the plan", "No, and tell Tinybird Code what to do"],
|
|
12
|
-
title="Do you want to implement the plan?",
|
|
13
|
-
)
|
|
14
|
-
|
|
15
|
-
if result is None: # Cancelled
|
|
16
|
-
return False
|
|
17
|
-
|
|
18
|
-
if result.startswith("Yes"):
|
|
19
|
-
return True
|
|
20
|
-
elif result.startswith("No"):
|
|
21
|
-
return False
|
|
22
|
-
|
|
23
|
-
return False
|
|
4
|
+
from tinybird.tb.modules.agent.utils import TinybirdAgentContext, show_confirmation, show_input
|
|
24
5
|
|
|
25
6
|
|
|
26
7
|
def plan(ctx: RunContext[TinybirdAgentContext], plan: str) -> str:
|
|
@@ -35,11 +16,20 @@ def plan(ctx: RunContext[TinybirdAgentContext], plan: str) -> str:
|
|
|
35
16
|
try:
|
|
36
17
|
ctx.deps.thinking_animation.stop()
|
|
37
18
|
click.echo(plan)
|
|
38
|
-
confirmation =
|
|
19
|
+
confirmation = show_confirmation(
|
|
20
|
+
title="Do you want to implement the plan?", skip_confirmation=ctx.deps.dangerously_skip_permissions
|
|
21
|
+
)
|
|
22
|
+
|
|
23
|
+
if confirmation == "review":
|
|
24
|
+
click.echo()
|
|
25
|
+
feedback = show_input(ctx.deps.workspace_name)
|
|
26
|
+
ctx.deps.thinking_animation.start()
|
|
27
|
+
return f"User did not confirm the proposed plan and gave the following feedback: {feedback}"
|
|
28
|
+
|
|
39
29
|
ctx.deps.thinking_animation.start()
|
|
40
30
|
|
|
41
|
-
if
|
|
42
|
-
return "
|
|
31
|
+
if confirmation == "cancel":
|
|
32
|
+
return "User cancelled the plan. Stop the process."
|
|
43
33
|
|
|
44
34
|
return "User confirmed the plan. Implementing..."
|
|
45
35
|
|
|
@@ -1,9 +1,11 @@
|
|
|
1
1
|
import json
|
|
2
2
|
from pathlib import Path
|
|
3
3
|
|
|
4
|
+
import click
|
|
4
5
|
from pydantic_ai import RunContext
|
|
5
6
|
|
|
6
7
|
from tinybird.tb.modules.agent.utils import TinybirdAgentContext
|
|
8
|
+
from tinybird.tb.modules.feedback_manager import FeedbackManager
|
|
7
9
|
|
|
8
10
|
|
|
9
11
|
def read_fixture_data(ctx: RunContext[TinybirdAgentContext], fixture_pathname: str):
|
|
@@ -15,12 +17,18 @@ def read_fixture_data(ctx: RunContext[TinybirdAgentContext], fixture_pathname: s
|
|
|
15
17
|
Returns:
|
|
16
18
|
str: The content of the fixture data file.
|
|
17
19
|
"""
|
|
20
|
+
ctx.deps.thinking_animation.stop()
|
|
21
|
+
click.echo(FeedbackManager.highlight(message=f"» Analyzing {fixture_pathname}..."))
|
|
18
22
|
fixture_path = Path(ctx.deps.folder) / fixture_pathname.lstrip("/")
|
|
19
23
|
|
|
20
24
|
if not fixture_path.exists():
|
|
25
|
+
click.echo(FeedbackManager.error(message=f"No fixture data found for {fixture_pathname}."))
|
|
26
|
+
ctx.deps.thinking_animation.start()
|
|
21
27
|
return f"No fixture data found for {fixture_pathname}. Please check the name of the fixture and try again."
|
|
22
28
|
|
|
23
29
|
response = ctx.deps.analyze_fixture(fixture_path=str(fixture_path))
|
|
30
|
+
click.echo(FeedbackManager.success(message="✓ Done!\n"))
|
|
31
|
+
ctx.deps.thinking_animation.start()
|
|
24
32
|
# limit content to first 10 rows
|
|
25
33
|
data = response["preview"]["data"][:10]
|
|
26
34
|
columns = response["analysis"]["columns"]
|
|
@@ -1,10 +1,12 @@
|
|
|
1
1
|
import difflib
|
|
2
2
|
import os
|
|
3
3
|
from contextlib import contextmanager
|
|
4
|
-
from typing import Any, Callable, List, Optional, Tuple
|
|
4
|
+
from typing import Any, Callable, List, Literal, Optional, Tuple
|
|
5
5
|
|
|
6
6
|
import click
|
|
7
|
+
from prompt_toolkit import prompt as prompt_toolkit_prompt
|
|
7
8
|
from prompt_toolkit.application import Application, get_app
|
|
9
|
+
from prompt_toolkit.cursor_shapes import CursorShape
|
|
8
10
|
from prompt_toolkit.filters import IsDone
|
|
9
11
|
from prompt_toolkit.key_binding import KeyBindings
|
|
10
12
|
from prompt_toolkit.layout import Layout
|
|
@@ -17,6 +19,8 @@ from prompt_toolkit.shortcuts import PromptSession
|
|
|
17
19
|
from prompt_toolkit.styles import Style as PromptStyle
|
|
18
20
|
from pydantic import BaseModel, Field
|
|
19
21
|
|
|
22
|
+
from tinybird.tb.modules.agent.memory import load_history
|
|
23
|
+
|
|
20
24
|
try:
|
|
21
25
|
from colorama import Back, Fore, Style, init
|
|
22
26
|
|
|
@@ -38,7 +42,12 @@ class TinybirdAgentContext(BaseModel):
|
|
|
38
42
|
mock_data: Callable[..., list[dict[str, Any]]]
|
|
39
43
|
append_data: Callable[..., None]
|
|
40
44
|
analyze_fixture: Callable[..., dict[str, Any]]
|
|
45
|
+
execute_cloud_query: Callable[..., str]
|
|
46
|
+
execute_local_query: Callable[..., str]
|
|
41
47
|
dangerously_skip_permissions: bool
|
|
48
|
+
token: str
|
|
49
|
+
user_token: str
|
|
50
|
+
host: str
|
|
42
51
|
|
|
43
52
|
|
|
44
53
|
default_style = PromptStyle.from_dict(
|
|
@@ -345,6 +354,9 @@ def show_options(options: List[str], title: str = "Select an option") -> Optiona
|
|
|
345
354
|
return result["option"]
|
|
346
355
|
|
|
347
356
|
|
|
357
|
+
ConfirmationResult = Literal["yes", "review", "cancel"]
|
|
358
|
+
|
|
359
|
+
|
|
348
360
|
def load_existing_resources(folder: str) -> str:
|
|
349
361
|
"""Load existing Tinybird resources from the workspace"""
|
|
350
362
|
existing_resources = []
|
|
@@ -677,3 +689,38 @@ def _process_line(
|
|
|
677
689
|
remaining = remaining[content_width:]
|
|
678
690
|
|
|
679
691
|
return processed
|
|
692
|
+
|
|
693
|
+
|
|
694
|
+
def show_input(workspace_name: str) -> str:
|
|
695
|
+
return prompt_toolkit_prompt(
|
|
696
|
+
[("class:prompt", f"tb ({workspace_name}) » ")],
|
|
697
|
+
history=load_history(),
|
|
698
|
+
cursor=CursorShape.BLOCK,
|
|
699
|
+
style=PromptStyle.from_dict(
|
|
700
|
+
{
|
|
701
|
+
"prompt": "#40a8a8 bold",
|
|
702
|
+
"": "", # Normal color for user input
|
|
703
|
+
}
|
|
704
|
+
),
|
|
705
|
+
)
|
|
706
|
+
|
|
707
|
+
|
|
708
|
+
def show_confirmation(title: str, skip_confirmation: bool = False) -> ConfirmationResult:
|
|
709
|
+
if skip_confirmation:
|
|
710
|
+
return "yes"
|
|
711
|
+
|
|
712
|
+
while True:
|
|
713
|
+
result = show_options(
|
|
714
|
+
options=["Yes, continue", "No, tell Tinybird Code what to do", "Cancel"],
|
|
715
|
+
title=title,
|
|
716
|
+
)
|
|
717
|
+
|
|
718
|
+
if result is None: # Cancelled
|
|
719
|
+
return "cancel"
|
|
720
|
+
|
|
721
|
+
if result.startswith("Yes"):
|
|
722
|
+
return "yes"
|
|
723
|
+
elif result.startswith("No"):
|
|
724
|
+
return "review"
|
|
725
|
+
|
|
726
|
+
return "cancel"
|
|
@@ -3,9 +3,9 @@ tinybird/context.py,sha256=FfqYfrGX_I7PKGTQo93utaKPDNVYWelg4Hsp3evX5wM,1291
|
|
|
3
3
|
tinybird/datatypes.py,sha256=r4WCvspmrXTJHiPjjyOTiZyZl31FO3Ynkwq4LQsYm6E,11059
|
|
4
4
|
tinybird/feedback_manager.py,sha256=1INQFfRfuMCb9lfB8KNf4r6qC2khW568hoHjtk-wshI,69305
|
|
5
5
|
tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
|
|
6
|
-
tinybird/prompts.py,sha256=
|
|
6
|
+
tinybird/prompts.py,sha256=4VmdaMX7oUFoqjseXe8QuF9wTtIabbDkdwVGmd34S7s,45502
|
|
7
7
|
tinybird/sql.py,sha256=BufnOgclQokDyihtuXesOwHBsebN6wRXIxO5wKRkOwE,48299
|
|
8
|
-
tinybird/sql_template.py,sha256=
|
|
8
|
+
tinybird/sql_template.py,sha256=AezE1o6_BzbHFi0J9OIqTrXQ5WvoX5eNVq4QCbFjGcs,100338
|
|
9
9
|
tinybird/sql_template_fmt.py,sha256=KUHdj5rYCYm_rKKdXYSJAE9vIyXUQLB0YSZnUXHeBlY,10196
|
|
10
10
|
tinybird/sql_toolset.py,sha256=M2rpLYkgV2W8NnYEYPC1tJdpy4uZHXVF64NBSKLQka4,19549
|
|
11
11
|
tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
|
|
@@ -17,7 +17,7 @@ tinybird/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1w
|
|
|
17
17
|
tinybird/datafile/parse_connection.py,sha256=tRyn2Rpr1TeWet5BXmMoQgaotbGdYep1qiTak_OqC5E,1825
|
|
18
18
|
tinybird/datafile/parse_datasource.py,sha256=ssW8QeFSgglVFi3sDZj_HgkJiTJ2069v2JgqnH3CkDE,1825
|
|
19
19
|
tinybird/datafile/parse_pipe.py,sha256=xf4m0Tw44QWJzHzAm7Z7FwUoUUtr7noMYjU1NiWnX0k,3880
|
|
20
|
-
tinybird/tb/__cli__.py,sha256=
|
|
20
|
+
tinybird/tb/__cli__.py,sha256=ldJiOCMSMd-ptHqIVlOItLL_kL3ehE--VamyNzl-O8E,247
|
|
21
21
|
tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
|
|
22
22
|
tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
|
|
23
23
|
tinybird/tb/client.py,sha256=pJbdkWMXGAqKseNAvdsRRnl_c7I-DCMB0dWCQnG82nU,54146
|
|
@@ -68,24 +68,26 @@ tinybird/tb/modules/watch.py,sha256=No0bK1M1_3CYuMaIgylxf7vYFJ72lTJe3brz6xQ-mJo,
|
|
|
68
68
|
tinybird/tb/modules/workspace.py,sha256=Q_8HcxMsNg8QG9aBlwcWS2umrDP5IkTIHqqz3sfmGuc,11341
|
|
69
69
|
tinybird/tb/modules/workspace_members.py,sha256=5JdkJgfuEwbq-t6vxkBhYwgsiTDxF790wsa6Xfif9nk,8608
|
|
70
70
|
tinybird/tb/modules/agent/__init__.py,sha256=i3oe3vDIWWPaicdCM0zs7D7BJ1W0k7th93ooskHAV00,54
|
|
71
|
-
tinybird/tb/modules/agent/agent.py,sha256=
|
|
72
|
-
tinybird/tb/modules/agent/animations.py,sha256=
|
|
71
|
+
tinybird/tb/modules/agent/agent.py,sha256=Uxo2FNNu-YW-J-AbZizZe0eKYKdZRwbPJCgA9EsXlgs,21208
|
|
72
|
+
tinybird/tb/modules/agent/animations.py,sha256=4WOC5_2BracttmMCrV0H91tXfWcUzQHBUaIJc5FA7tE,3490
|
|
73
73
|
tinybird/tb/modules/agent/banner.py,sha256=KX_e467uiy1gWOZ4ofTZt0GCFGQqHQ_8Ob27XLQqda0,3053
|
|
74
74
|
tinybird/tb/modules/agent/memory.py,sha256=H6SJK--2L5C87B7AJd_jMqsq3sCvFvZwZXmajuT0GBE,1171
|
|
75
75
|
tinybird/tb/modules/agent/models.py,sha256=LW1D27gjcd_jwFmghEzteCgToDfodX2B6B5S8BYbysw,735
|
|
76
|
-
tinybird/tb/modules/agent/prompts.py,sha256=
|
|
77
|
-
tinybird/tb/modules/agent/utils.py,sha256=
|
|
76
|
+
tinybird/tb/modules/agent/prompts.py,sha256=wbe6vUnm-fskceWgP13R5VW1v_YF7_wLDe-wBN6rlWw,6998
|
|
77
|
+
tinybird/tb/modules/agent/utils.py,sha256=dWdjtptsBRKUAtg6wz2SNf5fmmGKMG-WKNk9miNWZ-s,26520
|
|
78
78
|
tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
79
|
-
tinybird/tb/modules/agent/tools/append.py,sha256=
|
|
79
|
+
tinybird/tb/modules/agent/tools/append.py,sha256=Db576-ziulykJyaWl03zUp-KI-NVURh0WcFIAPNbNPw,1974
|
|
80
80
|
tinybird/tb/modules/agent/tools/build.py,sha256=LhzJMx6tbxC7gogIrxhfKJc-SDgoSR-FC6IunfaCdn8,758
|
|
81
|
-
tinybird/tb/modules/agent/tools/create_datafile.py,sha256=
|
|
82
|
-
tinybird/tb/modules/agent/tools/deploy.py,sha256=
|
|
81
|
+
tinybird/tb/modules/agent/tools/create_datafile.py,sha256=wcPcChACTIFKw0lKFTlhm0sWJKhQkMLPLnGNpKyeETA,2962
|
|
82
|
+
tinybird/tb/modules/agent/tools/deploy.py,sha256=WrsSlaufKGOBx0S13uoMQQH2DnKue5LQ231Rx4RXh2I,1443
|
|
83
83
|
tinybird/tb/modules/agent/tools/deploy_check.py,sha256=VqMYC7l3_cihmmM_pi8w1t8rJ3P0xDc7pHs_st9k-9Q,684
|
|
84
84
|
tinybird/tb/modules/agent/tools/explore.py,sha256=ihALc_kBcsjrKT3hZyicqyIowB0g_K3AtNNi-5uz9-8,412
|
|
85
|
-
tinybird/tb/modules/agent/tools/
|
|
86
|
-
tinybird/tb/modules/agent/tools/
|
|
85
|
+
tinybird/tb/modules/agent/tools/get_endpoint_stats.py,sha256=_3wAvDykJitIOb5BRnP7wCy6y06y1qlULHLWB-MvS2M,1705
|
|
86
|
+
tinybird/tb/modules/agent/tools/get_openapi_definition.py,sha256=9cQ-SUeB1NVhPJN1s8aQh9KQxqI9-DEEW1Ot5r2JbOk,1575
|
|
87
|
+
tinybird/tb/modules/agent/tools/mock.py,sha256=Omog_gdEdm8YuBXNrJdHwxHqjL_ji9UIr75mALF4ozI,3408
|
|
88
|
+
tinybird/tb/modules/agent/tools/plan.py,sha256=pr6LnItz6vlOeCG8GE459ExsrBEG0KLx-g02SZGNjXU,1217
|
|
87
89
|
tinybird/tb/modules/agent/tools/preview_datafile.py,sha256=e9q5fR0afApcrntzFrnuHmd10ex7MG_GM6T0Pwc9bRI,850
|
|
88
|
-
tinybird/tb/modules/agent/tools/read_fixture_data.py,sha256=
|
|
90
|
+
tinybird/tb/modules/agent/tools/read_fixture_data.py,sha256=rvTdVlZsu3rQTSWqXzpFt4LEwnBcMLIT8hlI5C7MVN4,1430
|
|
89
91
|
tinybird/tb/modules/datafile/build.py,sha256=NFKBrusFLU0WJNCXePAFWiEDuTaXpwc0lHlOQWEJ43s,51117
|
|
90
92
|
tinybird/tb/modules/datafile/build_common.py,sha256=2yNdxe49IMA9wNvl25NemY2Iaz8L66snjOdT64dm1is,4511
|
|
91
93
|
tinybird/tb/modules/datafile/build_datasource.py,sha256=Ra8pVQBDafbFRUKlhpgohhTsRyp_ADKZJVG8Gd69idY,17227
|
|
@@ -106,8 +108,8 @@ tinybird/tb_cli_modules/config.py,sha256=IsgdtFRnUrkY8-Zo32lmk6O7u3bHie1QCxLwgp4
|
|
|
106
108
|
tinybird/tb_cli_modules/exceptions.py,sha256=pmucP4kTF4irIt7dXiG-FcnI-o3mvDusPmch1L8RCWk,3367
|
|
107
109
|
tinybird/tb_cli_modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
|
|
108
110
|
tinybird/tb_cli_modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09K1A,10449
|
|
109
|
-
tinybird-0.0.1.
|
|
110
|
-
tinybird-0.0.1.
|
|
111
|
-
tinybird-0.0.1.
|
|
112
|
-
tinybird-0.0.1.
|
|
113
|
-
tinybird-0.0.1.
|
|
111
|
+
tinybird-0.0.1.dev251.dist-info/METADATA,sha256=7hx0f5-Lb5i3niyVL2AsZ9LcJzp3cTAoX0QHuATZv5A,1733
|
|
112
|
+
tinybird-0.0.1.dev251.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
|
|
113
|
+
tinybird-0.0.1.dev251.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
|
|
114
|
+
tinybird-0.0.1.dev251.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
|
|
115
|
+
tinybird-0.0.1.dev251.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|