From 2b92175ba944040c39abc4a87d2c4596047810d5 Mon Sep 17 00:00:00 2001
From: Andriy Mulyar <andriy.mulyar@gmail.com>
Date: Wed, 14 Aug 2024 16:46:18 -0400
Subject: [PATCH] Adds GPT4All as a local model option

---
 packages/tasks/src/local-apps.ts | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/packages/tasks/src/local-apps.ts b/packages/tasks/src/local-apps.ts
index e045d8378..1581021b1 100644
--- a/packages/tasks/src/local-apps.ts
+++ b/packages/tasks/src/local-apps.ts
@@ -250,6 +250,13 @@ export const LOCAL_APPS = {
 		displayOnModelPage: (model) => model.library_name === "diffusers" && model.pipeline_tag === "text-to-image",
 		deeplink: (model) => new URL(`https://models.invoke.ai/huggingface/${model.id}`),
 	},
+	gpt4all: {
+		prettyLabel: "GPT4All",
+		docsUrl: "https://nomic.ai/gpt4all",
+		mainTask: "text-generation",
+		displayOnModelPage: isGgufModel,
+		deeplink: (model) => new URL(`gpt4all://models/huggingface/${model.id}`),
+	},
 } satisfies Record<string, LocalApp>;
 
 export type LocalAppKey = keyof typeof LOCAL_APPS;