agenta 0.32.0a1__py3-none-any.whl → 0.33.0__py3-none-any.whl

agenta/client/backend/containers/client.py

@@ -34,13 +34,13 @@ class ContainersClient:
         Builds a Docker image from a tar file containing the application code.
 
         Args:
-        app_id (str): The ID of the application to build the image for.
-        base_name (str): The base name of the image to build.
-        tar_file (UploadFile): The tar file containing the application code.
-        stoken_session (SessionContainer): The session container for the user making the request.
+            app_id (str): The ID of the application to build the image for.
+            base_name (str): The base name of the image to build.
+            tar_file (UploadFile): The tar file containing the application code.
+            stoken_session (SessionContainer): The session container for the user making the request.
 
         Returns:
-        Image: The Docker image that was built.
+            Image: The Docker image that was built.
 
         Parameters
         ----------
@@ -124,7 +124,7 @@ class ContainersClient:
         Restart docker container.
 
         Args:
-        payload (RestartAppContainer) -- the required data (app_name and variant_name)
+            payload (RestartAppContainer) -- the required data (app_name and variant_name)
 
         Parameters
         ----------
@@ -159,7 +159,11 @@ class ContainersClient:
             headers={
                 "content-type": "application/json",
             },
-            request_options=request_options,
+            request_options=(
+                {**request_options, "timeout_in_seconds": 600}
+                if request_options
+                else {"timeout_in_seconds": 600}
+            ),
             omit=OMIT,
         )
         try:
@@ -222,7 +226,11 @@ class ContainersClient:
         _response = self._client_wrapper.httpx_client.request(
             "containers/templates",
             method="GET",
-            request_options=request_options,
+            request_options=(
+                {**request_options, "timeout_in_seconds": 600}
+                if request_options
+                else {"timeout_in_seconds": 600}
+            ),
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -249,15 +257,24 @@ class ContainersClient:
         Constructs the URL for an app container based on the provided base_id or variant_id.
 
         Args:
-        base_id (Optional[str]): The ID of the base to use for the app container.
-        variant_id (Optional[str]): The ID of the variant to use for the app container.
-        request (Request): The request object.
+
+
+
+            base_id (Optional[str]): The ID of the base to use for the app container.
+            variant_id (Optional[str]): The ID of the variant to use for the app container.
+            request (Request): The request object.
 
         Returns:
-        URI: The URI for the app container.
+
+
+
+            URI: The URI for the app container.
 
         Raises:
-        HTTPException: If the base or variant cannot be found or the user does not have access.
+
+
+
+            HTTPException: If the base or variant cannot be found or the user does not have access.
 
         Parameters
         ----------
@@ -290,7 +307,11 @@ class ContainersClient:
                 "base_id": base_id,
                 "variant_id": variant_id,
             },
-            request_options=request_options,
+            request_options=(
+                {**request_options, "timeout_in_seconds": 600}
+                if request_options
+                else {"timeout_in_seconds": 600}
+            ),
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -333,13 +354,13 @@ class AsyncContainersClient:
         Builds a Docker image from a tar file containing the application code.
 
         Args:
-        app_id (str): The ID of the application to build the image for.
-        base_name (str): The base name of the image to build.
-        tar_file (UploadFile): The tar file containing the application code.
-        stoken_session (SessionContainer): The session container for the user making the request.
+            app_id (str): The ID of the application to build the image for.
+            base_name (str): The base name of the image to build.
+            tar_file (UploadFile): The tar file containing the application code.
+            stoken_session (SessionContainer): The session container for the user making the request.
 
         Returns:
-        Image: The Docker image that was built.
+            Image: The Docker image that was built.
 
         Parameters
         ----------
@@ -431,7 +452,7 @@ class AsyncContainersClient:
         Restart docker container.
 
         Args:
-        payload (RestartAppContainer) -- the required data (app_name and variant_name)
+            payload (RestartAppContainer) -- the required data (app_name and variant_name)
 
         Parameters
         ----------
