1 files changed, 64 insertions, 0 deletions
diff --git a/frontend/lib/models/chat.dart b/frontend/lib/models/chat.dart
new file mode 100644
index 000000000..6002f147f
--- /dev/null
+++ b/frontend/lib/models/chat.dart
@@ -0,0 +1,64 @@
+import 'package:auto_gpt_flutter_client/models/artifact.dart';
+import 'package:auto_gpt_flutter_client/models/message_type.dart';
+
+/// Represents a chat message related to a specific task.
+class Chat {
+  final String id;
+  final String taskId;
+  final String message;
+  final DateTime timestamp;
+  final MessageType messageType;
+  final Map<String, dynamic>? jsonResponse;
+  final List<Artifact> artifacts;
+
+  Chat({
+    required this.id,
+    required this.taskId,
+    required this.message,
+    required this.timestamp,
+    required this.messageType,
+    this.jsonResponse,
+    required this.artifacts,
+  });
+
+  // Convert a Map (usually from JSON) to a Chat object
+  factory Chat.fromMap(Map<String, dynamic> map) {
+    return Chat(
+      id: map['id'],
+      taskId: map['taskId'],
+      message: map['message'],
+      timestamp: DateTime.parse(map['timestamp']),
+      messageType: MessageType.values.firstWhere(
+          (e) => e.toString() == 'MessageType.${map['messageType']}'),
+      artifacts: (map['artifacts'] as List)
+          .map(
+              (artifact) => Artifact.fromJson(artifact as Map<String, dynamic>))
+          .toList(),
+    );
+  }
+
+  @override
+  bool operator ==(Object other) =>
+      identical(this, other) ||
+      other is Chat &&
+          runtimeType == other.runtimeType &&
+          id == other.id &&
+          taskId == other.taskId &&
+          message == other.message &&
+          timestamp == other.timestamp &&
+          messageType == other.messageType &&
+          artifacts == other.artifacts;
+
+  @override
+  int get hashCode =>
+      id.hashCode ^
+      taskId.hashCode ^
+      message.hashCode ^
+      timestamp.hashCode ^
+      messageType.hashCode ^
+      artifacts.hashCode;
+
+  @override
+  String toString() =>
+      'Chat(id: $id, taskId: $taskId, message: $message, timestamp: $timestamp, messageType: $messageType, artifacts: $artifacts)'; // Added artifacts in toString method
+}