feat: assistant add icon and description

This commit is contained in:
Stan Girard 2024-04-11 22:34:49 +02:00
parent 9db8b63bb9
commit 41a9ec29f2
3 changed files with 5 additions and 5 deletions

View File

@ -5,8 +5,6 @@ from logger import get_logger
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(

View File

@ -47,6 +47,7 @@ class Outputs(BaseModel):
class AssistantOutput(BaseModel):
name: str
description: str
tags: Optional[List[str]] = []
input_description: str
output_description: str

View File

@ -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(