@vladimir-ks/aigile 0.2.4 → 0.2.5

package/dist/bin/aigile.js

@@ -970,12 +970,32 @@ function getFilesWithQualityIssues(sessionId) {
   );
 }
 function getUntaggedFiles(projectId, sessionId, options = {}) {
-  if (options.assignedOnly && options.chunkId) {
-    const chunk = getChunk(options.chunkId);
-    if (!chunk || !chunk.assigned_files) {
+  if (options.assignedOnly) {
+    let assignedFiles = [];
+    if (options.chunkId) {
+      const chunk = getChunk(options.chunkId);
+      if (chunk?.assigned_files) {
+        try {
+          assignedFiles = JSON.parse(chunk.assigned_files);
+        } catch {
+        }
+      }
+    } else {
+      const chunks = getSessionChunks(sessionId);
+      for (const chunk of chunks) {
+        if (chunk.assigned_files) {
+          try {
+            const files = JSON.parse(chunk.assigned_files);
+            assignedFiles.push(...files);
+          } catch {
+          }
+        }
+      }
+      assignedFiles = [...new Set(assignedFiles)];
+    }
+    if (assignedFiles.length === 0) {
       return [];
     }
-    const assignedFiles = JSON.parse(chunk.assigned_files);
     return queryAll(
       `SELECT d.path, d.id as document_id
        FROM documents d
@@ -1166,7 +1186,7 @@ var init_connection = __esm({
 import { Command as Command23 } from "commander";
 
 // src/index.ts
-var VERSION = true ? "0.2.4" : "0.0.0-dev";
+var VERSION = true ? "0.2.5" : "0.0.0-dev";
 
 // src/bin/aigile.ts
 init_connection();