tinybird 0.0.1.dev91__tar.gz → 0.0.1.dev92__tar.gz

{tinybird-0.0.1.dev91 → tinybird-0.0.1.dev92}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev91
+Version: 0.0.1.dev92
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird

{tinybird-0.0.1.dev91 → tinybird-0.0.1.dev92}/tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/cli/introduction.html'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev91'
-__revision__ = 'b965ddf'
+__version__ = '0.0.1.dev92'
+__revision__ = 'd99cccc'

{tinybird-0.0.1.dev91 → tinybird-0.0.1.dev92}/tinybird/tb/modules/deployment.py

@@ -4,7 +4,7 @@ import sys
 import time
 from datetime import datetime
 from pathlib import Path
-from typing import Optional, Union
+from typing import Any, Dict, Optional, Union
 
 import click
 import requests
@@ -330,7 +330,7 @@ def create_deployment(
     }
     project: Project = ctx.ensure_object(dict)["project"]
     client = ctx.ensure_object(dict)["client"]
-    config = ctx.ensure_object(dict)["config"]
+    config: Dict[str, Any] = ctx.ensure_object(dict)["config"]
     TINYBIRD_API_URL = f"{client.host}/v1/deploy"
     TINYBIRD_API_KEY = client.token
 

{tinybird-0.0.1.dev91 → tinybird-0.0.1.dev92}/tinybird/tb/modules/playground.py

@@ -62,7 +62,7 @@ def playground(
     """Build the project in Tinybird Local."""
     project: Project = ctx.ensure_object(dict)["project"]
     tb_client: TinyB = ctx.ensure_object(dict)["client"]
-    config: CLIConfig = ctx.ensure_object(dict)["config"]
+    config = CLIConfig.get_project_config()
     context.disable_template_security_validation.set(True)
 
     async def process(filenames: List[str], watch: bool = False):

{tinybird-0.0.1.dev91 → tinybird-0.0.1.dev92}/tinybird/tb/modules/test.py

@@ -5,6 +5,7 @@
 
 import difflib
 import glob
+import os
 import sys
 import urllib.parse
 from pathlib import Path
@@ -75,8 +76,8 @@ async def test_create(ctx: click.Context, name_or_filename: str, prompt: str) ->
     """
     project: Project = ctx.ensure_object(dict)["project"]
     client: TinyB = ctx.ensure_object(dict)["client"]
-    config: CLIConfig = ctx.ensure_object(dict)["config"]
-    root_path = Path(project.folder)
+    config = CLIConfig.get_project_config()
+    root_path = project.path
     folder = project.folder
     try:
         if ".pipe" in name_or_filename:
@@ -85,20 +86,17 @@ async def test_create(ctx: click.Context, name_or_filename: str, prompt: str) ->
                 raise CLIException(FeedbackManager.error(message=f"Pipe {name_or_filename} not found"))
         else:
             pipe_folders = ("endpoints", "copies", "materializations", "sinks", "pipes")
-            try:
-                pipe_path = next(
-                    root_path / folder / f"{name_or_filename}.pipe"
-                    for folder in pipe_folders
-                    if (root_path / folder / f"{name_or_filename}.pipe").exists()
-                )
-            except Exception:
+            pipe_path = next(
+                root_path / folder / f"{name_or_filename}.pipe"
+                for folder in pipe_folders
+                if os.path.exists(root_path / folder / f"{name_or_filename}.pipe")
+            )
+            if not pipe_path:
                 raise CLIException(FeedbackManager.error(message=f"Pipe {name_or_filename} not found"))
 
         pipe_name = pipe_path.stem
         click.echo(FeedbackManager.highlight(message=f"\n» Creating tests for {pipe_name} endpoint..."))
-        pipe_path = root_path / pipe_path
         pipe_content = pipe_path.read_text()
-
         pipe = await client._req(f"/v0/pipes/{pipe_name}")
         parameters = set([param["name"] for node in pipe["nodes"] for param in node["params"]])
 
@@ -112,7 +110,6 @@ async def test_create(ctx: click.Context, name_or_filename: str, prompt: str) ->
             raise CLIException(FeedbackManager.error(message="No user token found"))
 
         llm = LLM(user_token=user_token, host=config.get_client().host)
-        click.echo(FeedbackManager.highlight(message=f"\n» Creating test for {pipe_name} endpoint..."))
         response_llm = llm.ask(system_prompt=system_prompt, prompt=prompt)
         response_xml = extract_xml(response_llm, "response")
         tests_content = parse_xml(response_xml, "test")
@@ -216,9 +213,12 @@ async def test_update(ctx: click.Context, pipe: str) -> None:
 async def run_tests(ctx: click.Context, name: Tuple[str, ...]) -> None:
     click.echo(FeedbackManager.highlight(message="\n» Running tests"))
     client: TinyB = ctx.ensure_object(dict)["client"]
+    project: Project = ctx.ensure_object(dict)["project"]
     paths = [Path(n) for n in name]
-    endpoints = [f"./tests/{p.stem}.yaml" for p in paths]
-    test_files: List[str] = endpoints if len(endpoints) > 0 else glob.glob("./tests/**/*.y*ml", recursive=True)
+    endpoints = [f"{project.path}/tests/{p.stem}.yaml" for p in paths]
+    test_files: List[str] = (
+        endpoints if len(endpoints) > 0 else glob.glob(f"{project.path}/tests/**/*.y*ml", recursive=True)
+    )
 
     async def run_test(test_file):
         test_file_path = Path(test_file)

{tinybird-0.0.1.dev91 → tinybird-0.0.1.dev92}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev91
+Version: 0.0.1.dev92
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird