tinybird 0.0.1.dev2__py3-none-any.whl → 0.0.1.dev3__py3-none-any.whl

tinybird/__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.dev2'
-__revision__ = 'bdbc260'
+__version__ = '0.0.1.dev3'
+__revision__ = '50bec29'

tinybird/tb_cli_modules/llm.py

@@ -0,0 +1,31 @@
+from typing import List, Optional
+
+from openai import OpenAI
+from pydantic import BaseModel
+
+from tinybird.tb_cli_modules.prompts import create_project_prompt
+
+
+class DataFile(BaseModel):
+    name: str
+    content: str
+
+
+class DataProject(BaseModel):
+    datasources: List[DataFile]
+    pipes: List[DataFile]
+
+
+class LLM:
+    def __init__(self, model: Optional[str] = None, key: Optional[str] = None):
+        self.model = model
+        self.key = key
+
+    async def create_project(self, prompt: str) -> DataProject:
+        client = OpenAI(api_key=self.key)
+        completion = client.beta.chat.completions.parse(
+            model=self.model,
+            messages=[{"role": "system", "content": create_project_prompt}, {"role": "user", "content": prompt}],
+            response_format=DataProject,
+        )
+        return completion.choices[0].message.parsed or DataProject(datasources=[], pipes=[])

tinybird/tb_cli_modules/local.py

@@ -0,0 +1,108 @@
+import time
+
+import click
+import requests
+
+import docker
+from tinybird.feedback_manager import FeedbackManager
+from tinybird.tb_cli_modules.common import CLIException
+from tinybird.tb_cli_modules.config import CLIConfig
+
+# TODO: Use the official Tinybird image once it's available 'tinybirdco/tinybird-local:latest'
+TB_IMAGE_NAME = "registry.gitlab.com/tinybird/analytics/tinybird-local-jammy-3.11:latest"
+TB_CONTAINER_NAME = "tinybird-local"
+TB_LOCAL_PORT = 80
+TB_LOCAL_HOST = f"http://localhost:{TB_LOCAL_PORT}"
+
+
+def start_tinybird_local(
+    docker_client,
+):
+    """Start the Tinybird container."""
+    containers = docker_client.containers.list(all=True, filters={"name": TB_CONTAINER_NAME})
+    if containers:
+        # Container `start` is idempotent. It's safe to call it even if the container is already running.
+        container = containers[0]
+        container.start()
+    else:
+        pull_required = False
+        try:
+            local_image = docker_client.images.get(TB_IMAGE_NAME)
+            local_image_id = local_image.attrs["RepoDigests"][0].split("@")[1]
+            remote_image = docker_client.images.get_registry_data(TB_IMAGE_NAME)
+            pull_required = local_image_id != remote_image.id
+        except Exception:
+            pull_required = True
+
+        if pull_required:
+            click.echo(
+                FeedbackManager.info(message="** Downloading latest version of Tinybird development environment...")
+            )
+            docker_client.images.pull(TB_IMAGE_NAME, platform="linux/amd64")
+
+        container = docker_client.containers.run(
+            TB_IMAGE_NAME,
+            name=TB_CONTAINER_NAME,
+            detach=True,
+            ports={"80/tcp": TB_LOCAL_PORT},
+            remove=False,
+            platform="linux/amd64",
+        )
+
+    click.echo(FeedbackManager.info(message="** Waiting for Tinybird development environment to be ready..."))
+    for attempt in range(10):
+        try:
+            run = container.exec_run("tb --no-version-warning sql 'SELECT 1 AS healthcheck' --format json").output
+            # dont parse the json as docker sometimes returns warning messages
+            # todo: rafa, make this rigth
+            if b'"healthcheck": 1' in run:
+                break
+            raise RuntimeError("Unexpected response from Tinybird")
+        except Exception:
+            if attempt == 9:  # Last attempt
+                raise CLIException("Tinybird local environment not ready yet. Please try again in a few seconds.")
+            time.sleep(5)  # Wait 5 seconds before retrying
+
+
+def get_docker_client():
+    """Check if Docker is installed and running."""
+    try:
+        client = docker.from_env()
+        client.ping()
+        return client
+    except Exception:
+        raise CLIException("Docker is not running or installed. Please ensure Docker is installed and running.")
+
+
+def stop_tinybird_local(docker_client):
+    """Stop the Tinybird container."""
+    try:
+        container = docker_client.containers.get(TB_CONTAINER_NAME)
+        container.stop()
+    except Exception:
+        pass
+
+
+def remove_tinybird_local(docker_client):
+    """Remove the Tinybird container."""
+    try:
+        container = docker_client.containers.get(TB_CONTAINER_NAME)
+        container.remove(force=True)
+    except Exception:
+        pass
+
+
+def set_up_tinybird_local(docker_client):
+    """Set up the Tinybird local environment."""
+    start_tinybird_local(docker_client)
+    return get_tinybird_local_client()
+
+
+def get_tinybird_local_client():
+    """Get a Tinybird client connected to the local environment."""
+    config = CLIConfig.get_project_config()
+    tokens = requests.get(f"{TB_LOCAL_HOST}/tokens").json()
+    token = tokens["workspace_admin_token"]
+    config.set_token(token)
+    config.set_host(TB_LOCAL_HOST)
+    return config.get_client(host=TB_LOCAL_HOST, token=token)

