tinybird 0.0.1.dev281__tar.gz → 0.0.1.dev282__tar.gz

{tinybird-0.0.1.dev281 → tinybird-0.0.1.dev282}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev281
+Version: 0.0.1.dev282
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -22,7 +22,8 @@ Requires-Dist: humanfriendly~=8.2
 Requires-Dist: plotext==5.3.2
 Requires-Dist: prompt_toolkit==3.0.48
 Requires-Dist: pydantic~=2.11.7
-Requires-Dist: pydantic-ai-slim[anthropic]~=0.4.2
+Requires-Dist: pydantic-ai-slim[anthropic]~=0.5.0
+Requires-Dist: pydantic-ai-slim[retries]~=0.5.0
 Requires-Dist: pyperclip==1.8.2
 Requires-Dist: pyyaml<6.1,>=6.0
 Requires-Dist: requests<3,>=2.28.1

{tinybird-0.0.1.dev281 → tinybird-0.0.1.dev282}/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.dev281'
-__revision__ = 'a7365be'
+__version__ = '0.0.1.dev282'
+__revision__ = '77eb85f'

tinybird-0.0.1.dev282/tinybird/tb/modules/agent/models.py

@@ -0,0 +1,61 @@
+from typing import Optional
+
+from anthropic import AsyncAnthropic
+from httpx import AsyncClient, HTTPStatusError
+from pydantic_ai.models.anthropic import AnthropicModel, AnthropicModelName
+from pydantic_ai.providers.anthropic import AnthropicProvider
+from pydantic_ai.retries import AsyncTenacityTransport, wait_retry_after
+from tenacity import AsyncRetrying, retry_if_exception_type, stop_after_attempt, wait_exponential
+
+
+def create_retrying_client(token: str, workspace_id: str):
+    """Create a client with smart retry handling for multiple error types."""
+
+    def should_retry_status(response):
+        """Raise exceptions for retryable HTTP status codes."""
+        if response.status_code in (400, 429, 502, 503, 504):
+            response.raise_for_status()  # This will raise HTTPStatusError
+
+    transport = AsyncTenacityTransport(
+        controller=AsyncRetrying(
+            # Retry on HTTP errors and connection issues
+            retry=retry_if_exception_type((HTTPStatusError, ConnectionError)),
+            # Smart waiting: respects Retry-After headers, falls back to exponential backoff
+            wait=wait_retry_after(fallback_strategy=wait_exponential(multiplier=1, max=60), max_wait=300),
+            # Stop after 5 attempts
+            stop=stop_after_attempt(5),
+            # Re-raise the last exception if all retries fail
+            reraise=True,
+        ),
+        validate_response=should_retry_status,
+    )
+    return AsyncClient(transport=transport, params={"token": token, "workspace_id": workspace_id})
+
+
+def create_model(
+    token: str,
+    base_url: str,
+    workspace_id: str,
+    model: AnthropicModelName = "claude-4-sonnet-20250514",
+    run_id: Optional[str] = None,
+):
+    default_headers = {}
+    if run_id:
+        default_headers["X-Run-Id"] = run_id
+
+    client = AsyncAnthropic(
+        base_url=base_url,
+        http_client=create_retrying_client(token, workspace_id),
+        auth_token=token,
+        default_headers=default_headers,
+    )
+    return AnthropicModel(
+        model_name=model,
+        provider=AnthropicProvider(anthropic_client=client),
+    )
+
+
+model_costs = {
+    "input_cost_per_token": 3e-06,
+    "output_cost_per_token": 1.5e-05,
+}

{tinybird-0.0.1.dev281 → tinybird-0.0.1.dev282}/tinybird/tb/modules/login_common.py

@@ -257,19 +257,31 @@ def open_url(url: str, *, new_tab: bool = False) -> bool:
 
     # 2. Inside WSL, prefer `wslview` if the user has it (wslu package).
     if _running_in_wsl() and shutil.which("wslview"):
-        subprocess.Popen(["wslview", url])
-        return True
+        try:
+            subprocess.Popen(["wslview", url])
+            return True
+        except Exception:
+            # wslview not found, continue to next fallback
+            pass
 
     # 3. Secondary WSL fallback use Windows **start** through cmd.exe.
     #    Empty "" argument is required so long URLs are not treated as a window title.
     if _running_in_wsl():
-        subprocess.Popen(["cmd.exe", "/c", "start", "", url])
-        return True
+        try:
+            subprocess.Popen(["cmd.exe", "/c", "start", "", url])
+            return True
+        except Exception:
+            # cmd.exe not found, continue to next fallback
+            pass
 
     # 4. Unix last-ditch fallback xdg-open (most minimal container images have it)
     if shutil.which("xdg-open"):
-        subprocess.Popen(["xdg-open", url])
-        return True
+        try:
+            subprocess.Popen(["xdg-open", url])
+            return True
+        except Exception:
+            # xdg-open not found, continue to next fallback
+            pass
 
     # 5. If everything failed, let the caller know.
     return False

{tinybird-0.0.1.dev281 → tinybird-0.0.1.dev282}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev281
+Version: 0.0.1.dev282
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -22,7 +22,8 @@ Requires-Dist: humanfriendly~=8.2
 Requires-Dist: plotext==5.3.2
 Requires-Dist: prompt_toolkit==3.0.48
 Requires-Dist: pydantic~=2.11.7
-Requires-Dist: pydantic-ai-slim[anthropic]~=0.4.2
+Requires-Dist: pydantic-ai-slim[anthropic]~=0.5.0
+Requires-Dist: pydantic-ai-slim[retries]~=0.5.0
 Requires-Dist: pyperclip==1.8.2
 Requires-Dist: pyyaml<6.1,>=6.0
 Requires-Dist: requests<3,>=2.28.1

{tinybird-0.0.1.dev281 → tinybird-0.0.1.dev282}/tinybird.egg-info/requires.txt

@@ -13,7 +13,8 @@ humanfriendly~=8.2
 plotext==5.3.2
 prompt_toolkit==3.0.48
 pydantic~=2.11.7
-pydantic-ai-slim[anthropic]~=0.4.2
+pydantic-ai-slim[anthropic]~=0.5.0
+pydantic-ai-slim[retries]~=0.5.0
 pyperclip==1.8.2
 pyyaml<6.1,>=6.0
 requests<3,>=2.28.1

tinybird-0.0.1.dev281/tinybird/tb/modules/agent/models.py

@@ -1,35 +0,0 @@
-from typing import Optional
-
-from anthropic import AsyncAnthropic
-from httpx import AsyncClient
-from pydantic_ai.models.anthropic import AnthropicModel, AnthropicModelName
-from pydantic_ai.providers.anthropic import AnthropicProvider
-
-
-def create_model(
-    token: str,
-    base_url: str,
-    workspace_id: str,
-    model: AnthropicModelName = "claude-4-sonnet-20250514",
-    run_id: Optional[str] = None,
-):
-    default_headers = {}
-    if run_id:
-        default_headers["X-Run-Id"] = run_id
-
-    client = AsyncAnthropic(
-        base_url=base_url,
-        http_client=AsyncClient(params={"token": token, "workspace_id": workspace_id}),
-        auth_token=token,
-        default_headers=default_headers,
-    )
-    return AnthropicModel(
-        model_name=model,
-        provider=AnthropicProvider(anthropic_client=client),
-    )
-
-
-model_costs = {
-    "input_cost_per_token": 3e-06,
-    "output_cost_per_token": 1.5e-05,
-}