tinybird 0.0.1.dev289__tar.gz → 0.0.1.dev290__tar.gz

{tinybird-0.0.1.dev289 → tinybird-0.0.1.dev290}/PKG-INFO

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

{tinybird-0.0.1.dev289 → tinybird-0.0.1.dev290}/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.dev289'
-__revision__ = 'ac20837'
+__version__ = '0.0.1.dev290'
+__revision__ = '4d1cca4'

{tinybird-0.0.1.dev289 → tinybird-0.0.1.dev290}/tinybird/tb/modules/agent/agent.py

@@ -23,6 +23,7 @@ from tinybird.tb.modules.agent.banner import display_banner
 from tinybird.tb.modules.agent.command_agent import CommandAgent
 from tinybird.tb.modules.agent.compactor import compact_messages
 from tinybird.tb.modules.agent.explore_agent import ExploreAgent
+from tinybird.tb.modules.agent.file_agent import FileAgent
 from tinybird.tb.modules.agent.memory import (
     clear_history,
     clear_messages,
@@ -185,6 +186,16 @@ class TinybirdAgent:
             workspace_id=workspace_id,
             project=self.project,
         )
+        self.file_agent = FileAgent(
+            dangerously_skip_permissions=self.dangerously_skip_permissions,
+            prompt_mode=prompt_mode,
+            thinking_animation=self.thinking_animation,
+            token=self.token,
+            user_token=self.user_token,
+            host=self.host,
+            workspace_id=workspace_id,
+            project=self.project,
+        )
 
         @self.agent.tool
         def manage_tests(ctx: RunContext[TinybirdAgentContext], task: str) -> str:
@@ -251,6 +262,19 @@ class TinybirdAgent:
             result = self.mock_agent.run(user_input, deps=ctx.deps, usage=ctx.usage)
             return result.output or "No result returned"
 
