Merge branch 'main' into main

QuivrHQ · Apr 12, 2024 · bb45566 · bb45566
2 parents 77e5f33 + a8c5e18
commit bb45566
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/backend/modules/assistant/controller/assistant_routes.py b/backend/modules/assistant/controller/assistant_routes.py
@@ -5,8 +5,6 @@
 from middlewares.auth import AuthBearer, get_current_user
 from modules.assistant.dto.inputs import InputAssistant
 from modules.assistant.dto.outputs import AssistantOutput
-from modules.assistant.ito.audio_transcript import audio_transcript_inputs
-from modules.assistant.ito.crawler import crawler_inputs
 from modules.assistant.ito.summary import SummaryAssistant, summary_inputs
 from modules.assistant.service.assistant import Assistant
 from modules.user.entity.user_identity import UserIdentity
@@ -28,9 +26,9 @@ async def list_assistants(
     """
 
     summary = summary_inputs()
-    crawler = crawler_inputs()
-    audio_transcript = audio_transcript_inputs()
-    return [summary, crawler, audio_transcript]
+    # crawler = crawler_inputs()
+    # audio_transcript = audio_transcript_inputs()
+    return [summary]
 
 
 @assistant_router.post(

diff --git 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
+    description: str
     tags: Optional[List[str]] = []
     input_description: str
     output_description: str

diff --git a/backend/modules/assistant/ito/summary.py b/backend/modules/assistant/ito/summary.py
@@ -144,6 +144,7 @@ def summary_inputs():
         tags=["new"],
         input_description="One document to summarize",
         output_description="A summary of the document",
+        icon_url="https://quivr-cms.s3.eu-west-3.amazonaws.com/assistant_summary_434446a2aa.png",
         inputs=Inputs(
             files=[
                 InputFile(