apidash/lib/screens/common_widgets/ai/ai_model_selector.dart

import 'package:flutter/material.dart';
import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:apidash_core/apidash_core.dart';
import 'package:apidash/providers/providers.dart';
import 'ai_model_selector_button.dart';

class AIModelSelector extends ConsumerWidget {
  final AIRequestModel? readOnlyModel;

  const AIModelSelector({
    super.key,
    this.readOnlyModel,
  });

  @override
  Widget build(BuildContext context, WidgetRef ref) {
    AIRequestModel? aiRequestModel;
    if (readOnlyModel != null) {
      ref.watch(selectedIdStateProvider);
      aiRequestModel = ref.watch(selectedRequestModelProvider
          .select((value) => value?.aiRequestModel));
    } else {
      aiRequestModel = readOnlyModel;
    }

    if (aiRequestModel == null) {
      return Container();
    }

    return AIModelSelectorButton(
      readonly: (readOnlyModel != null),
      key: ValueKey(ref.watch(selectedIdStateProvider)),
      aiRequestModel: aiRequestModel,
      onModelUpdated: (newAIRequestModel) {
        ref
            .read(collectionStateNotifierProvider.notifier)
            .update(aiRequestModel: newAIRequestModel.copyWith());
      },
    );
  }
}