vibecore 0.3.0__py3-none-any.whl → 0.6.2__py3-none-any.whl

vibecore/tools/shell/tools.py

@@ -2,14 +2,14 @@
 
 from agents import RunContextWrapper, function_tool
 
-from vibecore.context import VibecoreContext
+from vibecore.context import PathValidatorContext
 
 from .executor import bash_executor, glob_files, grep_files, list_directory
 
 
 @function_tool
 async def bash(
-    ctx: RunContextWrapper[VibecoreContext],
+    ctx: RunContextWrapper[PathValidatorContext],
     command: str,
     timeout: int | None = None,
     description: str | None = None,
@@ -58,7 +58,7 @@ async def bash(
     Returns:
         The command output or error message
     """
-    output, exit_code = await bash_executor(command, timeout)
+    output, exit_code = await bash_executor(ctx, command, timeout)
     if exit_code != 0:
         return f"{output}\nExit code: {exit_code}"
     return output
@@ -66,7 +66,7 @@ async def bash(
 
 @function_tool
 async def glob(
-    ctx: RunContextWrapper[VibecoreContext],
+    ctx: RunContextWrapper[PathValidatorContext],
     pattern: str,
     path: str | None = None,
 ) -> str:
@@ -90,7 +90,7 @@ async def glob(
     Returns:
         List of matching file paths, one per line
     """
-    files = await glob_files(pattern, path)
+    files = await glob_files(ctx, pattern, path)
     if files and files[0].startswith("Error:"):
         return files[0]
     return "\n".join(files) if files else "No files found matching pattern"
@@ -98,7 +98,7 @@ async def glob(
 
 @function_tool
 async def grep(
-    ctx: RunContextWrapper[VibecoreContext],
+    ctx: RunContextWrapper[PathValidatorContext],
     pattern: str,
     path: str | None = None,
     include: str | None = None,
@@ -124,7 +124,7 @@ async def grep(
     Returns:
         List of file paths containing matches, one per line
     """
-    files = await grep_files(pattern, path, include)
+    files = await grep_files(ctx, pattern, path, include)
     if files and files[0].startswith("Error:"):
         return files[0]
     return "\n".join(files) if files else "No files found containing pattern"
@@ -132,7 +132,7 @@ async def grep(
 
 @function_tool
 async def ls(
-    ctx: RunContextWrapper[VibecoreContext],
+    ctx: RunContextWrapper[PathValidatorContext],
     path: str,
     ignore: list[str] | None = None,
 ) -> str:
@@ -150,7 +150,7 @@ async def ls(
     Returns:
         List of entries in the directory, one per line
     """
-    entries = await list_directory(path, ignore)
+    entries = await list_directory(ctx, path, ignore)
     if entries and entries[0].startswith("Error:"):
         return entries[0]
     return "\n".join(entries) if entries else "Empty directory"

vibecore/tools/task/executor.py

@@ -8,12 +8,12 @@ from agents import (
 from textual import log
 
 from vibecore.agents.task import create_task_agent
-from vibecore.context import VibecoreContext
+from vibecore.context import FullVibecoreContext
 from vibecore.settings import settings
 
 
 async def execute_task(
-    context: VibecoreContext,
+    context: FullVibecoreContext,
     description: str,
     prompt: str,
     tool_name: str,

vibecore/tools/task/tools.py

@@ -3,14 +3,14 @@
 from agents import function_tool
 from agents.tool_context import ToolContext
 
-from vibecore.context import VibecoreContext
+from vibecore.context import FullVibecoreContext
 
 from .executor import execute_task
 
 
 @function_tool
 async def task(
-    ctx: ToolContext[VibecoreContext],
+    ctx: ToolContext[FullVibecoreContext],
     description: str,
     prompt: str,
 ) -> str:

vibecore/tools/todo/manager.py

@@ -2,7 +2,7 @@
 
 from typing import Any
 
-from .models import TodoItem, TodoPriority, TodoStatus
+from .models import TodoItem
 
 
 class TodoManager:
@@ -20,12 +20,4 @@ class TodoManager:
 
     def write(self, todos: list[dict[str, Any]]) -> None:
         """Replace the entire todo list."""
-        self.todos = [
-            TodoItem(
-                id=todo["id"],
-                content=todo["content"],
-                status=TodoStatus(todo["status"]),
-                priority=TodoPriority(todo["priority"]),
-            )
-            for todo in todos
-        ]
+        self.todos = [TodoItem(**todo) for todo in todos]

vibecore/tools/todo/models.py

@@ -1,10 +1,9 @@
 """Todo data models."""
 
 import uuid
-from dataclasses import dataclass, field
 from enum import Enum
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
 
 class TodoStatus(str, Enum):
@@ -19,18 +18,10 @@ class TodoPriority(str, Enum):
     LOW = "low"
 
 
-@dataclass
-class TodoItem:
-    content: str
-    status: TodoStatus
-    priority: TodoPriority
-    id: str = field(default_factory=lambda: str(uuid.uuid4()))
-
-
-class TodoItemModel(BaseModel):
+class TodoItem(BaseModel):
     """Pydantic model for todo items."""
 
-    id: str
+    id: str = Field(default_factory=lambda: str(uuid.uuid4()))
     content: str
-    status: str
-    priority: str
+    status: TodoStatus
+    priority: TodoPriority

vibecore/tools/todo/tools.py

@@ -4,13 +4,13 @@ from typing import Any
 
 from agents import RunContextWrapper, function_tool
 
-from vibecore.context import VibecoreContext
+from vibecore.context import TodoToolContext
 
-from .models import TodoItemModel
+from .models import TodoItem
 
 
 @function_tool
-async def todo_read(ctx: RunContextWrapper[VibecoreContext]) -> list[dict[str, Any]]:
+async def todo_read(ctx: RunContextWrapper[TodoToolContext]) -> list[dict[str, Any]]:
     """Use this tool to read the current to-do list for the session. This tool should be used proactively and
     frequently to ensure that you are aware of the status of the current task list. You should make use of this
     tool as often as possible, especially in the following situations:
@@ -38,7 +38,7 @@ async def todo_read(ctx: RunContextWrapper[VibecoreContext]) -> list[dict[str, A
 
 
 @function_tool
-async def todo_write(ctx: RunContextWrapper[VibecoreContext], todos: list[TodoItemModel]) -> str:
+async def todo_write(ctx: RunContextWrapper[TodoToolContext], todos: list[TodoItem]) -> str:
     """Use this tool to create and manage a structured task list for your current coding session. This helps you
     track progress, organize complex tasks, and demonstrate thoroughness to the user. It also helps the user
     understand the progress of the task and overall progress of their requests.
@@ -106,6 +106,6 @@ async def todo_write(ctx: RunContextWrapper[VibecoreContext], todos: list[TodoIt
             Success message.
     """
     # Convert Pydantic models to dicts for the implementation
-    todos_dict = [todo.model_dump() for todo in todos]
+    todos_dict = [todo.model_dump() if isinstance(todo, TodoItem) else TodoItem(**todo).model_dump() for todo in todos]
     ctx.context.todo_manager.write(todos_dict)
     return "Todo list updated successfully."

vibecore/tools/webfetch/tools.py

@@ -1,8 +1,6 @@
 """Webfetch tool for Vibecore agents."""
 
-from agents import RunContextWrapper, function_tool
-
-from vibecore.context import VibecoreContext
+from agents import function_tool
 
 from .executor import fetch_url
 from .models import WebFetchParams
@@ -10,7 +8,6 @@ from .models import WebFetchParams
 
 @function_tool
 async def webfetch(
-    ctx: RunContextWrapper[VibecoreContext],
     url: str,
     timeout: int = 30,
     follow_redirects: bool = True,

vibecore/tools/websearch/ddgs/backend.py

@@ -31,7 +31,7 @@ class DDGSBackend(WebSearchBackend):
 
             # Perform search (synchronous call)
             # ddgs.text() expects 'query' as first positional argument
-            raw_results = ddgs.text(
+            raw_results = ddgs.text(  # type: ignore
                 query=params.query,
                 region=params.region,
                 safesearch=params.safesearch,

vibecore/tools/websearch/tools.py

@@ -1,8 +1,6 @@
 """Websearch tool for Vibecore agents."""
 
-from agents import RunContextWrapper, function_tool
-
-from vibecore.context import VibecoreContext
+from agents import function_tool
 
 from .executor import perform_websearch
 from .models import SearchParams
@@ -10,7 +8,6 @@ from .models import SearchParams
 
 @function_tool
 async def websearch(
-    ctx: RunContextWrapper[VibecoreContext],
     query: str,
     max_results: int = 5,
     region: str | None = None,

vibecore/widgets/core.py

@@ -5,7 +5,6 @@ from typing import ClassVar
 from textual import events
 from textual.app import ComposeResult
 from textual.containers import Horizontal, ScrollableContainer, Vertical
-from textual.geometry import Size
 from textual.message import Message
 from textual.widget import Widget
 from textual.widgets import Footer, ProgressBar, Static, TextArea
@@ -94,10 +93,10 @@ class MyTextArea(TextArea):
             history = []
 
             # First, load history from the session
-            if app.session:
+            if app.runner.session:
                 try:
                     # Get all items from the session
-                    session_items = await app.session.get_items()
+                    session_items = await app.runner.session.get_items()
                     for item in session_items:
                         # Filter for user messages
                         if isinstance(item, dict):
@@ -113,13 +112,6 @@ class MyTextArea(TextArea):
 
                     log(f"Error loading session history: {e}")
 
-            # Then add current session's messages (in memory, not yet persisted)
-            for item in app.input_items:
-                if isinstance(item, dict) and item.get("role") == "user":
-                    content = item.get("content")
-                    if isinstance(content, str) and content not in history:
-                        history.append(content)
-
             return history
         return []
 
@@ -218,13 +210,7 @@ class MyTextArea(TextArea):
 class MainScroll(ScrollableContainer):
     """A container with vertical layout and an automatic scrollbar on the Y axis."""
 
-    def watch_virtual_size(self, size: Size) -> None:
-        """Scroll to the bottom when resized = when new content is added."""
-        # If the scroll is near the end, keep the scroll sticky to the end
-        epsilon = 30
-        in_the_end = (size.height - (self.scroll_target_y + self.scrollable_size.height)) < epsilon
-        if size.height > self.scrollable_size.height and in_the_end:
-            self.scroll_end(animate=False)
+    ...
 
 
 class LoadingWidget(Widget):

vibecore/widgets/feedback.py

@@ -0,0 +1,164 @@
+"""Feedback widget for collecting user feedback."""
+
+from textual.app import ComposeResult
+from textual.containers import Horizontal, Vertical
+from textual.content import Content
+from textual.message import Message
+from textual.reactive import reactive
+from textual.widgets import Button, Checkbox, Static, TextArea
+
+from vibecore.widgets.messages import BaseMessage, MessageHeader, MessageStatus
+
+
+class FeedbackWidget(BaseMessage):
+    """A widget to collect user feedback with Good/Bad rating and optional text comment."""
+
+    class FeedbackSubmitted(Message):
+        """Event emitted when feedback is submitted."""
+
+        def __init__(self, rating: str, comment: str, criteria: dict[str, bool]) -> None:
+            """
+            Construct a FeedbackSubmitted message.
+
+            Args:
+                rating: The rating ("good" or "bad").
+                comment: Optional text comment from the user.
+                criteria: Dict of criterion name to boolean value.
+            """
+            super().__init__()
+            self.rating = rating
+            self.comment = comment
+            self.criteria = criteria
+
+    rating: reactive[str | None] = reactive(None)
+    comment: reactive[str] = reactive("")
+    submitted: reactive[bool] = reactive(False)
+    show_comment_input: reactive[bool] = reactive(False)
+    show_criteria: reactive[bool] = reactive(False)
+
+    def __init__(self, prompt: str = "How was this response?", **kwargs) -> None:
+        """
+        Construct a FeedbackWidget.
+
+        Args:
+            prompt: The prompt text to display.
+            **kwargs: Additional keyword arguments for Widget.
+        """
+        super().__init__(status=MessageStatus.IDLE, **kwargs)
+        self.prompt = prompt
+        self.add_class("feedback-widget")
+
+    def get_header_params(self) -> tuple[str, str, bool]:
+        """Get parameters for MessageHeader."""
+        return ("⏺", self.prompt, False)
+
+    def compose(self) -> ComposeResult:
+        """Create child widgets for the feedback widget."""
+        yield MessageHeader("⏺", self.prompt, status=self.status)
+
+        with Horizontal(classes="feedback-controls"):
+            yield Button("👍 Good", id="feedback-good", classes="feedback-button good-button", variant="success")
+            yield Button("👎 Bad", id="feedback-bad", classes="feedback-button bad-button", variant="error")
+
+        # Feedback form containing criteria, comment area, and submit button (shown after rating selection)
+        with Vertical(id="feedback-form", classes="feedback-form"):
+            # Structured feedback criteria checkboxes
+            with Vertical(classes="feedback-criteria"):
+                yield Static("Please check any criteria that apply:", classes="criteria-label")
+                yield Checkbox("Factual accuracy - Was the information correct?", id="criteria-accuracy")
+                yield Checkbox("Task completion - Did it fully address the request?", id="criteria-completion")
+                yield Checkbox(
+                    "Instruction following - Did it follow specific constraints or requirements?",
+                    id="criteria-instructions",
+                )
+                yield Checkbox(
+                    "Good format/structure - Are the format and structure appropriate?", id="criteria-format"
+                )
+
+            # Comment text area
+            with Vertical(classes="feedback-comment-area"):
+                yield Static("Optional comment:", classes="comment-label")
+                yield TextArea(id="feedback-textarea", classes="feedback-textarea", show_line_numbers=False)
+
+            # Submit button
+            yield Button("Submit", id="feedback-submit", classes="feedback-submit-button", variant="primary")
+
+        with Vertical(id="feedback-result", classes="feedback-result"):
+            yield Static("", id="feedback-result-text")
+
+    def on_mount(self) -> None:
+        """Initialize widget state on mount."""
+        # Hide feedback form and result initially
+        self.query_one("#feedback-form").display = False
+        self.query_one("#feedback-result").display = False
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button press events."""
+        button_id = event.button.id
+
+        if self.submitted:
+            return  # Ignore clicks after submission
+
+        if button_id == "feedback-good":
+            self.rating = "good"
+            self._show_feedback_form()
+            event.button.add_class("selected")
+            self.query_one("#feedback-bad").remove_class("selected")
+
+        elif button_id == "feedback-bad":
+            self.rating = "bad"
+            self._show_feedback_form()
+            event.button.add_class("selected")
+            self.query_one("#feedback-good").remove_class("selected")
+
+        elif button_id == "feedback-submit":
+            self._submit_feedback()
+
+    def _show_feedback_form(self) -> None:
+        """Show the feedback form (criteria, comment area, and submit button)."""
+        self.show_criteria = True
+        self.show_comment_input = True
+        feedback_form = self.query_one("#feedback-form")
+        feedback_form.display = True
+        feedback_form.refresh()
+
+    def _get_criteria_values(self) -> dict[str, bool]:
+        """Get the current state of all criteria checkboxes."""
+        return {
+            "accuracy": self.query_one("#criteria-accuracy", Checkbox).value,
+            "completion": self.query_one("#criteria-completion", Checkbox).value,
+            "instructions": self.query_one("#criteria-instructions", Checkbox).value,
+            "format": self.query_one("#criteria-format", Checkbox).value,
+        }
+
+    def _submit_feedback(self) -> None:
+        """Submit the feedback."""
+        if not self.rating:
+            # Require a rating before submission
+            return
+
+        self.submitted = True
+        self.comment = self.query_one("#feedback-textarea", TextArea).text
+        criteria = self._get_criteria_values()
+
+        # Hide controls and form, show result
+        self.query_one(".feedback-controls").display = False
+        self.query_one("#feedback-form").display = False
+
+        # Show result
+        result_container = self.query_one("#feedback-result")
+        result_text = self.query_one("#feedback-result-text", Static)
+
+        rating_emoji = "👍" if self.rating == "good" else "👎"
+        result_msg = f"{rating_emoji} Thank you for your feedback!"
+        if self.comment:
+            result_msg += f"\nComment: {self.comment}"
+
+        result_text.update(Content(result_msg))
+        result_container.display = True
+
+        # Update status to success
+        self.status = MessageStatus.SUCCESS
+
+        # Emit feedback event
+        self.post_message(self.FeedbackSubmitted(self.rating, self.comment, criteria))

vibecore/widgets/feedback.tcss

@@ -0,0 +1,121 @@
+/* Feedback widget styles */
+
+FeedbackWidget {
+    color: $text;
+
+    .feedback-controls {
+        height: auto;
+        width: 1fr;
+        padding: 0 0 0 4;
+        margin-top: 1;
+
+        .feedback-button {
+            height: 3;
+            min-width: 12;
+            margin-right: 1;
+
+            &.selected {
+                text-style: bold;
+                border: heavy;
+            }
+
+            &.good-button {
+                background: $success;
+                color: $text;
+
+                &:hover {
+                    background: $success-lighten-1;
+                }
+
+                &.selected {
+                    background: $success-darken-1;
+                }
+            }
+
+            &.bad-button {
+                background: $error;
+                color: $text;
+
+                &:hover {
+                    background: $error-lighten-1;
+                }
+
+                &.selected {
+                    background: $error-darken-1;
+                }
+            }
+        }
+
+    }
+
+    .feedback-form {
+        height: auto;
+        width: 1fr;
+        padding: 0 0 0 4;
+        margin-top: 1;
+
+        .feedback-criteria {
+            height: auto;
+            width: 1fr;
+
+            .criteria-label {
+                color: $text-muted;
+                margin-bottom: 0;
+                height: 1;
+            }
+
+            Checkbox {
+                border: none;
+                margin: 0;
+                padding: 0;
+                color: $text;
+
+                &:focus {
+                    background: $panel;
+                }
+            }
+        }
+
+        .feedback-comment-area {
+            height: auto;
+            width: 1fr;
+            margin-top: 1;
+
+            .comment-label {
+                color: $text-muted;
+                margin-bottom: 0;
+                height: 1;
+            }
+
+            .feedback-textarea {
+                height: 6;
+                width: 1fr;
+                border: tall $primary;
+            }
+        }
+
+        .feedback-submit-button {
+            height: 3;
+            min-width: 10;
+            margin-top: 1;
+            background: $primary;
+            color: $text;
+
+            &:hover {
+                background: $primary-lighten-1;
+            }
+        }
+    }
+
+    .feedback-result {
+        height: auto;
+        width: 1fr;
+        padding: 1 0 0 4;
+        margin-top: 1;
+        color: $success;
+
+        #feedback-result-text {
+            color: $success;
+        }
+    }
+}

vibecore/widgets/messages.py

@@ -1,10 +1,12 @@
+import os
 from enum import StrEnum
 
 from textual.app import ComposeResult
+from textual.containers import Horizontal
 from textual.content import Content
 from textual.reactive import reactive
 from textual.widget import Widget
-from textual.widgets import Markdown, Static
+from textual.widgets import Button, Markdown, Static
 
 
 class MessageStatus(StrEnum):
@@ -83,11 +85,16 @@ class MessageHeader(Widget):
         # self.query_one(".prefix").visible = self._prefix_visible
 
     def _on_mount(self, event) -> None:
+        disable_blink = bool(os.environ.get("TEXTUAL_SNAPSHOT_TEMPDIR"))
         self.blink_timer = self.set_interval(
             0.5,
             self._toggle_cursor_blink_visible,
-            pause=(self.status != MessageStatus.EXECUTING),
+            pause=(self.status != MessageStatus.EXECUTING) or disable_blink,
         )
+        # Ensure the prefix starts visible for executing statuses so snapshot tests
+        # and initial renders see the indicator before the first timer tick hides it.
+        if self.status == MessageStatus.EXECUTING:
+            self._prefix_visible = True
 
 
 class BaseMessage(Widget):
@@ -167,6 +174,19 @@ class AgentMessage(BaseMessage):
         """Get parameters for MessageHeader."""
         return ("⏺", self.text, True)
 
+    def compose(self) -> ComposeResult:
+        """Create child widgets for the agent message."""
+        prefix, text, use_markdown = self.get_header_params()
+        with Horizontal(classes="agent-message-header"):
+            yield MessageHeader(prefix, text, status=self.status, use_markdown=use_markdown)
+            yield Button("Copy", classes="copy-button", variant="primary")
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button press events."""
+        if event.button.has_class("copy-button"):
+            # Copy the markdown text to clipboard
+            self.app.copy_to_clipboard(self.text)
+
     def update(self, text: str, status: MessageStatus | None = None) -> None:
         """Update the text of the agent message."""
         self.text = text

vibecore/widgets/messages.tcss

@@ -62,6 +62,34 @@ UserMessage {
 
 AgentMessage {
     color: $text;
+
+    Horizontal.agent-message-header {
+        height: auto;
+        width: 1fr;
+        layers: main button;
+
+        .copy-button {
+            layer: button;
+            dock: right;
+            height: 1;
+            width: 8;
+            min-width: 8;
+            margin: 0;
+            padding: 0;
+            background: $secondary;
+            color: $text;
+            border: none;
+            
+            &:hover {
+                background: $secondary-lighten-1;
+            }
+            
+            &:focus {
+                background: $secondary-lighten-1;
+                text-style: bold;
+            }
+        }
+    }
 }
 
 SystemMessage {