From d63ed6e651d40d3a18ddbc265d8166fddb18cf98 Mon Sep 17 00:00:00 2001
From: Stan Girard <girard.stanislas@gmail.com>
Date: Thu, 18 Apr 2024 23:00:41 +0200
Subject: [PATCH] fix(assistants): brain id is null

---
 backend/modules/assistant/dto/inputs.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/backend/modules/assistant/dto/inputs.py b/backend/modules/assistant/dto/inputs.py
index cb77e2229..631f3e4fe 100644
--- a/backend/modules/assistant/dto/inputs.py
+++ b/backend/modules/assistant/dto/inputs.py
@@ -2,7 +2,7 @@ import json
 from typing import List, Optional
 from uuid import UUID
 
-from pydantic import BaseModel, model_validator
+from pydantic import BaseModel, model_validator, root_validator
 
 
 class EmailInput(BaseModel):
@@ -10,8 +10,15 @@ class EmailInput(BaseModel):
 
 
 class BrainInput(BaseModel):
-    activated: bool
-    value: UUID
+    activated: Optional[bool] = False
+    value: Optional[UUID] = None
+
+    @root_validator(pre=True)
+    def empty_string_to_none(cls, values):
+        for field, value in values.items():
+            if value == "":
+                values[field] = None
+        return values
 
 
 class FileInput(BaseModel):