flutter/apidash
1
0
Fork 0
mirror of https://github.com/foss42/apidash.git synced 2025-12-09 23:02:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: dahbot chat page

Browse Source
This commit is contained in:
Udhay-Adithya
2025-09-01 15:59:07 +05:30
parent 8b74dbbfcd
commit 0def6c1713
7 changed files with 711 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

226
packages/dashbot/lib/features/chat/viewmodel/chat_viewmodel.dart Normal file
View File

@@ -0,0 +1,226 @@
import 'dart:async';
import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:nanoid/nanoid.dart';
import '../../../core/constants/dashbot_prompts.dart' as dash;
import '../../../core/model/dashbot_request_context.dart';
import '../../../core/providers/dashbot_request_provider.dart';
import '../view/widgets/chat_bubble.dart';
import '../models/chat_models.dart';
import '../repository/chat_remote_repository.dart';
class ChatViewmodel extends StateNotifier<ChatState> {
ChatViewmodel(this._ref) : super(const ChatState());
final Ref _ref;
StreamSubscription<String>? _sub;
ChatRemoteRepository get _repo => _ref.read(chatRepositoryProvider);
DashbotRequestContext? get _ctx => _ref.read(dashbotRequestContextProvider);
List<ChatMessage> get currentMessages {
final id = _ctx?.requestId;
if (id == null) return const [];
return state.chatSessions[id] ?? const [];
}
Future<void> sendMessage({
required String text,
ChatMessageType type = ChatMessageType.general,
bool countAsUser = true,
}) async {
final ctx = _ctx;
final ai = ctx?.aiRequestModel;
if (text.trim().isEmpty && countAsUser) return;
if (ai == null) {
_appendSystem(
'AI model is not configured. Please set one in AI Request tab.',
type,
);
return;
}
final requestId = ctx?.requestId ?? 'global';
if (countAsUser) {
_addMessage(
requestId,
ChatMessage(
id: nanoid(),
content: text,
role: MessageRole.user,
timestamp: DateTime.now(),
messageType: type,
),
);
}
final systemPrompt = _composeSystemPrompt(ctx, type);
final enriched = ai.copyWith(
systemPrompt: systemPrompt,
userPrompt: text,
stream: true,
);
// start stream
_sub?.cancel();
state = state.copyWith(isGenerating: true, currentStreamingResponse: '');
_sub = _repo
.streamChat(request: enriched)
.listen(
(chunk) {
state = state.copyWith(
currentStreamingResponse:
state.currentStreamingResponse + (chunk),
);
},
onError: (e) {
state = state.copyWith(isGenerating: false);
_appendSystem('Error: $e', type);
},
onDone: () {
if (state.currentStreamingResponse.isNotEmpty) {
_addMessage(
requestId,
ChatMessage(
id: nanoid(),
content: state.currentStreamingResponse,
role: MessageRole.system,
timestamp: DateTime.now(),
messageType: type,
),
);
}
state = state.copyWith(
isGenerating: false,
currentStreamingResponse: '',
);
},
cancelOnError: true,
);
}
void cancel() {
_sub?.cancel();
state = state.copyWith(isGenerating: false);
}
// Helpers
void _addMessage(String requestId, ChatMessage m) {
final msgs = state.chatSessions[requestId] ?? const [];
state = state.copyWith(
chatSessions: {
...state.chatSessions,
requestId: [...msgs, m],
},
);
}
void _appendSystem(String text, ChatMessageType type) {
final id = _ctx?.requestId ?? 'global';
_addMessage(
id,
ChatMessage(
id: nanoid(),
content: text,
role: MessageRole.system,
timestamp: DateTime.now(),
messageType: type,
),
);
}
String _composeSystemPrompt(
DashbotRequestContext? ctx,
ChatMessageType type,
) {
final history = _buildHistoryBlock();
final contextBlock = _buildContextBlock(ctx);
final task = _buildTaskPrompt(ctx, type);
return [
if (task != null) task,
if (contextBlock != null) contextBlock,
if (history.isNotEmpty) history,
].join('\n\n');
}
String _buildHistoryBlock({int maxTurns = 8}) {
final id = _ctx?.requestId ?? 'global';
final messages = state.chatSessions[id] ?? const [];
if (messages.isEmpty) return '';
final start = messages.length > maxTurns ? messages.length - maxTurns : 0;
final recent = messages.sublist(start);
final buf = StringBuffer('''<conversation_context>
Only use the following short chat history to maintain continuity. Do not repeat it back.
''');
for (final m in recent) {
final role = m.role == MessageRole.user ? 'user' : 'assistant';
buf.writeln('- $role: ${m.content}');
}
buf.writeln('</conversation_context>');
return buf.toString();
}
String? _buildContextBlock(DashbotRequestContext? ctx) {
final http = ctx?.httpRequestModel;
if (ctx == null || http == null) return null;
final headers = http.headersMap.entries
.map((e) => '"${e.key}": "${e.value}"')
.join(', ');
return '''<request_context>
Request Name: ${ctx.requestName ?? ''}
URL: ${http.url}
Method: ${http.method.name.toUpperCase()}
Status: ${ctx.responseStatus ?? ''}
Content-Type: ${http.bodyContentType.name}
Headers: { $headers }
Body: ${http.body ?? ''}
Response: ${ctx.httpResponseModel?.body ?? ''}
</request_context>''';
}
String? _buildTaskPrompt(DashbotRequestContext? ctx, ChatMessageType type) {
if (ctx == null) return null;
final http = ctx.httpRequestModel;
final resp = ctx.httpResponseModel;
final prompts = dash.DashbotPrompts();
switch (type) {
case ChatMessageType.explainResponse:
return prompts.explainApiResponsePrompt(
url: http?.url,
method: http?.method.name.toUpperCase(),
responseStatus: ctx.responseStatus,
bodyContentType: http?.bodyContentType.name,
message: resp?.body,
headersMap: http?.headersMap,
body: http?.body,
);
case ChatMessageType.debugError:
return prompts.debugApiErrorPrompt(
url: http?.url,
method: http?.method.name.toUpperCase(),
responseStatus: ctx.responseStatus,
bodyContentType: http?.bodyContentType.name,
message: resp?.body,
headersMap: http?.headersMap,
body: http?.body,
);
case ChatMessageType.generateTest:
return prompts.generateTestCasesPrompt(
url: http?.url,
method: http?.method.name.toUpperCase(),
headersMap: http?.headersMap,
body: http?.body,
);
case ChatMessageType.general:
return null;
}
}
}
final chatViewmodelProvider = StateNotifierProvider<ChatViewmodel, ChatState>((
ref,
) {
return ChatViewmodel(ref);
});