From 79eb092c6146b115ef2b33e873b6fa44ffdec283 Mon Sep 17 00:00:00 2001
From: Merve Noyan <merveenoyan@gmail.com>
Date: Mon, 13 Jan 2025 11:09:57 +0100
Subject: [PATCH] add document-retrieval to Hub as a task

---
 packages/tasks/src/pipelines.ts                              | 5 +++++
 packages/tasks/src/tasks/index.ts                            | 2 ++
 .../src/lib/components/Icons/IconDocumentRetrieval.svelte    | 1 +
 3 files changed, 8 insertions(+)
 create mode 100644 packages/widgets/src/lib/components/Icons/IconDocumentRetrieval.svelte

diff --git a/packages/tasks/src/pipelines.ts b/packages/tasks/src/pipelines.ts
index e272d2eef..319fd4f0c 100644
--- a/packages/tasks/src/pipelines.ts
+++ b/packages/tasks/src/pipelines.ts
@@ -676,6 +676,11 @@ export const PIPELINE_DATA = {
 		color: "red",
 		hideInDatasets: true,
 	},
+	"document-retrieval": {
+		name: "Document Retrieval",
+		modality: "multimodal",
+		color: "yellow",
+	},
 	"any-to-any": {
 		name: "Any-to-Any",
 		modality: "multimodal",
diff --git a/packages/tasks/src/tasks/index.ts b/packages/tasks/src/tasks/index.ts
index c1df8116d..e54d5d304 100644
--- a/packages/tasks/src/tasks/index.ts
+++ b/packages/tasks/src/tasks/index.ts
@@ -173,6 +173,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
 	"text-to-3d": ["diffusers"],
 	"image-to-3d": ["diffusers"],
 	"any-to-any": ["transformers"],
+	"document-retrieval": ["transformers"],
 };
 
 /**
@@ -201,6 +202,7 @@ export const TASKS_DATA: Record<PipelineType, TaskData | undefined> = {
 	"automatic-speech-recognition": getData("automatic-speech-recognition", automaticSpeechRecognition),
 	"depth-estimation": getData("depth-estimation", depthEstimation),
 	"document-question-answering": getData("document-question-answering", documentQuestionAnswering),
+	"document-retrieval": getData("document-retrieval", placeholder),
 	"feature-extraction": getData("feature-extraction", featureExtraction),
 	"fill-mask": getData("fill-mask", fillMask),
 	"graph-ml": undefined,
diff --git a/packages/widgets/src/lib/components/Icons/IconDocumentRetrieval.svelte b/packages/widgets/src/lib/components/Icons/IconDocumentRetrieval.svelte
new file mode 100644
index 000000000..d0c226ff6
--- /dev/null
+++ b/packages/widgets/src/lib/components/Icons/IconDocumentRetrieval.svelte
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="1em" height="1em" viewBox="0 0 32 32" {...$$props}><path fill="currentColor" d="m27.4 14.7l-6.1-6.1C21 8.2 20.5 8 20 8h-8c-1.1 0-2 .9-2 2v18c0 1.1.9 2 2 2h14c1.1 0 2-.9 2-2V16.1c0-.5-.2-1-.6-1.4M20 10l5.9 6H20zm-8 18V10h6v6c0 1.1.9 2 2 2h6v10z"/><path fill="currentColor" d="M6 18H4V4c0-1.1.9-2 2-2h14v2H6z"/></svg>
\ No newline at end of file