From c130e3a116d56d2573eb2692307a105ea352ad90 Mon Sep 17 00:00:00 2001 From: Evzen Gasta Date: Thu, 21 Nov 2024 15:32:50 +0100 Subject: [PATCH] chore: changed imported model id to sha256 of model path Signed-off-by: Evzen Gasta --- packages/backend/src/managers/catalogManager.ts | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/packages/backend/src/managers/catalogManager.ts b/packages/backend/src/managers/catalogManager.ts index e8d64e595..df3cc4815 100644 --- a/packages/backend/src/managers/catalogManager.ts +++ b/packages/backend/src/managers/catalogManager.ts @@ -19,11 +19,12 @@ import type { ApplicationCatalog } from '@shared/src/models/IApplicationCatalog'; import fs, { promises } from 'node:fs'; import path from 'node:path'; +import crypto from 'node:crypto'; import defaultCatalog from '../assets/ai.json'; import type { Recipe } from '@shared/src/models/IRecipe'; import type { ModelInfo } from '@shared/src/models/IModelInfo'; import { Messages } from '@shared/Messages'; -import { type Disposable, type Event, EventEmitter, type Webview, window, env } from '@podman-desktop/api'; +import { type Disposable, type Event, EventEmitter, type Webview, window } from '@podman-desktop/api'; import { JsonWatcher } from '../utils/JsonWatcher'; import { Publisher } from '../utils/Publisher'; import type { LocalModelImportInfo } from '@shared/src/models/ILocalModelInfo'; @@ -211,11 +212,9 @@ export class CatalogManager extends Publisher implements Dis const models: ModelInfo[] = await Promise.all( localModels.map(async local => { const statFile = await promises.stat(local.path); + const sha256 = crypto.createHash('sha256').update(local.path).digest('hex'); return { - // on Linux the path is in the format "/home/user/folder/model1" and the routes using the "/" as well. - // Due to this would path to some models looks e.g. like this "/model/id//home/user/folder/model1", - // which is wrong. Replacing the "/" to "\" solves this issue - id: env.isLinux ? local.path.replace(/\//g, '\\') : local.path, + id: sha256, name: local.name, description: `Model imported from ${local.path}`, file: {