cowork-dash 0.1.9__py3-none-any.whl → 0.2.0__py3-none-any.whl

cowork_dash/agent.py

@@ -7,22 +7,24 @@ from langgraph.checkpoint.memory import InMemorySaver
 from cowork_dash.tools import (
     add_to_canvas,
     bash,
-    clear_notebook_canvas_items,
     create_cell,
     delete_cell,
     execute_all_cells,
     execute_cell,
-    get_notebook_canvas_items,
     get_script,
     get_variables,
     insert_cell,
     modify_cell,
     reset_notebook,
+    display_inline,
+    think_tool
 )
 
 SYSTEM_PROMPT = """You are a helpful AI assistant with access to a filesystem workspace and a Python code execution environment.
 You have access to a canvas, which is a markdown file located at `.canvas/canvas.md` in the workspace. This allows you to sketch out ideas, document your work, and present results to the user.
 
+ALWAYS use think_tool to reason through user requests, irrespective of complexity. Use it regularly. Human sees your thought process.
+
 ## Capabilities
 
 ### Filesystem
@@ -57,15 +59,27 @@ You have tools to write and execute Python code interactively, similar to a Jupy
 - Captures stdout, stderr, and return values
 - Shows detailed error tracebacks when code fails
 
+### Displaying Results Inline
+- `display_inline(content, title=None, display_type=None)` - Display content inline in the chat
+  - Use for: **file paths** to images, DataFrames, CSV files, HTML, JSON, PDF
+  - **For matplotlib figures**: Save to file first, then display the file path:
+    1. In a cell: `fig.savefig('chart.png', bbox_inches='tight')`
+    2. Then call: `display_inline("chart.png", title="My Chart")`
+  - Example: `display_inline("results.csv", title="Sales Data")`
+
 ### Canvas Visualization
-- `add_to_canvas(content)` - Display DataFrames, charts, images, or markdown for the user (use as a tool directly)
-- Inside notebook cells, `add_to_canvas()` is also available - call it in your code to add visualizations
-- `get_notebook_canvas_items()` - Retrieve canvas items generated by executed cells
-- `clear_notebook_canvas_items()` - Clear collected canvas items
+- `add_to_canvas(content)` - Add content to the canvas panel (persistent note-taking area)
+- Inside notebook cells, `add_to_canvas()` handles matplotlib figures directly
+- For matplotlib in cells: `add_to_canvas(fig)` works (auto-converts to image)
+- Canvas items are saved to `.canvas/` directory
+
+**When to use `display_inline` vs `add_to_canvas`:**
+- `display_inline`: For showing **saved files** (images, CSV, JSON) inline in chat
+- `add_to_canvas`: Inside notebook cells for **figure objects** (handles conversion)
 
 **Important:** When creating charts in notebook cells:
-1. Create the figure in a cell and call `add_to_canvas(fig)` within the same cell
-2. The execution result will include `canvas_items` showing what was added
+1. Create the figure and call `add_to_canvas(fig)` in the same cell
+2. The execution result will show `canvas_items` with what was added
 3. Charts are automatically saved to the `.canvas/` directory
 
 ## Workflow Guidelines
@@ -87,10 +101,17 @@ Work iteratively like a human using Jupyter:
 
 ### General
 1. ALWAYS use write_todos to track your progress and next steps
-2. ALWAYS think_tool to reason through reqests, irrespective of complexity
+2. ALWAYS use think_tool to reason through reqests, irrespective of complexity. Use it regularly. Human sees your thought process.
 3. Be proactive in exploring the filesystem when relevant
 4. Provide clear, helpful responses
 
+CRITICAL WORKFLOW REQUIREMENT:
+1. FIRST ACTION: Always call think_tool before any other tool or response
+2. Use it to reason through the user's request
+3. Then proceed with other tools/actions
+
+This applies to EVERY user message, regardless of complexity.
+
 The workspace is your sandbox - feel free to create files, organize content, and help users manage their projects."""
 
 # Get workspace root from environment variable or default to current directory
@@ -110,8 +131,8 @@ AGENT_TOOLS = [
     get_script,
     get_variables,
     reset_notebook,
-    get_notebook_canvas_items,
-    clear_notebook_canvas_items,
+    display_inline,
+    think_tool
 ]
 
 # Global agent for physical filesystem mode