Function TestChatModelOverrides

coderd/exp_chats_test.go:11255–11509

nolint:tparallel,paralleltest // Setting subtests share per-setting coderdtest instances.

(t *testing.T)

//nolint:tparallel,paralleltest // Setting subtests share per-setting coderdtest instances.
func TestChatModelOverrides(t *testing.T) {
	t.Parallel()

	type overrideResponse struct {
		context       codersdk.ChatModelOverrideContext
		modelConfigID string
		isMalformed   bool
	}

	type settingTest struct {
		name     string
		context  codersdk.ChatModelOverrideContext
		dbGet    func(context.Context, database.Store) (string, error)
		dbUpsert func(context.Context, database.Store, string) error
	}

	settingPath := func(overrideContext codersdk.ChatModelOverrideContext) string {
		return "/api/experimental/chats/config/model-override/" + string(overrideContext)
	}

	getOverride := func(
		ctx context.Context,
		client *codersdk.ExperimentalClient,
		overrideContext codersdk.ChatModelOverrideContext,
	) (overrideResponse, error) {
		resp, err := client.GetChatModelOverride(ctx, overrideContext)
		if err != nil {
			return overrideResponse{}, err
		}
		return overrideResponse{
			context:       resp.Context,
			modelConfigID: resp.ModelConfigID,
			isMalformed:   resp.IsMalformed,
		}, nil
	}

	putOverride := func(
		ctx context.Context,
		client *codersdk.ExperimentalClient,
		overrideContext codersdk.ChatModelOverrideContext,
		modelConfigID string,
	) error {
		return client.UpdateChatModelOverride(
			ctx,
			overrideContext,
			codersdk.UpdateChatModelOverrideRequest{ModelConfigID: modelConfigID},
		)
	}

	settings := []settingTest{
		{
			name:    "General",
			context: codersdk.ChatModelOverrideContextGeneral,
			dbGet: func(ctx context.Context, db database.Store) (string, error) {
				return db.GetChatGeneralModelOverride(dbauthz.AsSystemRestricted(ctx))
			},
			dbUpsert: func(ctx context.Context, db database.Store, value string) error {
				return db.UpsertChatGeneralModelOverride(dbauthz.AsSystemRestricted(ctx), value)