Function TestAdvisorGating_RootChat

coderd/x/chatd/chatd_test.go:9722–9850

(t *testing.T)

}

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

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

	var streamedCallCount atomic.Int32
	var streamedCallsMu sync.Mutex
	var firstCallTools []string
	var firstCallMessages []chattest.OpenAIMessage
	var secondCallMessages []chattest.OpenAIMessage

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

		switch streamedCallCount.Add(1) {
		case 1:
			names := make([]string, 0, len(req.Tools))
			for _, tool := range req.Tools {
				names = append(names, tool.Function.Name)
			}
			streamedCallsMu.Lock()
			firstCallTools = names
			firstCallMessages = append([]chattest.OpenAIMessage(nil), req.Messages...)
			streamedCallsMu.Unlock()

			advisorChunk := chattest.OpenAIToolCallChunk(
				"advisor",
				`{"question":"help me plan"}`,
			)
			readChunk := chattest.OpenAIToolCallChunk(
				"read_file",
				`{"path":"/tmp/test.txt"}`,
			)
			mergedChunk := advisorChunk
			readCall := readChunk.Choices[0].ToolCalls[0]
			readCall.Index = 1
			mergedChunk.Choices[0].ToolCalls = append(
				mergedChunk.Choices[0].ToolCalls,
				readCall,
			)
			return chattest.OpenAIStreamingResponse(mergedChunk)
		case 2:
			streamedCallsMu.Lock()
			secondCallMessages = append([]chattest.OpenAIMessage(nil), req.Messages...)
			streamedCallsMu.Unlock()
		}

		return chattest.OpenAIStreamingResponse(
			chattest.OpenAITextChunks("done")...,
		)
	})

	user, org, model := seedChatDependenciesWithProvider(t, db, "openai-compat", openAIURL)
	seedAdvisorConfig(ctx, t, db, codersdk.AdvisorConfig{
		Enabled:         true,