ars/iiEsaywebUIapp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: sources count

Browse Source
This commit is contained in:
cogwheel0
2025-09-28 15:15:35 +05:30
parent b8856679da
commit 0ff48eeb38
4 changed files with 243 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
lib/core/services/streaming_helper.dart
View File

@@ -12,6 +12,7 @@ import 'navigation_service.dart';
import '../../shared/widgets/themed_dialogs.dart';
import '../../shared/theme/theme_extensions.dart';
import '../utils/debug_logger.dart';
import '../utils/openwebui_source_parser.dart';
// Keep local verbosity toggle for socket logs
const bool kSocketVerboseLogging = false;
@@ -705,38 +706,17 @@ ActiveSocketStream attachUnifiedChunkedStreaming({
} catch (_) {}
} else {
try {
// Handle nested source structure from OpenWebUI
final sourceData = map['source'];
final ChatSourceReference source;
if (sourceData is Map<String, dynamic>) {
// Extract the actual source information from nested structure
final sourceMap = Map<String, dynamic>.from(sourceData);
// Add additional metadata from the outer structure if available
if (map.containsKey('document') && map['document'] is List) {
final documents = map['document'] as List;
if (documents.isNotEmpty) {
sourceMap['snippet'] = documents.first?.toString();
final sources = parseOpenWebUISourceList([map]);
if (sources.isNotEmpty) {
final targetId = _resolveTargetMessageId(
messageId,
getMessages,
);
if (targetId != null) {
for (final source in sources) {
appendSourceReference(targetId, source);
}
}
if (map.containsKey('metadata') && map['metadata'] is List) {
final metadata = map['metadata'] as List;
if (metadata.isNotEmpty && metadata.first is Map) {
sourceMap['metadata'] = metadata.first;
}
}
source = ChatSourceReference.fromJson(sourceMap);
} else {
// Fallback: treat the entire map as a source (for backward compatibility)
source = ChatSourceReference.fromJson(map);
}
final targetId = _resolveTargetMessageId(messageId, getMessages);
if (targetId != null) {
appendSourceReference(targetId, source);
}
} catch (_) {}
}