+        @self.agent.tool
+        def manage_files(ctx: RunContext[TinybirdAgentContext], task: str) -> str:
+            """List file/folders and read any type of file in the current directory. Use this tool when you need to list or read non Tinybird project files (e.g. .txt, .md).
+
+            Args:
+                task (str): The task to solve. Required.
+
+            Returns:
+                str: The result of the task.
+            """
+            result = self.file_agent.run(task, deps=ctx.deps, usage=ctx.usage)
+            return result.output or "No result returned"
+
         @self.agent.instructions
         def get_local_host(ctx: RunContext[TinybirdAgentContext]) -> str:
             return f"""

tinybird-0.0.1.dev290/tinybird/tb/modules/agent/file_agent.py

@@ -0,0 +1,64 @@
+from pydantic_ai import Agent, Tool
+from pydantic_ai.messages import ModelMessage
+from pydantic_ai.usage import Usage
+
+from tinybird.tb.modules.agent.animations import ThinkingAnimation
+from tinybird.tb.modules.agent.models import create_model
+from tinybird.tb.modules.agent.prompts import tone_and_style_instructions
+from tinybird.tb.modules.agent.tools.file import list_files, read_file
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.project import Project
+
+
+class FileAgent:
+    def __init__(
+        self,
+        token: str,
+        user_token: str,
+        host: str,
+        workspace_id: str,
+        project: Project,
+        dangerously_skip_permissions: bool,
+        prompt_mode: bool,
+        thinking_animation: ThinkingAnimation,
+    ):
+        self.token = token
+        self.user_token = user_token
+        self.host = host
+        self.workspace_id = workspace_id
+        self.dangerously_skip_permissions = dangerously_skip_permissions or prompt_mode
+        self.project = project
+        self.thinking_animation = thinking_animation
+        self.messages: list[ModelMessage] = []
+        self.agent = Agent(
+            model=create_model(user_token, host, workspace_id),
+            deps_type=TinybirdAgentContext,
+            instructions=[
+                """
+You are part of Tinybird Code, an agentic CLI that can help users to work with Tinybird.                 
+You are a sub-agent of the main Tinybird Code agent. You are responsible for managing files.
+You can do the following:
+- List files in a directory.
+- Read a file.
+
+IMPORTANT: Use the `list_files` tool always before `read_file` to get the correct file path.
+""",
+                tone_and_style_instructions,
+            ],
+            tools=[
+                Tool(list_files, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+                Tool(read_file, docstring_format="google", require_parameter_descriptions=True, takes_ctx=True),
+            ],
+        )
+
+    def run(self, task: str, deps: TinybirdAgentContext, usage: Usage):
+        result = self.agent.run_sync(
+            task,
+            deps=deps,
+            usage=usage,
+            message_history=self.messages,
+            model=create_model(self.user_token, self.host, self.workspace_id, run_id=deps.run_id),
+        )
+        new_messages = result.new_messages()
+        self.messages.extend(new_messages)
+        return result

{tinybird-0.0.1.dev289 → tinybird-0.0.1.dev290}/tinybird/tb/modules/agent/tools/append.py

@@ -24,7 +24,7 @@ def append_file(
     Args:
         datasource_name: Name of the datasource to append fixture to
         fixture_pathname: Path to the fixture file to append
-        cloud: Whether to append the fixture to the cloud or local environment. If None (user didn't specify), will ask user to clarify. Defaults to local (False) in dangerous skip permissions mode.
+        cloud: Whether to append the fixture to the cloud or local environment. If None (user didn't specify in a previous step), will ask user to clarify. Defaults to local (False) in dangerous skip permissions mode.
 
     Returns:
         str: Message indicating the success or failure of the appending
@@ -54,6 +54,7 @@ def append_file(
 
         cloud_or_local = "Cloud" if cloud else "Local"
         active_plan = ctx.deps.get_plan() is not None and not cloud
+        ctx.deps.thinking_animation.stop()
         confirmation = show_confirmation(
             title=f"Append fixture {fixture_pathname} to datasource '{datasource_name}' in Tinybird {cloud_or_local}?",
             skip_confirmation=ctx.deps.dangerously_skip_permissions or active_plan,

tinybird-0.0.1.dev290/tinybird/tb/modules/agent/tools/file.py

@@ -0,0 +1,82 @@
+from glob import glob
+from pathlib import Path
+
+import click
+from pydantic_ai import RunContext
+
+from tinybird.tb.modules.agent.utils import TinybirdAgentContext
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+
+def list_files(
+    ctx: RunContext[TinybirdAgentContext],
+    folder_path: str = ".",
+    search_pattern: str = "*",
+) -> str:
+    """Lists all files and folders in a directory.
+
+    Args:
+        folder_path (str): The path to the directory to list. If not provided, it will list the files in the current directory. Default: "."
+        search_pattern (str): The pattern to search for in the files and folders. Default: "*"
+    Returns:
+        str: The list of files in the directory.
+    """
+    try:
+        ctx.deps.thinking_animation.stop()
+        if folder_path:
+            path = Path(folder_path)
+        else:
+            path = Path.cwd()
+
+        display_folder = "current directory" if (folder_path == "." or not folder_path) else folder_path
+
+        click.echo(
+            FeedbackManager.highlight(message=f"» Searching in {display_folder} with pattern '{search_pattern}'")
+        )
+
+        if not path.exists():
+            ctx.deps.thinking_animation.start()
+            return f"Folder {folder_path} does not exist"
+
+        result = glob(f"{path}/**/{search_pattern}", recursive=True)
+        folders = [f for f in result if Path(f).is_dir()]
+        files = [f for f in result if Path(f).is_file()]
+
+        ctx.deps.thinking_animation.start()
+        return f"Folders: {folders}\nFiles: {files}"
+    except Exception as e:
+        ctx.deps.thinking_animation.start()
+        return f"Error listing files: {e}"
+
+
+def read_file(ctx: RunContext[TinybirdAgentContext], file_path: str) -> str:
+    """Reads a file from the local filesystem. The file_path parameter must be an absolute path, not a relative path. Any lines longer than 2000 characters will be truncated.
+
+    Args:
+        file_path (str): The path to the file to read.
+
+    Returns:
+        str: The content of the file.
+    """
+    try:
+        ctx.deps.thinking_animation.stop()
+        click.echo(FeedbackManager.highlight(message=f"» Reading file {file_path}"))
+        path = Path(file_path)
+
+        if not path.exists():
+            ctx.deps.thinking_animation.start()
+            return f"File {file_path} does not exist"
+
+        content = path.read_text()
+        chunks = content.split("\n")
+
+        if len(chunks) > 2000:
+            first_2000 = "\n".join(chunks[0:2000])
+            ctx.deps.thinking_animation.start()
+            return f"This file is too large to be read by the agent. Showing first 2000 lines instead:\n\n{first_2000}"
+
+        ctx.deps.thinking_animation.start()
+        return content
+    except Exception as e:
+        ctx.deps.thinking_animation.start()
+        return f"Error reading file: {e}"

{tinybird-0.0.1.dev289 → tinybird-0.0.1.dev290}/tinybird.egg-info/PKG-INFO

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

{tinybird-0.0.1.dev289 → tinybird-0.0.1.dev290}/tinybird.egg-info/SOURCES.txt

@@ -87,6 +87,7 @@ tinybird/tb/modules/agent/banner.py
 tinybird/tb/modules/agent/command_agent.py
 tinybird/tb/modules/agent/compactor.py
 tinybird/tb/modules/agent/explore_agent.py
+tinybird/tb/modules/agent/file_agent.py
 tinybird/tb/modules/agent/memory.py
 tinybird/tb/modules/agent/mock_agent.py
 tinybird/tb/modules/agent/models.py
@@ -102,6 +103,7 @@ tinybird/tb/modules/agent/tools/deploy.py
 tinybird/tb/modules/agent/tools/deploy_check.py
 tinybird/tb/modules/agent/tools/diff_resource.py
 tinybird/tb/modules/agent/tools/execute_query.py
+tinybird/tb/modules/agent/tools/file.py
 tinybird/tb/modules/agent/tools/get_endpoint_stats.py
 tinybird/tb/modules/agent/tools/get_openapi_definition.py
 tinybird/tb/modules/agent/tools/mock.py