AI/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2024-10-01 01:06:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Clean up settings properly for removed models and also when user manually deletes.

Browse Source 
Signed-off-by: Adam Treat <treat.adam@gmail.com>
This commit is contained in:
Adam Treat 2024-03-09 10:03:31 -05:00 committed by AT
parent 0cc5a80656
commit f36a2874eb
3 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
gpt4all-chat/download.cpp
View File

@ -203,6 +203,7 @@ void Download::removeModel(const QString &modelFile)
QFile file(filePath);
if (file.exists()) {
const ModelInfo info = ModelList::globalInstance()->modelInfoByFilename(modelFile);
MySettings::globalInstance()->eraseModel(info);
shouldRemoveInstalled = info.installed && !info.isClone() && (info.isDiscovered() || info.description() == "" /*indicates sideloaded*/);
if (shouldRemoveInstalled)
ModelList::globalInstance()->removeInstalled(info);

37
gpt4all-chat/modellist.cpp
View File

@ -1137,27 +1137,18 @@ QString ModelList::uniqueModelName(const ModelInfo &model) const
return baseName;
}
QString ModelList::modelDirPath(const QString &modelName, bool isOnline)
bool ModelList::modelExists(const QString &modelFilename) const
{
QVector<QString> possibleFilePaths;
if (isOnline)
possibleFilePaths << "/" + modelName + ".txt";
else {
possibleFilePaths << "/ggml-" + modelName + ".bin";
possibleFilePaths << "/" + modelName + ".bin";
}
for (const QString &modelFilename : possibleFilePaths) {
QString appPath = QCoreApplication::applicationDirPath() + modelFilename;
QFileInfo infoAppPath(appPath);
if (infoAppPath.exists())
return QCoreApplication::applicationDirPath();
QString appPath = QCoreApplication::applicationDirPath() + modelFilename;
QFileInfo infoAppPath(appPath);
if (infoAppPath.exists())
return true;
QString downloadPath = MySettings::globalInstance()->modelPath() + modelFilename;
QFileInfo infoLocalPath(downloadPath);
if (infoLocalPath.exists())
return MySettings::globalInstance()->modelPath();
}
return QString();
QString downloadPath = MySettings::globalInstance()->modelPath() + modelFilename;
QFileInfo infoLocalPath(downloadPath);
if (infoLocalPath.exists())
return true;
return false;
}
void ModelList::updateModelsFromDirectory()
@ -1569,6 +1560,14 @@ void ModelList::updateModelsFromSettings()
if (contains(id))
continue;
// If we can't find the corresponding file, then delete it from settings as this reflects a
// stale model. The file could have been deleted manually by the user for instance.
if (!settings.contains(g + "/filename") || !modelExists(settings.value(g + "/filename").toString())) {
settings.remove(g);
settings.sync();
continue;
}
addModel(id);
QVector<QPair<int, QVariant>> data;

2
gpt4all-chat/modellist.h
View File

@ -462,7 +462,7 @@ private Q_SLOTS:
private:
void removeInternal(const ModelInfo &model);
void clearDiscoveredModels();
QString modelDirPath(const QString &modelName, bool isOnline);
bool modelExists(const QString &fileName) const;
int indexForModel(ModelInfo *model);
QVariant dataInternal(const ModelInfo *info, int role) const;
static bool lessThan(const ModelInfo* a, const ModelInfo* b, DiscoverSort s, int d);