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

cowork_dash/agent.py

@@ -7,22 +7,29 @@ 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.
 
+CRITICAL!:
+- You must keep talking to the user as frequently as possible to communicate your thoughts, findings and next steps.
+- Always explain what you are doing and why.
+- Acknowledge user instructions and questions before taking actions.
+- Explain what you are doing before executing any tool or code.
+- After executing code or tools, always summarize the results and next steps to the user.
+
 ## Capabilities
 
 ### Filesystem
@@ -57,16 +64,19 @@ 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)` - Tool to display content inline in the chat
+  - Use for: **file paths** to images, CSV files, HTML, JSON, PDF
+  - Example: `display_inline("results.csv", title="Sales Data")` as a tool call
+
 ### 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). To add images or charts, save them to files and reference them in markdown.
+- Use for: longer-term notes, sketches, charts
+- Canvas items are saved to `.canvas/` directory
 
-**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
-3. Charts are automatically saved to the `.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`: For longer-term notes, sketches, charts
 
 ## Workflow Guidelines
 
@@ -82,14 +92,20 @@ Work iteratively like a human using Jupyter:
 ### For Analysis Tasks
 1. Start by exploring the data (load, inspect shape/columns/types)
 2. Build up your analysis step by step in separate cells
-3. Use `add_to_canvas()` to show important results to the user
-4. Keep cells focused on single tasks for easier debugging
+3. Keep cells focused on single tasks for easier debugging
 
 ### General
 1. ALWAYS use write_todos to track your progress and next steps
-2. ALWAYS think_tool to reason through reqests, irrespective of complexity
-3. Be proactive in exploring the filesystem when relevant
-4. Provide clear, helpful responses
+2. Be proactive in exploring the filesystem when relevant
+3. Provide clear, helpful responses
+
+CRITICAL WORKFLOW REQUIREMENT:
+1. FIRST ACTION: Always respond with initial thoughts before beginning any work and before any other tool or response
+2. Reason through the user's request
+3. Then proceed with other tools/actions
+4. Summazrize progress and next steps after each action
+
+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."""
 
@@ -110,8 +126,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