scale-gp-beta 0.1.0a8__py3-none-any.whl → 0.1.0a10__py3-none-any.whl

scale_gp_beta/resources/evaluations.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Dict, Iterable, Optional
-from typing_extensions import overload
+from typing import Dict, List, Iterable, Optional
+from typing_extensions import Literal, overload
 
 import httpx
 
@@ -58,6 +58,7 @@ class EvaluationsResource(SyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -72,6 +73,8 @@ class EvaluationsResource(SyncAPIResource):
         Args:
           data: Items to be evaluated
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -92,6 +95,7 @@ class EvaluationsResource(SyncAPIResource):
         name: str,
         data: Iterable[evaluation_create_params.EvaluationFromDatasetCreateRequestData] | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -108,6 +112,8 @@ class EvaluationsResource(SyncAPIResource):
 
           data: Items to be evaluated, including references to the input dataset items
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -128,6 +134,7 @@ class EvaluationsResource(SyncAPIResource):
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset,
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -144,6 +151,8 @@ class EvaluationsResource(SyncAPIResource):
 
           dataset: Create a reusable dataset from items in the `data` field
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -165,6 +174,7 @@ class EvaluationsResource(SyncAPIResource):
         | NotGiven = NOT_GIVEN,
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         dataset_id: str | NotGiven = NOT_GIVEN,
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset | NotGiven = NOT_GIVEN,
@@ -182,6 +192,7 @@ class EvaluationsResource(SyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "tags": tags,
                     "tasks": tasks,
                     "dataset_id": dataset_id,
                     "dataset": dataset,
@@ -199,6 +210,7 @@ class EvaluationsResource(SyncAPIResource):
         evaluation_id: str,
         *,
         include_archived: bool | NotGiven = NOT_GIVEN,
+        views: List[Literal["tasks"]] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -228,7 +240,11 @@ class EvaluationsResource(SyncAPIResource):
                 extra_body=extra_body,
                 timeout=timeout,
                 query=maybe_transform(
-                    {"include_archived": include_archived}, evaluation_retrieve_params.EvaluationRetrieveParams
+                    {
+                        "include_archived": include_archived,
+                        "views": views,
+                    },
+                    evaluation_retrieve_params.EvaluationRetrieveParams,
                 ),
             ),
             cast_to=Evaluation,
@@ -241,6 +257,7 @@ class EvaluationsResource(SyncAPIResource):
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
         starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        views: List[Literal["tasks"]] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -274,6 +291,7 @@ class EvaluationsResource(SyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "starting_after": starting_after,
+                        "views": views,
                     },
                     evaluation_list_params.EvaluationListParams,
                 ),
@@ -342,6 +360,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         data: Iterable[Dict[str, object]],
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -356,6 +375,8 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         Args:
           data: Items to be evaluated
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -376,6 +397,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         name: str,
         data: Iterable[evaluation_create_params.EvaluationFromDatasetCreateRequestData] | NotGiven = NOT_GIVEN,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -392,6 +414,8 @@ class AsyncEvaluationsResource(AsyncAPIResource):
 
           data: Items to be evaluated, including references to the input dataset items
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -412,6 +436,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset,
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -428,6 +453,8 @@ class AsyncEvaluationsResource(AsyncAPIResource):
 
           dataset: Create a reusable dataset from items in the `data` field
 
+          tags: The tags associated with the entity
+
           tasks: Tasks allow you to augment and evaluate your data
 
           extra_headers: Send extra headers
@@ -449,6 +476,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         | NotGiven = NOT_GIVEN,
         name: str,
         description: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         tasks: Iterable[EvaluationTaskParam] | NotGiven = NOT_GIVEN,
         dataset_id: str | NotGiven = NOT_GIVEN,
         dataset: evaluation_create_params.EvaluationWithDatasetCreateRequestDataset | NotGiven = NOT_GIVEN,
@@ -466,6 +494,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                     "data": data,
                     "name": name,
                     "description": description,
+                    "tags": tags,
                     "tasks": tasks,
                     "dataset_id": dataset_id,
                     "dataset": dataset,
@@ -483,6 +512,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         evaluation_id: str,
         *,
         include_archived: bool | NotGiven = NOT_GIVEN,
+        views: List[Literal["tasks"]] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -512,7 +542,11 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                 extra_body=extra_body,
                 timeout=timeout,
                 query=await async_maybe_transform(
-                    {"include_archived": include_archived}, evaluation_retrieve_params.EvaluationRetrieveParams
+                    {
+                        "include_archived": include_archived,
+                        "views": views,
+                    },
+                    evaluation_retrieve_params.EvaluationRetrieveParams,
                 ),
             ),
             cast_to=Evaluation,
@@ -525,6 +559,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
         starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        views: List[Literal["tasks"]] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -558,6 +593,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "starting_after": starting_after,
+                        "views": views,
                     },
                     evaluation_list_params.EvaluationListParams,
                 ),