Function CreateOpenAICompatChatModelConfig

coderd/coderdtest/chat.go:53–84

CreateOpenAICompatChatModelConfig creates the default provider and model config used by chat runtime tests. Tests can pass a baseURL to route chat work to a specific local provider. If baseURL is empty, this helper starts a fake OpenAI-compatible provider.

(
	t testing.TB,
	client *codersdk.ExperimentalClient,
	baseURL string,
)

// to a specific local provider. If baseURL is empty, this helper starts a fake
// OpenAI-compatible provider.
func CreateOpenAICompatChatModelConfig(
	t testing.TB,
	client *codersdk.ExperimentalClient,
	baseURL string,
) codersdk.ChatModelConfig {
	t.Helper()

	if baseURL == "" {
		baseURL = chattest.OpenAI(t)
	}

	ctx := testutil.Context(t, testutil.WaitLong)
	provider, err := client.CreateAIProvider(ctx, codersdk.CreateAIProviderRequest{
		Type:    codersdk.AIProviderType(TestChatProviderOpenAICompat),
		Name:    "test-" + uuid.NewString(),
		BaseURL: baseURL,
		Enabled: true,
		APIKeys: []string{TestChatProviderAPIKey},
	})
	require.NoError(t, err)
	contextLimit := int64(4096)
	isDefault := true
	modelConfig, err := client.CreateChatModelConfig(ctx, codersdk.CreateChatModelConfigRequest{
		Provider:     TestChatProviderOpenAICompat,
		AIProviderID: &provider.ID,
		Model:        TestChatModelOpenAICompat,
		ContextLimit: &contextLimit,
		IsDefault:    &isDefault,
	})
	require.NoError(t, err)
	return modelConfig
}

// WaitForChatSettled waits for a chat to leave active processing and drains
// tracked chat daemon work before returning the final row.