@@ -474,7 +495,11 @@ class AsyncContainersClient:
             headers={
                 "content-type": "application/json",
             },
-            request_options=request_options,
+            request_options=(
+                {**request_options, "timeout_in_seconds": 600}
+                if request_options
+                else {"timeout_in_seconds": 600}
+            ),
             omit=OMIT,
         )
         try:
@@ -545,7 +570,11 @@ class AsyncContainersClient:
         _response = await self._client_wrapper.httpx_client.request(
             "containers/templates",
             method="GET",
-            request_options=request_options,
+            request_options=(
+                {**request_options, "timeout_in_seconds": 600}
+                if request_options
+                else {"timeout_in_seconds": 600}
+            ),
         )
         try:
             if 200 <= _response.status_code < 300:
@@ -572,15 +601,24 @@ class AsyncContainersClient:
         Constructs the URL for an app container based on the provided base_id or variant_id.
 
         Args:
-        base_id (Optional[str]): The ID of the base to use for the app container.
-        variant_id (Optional[str]): The ID of the variant to use for the app container.
-        request (Request): The request object.
+
+
+
+            base_id (Optional[str]): The ID of the base to use for the app container.
+            variant_id (Optional[str]): The ID of the variant to use for the app container.
+            request (Request): The request object.
 
         Returns:
-        URI: The URI for the app container.
+
+
+
+            URI: The URI for the app container.
 
         Raises:
-        HTTPException: If the base or variant cannot be found or the user does not have access.
+
+
+
+            HTTPException: If the base or variant cannot be found or the user does not have access.
 
         Parameters
         ----------
@@ -621,7 +659,11 @@ class AsyncContainersClient:
                 "base_id": base_id,
                 "variant_id": variant_id,
             },
-            request_options=request_options,
+            request_options=(
+                {**request_options, "timeout_in_seconds": 600}
+                if request_options
+                else {"timeout_in_seconds": 600}
+            ),
         )
         try:
             if 200 <= _response.status_code < 300:

agenta/client/backend/core/http_client.py

