feat: Add response_language parameter support to Maestro run methods (#311)

Add response_language parameter to BaseMaestroRun and MaestroRun classes
to enable language-specific responses in Maestro API calls.

Co-authored-by: benshuk <[email protected]>

Author: benshuk

ai21/clients/common/maestro/run.py

@@ -31,6 +31,7 @@ def _create_body(
         requirements: List[Requirement] | NotGiven,
         budget: Budget | NotGiven,
         include: List[OutputOptions] | NotGiven,
+        response_language: str | NotGiven,
         **kwargs,
     ) -> dict:
         return remove_not_given(
@@ -42,6 +43,7 @@ def _create_body(
                 "requirements": requirements,
                 "budget": budget,
                 "include": include,
+                "response_language": response_language,
                 **kwargs,
             }
         )
@@ -57,6 +59,7 @@ def create(
         requirements: List[Requirement] | NotGiven = NOT_GIVEN,
         budget: Budget | NotGiven = NOT_GIVEN,
         include: List[OutputOptions] | NotGiven = NOT_GIVEN,
+        response_language: str | NotGiven = NOT_GIVEN,
         **kwargs,
     ) -> RunResponse:
         pass
@@ -80,6 +83,7 @@ def create_and_poll(
         requirements: List[Requirement] | NotGiven = NOT_GIVEN,
         budget: Budget | NotGiven = NOT_GIVEN,
         include: List[OutputOptions] | NotGiven = NOT_GIVEN,
+        response_language: str | NotGiven = NOT_GIVEN,
         poll_interval_sec: float = DEFAULT_RUN_POLL_INTERVAL,
         poll_timeout_sec: float = DEFAULT_RUN_POLL_TIMEOUT,
         **kwargs,

ai21/clients/studio/resources/maestro/run.py

@@ -32,6 +32,7 @@ def create(
         requirements: List[Requirement] | NotGiven = NOT_GIVEN,
         budget: Budget | NotGiven = NOT_GIVEN,
         include: List[OutputOptions] | NotGiven = NOT_GIVEN,
+        response_language: str | NotGiven = NOT_GIVEN,
         **kwargs,
     ) -> RunResponse:
         body = self._create_body(
@@ -42,6 +43,7 @@ def create(
             requirements=requirements,
             budget=budget,
             include=include,
+            response_language=response_language,
             **kwargs,
         )
 
@@ -77,6 +79,7 @@ def create_and_poll(
         requirements: List[Requirement] | NotGiven = NOT_GIVEN,
         budget: Budget | NotGiven = NOT_GIVEN,
         include: List[OutputOptions] | NotGiven = NOT_GIVEN,
+        response_language: str | NotGiven = NOT_GIVEN,
         poll_interval_sec: float = DEFAULT_RUN_POLL_INTERVAL,
         poll_timeout_sec: float = DEFAULT_RUN_POLL_TIMEOUT,
         **kwargs,
@@ -89,6 +92,7 @@ def create_and_poll(
             requirements=requirements,
             budget=budget,
             include=include,
+            response_language=response_language,
             **kwargs,
         )
 
@@ -108,6 +112,7 @@ async def create(
         requirements: List[Requirement] | NotGiven = NOT_GIVEN,
         budget: Budget | NotGiven = NOT_GIVEN,
         include: List[OutputOptions] | NotGiven = NOT_GIVEN,
+        response_language: str | NotGiven = NOT_GIVEN,
         **kwargs,
     ) -> RunResponse:
         body = self._create_body(
@@ -118,6 +123,7 @@ async def create(
             requirements=requirements,
             budget=budget,
             include=include,
+            response_language=response_language,
             **kwargs,
         )
 
@@ -155,6 +161,7 @@ async def create_and_poll(
         include: List[OutputOptions] | NotGiven = NOT_GIVEN,
         poll_interval_sec: float = DEFAULT_RUN_POLL_INTERVAL,
         poll_timeout_sec: float = DEFAULT_RUN_POLL_TIMEOUT,
+        response_language: str | NotGiven = NOT_GIVEN,
         **kwargs,
     ) -> RunResponse:
         run = await self.create(
@@ -165,6 +172,7 @@ async def create_and_poll(
             requirements=requirements,
             budget=budget,
             include=include,
+            response_language=response_language,
             **kwargs,
         )