Function TestDynamicToolCallMixedWithBuiltIn

coderd/x/chatd/chatd_test.go:4561–4727

(t *testing.T)

}

func TestDynamicToolCallMixedWithBuiltIn(t *testing.T) {
	t.Parallel()

	db, ps := dbtestutil.NewDB(t)
	ctx := testutil.Context(t, testutil.WaitLong)

	// Track streaming calls to the mock LLM.
	var streamedCallCount atomic.Int32
	var streamedCallsMu sync.Mutex
	streamedCalls := make([]chattest.OpenAIRequest, 0, 2)

	openAIURL := chattest.NewOpenAI(t, func(req *chattest.OpenAIRequest) chattest.OpenAIResponse {
		if !req.Stream {
			return chattest.OpenAINonStreamingResponse("Mixed tool test")
		}

		streamedCallsMu.Lock()
		streamedCalls = append(streamedCalls, chattest.OpenAIRequest{
			Messages: append([]chattest.OpenAIMessage(nil), req.Messages...),
			Tools:    append([]chattest.OpenAITool(nil), req.Tools...),
			Stream:   req.Stream,
		})
		streamedCallsMu.Unlock()

		if streamedCallCount.Add(1) == 1 {
			// First call: return TWO tool calls in one
			// response: a built-in tool (read_file) and a
			// dynamic tool (my_dynamic_tool).
			builtinChunk := chattest.OpenAIToolCallChunk(
				"read_file",
				`{"path":"/tmp/test.txt"}`,
			)
			dynamicChunk := chattest.OpenAIToolCallChunk(
				"my_dynamic_tool",
				`{"input":"hello world"}`,
			)
			// Merge both tool calls into one chunk with
			// separate indices so the LLM appears to have
			// requested both tools simultaneously.
			mergedChunk := builtinChunk
			dynCall := dynamicChunk.Choices[0].ToolCalls[0]
			dynCall.Index = 1
			mergedChunk.Choices[0].ToolCalls = append(
				mergedChunk.Choices[0].ToolCalls,
				dynCall,
			)
			return chattest.OpenAIStreamingResponse(mergedChunk)
		}
		// Second call (after tool results): normal text
		// response.
		return chattest.OpenAIStreamingResponse(
			chattest.OpenAITextChunks("All done.")...,
		)
	})

	user, org, model := seedChatDependenciesWithProvider(t, db, "openai-compat", openAIURL)
	server := newActiveTestServer(t, db, ps)