From 91ba90c9a42f2f7a4e91d820ce2fbed7350c433d Mon Sep 17 00:00:00 2001
From: Stan Girard <girard.stanislas@gmail.com>
Date: Thu, 11 Apr 2024 18:41:09 +0200
Subject: [PATCH] feat: Add tags to AssistantOutput classes

---
 backend/modules/assistant/dto/outputs.py          | 1 +
 backend/modules/assistant/ito/audio_transcript.py | 1 +
 backend/modules/assistant/ito/crawler.py          | 1 +
 backend/modules/assistant/ito/summary.py          | 1 +
 4 files changed, 4 insertions(+)

diff --git a/backend/modules/assistant/dto/outputs.py b/backend/modules/assistant/dto/outputs.py
index 0d3dc7f56..2935d137e 100644
--- a/backend/modules/assistant/dto/outputs.py
+++ b/backend/modules/assistant/dto/outputs.py
@@ -47,6 +47,7 @@ class Outputs(BaseModel):
 
 class AssistantOutput(BaseModel):
     name: str
+    tags: Optional[List[str]] = []
     input_description: str
     output_description: str
     inputs: Inputs
diff --git a/backend/modules/assistant/ito/audio_transcript.py b/backend/modules/assistant/ito/audio_transcript.py
index c298fc436..8ae8c2044 100644
--- a/backend/modules/assistant/ito/audio_transcript.py
+++ b/backend/modules/assistant/ito/audio_transcript.py
@@ -59,6 +59,7 @@ def audio_transcript_inputs():
     output = AssistantOutput(
         name="Audio Transcript",
         description="Transcribes an audio file",
+        tags=["new"],
         input_description="One audio file to transcribe",
         output_description="Transcription of the audio file",
         inputs=Inputs(
diff --git a/backend/modules/assistant/ito/crawler.py b/backend/modules/assistant/ito/crawler.py
index b564cc94f..891421e4b 100644
--- a/backend/modules/assistant/ito/crawler.py
+++ b/backend/modules/assistant/ito/crawler.py
@@ -45,6 +45,7 @@ def crawler_inputs():
     output = AssistantOutput(
         name="Crawler",
         description="Crawls a website and extracts the text from the pages",
+        tags=["new"],
         input_description="One URL to crawl",
         output_description="Text extracted from the pages",
         inputs=Inputs(
diff --git a/backend/modules/assistant/ito/summary.py b/backend/modules/assistant/ito/summary.py
index 4df4ad52c..d128764cd 100644
--- a/backend/modules/assistant/ito/summary.py
+++ b/backend/modules/assistant/ito/summary.py
@@ -141,6 +141,7 @@ def summary_inputs():
     output = AssistantOutput(
         name="Summary",
         description="Summarize a set of documents",
+        tags=["new"],
         input_description="One document to summarize",
         output_description="A summary of the document",
         inputs=Inputs(