Function TestChatToolCallMultipleByIndex

internal/llm/llm_test.go:201–232

TestChatToolCallMultipleByIndex: two tool calls interleaved across chunks. Each fragment must route to its slot by `index`, not by slice position.

(t *testing.T)

// TestChatToolCallMultipleByIndex: two tool calls interleaved across chunks.
// Each fragment must route to its slot by `index`, not by slice position.
func TestChatToolCallMultipleByIndex(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
		sseOK(w, []string{
			`{"choices":[{"delta":{"tool_calls":[{"index":0,"id":"c1","function":{"name":"bash"}}]}}]}`,
			`{"choices":[{"delta":{"tool_calls":[{"index":1,"id":"c2","function":{"name":"python"}}]}}]}`,
			`{"choices":[{"delta":{"tool_calls":[{"index":0,"function":{"arguments":"{\"cmd\":\"ls\"}"}}]}}]}`,
			`{"choices":[{"delta":{"tool_calls":[{"index":1,"function":{"arguments":"{\"cmd\":\"print()\"}"}}]}}]}`,
			`{"choices":[{"delta":{},"finish_reason":"tool_calls"}]}`,
		})
	}))
	defer srv.Close()
	c := New(srv.URL, "m", "")
	var calls []chmctx.ToolCall
	for _, e := range collect(c.Chat(context.Background(), nil, nil)) {
		if e.Kind == EventToolCall {
			calls = append(calls, *e.ToolCall)
		}
	}
	if len(calls) != 2 {
		t.Fatalf("want 2 tool-call events, got %d: %+v", len(calls), calls)
	}
	byName := map[string]map[string]any{}
	for _, c := range calls {
		byName[c.Name] = c.Arguments
	}
	if cmd, _ := byName["bash"]["cmd"].(string); cmd != "ls" {
		t.Fatalf("bash args wrong: %+v", byName["bash"])
	}
	if cmd, _ := byName["python"]["cmd"].(string); cmd != "print()" {
		t.Fatalf("python args wrong: %+v", byName["python"])
	}
}

// TestChatDispatchesToolCallsOnFinishStop: Ollama's /v1 shim sometimes emits
// finish_reason="stop" even after streaming tool_calls. The client must still