Function TestChatToolArgsStreamLive

internal/llm/llm_test.go:162–197

TestChatToolArgsStreamLive: each tool-call arguments fragment is forwarded as an EventToolArgs as it arrives, so the UI can tick its live token estimate while a file streams into write_file, not just once at stream end. Fragments concatenate to the full arguments and all precede the resolved EventTo

(t *testing.T)

		switch e.Kind {
		case EventToolCall:
			calls = append(calls, *e.ToolCall)
		case EventDone:
			final = e.Final
		}
	}
	if len(calls) != 1 || calls[0].Name != "bash" || calls[0].ID != "call_1" {
		t.Fatalf("want one bash call with call_id, got %+v", calls)
	}
	if cmd, _ := calls[0].Arguments["cmd"].(string); cmd != "ls" {
		t.Fatalf("tool args wrong (done must replace, not append to, the deltas): %+v", calls[0].Arguments)
	}
	if final == nil || len(final.ToolCalls) != 1 || final.ToolCalls[0].Name != "bash" {
		t.Fatalf("Final.ToolCalls should carry the bash call: %+v", final)
	}
}

// TestChatToolCallFromDoneOnly: a server that streams no argument deltas and
// no arguments.done (only the finished item) must still resolve the call.
func TestChatToolCallFromDoneOnly(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
		sseOK(w, []string{
			callDone(0, "c1", "bash", `{"cmd":"ls"}`),
			completed(5, 0),
		})
	}))
	defer srv.Close()
	var got *chmctx.ToolCall
	for _, e := range collect(New(srv.URL, "m", "").Chat(context.Background(), nil, nil)) {
		if e.Kind == EventToolCall {
			got = e.ToolCall
		}
	}
	if got == nil || got.Name != "bash" || got.ID != "c1" {
		t.Fatalf("tool call missing: %+v", got)
	}
	if cmd, _ := got.Arguments["cmd"].(string); cmd != "ls" {
		t.Fatalf("args wrong: %+v", got.Arguments)
	}
}