@@ -148,9 +148,9 @@ def get_request_body(
         json_body = maybe_filter_request_body(json, request_options, omit)
 
     # If you have an empty JSON body, you should just send None
-    return (json_body if json_body != {} else None), (
-        data_body if data_body != {} else None
-    )
+    return (
+        json_body if json_body != {} else None
+    ), data_body if data_body != {} else None
 
 
 class HttpClient:

agenta/client/backend/environments/client.py

@@ -29,12 +29,12 @@ class EnvironmentsClient:
         Deploys a given variant to an environment
 
         Args:
-        environment_name: Name of the environment to deploy to.
-        variant_id: variant id to deploy.
-        stoken_session: . Defaults to Depends(verify_session()).
+            environment_name: Name of the environment to deploy to.
+            variant_id: variant id to deploy.
+            stoken_session: . Defaults to Depends(verify_session()).
 
         Raises:
-        HTTPException: If the deployment fails.
+            HTTPException: If the deployment fails.
 
         Parameters
         ----------
@@ -116,12 +116,12 @@ class AsyncEnvironmentsClient:
         Deploys a given variant to an environment
 
         Args:
-        environment_name: Name of the environment to deploy to.
-        variant_id: variant id to deploy.
-        stoken_session: . Defaults to Depends(verify_session()).
+            environment_name: Name of the environment to deploy to.
+            variant_id: variant id to deploy.
+            stoken_session: . Defaults to Depends(verify_session()).
 
         Raises:
-        HTTPException: If the deployment fails.
+            HTTPException: If the deployment fails.
 
         Parameters
         ----------

agenta/client/backend/evaluations/client.py

@@ -34,15 +34,15 @@ class EvaluationsClient:
         Fetches evaluation ids for a given resource type and id.
 
         Arguments:
-        app_id (str): The ID of the app for which to fetch evaluations.
-        resource_type (str): The type of resource for which to fetch evaluations.
-        resource_ids List[ObjectId]: The IDs of resource for which to fetch evaluations.
+            app_id (str): The ID of the app for which to fetch evaluations.
+            resource_type (str): The type of resource for which to fetch evaluations.
+            resource_ids List[ObjectId]: The IDs of resource for which to fetch evaluations.
 
         Raises:
-        HTTPException: If the resource_type is invalid or access is denied.
+            HTTPException: If the resource_type is invalid or access is denied.
 
         Returns:
-        List[str]: A list of evaluation ids.
+            List[str]: A list of evaluation ids.
 
         Parameters
         ----------
@@ -110,10 +110,10 @@ class EvaluationsClient:
         Fetches a list of evaluations, optionally filtered by an app ID.
 
         Args:
-        app_id (Optional[str]): An optional app ID to filter the evaluations.
+            app_id (Optional[str]): An optional app ID to filter the evaluations.
 
         Returns:
-        List[Evaluation]: A list of evaluations.
+            List[Evaluation]: A list of evaluations.
 
         Parameters
         ----------
@@ -188,9 +188,9 @@ class EvaluationsClient:
         """
         Creates a new comparison table document
         Raises:
-        HTTPException: _description_
+            HTTPException: _description_
         Returns:
-        _description_
+            _description_
 
         Parameters
         ----------
@@ -326,9 +326,6 @@ class EvaluationsClient:
             json={
                 "evaluations_ids": evaluations_ids,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -366,11 +363,11 @@ class EvaluationsClient:
         Fetches the status of the evaluation.
 
         Args:
-        evaluation_id (str): the evaluation id
-        request (Request): the request object
+            evaluation_id (str): the evaluation id
+            request (Request): the request object
 
         Returns:
-        (str): the evaluation status
+            (str): the evaluation status
 
         Parameters
         ----------
@@ -435,11 +432,11 @@ class EvaluationsClient:
         Fetches the results of the evaluation
 
         Args:
-        evaluation_id (str): the evaluation id
-        request (Request): the request object
+            evaluation_id (str): the evaluation id
+            request (Request): the request object
 
         Returns:
-        _type_: _description_
+            _type_: _description_
 
         Parameters
         ----------
@@ -504,13 +501,13 @@ class EvaluationsClient:
         Fetches evaluation scenarios for a given evaluation ID.
 
         Arguments:
-        evaluation_id (str): The ID of the evaluation for which to fetch scenarios.
+            evaluation_id (str): The ID of the evaluation for which to fetch scenarios.
 
         Raises:
-        HTTPException: If the evaluation is not found or access is denied.
+            HTTPException: If the evaluation is not found or access is denied.
 
         Returns:
-        List[EvaluationScenario]: A list of evaluation scenarios.
+            List[EvaluationScenario]: A list of evaluation scenarios.
 
         Parameters
         ----------
@@ -575,10 +572,10 @@ class EvaluationsClient:
         Fetches a single evaluation based on its ID.
 
         Args:
-        evaluation_id (str): The ID of the evaluation to fetch.
+            evaluation_id (str): The ID of the evaluation to fetch.
 
         Returns:
-        Evaluation: The fetched evaluation.
+            Evaluation: The fetched evaluation.
 
         Parameters
         ----------
@@ -643,13 +640,13 @@ class EvaluationsClient:
         Fetches evaluation scenarios for a given evaluation ID.
 
         Arguments:
-        evaluation_id (str): The ID of the evaluation for which to fetch scenarios.
+            evaluation_id (str): The ID of the evaluation for which to fetch scenarios.
 
         Raises:
-        HTTPException: If the evaluation is not found or access is denied.
+            HTTPException: If the evaluation is not found or access is denied.
 
         Returns:
-        List[EvaluationScenario]: A list of evaluation scenarios.
+            List[EvaluationScenario]: A list of evaluation scenarios.
 
         Parameters
         ----------
@@ -723,15 +720,15 @@ class AsyncEvaluationsClient:
         Fetches evaluation ids for a given resource type and id.
 
         Arguments:
-        app_id (str): The ID of the app for which to fetch evaluations.
-        resource_type (str): The type of resource for which to fetch evaluations.
-        resource_ids List[ObjectId]: The IDs of resource for which to fetch evaluations.
+            app_id (str): The ID of the app for which to fetch evaluations.
+            resource_type (str): The type of resource for which to fetch evaluations.
+            resource_ids List[ObjectId]: The IDs of resource for which to fetch evaluations.
 
         Raises:
-        HTTPException: If the resource_type is invalid or access is denied.
+            HTTPException: If the resource_type is invalid or access is denied.
 
         Returns:
-        List[str]: A list of evaluation ids.
+            List[str]: A list of evaluation ids.
 
         Parameters
         ----------
@@ -807,10 +804,10 @@ class AsyncEvaluationsClient:
         Fetches a list of evaluations, optionally filtered by an app ID.
 
         Args:
-        app_id (Optional[str]): An optional app ID to filter the evaluations.
+            app_id (Optional[str]): An optional app ID to filter the evaluations.
 
         Returns:
-        List[Evaluation]: A list of evaluations.
+            List[Evaluation]: A list of evaluations.
 
         Parameters
         ----------
@@ -893,9 +890,9 @@ class AsyncEvaluationsClient:
         """
         Creates a new comparison table document
         Raises:
-        HTTPException: _description_
+            HTTPException: _description_
         Returns:
-        _description_
+            _description_
 
         Parameters
         ----------
@@ -1047,9 +1044,6 @@ class AsyncEvaluationsClient:
             json={
                 "evaluations_ids": evaluations_ids,
             },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1087,11 +1081,11 @@ class AsyncEvaluationsClient:
         Fetches the status of the evaluation.
 
         Args:
-        evaluation_id (str): the evaluation id
-        request (Request): the request object
+            evaluation_id (str): the evaluation id
+            request (Request): the request object
 
         Returns:
-        (str): the evaluation status
+            (str): the evaluation status
 
         Parameters
         ----------
@@ -1164,11 +1158,11 @@ class AsyncEvaluationsClient:
         Fetches the results of the evaluation
 
         Args:
-        evaluation_id (str): the evaluation id
-        request (Request): the request object
+            evaluation_id (str): the evaluation id
+            request (Request): the request object
 
         Returns:
-        _type_: _description_
+            _type_: _description_
 
         Parameters
         ----------
@@ -1241,13 +1235,13 @@ class AsyncEvaluationsClient:
         Fetches evaluation scenarios for a given evaluation ID.
 
         Arguments:
-        evaluation_id (str): The ID of the evaluation for which to fetch scenarios.
+            evaluation_id (str): The ID of the evaluation for which to fetch scenarios.
 
         Raises:
-        HTTPException: If the evaluation is not found or access is denied.
+            HTTPException: If the evaluation is not found or access is denied.
 
         Returns:
-        List[EvaluationScenario]: A list of evaluation scenarios.
+            List[EvaluationScenario]: A list of evaluation scenarios.
 
         Parameters
         ----------
@@ -1320,10 +1314,10 @@ class AsyncEvaluationsClient:
         Fetches a single evaluation based on its ID.
 
         Args:
-        evaluation_id (str): The ID of the evaluation to fetch.
+            evaluation_id (str): The ID of the evaluation to fetch.
 
         Returns:
-        Evaluation: The fetched evaluation.
+            Evaluation: The fetched evaluation.
 
         Parameters
         ----------
@@ -1396,13 +1390,13 @@ class AsyncEvaluationsClient:
         Fetches evaluation scenarios for a given evaluation ID.
 
         Arguments:
-        evaluation_id (str): The ID of the evaluation for which to fetch scenarios.
+            evaluation_id (str): The ID of the evaluation for which to fetch scenarios.
 
         Raises:
-        HTTPException: If the evaluation is not found or access is denied.
+            HTTPException: If the evaluation is not found or access is denied.
 
         Returns:
-        List[EvaluationScenario]: A list of evaluation scenarios.
+            List[EvaluationScenario]: A list of evaluation scenarios.
 
         Parameters
         ----------