Refactoring

lib/data/database/daos/analysis_dao.dart

@@ -27,6 +27,20 @@ class AnalysisDao extends DatabaseAccessor<AppDatabase>
   Future<void> insertAnnotation(AnnotationsCompanion entry) =>
       into(annotations).insert(entry);
 
+  Future<void> updateAnnotation({
+    required String id,
+    required String name,
+    required String description,
+    required String color,
+  }) =>
+      (update(annotations)..where((t) => t.id.equals(id))).write(
+        AnnotationsCompanion(
+          name: Value(name),
+          description: Value(description),
+          color: Value(color),
+        ),
+      );
+
   Future<void> deleteAnnotation(String id) =>
       (delete(annotations)..where((t) => t.id.equals(id))).go();
 }

lib/data/database/daos/knowledge_chunk_dao.dart

@@ -0,0 +1,35 @@
+import 'package:drift/drift.dart';
+import 'package:trainhub_flutter/data/database/app_database.dart';
+
+part 'knowledge_chunk_dao.g.dart';
+
+@DriftAccessor(tables: [KnowledgeChunks])
+class KnowledgeChunkDao extends DatabaseAccessor<AppDatabase>
+    with _$KnowledgeChunkDaoMixin {
+  KnowledgeChunkDao(super.db);
+
+  Future<void> insertChunk(KnowledgeChunksCompanion entry) =>
+      into(knowledgeChunks).insert(entry);
+
+  /// Returns every stored chunk, including its JSON-encoded embedding.
+  /// Loaded into memory for in-process cosine similarity scoring.
+  Future<List<KnowledgeChunk>> getAllChunks() =>
+      (select(knowledgeChunks)
+            ..orderBy([
+              (t) =>
+                  OrderingTerm(expression: t.createdAt, mode: OrderingMode.asc)
+            ]))
+          .get();
+
+  Future<int> getCount() async {
+    final rows = await select(knowledgeChunks).get();
+    return rows.length;
+  }
+
+  Future<void> deleteAll() => delete(knowledgeChunks).go();
+
+  Future<void> deleteBySourceId(String sourceId) =>
+      (delete(knowledgeChunks)
+            ..where((t) => t.sourceId.equals(sourceId)))
+          .go();
+}

lib/data/database/daos/knowledge_chunk_dao.g.dart

@@ -0,0 +1,8 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+part of 'knowledge_chunk_dao.dart';
+
+// ignore_for_file: type=lint
+mixin _$KnowledgeChunkDaoMixin on DatabaseAccessor<AppDatabase> {
+  $KnowledgeChunksTable get knowledgeChunks => attachedDatabase.knowledgeChunks;
+}