cowork-dash 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

cowork_dash/agent.py

@@ -1,3 +1,4 @@
+import os
 import uuid
 from deepagents import create_deep_agent
 from langgraph.checkpoint.memory import InMemorySaver
@@ -85,14 +86,16 @@ Work iteratively like a human using Jupyter:
 4. Keep cells focused on single tasks for easier debugging
 
 ### General
-1. Use write_todos to track your progress and next steps
-2. Use think_tool to reason through complex problems
+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
 
 The workspace is your sandbox - feel free to create files, organize content, and help users manage their projects."""
 
-backend = FilesystemBackend(root_dir=str("./"), virtual_mode=True)
+# Get workspace root from environment variable or default to current directory
+workspace_root = os.getenv("DEEPAGENT_WORKSPACE_ROOT", os.getcwd())
+backend = FilesystemBackend(root_dir=workspace_root, virtual_mode=True)
 
 agent = create_deep_agent(
     system_prompt=SYSTEM_PROMPT,
@@ -113,6 +116,6 @@ agent = create_deep_agent(
         get_notebook_canvas_items,
         clear_notebook_canvas_items,
     ],
-    # interrupt_on=dict(bash=True),
+    interrupt_on=dict(bash=True),
     checkpointer=InMemorySaver()
 )