From 6fa583b6a86dfc9a45a9ced826a339be19b5c9df Mon Sep 17 00:00:00 2001
From: Stan Girard <girard.stanislas@gmail.com>
Date: Mon, 15 Apr 2024 23:21:03 +0200
Subject: [PATCH] feat(assistants):  Add new input models for boolean, number,
 select text, and select number

---
 backend/modules/assistant/dto/inputs.py  | 24 +++++++++++++++
 backend/modules/assistant/dto/outputs.py | 37 ++++++++++++++++++++++++
 2 files changed, 61 insertions(+)

diff --git a/backend/modules/assistant/dto/inputs.py b/backend/modules/assistant/dto/inputs.py
index 512b870b5..cb77e2229 100644
--- a/backend/modules/assistant/dto/inputs.py
+++ b/backend/modules/assistant/dto/inputs.py
@@ -29,10 +29,34 @@ class TextInput(BaseModel):
     value: str
 
 
+class InputBoolean(BaseModel):
+    key: str
+    value: bool
+
+
+class InputNumber(BaseModel):
+    key: str
+    value: int
+
+
+class InputSelectText(BaseModel):
+    key: str
+    value: str
+
+
+class InputSelectNumber(BaseModel):
+    key: str
+    value: int
+
+
 class Inputs(BaseModel):
     files: Optional[List[FileInput]] = None
     urls: Optional[List[UrlInput]] = None
     texts: Optional[List[TextInput]] = None
+    booleans: Optional[List[InputBoolean]] = None
+    numbers: Optional[List[InputNumber]] = None
+    select_texts: Optional[List[InputSelectText]] = None
+    select_numbers: Optional[List[InputSelectNumber]] = None
 
 
 class Outputs(BaseModel):
diff --git a/backend/modules/assistant/dto/outputs.py b/backend/modules/assistant/dto/outputs.py
index a13af7370..3254e030b 100644
--- a/backend/modules/assistant/dto/outputs.py
+++ b/backend/modules/assistant/dto/outputs.py
@@ -20,12 +20,49 @@ class InputText(BaseModel):
     key: str
     required: Optional[bool] = True
     description: str
+    validation_regex: Optional[str] = None
+
+
+class InputBoolean(BaseModel):
+    key: str
+    required: Optional[bool] = True
+    description: str
+
+
+class InputNumber(BaseModel):
+    key: str
+    required: Optional[bool] = True
+    description: str
+    min: Optional[int] = None
+    max: Optional[int] = None
+    increment: Optional[int] = None
+    default: Optional[int] = None
+
+
+class InputSelectText(BaseModel):
+    key: str
+    required: Optional[bool] = True
+    description: str
+    options: List[str]
+    default: Optional[str] = None
+
+
+class InputSelectNumber(BaseModel):
+    key: str
+    required: Optional[bool] = True
+    description: str
+    options: List[int]
+    default: Optional[int] = None
 
 
 class Inputs(BaseModel):
     files: Optional[List[InputFile]] = None
     urls: Optional[List[InputUrl]] = None
     texts: Optional[List[InputText]] = None
+    booleans: Optional[List[InputBoolean]] = None
+    numbers: Optional[List[InputNumber]] = None
+    select_texts: Optional[List[InputSelectText]] = None
+    select_numbers: Optional[List[InputSelectNumber]] = None
 
 
 class OutputEmail(BaseModel):