💄 Optimize thinking
This commit is contained in:
535
lib/widgets/thought/shared_widgets.dart
Normal file
535
lib/widgets/thought/shared_widgets.dart
Normal file
@@ -0,0 +1,535 @@
|
||||
import "dart:convert";
|
||||
import "package:easy_localization/easy_localization.dart";
|
||||
import "package:flutter/material.dart";
|
||||
import "package:flutter/services.dart";
|
||||
import "package:gap/gap.dart";
|
||||
import "package:google_fonts/google_fonts.dart";
|
||||
import "package:island/models/thought.dart";
|
||||
import "package:island/services/time.dart";
|
||||
import "package:island/widgets/alert.dart";
|
||||
import "package:island/widgets/content/markdown.dart";
|
||||
import "package:island/widgets/post/compose_dialog.dart";
|
||||
import "package:island/screens/posts/compose.dart";
|
||||
import "package:material_symbols_icons/material_symbols_icons.dart";
|
||||
import "package:styled_widget/styled_widget.dart";
|
||||
import "package:markdown/markdown.dart" as markdown;
|
||||
import "package:markdown_widget/markdown_widget.dart";
|
||||
|
||||
// Common functions
|
||||
List<Map<String, String>> extractProposals(String content) {
|
||||
final proposalRegex = RegExp(
|
||||
r'<proposal\s+type="([^"]+)">(.*?)<\/proposal>',
|
||||
dotAll: true,
|
||||
);
|
||||
final matches = proposalRegex.allMatches(content);
|
||||
return matches.map((match) {
|
||||
return {'type': match.group(1)!, 'content': match.group(2)!};
|
||||
}).toList();
|
||||
}
|
||||
|
||||
void handleProposalAction(BuildContext context, Map<String, String> proposal) {
|
||||
switch (proposal['type']) {
|
||||
case 'post_create':
|
||||
// Show post creation dialog with the proposal content
|
||||
PostComposeDialog.show(
|
||||
context,
|
||||
initialState: PostComposeInitialState(
|
||||
content: (proposal['content'] ?? '').trim(),
|
||||
),
|
||||
);
|
||||
break;
|
||||
default:
|
||||
// Show a snackbar for unsupported proposal types
|
||||
showSnackBar('Unsupported proposal type: ${proposal['type']}');
|
||||
}
|
||||
}
|
||||
|
||||
// Common widgets
|
||||
Widget buildChunkTiles(List<SnThinkingChunk> chunks, BuildContext context) {
|
||||
return Column(
|
||||
children: [
|
||||
...chunks
|
||||
.where((chunk) => chunk.type == ThinkingChunkType.reasoning)
|
||||
.map(
|
||||
(chunk) => Card(
|
||||
margin: const EdgeInsets.only(bottom: 8),
|
||||
child: Theme(
|
||||
data: Theme.of(
|
||||
context,
|
||||
).copyWith(dividerColor: Colors.transparent),
|
||||
child: ExpansionTile(
|
||||
shape: RoundedRectangleBorder(
|
||||
borderRadius: BorderRadius.circular(8),
|
||||
),
|
||||
title: Text(
|
||||
'Reasoning',
|
||||
style: Theme.of(context).textTheme.titleSmall,
|
||||
),
|
||||
children: [
|
||||
Padding(
|
||||
padding: const EdgeInsets.all(8.0),
|
||||
child: Text(
|
||||
chunk.data?['content'] ?? '',
|
||||
style: Theme.of(context).textTheme.bodyMedium,
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
),
|
||||
),
|
||||
),
|
||||
...chunks
|
||||
.where((chunk) => chunk.type == ThinkingChunkType.functionCall)
|
||||
.map(
|
||||
(chunk) => Card(
|
||||
margin: const EdgeInsets.only(bottom: 8),
|
||||
child: Theme(
|
||||
data: Theme.of(
|
||||
context,
|
||||
).copyWith(dividerColor: Colors.transparent),
|
||||
child: ExpansionTile(
|
||||
shape: RoundedRectangleBorder(
|
||||
borderRadius: BorderRadius.circular(8),
|
||||
),
|
||||
visualDensity: VisualDensity.compact,
|
||||
title: Text(
|
||||
'Function Call',
|
||||
style: Theme.of(context).textTheme.titleSmall,
|
||||
),
|
||||
children: [
|
||||
Padding(
|
||||
padding: const EdgeInsets.all(8.0),
|
||||
child: SelectableText(
|
||||
JsonEncoder.withIndent(' ').convert(chunk.data),
|
||||
style: GoogleFonts.robotoMono(),
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
),
|
||||
),
|
||||
),
|
||||
],
|
||||
);
|
||||
}
|
||||
|
||||
Widget thoughtItem(SnThinkingThought thought, int index, BuildContext context) {
|
||||
final key = Key('thought-${thought.id}');
|
||||
|
||||
// Extract proposals from thought content
|
||||
final proposals =
|
||||
thought.content != null ? extractProposals(thought.content!) : [];
|
||||
|
||||
final thoughtWidget = Container(
|
||||
margin: const EdgeInsets.symmetric(vertical: 8, horizontal: 16),
|
||||
padding: const EdgeInsets.all(12),
|
||||
decoration: BoxDecoration(
|
||||
color:
|
||||
thought.role == ThinkingThoughtRole.assistant
|
||||
? Theme.of(context).colorScheme.surfaceContainerHighest
|
||||
: Theme.of(context).colorScheme.primaryContainer,
|
||||
borderRadius: BorderRadius.circular(12),
|
||||
),
|
||||
child: Column(
|
||||
crossAxisAlignment: CrossAxisAlignment.start,
|
||||
children: [
|
||||
Row(
|
||||
children: [
|
||||
Icon(
|
||||
thought.role == ThinkingThoughtRole.assistant
|
||||
? Symbols.smart_toy
|
||||
: Symbols.person,
|
||||
size: 20,
|
||||
),
|
||||
const Gap(8),
|
||||
Expanded(
|
||||
child: Row(
|
||||
crossAxisAlignment: CrossAxisAlignment.baseline,
|
||||
textBaseline: TextBaseline.ideographic,
|
||||
spacing: 8,
|
||||
children: [
|
||||
Text(
|
||||
thought.role == ThinkingThoughtRole.assistant
|
||||
? 'thoughtAiName'.tr()
|
||||
: 'thoughtUserName'.tr(),
|
||||
style: Theme.of(context).textTheme.titleSmall,
|
||||
),
|
||||
Tooltip(
|
||||
message: thought.createdAt.formatSystem(),
|
||||
child: Text(
|
||||
thought.createdAt.formatRelative(context),
|
||||
style: Theme.of(context).textTheme.bodySmall?.copyWith(
|
||||
color: Theme.of(context).colorScheme.onSurfaceVariant,
|
||||
),
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
),
|
||||
if (thought.role == ThinkingThoughtRole.assistant)
|
||||
SizedBox(
|
||||
height: 20,
|
||||
width: 20,
|
||||
child: IconButton(
|
||||
visualDensity: VisualDensity(horizontal: -4, vertical: -4),
|
||||
padding: EdgeInsets.zero,
|
||||
iconSize: 16,
|
||||
icon: Icon(Symbols.content_copy),
|
||||
onPressed: () {
|
||||
Clipboard.setData(
|
||||
ClipboardData(text: thought.content ?? ''),
|
||||
);
|
||||
showSnackBar('copiedToClipboard'.tr());
|
||||
},
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
const Gap(8),
|
||||
if (thought.chunks.isNotEmpty) ...[
|
||||
buildChunkTiles(thought.chunks, context),
|
||||
const Gap(8),
|
||||
],
|
||||
if (thought.content != null)
|
||||
MarkdownTextContent(
|
||||
isSelectable: true,
|
||||
content: thought.content!,
|
||||
extraBlockSyntaxList: [ProposalBlockSyntax()],
|
||||
textStyle: Theme.of(context).textTheme.bodyMedium,
|
||||
extraGenerators: [
|
||||
ProposalGenerator(
|
||||
backgroundColor:
|
||||
Theme.of(context).colorScheme.secondaryContainer,
|
||||
foregroundColor:
|
||||
Theme.of(context).colorScheme.onSecondaryContainer,
|
||||
borderColor: Theme.of(context).colorScheme.outline,
|
||||
),
|
||||
],
|
||||
),
|
||||
if (thought.role == ThinkingThoughtRole.assistant &&
|
||||
(thought.tokenCount != null || thought.modelName != null)) ...[
|
||||
const Gap(8),
|
||||
Row(
|
||||
children: [
|
||||
if (thought.modelName != null) ...[
|
||||
const Icon(Symbols.neurology, size: 16),
|
||||
const Gap(4),
|
||||
Text(
|
||||
'${thought.modelName}',
|
||||
style: Theme.of(context).textTheme.bodySmall?.copyWith(
|
||||
color: Theme.of(context).colorScheme.onSurfaceVariant,
|
||||
),
|
||||
),
|
||||
const Gap(16),
|
||||
],
|
||||
if (thought.tokenCount != null)
|
||||
...([
|
||||
const Icon(Symbols.token, size: 16),
|
||||
const Gap(4),
|
||||
Text(
|
||||
'${thought.tokenCount} tokens',
|
||||
style: Theme.of(context).textTheme.bodySmall?.copyWith(
|
||||
color: Theme.of(context).colorScheme.onSurfaceVariant,
|
||||
),
|
||||
),
|
||||
]),
|
||||
],
|
||||
),
|
||||
],
|
||||
if (proposals.isNotEmpty &&
|
||||
thought.role == ThinkingThoughtRole.assistant) ...[
|
||||
const Gap(12),
|
||||
Wrap(
|
||||
spacing: 8,
|
||||
runSpacing: 8,
|
||||
children:
|
||||
proposals.map((proposal) {
|
||||
return ElevatedButton.icon(
|
||||
onPressed: () => handleProposalAction(context, proposal),
|
||||
icon: Icon(switch (proposal['type']) {
|
||||
'post_create' => Symbols.add,
|
||||
_ => Symbols.lightbulb,
|
||||
}, size: 16),
|
||||
label: Text(switch (proposal['type']) {
|
||||
'post_create' => 'Create Post',
|
||||
_ => proposal['type'] ?? 'Action',
|
||||
}),
|
||||
style: ElevatedButton.styleFrom(
|
||||
backgroundColor:
|
||||
Theme.of(context).colorScheme.primaryContainer,
|
||||
foregroundColor:
|
||||
Theme.of(context).colorScheme.onPrimaryContainer,
|
||||
padding: const EdgeInsets.symmetric(
|
||||
horizontal: 12,
|
||||
vertical: 8,
|
||||
),
|
||||
),
|
||||
);
|
||||
}).toList(),
|
||||
),
|
||||
],
|
||||
],
|
||||
),
|
||||
);
|
||||
|
||||
return TweenAnimationBuilder<double>(
|
||||
key: key,
|
||||
tween: Tween<double>(begin: 0.0, end: 1.0),
|
||||
duration: Duration(milliseconds: 400 + (index % 5) * 50), // Staggered delay
|
||||
curve: Curves.easeOutCubic,
|
||||
builder: (context, animationValue, child) {
|
||||
return Transform.translate(
|
||||
offset: Offset(0, 20 * (1 - animationValue)), // Slide up from bottom
|
||||
child: Opacity(opacity: animationValue, child: child),
|
||||
);
|
||||
},
|
||||
child: thoughtWidget,
|
||||
);
|
||||
}
|
||||
|
||||
Widget streamingThoughtItem(
|
||||
String streamingText,
|
||||
List<String> reasoningChunks,
|
||||
List<String> functionCalls,
|
||||
BuildContext context,
|
||||
) {
|
||||
return Container(
|
||||
margin: const EdgeInsets.symmetric(vertical: 8, horizontal: 16),
|
||||
padding: const EdgeInsets.all(12),
|
||||
decoration: BoxDecoration(
|
||||
color: Theme.of(context).colorScheme.surfaceContainerHighest,
|
||||
borderRadius: BorderRadius.circular(12),
|
||||
),
|
||||
child: Column(
|
||||
crossAxisAlignment: CrossAxisAlignment.start,
|
||||
children: [
|
||||
Row(
|
||||
children: [
|
||||
Icon(Symbols.smart_toy, size: 20),
|
||||
const Gap(8),
|
||||
Text(
|
||||
'thoughtAiName'.tr(),
|
||||
style: Theme.of(context).textTheme.titleSmall,
|
||||
),
|
||||
const Spacer(),
|
||||
SizedBox(
|
||||
width: 16,
|
||||
height: 16,
|
||||
child: CircularProgressIndicator(strokeWidth: 2),
|
||||
),
|
||||
],
|
||||
),
|
||||
const Gap(8),
|
||||
MarkdownTextContent(
|
||||
content: streamingText,
|
||||
textStyle: Theme.of(context).textTheme.bodyMedium,
|
||||
extraBlockSyntaxList: [ProposalBlockSyntax()],
|
||||
extraGenerators: [
|
||||
ProposalGenerator(
|
||||
backgroundColor: Theme.of(context).colorScheme.secondaryContainer,
|
||||
foregroundColor:
|
||||
Theme.of(context).colorScheme.onSecondaryContainer,
|
||||
borderColor: Theme.of(context).colorScheme.outline,
|
||||
),
|
||||
],
|
||||
),
|
||||
if (reasoningChunks.isNotEmpty || functionCalls.isNotEmpty) ...[
|
||||
const Gap(8),
|
||||
Column(
|
||||
children: [
|
||||
...reasoningChunks.map(
|
||||
(chunk) => Card(
|
||||
margin: const EdgeInsets.only(bottom: 8),
|
||||
child: Theme(
|
||||
data: Theme.of(
|
||||
context,
|
||||
).copyWith(dividerColor: Colors.transparent),
|
||||
child: ExpansionTile(
|
||||
shape: RoundedRectangleBorder(
|
||||
borderRadius: BorderRadius.circular(8),
|
||||
),
|
||||
title: Text(
|
||||
'Reasoning',
|
||||
style: Theme.of(context).textTheme.titleSmall,
|
||||
),
|
||||
children: [
|
||||
Padding(
|
||||
padding: const EdgeInsets.all(8.0),
|
||||
child: Text(
|
||||
chunk,
|
||||
style: Theme.of(context).textTheme.bodyMedium,
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
),
|
||||
),
|
||||
),
|
||||
...functionCalls.map(
|
||||
(call) => Card(
|
||||
margin: const EdgeInsets.only(bottom: 8),
|
||||
child: Theme(
|
||||
data: Theme.of(
|
||||
context,
|
||||
).copyWith(dividerColor: Colors.transparent),
|
||||
child: ExpansionTile(
|
||||
shape: RoundedRectangleBorder(
|
||||
borderRadius: BorderRadius.circular(8),
|
||||
),
|
||||
visualDensity: VisualDensity.compact,
|
||||
title: Text(
|
||||
'Function Call',
|
||||
style: Theme.of(context).textTheme.titleSmall,
|
||||
),
|
||||
children: [
|
||||
Padding(
|
||||
padding: const EdgeInsets.all(8.0),
|
||||
child: SelectableText(
|
||||
call,
|
||||
style: GoogleFonts.robotoMono(),
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
),
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
],
|
||||
],
|
||||
),
|
||||
);
|
||||
}
|
||||
|
||||
class ProposalBlockSyntax extends markdown.BlockSyntax {
|
||||
@override
|
||||
RegExp get pattern => RegExp(r'^<proposal', caseSensitive: false);
|
||||
|
||||
@override
|
||||
bool canParse(markdown.BlockParser parser) {
|
||||
return pattern.hasMatch(parser.current.content);
|
||||
}
|
||||
|
||||
@override
|
||||
bool canEndBlock(markdown.BlockParser parser) {
|
||||
return parser.current.content.contains('</proposal>');
|
||||
}
|
||||
|
||||
@override
|
||||
markdown.Node parse(markdown.BlockParser parser) {
|
||||
final childLines = <String>[];
|
||||
|
||||
// Extract type from opening tag
|
||||
final openingLine = parser.current.content;
|
||||
final attrsMatch = RegExp(
|
||||
r'<proposal(\s[^>]*)?>',
|
||||
caseSensitive: false,
|
||||
).firstMatch(openingLine);
|
||||
final attrs = attrsMatch?.group(1) ?? '';
|
||||
final typeMatch = RegExp(r'type="([^"]*)"').firstMatch(attrs);
|
||||
final type = typeMatch?.group(1) ?? '';
|
||||
|
||||
// Collect all lines until closing tag
|
||||
while (!parser.isDone) {
|
||||
childLines.add(parser.current.content);
|
||||
if (canEndBlock(parser)) {
|
||||
parser.advance();
|
||||
break;
|
||||
}
|
||||
parser.advance();
|
||||
}
|
||||
|
||||
// Extract content between tags
|
||||
final fullContent = childLines.join('\n');
|
||||
final contentMatch = RegExp(
|
||||
r'<proposal[^>]*>(.*?)</proposal>',
|
||||
dotAll: true,
|
||||
caseSensitive: false,
|
||||
).firstMatch(fullContent);
|
||||
final content = contentMatch?.group(1)?.trim() ?? '';
|
||||
|
||||
final element = markdown.Element('proposal', [markdown.Text(content)])
|
||||
..attributes['type'] = type;
|
||||
|
||||
return element;
|
||||
}
|
||||
}
|
||||
|
||||
class ProposalGenerator extends SpanNodeGeneratorWithTag {
|
||||
ProposalGenerator({
|
||||
required Color backgroundColor,
|
||||
required Color foregroundColor,
|
||||
required Color borderColor,
|
||||
}) : super(
|
||||
tag: 'proposal',
|
||||
generator: (
|
||||
markdown.Element element,
|
||||
MarkdownConfig config,
|
||||
WidgetVisitor visitor,
|
||||
) {
|
||||
return ProposalSpanNode(
|
||||
text: element.textContent,
|
||||
type: element.attributes['type'] ?? '',
|
||||
backgroundColor: backgroundColor,
|
||||
foregroundColor: foregroundColor,
|
||||
borderColor: borderColor,
|
||||
);
|
||||
},
|
||||
);
|
||||
}
|
||||
|
||||
class ProposalSpanNode extends SpanNode {
|
||||
final String text;
|
||||
final String type;
|
||||
final Color backgroundColor;
|
||||
final Color foregroundColor;
|
||||
final Color borderColor;
|
||||
|
||||
ProposalSpanNode({
|
||||
required this.text,
|
||||
required this.type,
|
||||
required this.backgroundColor,
|
||||
required this.foregroundColor,
|
||||
required this.borderColor,
|
||||
});
|
||||
|
||||
@override
|
||||
InlineSpan build() {
|
||||
return WidgetSpan(
|
||||
child: Container(
|
||||
padding: const EdgeInsets.symmetric(horizontal: 8, vertical: 4),
|
||||
decoration: BoxDecoration(
|
||||
color: backgroundColor,
|
||||
border: Border.all(color: borderColor, width: 1),
|
||||
borderRadius: BorderRadius.circular(8),
|
||||
),
|
||||
child: Column(
|
||||
mainAxisSize: MainAxisSize.min,
|
||||
spacing: 6,
|
||||
children: [
|
||||
Row(
|
||||
spacing: 6,
|
||||
children: [
|
||||
Icon(Symbols.lightbulb, size: 16, color: foregroundColor),
|
||||
Text(
|
||||
'SN-chan suggest you to ${type.split('_').reversed.join(' ')}',
|
||||
).fontSize(13).opacity(0.8),
|
||||
],
|
||||
).padding(top: 3, bottom: 4),
|
||||
Flexible(
|
||||
child: Text(
|
||||
text,
|
||||
style: TextStyle(
|
||||
color: foregroundColor,
|
||||
fontSize: 14,
|
||||
fontWeight: FontWeight.w500,
|
||||
),
|
||||
),
|
||||
),
|
||||
],
|
||||
),
|
||||
),
|
||||
);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user