Skip to content

feat: Enable adding new AI model #1668

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
Manar-Elhabbal7 wants to merge 3 commits into
base: main
Choose a base branch
from
Open

feat: Enable adding new AI model #1668

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
0aabfba
enhance editor code to be like the real IDE
Manar-Elhabbal7 Apr 20, 2026
5d82f77
feat:add model done
Manar-Elhabbal7 Apr 20, 2026
62753e7
Revert changes to lib/widgets/editor_code.dart
Manar-Elhabbal7 Apr 20, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions lib/screens/common_widgets/ai/ai.dart
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
export 'ai_model_selector_button.dart';
export 'ai_model_selector_dialog.dart';
export 'ai_model_selector.dart';
export 'dialog_add_ai_model.dart';
261 changes: 145 additions & 116 deletions lib/screens/common_widgets/ai/ai_model_selector_dialog.dart
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
// import 'package:apidash/providers/providers.dart';
import 'ai.dart';
import 'package:apidash/widgets/widgets.dart';
import 'package:apidash/consts.dart';
import 'package:apidash_core/apidash_core.dart';
Expand All @@ -16,7 +16,8 @@ class AIModelSelectorDialog extends ConsumerStatefulWidget {
}

class _AIModelSelectorDialogState extends ConsumerState<AIModelSelectorDialog> {
late final Future<AvailableModels> aM;
AvailableModels? availableModels;
bool isLoading = true;
ModelAPIProvider? selectedProvider;
AIRequestModel? newAIRequestModel;

Expand All @@ -27,154 +28,163 @@ class _AIModelSelectorDialogState extends ConsumerState<AIModelSelectorDialog> {
if (selectedProvider != null && widget.aiRequestModel?.model != null) {
newAIRequestModel = widget.aiRequestModel?.copyWith();
}
aM = ModelManager.fetchAvailableModels();
ModelManager.fetchAvailableModels().then((value) {
if (mounted) {
setState(() {
availableModels = value;
isLoading = false;
});
}
});
}

@override
Widget build(BuildContext context) {
// ref.watch(aiApiCredentialProvider);
if (isLoading) {
return const Center(child: CircularProgressIndicator());
}
if (availableModels == null) {
return const Center(child: Text("Error loading models"));
}

final data = availableModels!;
final mappedData = data.map;
final width = MediaQuery.of(context).size.width * 0.8;
return FutureBuilder(
future: aM,
builder: (context, snapshot) {
if (snapshot.connectionState == ConnectionState.done &&
snapshot.hasData &&
snapshot.data != null) {
final data = snapshot.data!;
final mappedData = data.map;
if (context.isMediumWindow) {
return Container(
padding: kP20,
width: width,

if (context.isMediumWindow) {
return Container(
padding: kP20,
width: width,
child: Column(
crossAxisAlignment: CrossAxisAlignment.start,
children: [
ElevatedButton(
onPressed: () async {
setState(() {
isLoading = true;
});
final models = await ModelManager.fetchAvailableModels();
if (mounted) {
setState(() {
availableModels = models;
isLoading = false;
});
}
},
child: Text(kLabelUpdateModels),
),
kVSpacer10,
Row(
children: [
Text(kLabelSelectModelProvider),
kHSpacer20,
Expanded(
child: ADDropdownButton<ModelAPIProvider>(
onChanged: (x) {
setState(() {
selectedProvider = x;
newAIRequestModel =
mappedData[selectedProvider]?.toAiRequestModel();
});
},
value: selectedProvider,
values: data.modelProviders.map(
(e) => (e.providerId!, e.providerName),
),
),
),
],
),
kVSpacer10,
_buildModelSelector(mappedData[selectedProvider]),
],
),
);
}

return Container(
padding: kP20,
width: width,
child: Row(
crossAxisAlignment: CrossAxisAlignment.start,
children: [
Flexible(
flex: 1,
child: SingleChildScrollView(
child: Column(
crossAxisAlignment: CrossAxisAlignment.start,
children: [
ElevatedButton(
onPressed: null,
// TODO: Add update model logic
//() async {
// await LLMManager.fetchAvailableLLMs();
// setState(() {});
//},
onPressed: () async {
setState(() {
isLoading = true;
});
final models = await ModelManager.fetchAvailableModels();
if (mounted) {
setState(() {
availableModels = models;
isLoading = false;
});
}
},
child: Text(kLabelUpdateModels),
),
kVSpacer10,
Row(
children: [
Text(kLabelSelectModelProvider),
kHSpacer20,
Expanded(
child: ADDropdownButton<ModelAPIProvider>(
onChanged: (x) {
setState(() {
selectedProvider = x;
newAIRequestModel = mappedData[selectedProvider]
?.toAiRequestModel();
});
},
value: selectedProvider,
values: data.modelProviders.map(
(e) => (e.providerId!, e.providerName),
),
),
),
],
SizedBox(height: 20),
...data.modelProviders.map(
(x) => ListTile(
title: Text(x.providerName ?? ""),
trailing: selectedProvider != x.providerId
? null
: const CircleAvatar(
radius: 5,
backgroundColor: Colors.green,
),
onTap: () {
setState(() {
selectedProvider = x.providerId;
newAIRequestModel =
mappedData[selectedProvider]?.toAiRequestModel();
});
},
),
),
kVSpacer10,
_buildModelSelector(mappedData[selectedProvider]),
],
),
);
}

return Container(
padding: kP20,
width: width,
child: Row(
crossAxisAlignment: CrossAxisAlignment.start,
children: [
Flexible(
flex: 1,
child: SingleChildScrollView(
child: Column(
crossAxisAlignment: CrossAxisAlignment.start,
children: [
ElevatedButton(
onPressed: null,
// TODO: Add update model logic
//() async {
// await LLMManager.fetchAvailableLLMs();
// setState(() {});
//},
child: Text(kLabelUpdateModels),
),
SizedBox(height: 20),
...data.modelProviders.map(
(x) => ListTile(
title: Text(x.providerName ?? ""),
trailing: selectedProvider != x.providerId
? null
: CircleAvatar(
radius: 5,
backgroundColor: Colors.green,
),
onTap: () {
setState(() {
selectedProvider = x.providerId;
newAIRequestModel = mappedData[selectedProvider]
?.toAiRequestModel();
});
},
),
),
],
),
),
),
SizedBox(width: 40),
Flexible(
flex: 3,
child: _buildModelSelector(mappedData[selectedProvider]),
),
],
),
);
}
return Center(child: CircularProgressIndicator());
},
),
const SizedBox(width: 40),
Flexible(
flex: 3,
child: _buildModelSelector(mappedData[selectedProvider]),
),
],
),
);
}

_buildModelSelector(AIModelProvider? aiModelProvider) {
Widget _buildModelSelector(AIModelProvider? aiModelProvider) {
if (aiModelProvider == null) {
return Center(child: Text(kLabelSelectAIProvider));
return const Center(child: Text(kLabelSelectAIProvider));
}
// final currentCredential =
// ref.watch(aiApiCredentialProvider)[aiModelProvider.providerId!] ?? "";
return Column(
crossAxisAlignment: CrossAxisAlignment.start,
mainAxisSize: MainAxisSize.max,
children: [
Text(
aiModelProvider.providerName ?? "",
style: TextStyle(fontSize: 28),
style: const TextStyle(fontSize: 28),
),
SizedBox(height: 20),
const SizedBox(height: 20),
if (aiModelProvider.providerId != ModelAPIProvider.ollama) ...[
Text(kLabelApiKeyCredential),
kVSpacer8,
BoundedTextField(
onChanged: (x) {
// ref.read(aiApiCredentialProvider.notifier).state = {
// ...ref.read(aiApiCredentialProvider),
// aiModelProvider.providerId!: x
// };
setState(() {
newAIRequestModel = newAIRequestModel?.copyWith(apiKey: x);
});
},
value: newAIRequestModel?.apiKey ?? "",
// value: currentCredential,
),
kVSpacer10,
],
Expand All @@ -194,8 +204,27 @@ class _AIModelSelectorDialogState extends ConsumerState<AIModelSelectorDialog> {
mainAxisAlignment: MainAxisAlignment.spaceBetween,
children: [
Text(kLabelModels),
// IconButton(
// onPressed: () => addNewModel(context), icon: Icon(Icons.add))
IconButton(
onPressed: () async {
final newModel = await addNewModel(context);
if (!mounted) return;
if (newModel != null && availableModels != null) {
setState(() {
final providerId = selectedProvider;
final providers = availableModels!.modelProviders.map((p) {
if (p.providerId == providerId) {
return p.copyWith(
models: [...(p.models ?? []), newModel]);
}
return p;
}).toList();
availableModels =
availableModels!.copyWith(modelProviders: providers);
});
}
},
icon: const Icon(Icons.add),
)
],
),
kVSpacer8,
Expand All @@ -218,7 +247,7 @@ class _AIModelSelectorDialogState extends ConsumerState<AIModelSelectorDialog> {
mainAxisSize: MainAxisSize.min,
children: [
if (newAIRequestModel?.model == x.id)
CircleAvatar(
const CircleAvatar(
radius: 5,
backgroundColor: Colors.green,
),
Expand Down
10 changes: 5 additions & 5 deletions lib/screens/common_widgets/ai/dialog_add_ai_model.dart
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,11 +1,12 @@
import 'package:apidash_core/apidash_core.dart' as ref;
import 'package:apidash_design_system/apidash_design_system.dart';
import 'package:apidash/consts.dart';
import 'package:flutter/material.dart';

Future<void> addNewModel(BuildContext context) async {
Future<ref.Model?> addNewModel(BuildContext context) async {
TextEditingController iC = TextEditingController();
TextEditingController nC = TextEditingController();
final z = await showDialog(
final z = await showDialog<List<String>>(
context: context,
builder: (context) {
return AlertDialog(
Expand Down Expand Up @@ -36,7 +37,6 @@ Future<void> addNewModel(BuildContext context) async {
);
iC.dispose();
nC.dispose();
if (z == null) return;
// TODO: Add logic to add a new model
// setState(() {});
if (z == null || z.length < 2) return null;
return ref.Model(id: z[0], name: z[1]);
}