tinybird 0.0.1.dev240__py3-none-any.whl → 0.0.1.dev241__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.dev240'
-__revision__ = '268407c'
+__version__ = '0.0.1.dev241'
+__revision__ = 'b44ab3b'

tinybird/tb/modules/agent/agent.py

@@ -21,8 +21,8 @@ from tinybird.prompts import (
     s3_connection_example,
     sink_pipe_instructions,
 )
+from tinybird.tb.client import TinyB
 from tinybird.tb.modules.agent.banner import display_banner
-from tinybird.tb.modules.agent.client import TinybirdClient
 from tinybird.tb.modules.agent.memory import clear_history, load_history
 from tinybird.tb.modules.agent.models import create_model
 from tinybird.tb.modules.agent.prompts import datafile_instructions, plan_instructions, sql_instructions
@@ -134,10 +134,12 @@ Today is {datetime.now().strftime("%Y-%m-%d")}
         return self.messages[-5:] if len(self.messages) > 5 else self.messages
 
     def run(self, user_prompt: str) -> None:
+        client = TinyB(token=self.token, host=self.host)
         result = self.agent.run_sync(
             user_prompt,
             deps=TinybirdAgentContext(
-                client=TinybirdClient(token=self.token, host=self.host),
+                # context does not support the whole client, so we need to pass only the functions we need
+                explore_data=client.explore_data,
                 folder=self.folder,
             ),
             message_history=self.messages,

tinybird/tb/modules/agent/models.py

@@ -6,7 +6,7 @@ from pydantic_ai.providers.anthropic import AnthropicProvider
 
 def create_model(
     token: str,
-    base_url: str = "https://api.wadus3.aws.tinybird.co",
+    base_url: str,
     model: AnthropicModelName = "claude-4-sonnet-20250514",
 ):
     client = AsyncAnthropic(

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

@@ -12,4 +12,4 @@ def explore_data(ctx: RunContext[TinybirdAgentContext], prompt: str):
     Returns:
         str: The result of the exploration.
     """
-    return ctx.deps.client.explore_data(prompt)
+    return ctx.deps.explore_data(prompt)

tinybird/tb/modules/agent/utils.py

@@ -1,6 +1,6 @@
 import os
 from contextlib import contextmanager
-from typing import Any, List, Optional
+from typing import Any, Callable, List, Optional
 
 import click
 from prompt_toolkit.application import Application, get_app
@@ -16,11 +16,9 @@ from prompt_toolkit.shortcuts import PromptSession
 from prompt_toolkit.styles import Style
 from pydantic import BaseModel, Field
 
-from tinybird.tb.modules.agent.client import TinybirdClient
-
 
 class TinybirdAgentContext(BaseModel):
-    client: TinybirdClient
+    explore_data: Callable[[str], str]
     folder: str
 
 

{tinybird-0.0.1.dev240.dist-info → tinybird-0.0.1.dev241.dist-info}/METADATA

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

{tinybird-0.0.1.dev240.dist-info → tinybird-0.0.1.dev241.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=OB0M6HqPur6rsd93Mf9BRpLf5zvrZ_aq4HvYd7nAacc,247
+tinybird/tb/__cli__.py,sha256=QWlsNI8ondzuZlYrE_y6bh1Cfz6_5AfBIndh4z9wk60,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
@@ -63,16 +63,15 @@ 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=98X8rqUszArhOnrWWGAJRNwAK-me6Ql2UGNGFxcuc2E,10205
+tinybird/tb/modules/agent/agent.py,sha256=nUvSfgi7OdrWXRxLjLHzepsrICiduAZcW7iFE0x5Q5A,10323
 tinybird/tb/modules/agent/banner.py,sha256=KX_e467uiy1gWOZ4ofTZt0GCFGQqHQ_8Ob27XLQqda0,3053
-tinybird/tb/modules/agent/client.py,sha256=ZNE_0xtbrfhXXRZ7nx3Ze0gHb1IK7zXRLTlGudAAuUg,1264
 tinybird/tb/modules/agent/memory.py,sha256=H6SJK--2L5C87B7AJd_jMqsq3sCvFvZwZXmajuT0GBE,1171
-tinybird/tb/modules/agent/models.py,sha256=jUEO7i_JwlQVZi6QEUeGyFJMFkPZ4ikyDZdei6s6f3M,628
+tinybird/tb/modules/agent/models.py,sha256=mf8dRCdof6uEFZWh5xQ_D_FStk7eDds7qWRNSbDklUM,589
 tinybird/tb/modules/agent/prompts.py,sha256=j7nkx1E2BnPP6Ra718FvSDGRWucyFgSX07V9QAiaiGg,4864
-tinybird/tb/modules/agent/utils.py,sha256=E785Sq_TLVBi1gaBR53YlHvyP_Kbh4TmRouYl3KHFpo,13080
+tinybird/tb/modules/agent/utils.py,sha256=BWbS8m8HYrbbm1OnH7kLYLxL37-q1yVKa1ObWw7klfc,13041
 tinybird/tb/modules/agent/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tinybird/tb/modules/agent/tools/create_datafile.py,sha256=06HK33h86T1d54zvnJXJz_1su3zIS7UJDPyBqhgo81c,1771
-tinybird/tb/modules/agent/tools/explore.py,sha256=iEVsi85CRPCPhWcl-fC1J4hYK22TfunxyqNu5pDxdDU,419
+tinybird/tb/modules/agent/tools/explore.py,sha256=ihALc_kBcsjrKT3hZyicqyIowB0g_K3AtNNi-5uz9-8,412
 tinybird/tb/modules/agent/tools/plan.py,sha256=acMbyjqI6cN7IwmMAC8-MPNSg0PEZVKGEk2NOY4gMi8,1162
 tinybird/tb/modules/agent/tools/preview_datafile.py,sha256=e9q5fR0afApcrntzFrnuHmd10ex7MG_GM6T0Pwc9bRI,850
 tinybird/tb/modules/datafile/build.py,sha256=NFKBrusFLU0WJNCXePAFWiEDuTaXpwc0lHlOQWEJ43s,51117
@@ -95,8 +94,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.dev240.dist-info/METADATA,sha256=XT_HpNQ7KOTyQt5ICua7m_CWsFaKQGB0tMvg43YO4XM,1733
-tinybird-0.0.1.dev240.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev240.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev240.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev240.dist-info/RECORD,,
+tinybird-0.0.1.dev241.dist-info/METADATA,sha256=urGByLwa4s7Sstte1j0r7UZ6iYFUmnpX4ujKWztx9II,1733
+tinybird-0.0.1.dev241.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev241.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev241.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev241.dist-info/RECORD,,

tinybird/tb/modules/agent/client.py

@@ -1,42 +0,0 @@
-from dataclasses import dataclass
-from typing import Any, Dict, Optional
-
-import httpx
-
-
-@dataclass
-class TinybirdClient:
-    def __init__(self, host: str, token: str):
-        self.host = host
-        self.token = token
-        self.client = httpx.Client(
-            timeout=30.0,
-            headers={"Accept": "application/json", "User-Agent": "Python/APIClient"},
-        )
-        self.insights: list[str] = []
-
-    def __aenter__(self):
-        return self
-
-    def __aexit__(self, exc_type, exc_val, exc_tb):
-        self.close()
-
-    def close(self):
-        """Close the underlying HTTP client."""
-        self.client.close()
-
-    def _get(self, endpoint: str, params: Optional[Dict[str, Any]] = None) -> str:
-        if params is None:
-            params = {}
-        params["token"] = self.token
-        url = f"{self.host}{endpoint}"
-        response = self.client.get(url, params=params)
-        try:
-            response.raise_for_status()
-        except Exception as e:
-            raise Exception(response.json().get("error", str(e))) from e
-        return response.text
-
-    def explore_data(self, prompt: str) -> str:
-        params = {"prompt": prompt, "host": self.host, "origin": "cli"}
-        return self._get("/v1/agents/explore", params)