tinybird 0.0.1.dev240__tar.gz → 0.0.1.dev241__tar.gz

{tinybird-0.0.1.dev240 → tinybird-0.0.1.dev241}/PKG-INFO

@@ -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 → tinybird-0.0.1.dev241}/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-0.0.1.dev240 → tinybird-0.0.1.dev241}/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-0.0.1.dev240 → tinybird-0.0.1.dev241}/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-0.0.1.dev240 → tinybird-0.0.1.dev241}/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-0.0.1.dev240 → tinybird-0.0.1.dev241}/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 → tinybird-0.0.1.dev241}/tinybird.egg-info/PKG-INFO

@@ -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 → tinybird-0.0.1.dev241}/tinybird.egg-info/SOURCES.txt

@@ -76,7 +76,6 @@ tinybird/tb/modules/workspace_members.py
 tinybird/tb/modules/agent/__init__.py
 tinybird/tb/modules/agent/agent.py
 tinybird/tb/modules/agent/banner.py
-tinybird/tb/modules/agent/client.py
 tinybird/tb/modules/agent/memory.py
 tinybird/tb/modules/agent/models.py
 tinybird/tb/modules/agent/prompts.py

tinybird-0.0.1.dev240/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)