{tinybird-0.0.1.dev2.dist-info → tinybird-0.0.1.dev3.dist-info}/METADATA

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

{tinybird-0.0.1.dev2.dist-info → tinybird-0.0.1.dev3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tinybird/__cli__.py,sha256=PHs2gdn0u2epRONYAzC2TOYLhI2SR6FPDDS5NDrS0TY,250
+tinybird/__cli__.py,sha256=bgGjpK7NkAkSj1v-IXZIG9nlQnRLi4faLhm9wfdnePo,250
 tinybird/check_pypi.py,sha256=_4NkharLyR_ELrAdit-ftqIWvOf7jZNPt3i76frlo9g,975
 tinybird/client.py,sha256=nd97gD2-8Ap8yDonBcVwk9eXDAL43hmIYdo-Pse43RE,50738
 tinybird/config.py,sha256=Z-BX9FrjgsLw1YwcCdF0IztLB97Zpc70VVPplO_pDSY,6089
@@ -30,6 +30,8 @@ tinybird/tb_cli_modules/datasource.py,sha256=BVYwPkKvdnEv2YMkxof7n7FOffyJm-QK7Pk
 tinybird/tb_cli_modules/exceptions.py,sha256=pmucP4kTF4irIt7dXiG-FcnI-o3mvDusPmch1L8RCWk,3367
 tinybird/tb_cli_modules/fmt.py,sha256=_xL2AyGo3us7NpmfAXtSIhn1cGAC-A4PspdhUjm58jY,3382
 tinybird/tb_cli_modules/job.py,sha256=AG69LPb9MbobA1awwJFZJvxqarDKfRlsBjw2V1zvYqc,2964
+tinybird/tb_cli_modules/llm.py,sha256=RyUBtyLJcG5DFBjkalFLgwwX-415_Uom8Hq_oKiiyBs,904
+tinybird/tb_cli_modules/local.py,sha256=uamCRYoU0MhCaPUXtvCFfbzHJGwTaPMTTHRQyw0Da3k,3870
 tinybird/tb_cli_modules/pipe.py,sha256=BCLAQ3ZuWKGAih2mupnw_Y2S5B5cNS-epF317whsSEE,30989
 tinybird/tb_cli_modules/prompts.py,sha256=Esjhet-KUM1bT6RStk2voCCVjvAOUD4hDDiKpn-AYNY,7227
 tinybird/tb_cli_modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
@@ -41,8 +43,8 @@ tinybird/tb_cli_modules/workspace.py,sha256=N8f1dl4BYc34ucmJ6oNZc4XZMGKd8m0Fhq7o
 tinybird/tb_cli_modules/workspace_members.py,sha256=ksXsjd233y9-sNlz4Qb-meZbX4zn1B84e_bSm2i8rhg,8731
 tinybird/tb_cli_modules/tinyunit/tinyunit.py,sha256=_ydOD4WxmKy7_h-d3fG62w_0_lD0uLl9w4EbEYtwd0U,11720
 tinybird/tb_cli_modules/tinyunit/tinyunit_lib.py,sha256=hGh1ZaXC1af7rKnX7222urkj0QJMhMWclqMy59dOqwE,1922
-tinybird-0.0.1.dev2.dist-info/METADATA,sha256=C0w-b-Ur4VfPVJya8uxCZvv2cBxlgaDePZnV2m2-Y9k,2404
-tinybird-0.0.1.dev2.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tinybird-0.0.1.dev2.dist-info/entry_points.txt,sha256=PKPKuPmA4IfJYnCFHHUiw-aAWZuBomFvwCklv1OyCjE,43
-tinybird-0.0.1.dev2.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
-tinybird-0.0.1.dev2.dist-info/RECORD,,
+tinybird-0.0.1.dev3.dist-info/METADATA,sha256=6J__4w3BceyDNH4mybjldWRwFGgNaDLSyvugF9Se_E4,2404
+tinybird-0.0.1.dev3.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tinybird-0.0.1.dev3.dist-info/entry_points.txt,sha256=PKPKuPmA4IfJYnCFHHUiw-aAWZuBomFvwCklv1OyCjE,43
+tinybird-0.0.1.dev3.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
+tinybird-0.0.1.dev3.dist-info/RECORD,,