taskflow-agent 0.3.0__tar.gz → 0.4.0__tar.gz

{taskflow_agent-0.3.0 → taskflow_agent-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: taskflow-agent
-Version: 0.3.0
+Version: 0.4.0
 Summary: Lightweight project and task manager with MCP tools for Claude Code
 Project-URL: Repository, https://github.com/henrysouchien/taskflow-agent
 Author: Henry Chien

{taskflow_agent-0.3.0 → taskflow_agent-0.4.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "taskflow-agent"
-version = "0.3.0"
+version = "0.4.0"
 description = "Lightweight project and task manager with MCP tools for Claude Code"
 readme = "README.md"
 requires-python = ">=3.10"

{taskflow_agent-0.3.0 → taskflow_agent-0.4.0}/src/web.py

@@ -1241,7 +1241,6 @@ MUTATING_TOOL_NAMES = {
 _SUB_AGENT_EXCLUDED_TOOLS: set[str] = {
     "run_agent",
     "run_shell",
-    "load_tools",
     "tf_memory_read",
     "tf_memory_update",
 } | MUTATING_TOOL_NAMES
@@ -1261,7 +1260,7 @@ _SUB_AGENT_MAX_TURNS = 15
 _SUB_AGENT_TIMEOUT = int(os.getenv("SUB_AGENT_TIMEOUT", "300"))
 _SUB_AGENT_CLIENT_TIMEOUT = 90
 _SUB_AGENT_MAX_TOKENS = 32_000
-_SUB_AGENT_DEFAULT_MODEL = "claude-sonnet-4-6"
+_SUB_AGENT_DEFAULT_MODEL = "claude-haiku-4-5-20251001"
 
 mcp_manager = McpClientManager(
     allowed_servers=None,
@@ -1819,6 +1818,7 @@ def make_run_agent_handler(
     runner_ref: list[Any],
     local_tool_handlers: dict[str, Any],
     mcp_manager: McpClientManager,
+    event_log: EventLog,
 ):
     async def _handle_run_agent(
         tool_input: dict,
@@ -1834,7 +1834,7 @@ def make_run_agent_handler(
             return None, {"code": "invalid_input", "message": "task is required"}
 
         raw_model = tool_input.get("model")
-        allowed = {"claude-sonnet-4-6", "claude-opus-4-6"}
+        allowed = {"claude-haiku-4-5-20251001", "claude-sonnet-4-6", "claude-opus-4-6"}
         if raw_model is not None and raw_model not in allowed:
             return None, {"code": "invalid_input", "message": f"Invalid model: {raw_model}"}
 
@@ -1855,6 +1855,43 @@ def make_run_agent_handler(
             needs_approval=lambda _: False,
         )
 
+        def on_sub_event(event: dict[str, Any], session_id: str) -> None:
+            del session_id
+            event_type = event.get("type")
+            if event_type == "tool_call_start":
+                event_log.append(
+                    {
+                        "type": "sub_agent_progress",
+                        "call_index": call_index,
+                        "sub_event": "tool_start",
+                        "tool_name": event.get("tool_name", ""),
+                        "sub_tool_call_id": event.get("tool_call_id", ""),
+                    }
+                )
+                return
+            if event_type == "tool_call_complete":
+                event_log.append(
+                    {
+                        "type": "sub_agent_progress",
+                        "call_index": call_index,
+                        "sub_event": "tool_done",
+                        "tool_name": event.get("tool_name", ""),
+                        "sub_tool_call_id": event.get("tool_call_id", ""),
+                        "duration_ms": event.get("duration_ms"),
+                        "error": bool(event.get("error")),
+                    }
+                )
+                return
+            if event_type == "error":
+                event_log.append(
+                    {
+                        "type": "sub_agent_progress",
+                        "call_index": call_index,
+                        "sub_event": "error",
+                        "error_message": str(event.get("error", "Sub-agent error")),
+                    }
+                )
+
         result, error = await runner.spawn_sub_agent(
             task,
             model=effective_model,
@@ -1866,6 +1903,7 @@ def make_run_agent_handler(
             client_timeout=_SUB_AGENT_CLIENT_TIMEOUT,
             max_tokens=_SUB_AGENT_MAX_TOKENS,
             call_index=call_index,
+            on_sub_event=on_sub_event,
         )
         return result, error
 
@@ -2248,6 +2286,7 @@ async def sse_generator(
     runner_task: asyncio.Task[None],
     request_id: str,
 ):
+    run_agent_map: dict[int, str] = {}
     tool_inputs: dict[str, dict[str, Any]] = {}
     tool_calls: dict[str, dict[str, Any]] = {}
     turn_tools: list[dict[str, Any]] = []
@@ -2292,6 +2331,10 @@ async def sse_generator(
                 }
                 tool_calls[tool_call_id] = tool_entry
                 turn_tools.append(tool_entry)
+                if tool_name == "run_agent":
+                    call_index = event.get("call_index")
+                    if call_index is not None:
+                        run_agent_map[call_index] = tool_call_id
                 yield _sse(
                     {
                         "type": "tool_call_start",
@@ -2334,6 +2377,24 @@ async def sse_generator(
                     pending_mutations.append(mutation)
                 continue
 
+            if event_type == "sub_agent_progress":
+                call_index = event.get("call_index")
+                parent_tool_call_id = run_agent_map.get(call_index) if call_index is not None else None
+                if parent_tool_call_id:
+                    yield _sse(
+                        {
+                            "type": "sub_agent_progress",
+                            "parent_tool_call_id": parent_tool_call_id,
+                            "sub_event": event.get("sub_event"),
+                            "tool_name": event.get("tool_name", ""),
+                            "sub_tool_call_id": event.get("sub_tool_call_id", ""),
+                            "duration_ms": event.get("duration_ms"),
+                            "error": event.get("error"),
+                            "error_message": event.get("error_message"),
+                        }
+                    )
+                continue
+
             if event_type == "stream_complete":
                 assistant_text = "".join(assistant_chunks).strip()
                 full_content = assistant_text + _build_tool_summary(turn_tools)
@@ -2938,6 +2999,7 @@ async def chat(body: ChatRequest, request: Request):
             runner_ref,
             request_handlers,
             mcp_manager,
+            event_log,
         )
         dispatcher = ToolDispatcher(
             mcp_client=mcp_manager,

{taskflow_agent-0.3.0 → taskflow_agent-0.4.0}/static/index.html

@@ -1218,6 +1218,22 @@ button { cursor: pointer; }
   white-space: pre-wrap;
   word-break: break-word;
 }
+.chat-sub-progress {
+  padding: 4px 12px 8px;
+  font-family: var(--font-mono);
+  font-size: 10px;
+  line-height: 1.8;
+  color: var(--text-muted);
+}
+.chat-sub-progress-line {
+  display: flex;
+  align-items: center;
+  gap: 4px;
+}
+.chat-sub-progress-line .marker { color: var(--accent); }
+.chat-sub-progress-line .marker.done { color: var(--green); }
+.chat-sub-progress-line .marker.err { color: var(--red); }
+.chat-sub-progress-line .dur { opacity: 0.6; }
 
 .chat-compose {
   border-top: 1px solid var(--border-subtle);
@@ -2744,6 +2760,102 @@ function completeToolCard(toolCallId, toolName, result, error) {
   scrollChatToBottom()
 }
 
+function _ensureSubProgressContainer(card) {
+  let container = card.querySelector('.chat-sub-progress')
+  if (container) return container
+  container = document.createElement('div')
+  container.className = 'chat-sub-progress'
+  const bodyEl = card.querySelector('.chat-tool-body')
+  if (bodyEl) {
+    card.insertBefore(container, bodyEl)
+  } else {
+    card.appendChild(container)
+  }
+  return container
+}
+
+function _findSubProgressLine(container, subToolCallId) {
+  if (!subToolCallId) return null
+  return Array.from(container.querySelectorAll('.chat-sub-progress-line'))
+    .find(line => line.dataset.subCallId === subToolCallId) || null
+}
+
+function _formatSubProgressDuration(durationMs) {
+  const ms = Number(durationMs)
+  if (!Number.isFinite(ms) || ms < 0) return ''
+  return `${(ms / 1000).toFixed(1)}s`
+}
+
+function _renderSubProgressLine(line, markerText, markerClass, text, durationText = '') {
+  line.textContent = ''
+
+  const markerEl = document.createElement('span')
+  markerEl.className = markerClass
+  markerEl.textContent = markerText
+  line.appendChild(markerEl)
+
+  const textEl = document.createElement('span')
+  textEl.textContent = text || 'tool'
+  line.appendChild(textEl)
+
+  if (durationText) {
+    const durationEl = document.createElement('span')
+    durationEl.className = 'dur'
+    durationEl.textContent = durationText
+    line.appendChild(durationEl)
+  }
+}
+
+function updateSubAgentProgress(parentToolCallId, subEvent, toolName, subToolCallId, durationMs, error, errorMessage) {
+  const card = S.chat.toolCards.get(parentToolCallId)
+  if (!card) return
+
+  const container = _ensureSubProgressContainer(card)
+  const label = toolName || 'tool'
+
+  if (subEvent === 'tool_start') {
+    if (!subToolCallId) return
+    let line = _findSubProgressLine(container, subToolCallId)
+    if (!line) {
+      line = document.createElement('div')
+      line.className = 'chat-sub-progress-line'
+      line.dataset.subCallId = subToolCallId
+      container.appendChild(line)
+    }
+    _renderSubProgressLine(line, '▸', 'marker', label)
+    scrollChatToBottom()
+    return
+  }
+
+  if (subEvent === 'tool_done') {
+    if (!subToolCallId) return
+    let line = _findSubProgressLine(container, subToolCallId)
+    if (!line) {
+      line = document.createElement('div')
+      line.className = 'chat-sub-progress-line'
+      line.dataset.subCallId = subToolCallId
+      container.appendChild(line)
+    }
+    _renderSubProgressLine(
+      line,
+      error ? '✗' : '✓',
+      error ? 'marker err' : 'marker done',
+      label,
+      _formatSubProgressDuration(durationMs),
+    )
+    scrollChatToBottom()
+    return
+  }
+
+  if (subEvent === 'error') {
+    const line = document.createElement('div')
+    line.className = 'chat-sub-progress-line'
+    _renderSubProgressLine(line, '✗', 'marker err', errorMessage || 'Sub-agent error')
+    container.appendChild(line)
+    scrollChatToBottom()
+  }
+}
+
 function formatToolPayload(value) {
   if (value == null) return '{}'
   let text = typeof value === 'string' ? value : JSON.stringify(value, null, 2)
@@ -3040,6 +3152,19 @@ async function handleChatEvent(event) {
     return false
   }
 
+  if (event.type === 'sub_agent_progress') {
+    updateSubAgentProgress(
+      event.parent_tool_call_id,
+      event.sub_event,
+      event.tool_name,
+      event.sub_tool_call_id,
+      event.duration_ms,
+      event.error,
+      event.error_message,
+    )
+    return false
+  }
+
   if (event.type === 'taskflow_mutation') {
     await handleChatMutation(event)
     return false