From e263bd69d2beb7a21e4efd4a11e1e09dd6149e13 Mon Sep 17 00:00:00 2001
From: nazaranton <97673072+nazaranton@users.noreply.github.com>
Date: Thu, 6 Nov 2025 16:28:24 +0200
Subject: [PATCH] feat: add 'system' to Maestro message role

---
 ai21/models/maestro/run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ai21/models/maestro/run.py b/ai21/models/maestro/run.py
index 6cfdab76..4b206437 100644
--- a/ai21/models/maestro/run.py
+++ b/ai21/models/maestro/run.py
@@ -5,7 +5,7 @@
 from ai21.models.ai21_base_model import AI21BaseModel
 
 Budget = Literal["low", "medium", "high"]
-Role = Literal["user", "assistant"]
+Role = Literal["user", "assistant", "system"]
 RunStatus = Literal["completed", "failed", "in_progress", "requires_action"]
 OutputOptions = Literal["data_sources", "requirements_result"]
 PrimitiveTypes = Union[Type[str], Type[int], Type[float], Type[bool]]