modellist: fix a few issues with loading remote models (#2875)

Signed-off-by: Jared Van Bortel <jared@nomic.ai>

gpt4all-chat/CHANGELOG.md

@@ -23,8 +23,9 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/).
 - Set the window icon on Linux ([#2880](https://github.com/nomic-ai/gpt4all/pull/2880))
 - Corrections to the Romanian translation (by [@SINAPSA-IC](https://github.com/SINAPSA-IC) in [#2890](https://github.com/nomic-ai/gpt4all/pull/2890))
 - Fix singular/plural forms of LocalDocs "x Sources" (by [@cosmic-snow](https://github.com/cosmic-snow) in [#2885](https://github.com/nomic-ai/gpt4all/pull/2885))
-- Fixed typo in several files. (by [@3Simplex](https://github.com/3Simplex) in [#2916](https://github.com/nomic-ai/gpt4all/pull/2916))
+- Fix a typo in Model Settings (by [@3Simplex](https://github.com/3Simplex) in [#2916](https://github.com/nomic-ai/gpt4all/pull/2916))
 - Fix the antenna icon tooltip when using the local server ([#2922](https://github.com/nomic-ai/gpt4all/pull/2922))
+- Fix a few issues with locating files and handling errors when loading remote models on startup ([#2875](https://github.com/nomic-ai/gpt4all/pull/2875))
 
 ## [3.2.1] - 2024-08-13
 

gpt4all-chat/src/modellist.cpp

@@ -1208,51 +1208,55 @@ bool ModelList::modelExists(const QString &modelFilename) const
     return false;
 }
 
-void ModelList::updateModelsFromDirectory()
+void ModelList::updateOldRemoteModels(const QString &path)
 {
-    const QString exePath = QCoreApplication::applicationDirPath() + QDir::separator();
-    const QString localPath = MySettings::globalInstance()->modelPath();
-
-    auto updateOldRemoteModels = [&](const QString& path) {
-        QDirIterator it(path, QDirIterator::Subdirectories);
+    QDirIterator it(path, QDir::Files, QDirIterator::Subdirectories);
     while (it.hasNext()) {
-            it.next();
-            if (!it.fileInfo().isDir()) {
+        QFileInfo info = it.nextFileInfo();
         QString filename = it.fileName();
-                if (filename.startsWith("chatgpt-") && filename.endsWith(".txt")) {
+        if (!filename.startsWith("chatgpt-") || !filename.endsWith(".txt"))
+            continue;
+
         QString apikey;
         QString modelname(filename);
         modelname.chop(4); // strip ".txt" extension
         modelname.remove(0, 8); // strip "chatgpt-" prefix
-                    QFile file(path + filename);
-                    if (file.open(QIODevice::ReadWrite)) {
+        QFile file(info.filePath());
+        if (!file.open(QIODevice::ReadOnly)) {
+            qWarning().noquote() << tr("cannot open \"%1\": %2").arg(file.fileName(), file.errorString());
+            continue;
+        }
+
+        {
             QTextStream in(&file);
             apikey = in.readAll();
             file.close();
         }
 
-                    QJsonObject obj;
-                    obj.insert("apiKey", apikey);
-                    obj.insert("modelName", modelname);
-                    QJsonDocument doc(obj);
-
-                    auto newfilename = u"gpt4all-%1.rmodel"_s.arg(modelname);
-                    QFile newfile(path + newfilename);
-                    if (newfile.open(QIODevice::ReadWrite)) {
-                        QTextStream out(&newfile);
-                        out << doc.toJson();
-                        newfile.close();
+        QFile newfile(u"%1/gpt4all-%2.rmodel"_s.arg(info.dir().path(), modelname));
+        if (!newfile.open(QIODevice::ReadWrite)) {
+            qWarning().noquote() << tr("cannot create \"%1\": %2").arg(newfile.fileName(), file.errorString());
+            continue;
         }
+
+        QJsonObject obj {
+            { "apiKey",    apikey    },
+            { "modelName", modelname },
+        };
+
+        QTextStream out(&newfile);
+        out << QJsonDocument(obj).toJson();
+        newfile.close();
+
         file.remove();
     }
 }
-        }
-    };
 
-    auto processDirectory = [&](const QString& path) {
+void ModelList::processModelDirectory(const QString &path)
+{
     QDirIterator it(path, QDir::Files, QDirIterator::Subdirectories);
     while (it.hasNext()) {
-            it.next();
+        QFileInfo info = it.nextFileInfo();
 
         QString filename = it.fileName();
         if (filename.startsWith("incomplete") || FILENAME_BLACKLIST.contains(filename))
@@ -1260,22 +1264,6 @@ void ModelList::updateModelsFromDirectory()
         if (!filename.endsWith(".gguf") && !filename.endsWith(".rmodel"))
             continue;
 
-            QVector<QString> modelsById;
-            {
-                QMutexLocker locker(&m_mutex);
-                for (ModelInfo *info : m_models)
-                    if (info->filename() == filename)
-                        modelsById.append(info->id());
-            }
-
-            if (modelsById.isEmpty()) {
-                if (!contains(filename))
-                    addModel(filename);
-                modelsById.append(filename);
-            }
-
-            QFileInfo info = it.fileInfo();
-
         bool isOnline(filename.endsWith(".rmodel"));
         bool isCompatibleApi(filename.endsWith("-capi.rmodel"));
 
@@ -1284,10 +1272,11 @@ void ModelList::updateModelsFromDirectory()
         if (isCompatibleApi) {
             QJsonObject obj;
             {
-                    QFile file(path + filename);
-                    bool success = file.open(QIODeviceBase::ReadOnly);
-                    (void)success;
-                    Q_ASSERT(success);
+                QFile file(info.filePath());
+                if (!file.open(QIODeviceBase::ReadOnly)) {
+                    qWarning().noquote() << tr("cannot open \"%1\": %2").arg(file.fileName(), file.errorString());
+                    continue;
+                }
                 QJsonDocument doc = QJsonDocument::fromJson(file.readAll());
                 obj = doc.object();
             }
@@ -1305,6 +1294,20 @@ void ModelList::updateModelsFromDirectory()
             }
         }
 
+        QVector<QString> modelsById;
+        {
+            QMutexLocker locker(&m_mutex);
+            for (ModelInfo *info : m_models)
+                if (info->filename() == filename)
+                    modelsById.append(info->id());
+        }
+
+        if (modelsById.isEmpty()) {
+            if (!contains(filename))
+                addModel(filename);
+            modelsById.append(filename);
+        }
+
         for (const QString &id : modelsById) {
             QVector<QPair<int, QVariant>> data {
                 { InstalledRole, true },
@@ -1326,14 +1329,18 @@ void ModelList::updateModelsFromDirectory()
             updateData(id, data);
         }
     }
-    };
+}
 
+void ModelList::updateModelsFromDirectory()
+{
+    const QString exePath = QCoreApplication::applicationDirPath() + QDir::separator();
+    const QString localPath = MySettings::globalInstance()->modelPath();
 
     updateOldRemoteModels(exePath);
-    processDirectory(exePath);
+    processModelDirectory(exePath);
     if (localPath != exePath) {
         updateOldRemoteModels(localPath);
-        processDirectory(localPath);
+        processModelDirectory(localPath);
     }
 }
 

gpt4all-chat/src/modellist.h

@@ -502,6 +502,8 @@ private:
     void parseModelsJsonFile(const QByteArray &jsonData, bool save);
     void parseDiscoveryJsonFile(const QByteArray &jsonData);
     QString uniqueModelName(const ModelInfo &model) const;
+    void updateOldRemoteModels(const QString &path);
+    void processModelDirectory(const QString &path);
 
 private:
     mutable QMutex m_mutex;