adaptive-sdk 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

adaptive_sdk/output_types/job_wrapper.py

@@ -0,0 +1,106 @@
+"""
+Wrapper for JobData that adds eval_result functionality without modifying autogenerated code.
+"""
+
+from __future__ import annotations
+from typing import TYPE_CHECKING
+from pydantic import BaseModel, Field
+
+from adaptive_sdk.graphql_client import JobData, JobArtifactKind, JobDataDetailsArtifactsByproductsEvaluationByproducts
+
+if TYPE_CHECKING:
+    pass
+
+
+class EvalMetricScore(BaseModel):
+    """A single metric score for a model in an evaluation."""
+
+    metric_key: str = Field(description="The key of the metric")
+    metric_name: str = Field(description="The human-readable name of the metric")
+    mean: float = Field(description="The mean score for this metric")
+    min: float = Field(description="The minimum score for this metric")
+    max: float = Field(description="The maximum score for this metric")
+    stddev: float = Field(description="The standard deviation of scores")
+    count: int = Field(description="The number of samples evaluated")
+    feedback_count: int = Field(description="The number of feedbacks received")
+
+
+class ModelEvalResult(BaseModel):
+    """Evaluation results for a single model."""
+
+    model_key: str = Field(description="The key of the model that was evaluated")
+    model_name: str = Field(description="The human-readable name of the model")
+    scores: list[EvalMetricScore] = Field(description="List of metric scores for this model", default_factory=list)
+
+
+class JobDataPlus(JobData):
+    """
+    Extended JobData that adds eval_result() method.
+    Inherits all attributes and methods from JobData.
+    """
+
+    @classmethod
+    def from_job_data(cls, job_data: JobData) -> "JobDataPlus":
+        """
+        Create a JobDataWrapper from a JobData instance.
+
+        Args:
+            job_data: The JobData instance to wrap
+
+        Returns:
+            A new JobDataWrapper with all the same data
+        """
+        # Use model_validate to create a new instance with the same data
+        return cls.model_validate(job_data.model_dump())
+
+    @property
+    def eval_results(self) -> list[ModelEvalResult] | None:
+        """
+        Extract and organize evaluation results from this job.
+
+        Returns:
+            A list of ModelEvalResult objects (one per model), or None if no eval results exist
+        """
+        if not self.details or not self.details.artifacts:
+            return None
+
+        # Dictionary to group scores by model
+        model_results: dict[str, ModelEvalResult] = {}
+
+        # Iterate through all artifacts looking for evaluation byproducts
+        for artifact in self.details.artifacts:
+            if not artifact.byproducts:
+                continue
+
+            # Check if this is an EvaluationByproducts with eval_results
+            if artifact.kind == JobArtifactKind.EVALUATION:
+                if artifact.byproducts is not None:
+                    assert isinstance(artifact.byproducts, JobDataDetailsArtifactsByproductsEvaluationByproducts)
+
+                    for eval_result in artifact.byproducts.eval_results:
+                        model_key = eval_result.model_service.key
+                        model_name = eval_result.model_service.name
+
+                        # Create or get the model result entry
+                        if model_key not in model_results:
+                            model_results[model_key] = ModelEvalResult(
+                                model_key=model_key, model_name=model_name, scores=[]
+                            )
+
+                        # Add the metric score to this model's results
+                        metric_score = EvalMetricScore(
+                            metric_key=eval_result.metric.key,
+                            metric_name=eval_result.metric.name,
+                            mean=eval_result.mean,
+                            min=eval_result.min,
+                            max=eval_result.max,
+                            stddev=eval_result.stddev,
+                            count=eval_result.count,
+                            feedback_count=eval_result.feedback_count,
+                        )
+                        model_results[model_key].scores.append(metric_score)
+
+        if not model_results:
+            return None
+
+        return list(model_results.values())

adaptive_sdk/resources/datasets.py

@@ -14,7 +14,10 @@ from adaptive_sdk.graphql_client import (
     DatasetCreateFromMultipartUpload,
 )
 from adaptive_sdk.graphql_client.custom_mutations import Mutation
-from adaptive_sdk.graphql_client.custom_fields import DatasetFields
+from adaptive_sdk.graphql_client.custom_fields import (
+    DatasetFields,
+    DatasetUploadProcessingStatusFields,
+)
 from adaptive_sdk.rest import rest_types
 from adaptive_sdk.error_handling import rest_error_handler
 
@@ -178,12 +181,12 @@ class Datasets(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
 
             mutation_field = Mutation.create_dataset_from_multipart_upload(input=input)
             mutation_field.fields(
-                DatasetFields.id,
-                DatasetFields.key,
-                DatasetFields.name,
-                DatasetFields.created_at,
-                DatasetFields.kind,
-                DatasetFields.records,
+                DatasetUploadProcessingStatusFields.dataset_id,
+                DatasetUploadProcessingStatusFields.total_parts,
+                DatasetUploadProcessingStatusFields.processed_parts,
+                DatasetUploadProcessingStatusFields.progress,
+                DatasetUploadProcessingStatusFields.error,
+                DatasetUploadProcessingStatusFields.status,
             )
             result = self._gql_client.mutation(
                 mutation_field, operation_name="CreateDatasetFromMultipartUpload"
@@ -331,12 +334,12 @@ class AsyncDatasets(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
 
             mutation_field = Mutation.create_dataset_from_multipart_upload(input=input)
             mutation_field.fields(
-                DatasetFields.id,
-                DatasetFields.key,
-                DatasetFields.name,
-                DatasetFields.created_at,
-                DatasetFields.kind,
-                DatasetFields.records,
+                DatasetUploadProcessingStatusFields.dataset_id,
+                DatasetUploadProcessingStatusFields.total_parts,
+                DatasetUploadProcessingStatusFields.processed_parts,
+                DatasetUploadProcessingStatusFields.progress,
+                DatasetUploadProcessingStatusFields.error,
+                DatasetUploadProcessingStatusFields.status,
             )
             result = await self._gql_client.mutation(
                 mutation_field, operation_name="CreateDatasetFromMultipartUpload"

adaptive_sdk/resources/jobs.py

@@ -1,17 +1,18 @@
 from __future__ import annotations
-from typing import TYPE_CHECKING, Literal, Any
+from typing import TYPE_CHECKING, Literal, Any, cast
 
 from adaptive_sdk.graphql_client import (
     CursorPageInput,
     JobInput,
     ListJobsJobs,
     ListJobsFilterInput,
-    JobData,
     JobKind,
+    ListJobsJobsNodes,
 )
 
 
 from .base_resource import SyncAPIResource, AsyncAPIResource, UseCaseResource
+from adaptive_sdk.output_types import JobDataPlus
 
 if TYPE_CHECKING:
     from adaptive_sdk.client import Adaptive, AsyncAdaptive
@@ -26,8 +27,9 @@ class Jobs(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
         SyncAPIResource.__init__(self, client)
         UseCaseResource.__init__(self, client)
 
-    def get(self, job_id: str) -> JobData | None:
-        return self._gql_client.describe_job(id=job_id).job
+    def get(self, job_id: str) -> JobDataPlus | None:
+        job_data = self._gql_client.describe_job(id=job_id).job
+        return JobDataPlus.from_job_data(job_data) if job_data else None
 
     def list(
         self,
@@ -55,7 +57,13 @@ class Jobs(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
             useCase=use_case,
             kind=[JobKind(k) for k in kind] if kind else None,
         )
-        return self._gql_client.list_jobs(page=page, filter=validated_filter, order=[]).jobs
+        jobs = self._gql_client.list_jobs(page=page, filter=validated_filter, order=[]).jobs
+        new_nodes: list[JobDataPlus] = []
+        for i in jobs.nodes:
+            new_nodes.append(JobDataPlus.from_job_data(i))
+        jobs.nodes = cast(list[ListJobsJobsNodes], new_nodes)
+
+        return jobs
 
     def run(
         self,
@@ -65,8 +73,8 @@ class Jobs(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
         name: str | None = None,
         use_case: str | None = None,
         compute_pool: str | None = None,
-    ) -> JobData:
-        return self._gql_client.create_job(
+    ) -> JobDataPlus:
+        job_data = self._gql_client.create_job(
             input=JobInput(
                 recipe=recipe_key,
                 args=args,
@@ -76,9 +84,11 @@ class Jobs(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
                 numGpus=num_gpus,
             )
         ).create_job
+        return JobDataPlus.from_job_data(job_data)
 
-    def cancel(self, job_id: str) -> JobData:
-        return self._gql_client.cancel_job(job_id=job_id).cancel_job
+    def cancel(self, job_id: str) -> JobDataPlus:
+        job_data = self._gql_client.cancel_job(job_id=job_id).cancel_job
+        return JobDataPlus.from_job_data(job_data)
 
 
 class AsyncJobs(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
@@ -90,8 +100,9 @@ class AsyncJobs(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
         AsyncAPIResource.__init__(self, client)
         UseCaseResource.__init__(self, client)
 
-    async def get(self, job_id: str) -> JobData | None:
-        return (await self._gql_client.describe_job(id=job_id)).job
+    async def get(self, job_id: str) -> JobDataPlus | None:
+        job_data = (await self._gql_client.describe_job(id=job_id)).job
+        return JobDataPlus.from_job_data(job_data) if job_data else None
 
     async def list(
         self,
@@ -118,7 +129,12 @@ class AsyncJobs(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
             useCase=self.use_case_key(use_case),
             kind=[JobKind[k] for k in kind] if kind else None,
         )
-        return (await self._gql_client.list_jobs(page=page, filter=validated_filter)).jobs
+        jobs = (await self._gql_client.list_jobs(page=page, filter=validated_filter)).jobs
+        new_nodes: list[JobDataPlus] = []
+        for i in jobs.nodes:
+            new_nodes.append(JobDataPlus.from_job_data(i))
+        jobs.nodes = cast(list[ListJobsJobsNodes], new_nodes)
+        return jobs
 
     async def run(
         self,
@@ -128,8 +144,8 @@ class AsyncJobs(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
         name: str | None = None,
         use_case: str | None = None,
         compute_pool: str | None = None,
-    ) -> JobData:
-        return (
+    ) -> JobDataPlus:
+        job_data = (
             await self._gql_client.create_job(
                 input=JobInput(
                     recipe=recipe_key,
@@ -141,6 +157,8 @@ class AsyncJobs(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
                 )
             )
         ).create_job
+        return JobDataPlus.from_job_data(job_data)
 
-    async def cancel(self, job_id: str) -> JobData:
-        return (await self._gql_client.cancel_job(job_id=job_id)).cancel_job
+    async def cancel(self, job_id: str) -> JobDataPlus:
+        job_data = (await self._gql_client.cancel_job(job_id=job_id)).cancel_job
+        return JobDataPlus.from_job_data(job_data)

{adaptive_sdk-0.1.6.dist-info → adaptive_sdk-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: adaptive-sdk
-Version: 0.1.6
+Version: 0.1.8
 Summary: Python SDK for Adaptive Engine
 Author-email: Vincent Debergue <vincent@adaptive-ml.com>, Joao Moura <joao@adaptive-ml.com>, Yacine Bouraoui <yacine@adaptive-ml.com>
 Requires-Python: >=3.10

{adaptive_sdk-0.1.6.dist-info → adaptive_sdk-0.1.8.dist-info}/RECORD

@@ -1,6 +1,6 @@
 adaptive_sdk/__init__.py,sha256=RQWUHUyEWFSqvTKQpM7Q98nRPnisXRgIZlYd65aJucs,308
-adaptive_sdk/base_client.py,sha256=gHLNOdiPRE53JGCo6jA8D8sTS1WQJ5-UCJOpKMyEhxY,4255
-adaptive_sdk/client.py,sha256=xhAKU_zSSeQoBaUbtwHqHyZgtRQqYut4hNV8cTMoAw4,5077
+adaptive_sdk/base_client.py,sha256=kDeRwwsoue3MqKSXT-k-aUbUdAUb1Abow8oc_AULTnU,4935
+adaptive_sdk/client.py,sha256=p-kMH1fboIF6fpHx3sSyRqmlN3gbx9mxjEkDEfrgs_U,5509
 adaptive_sdk/error_handling.py,sha256=emicEqdEOCgWwCgMkQyvB7VWdKRMnfrWA2urEthQ_g0,3345
 adaptive_sdk/openapi.json,sha256=1zCNC1PsbRnfWy_2TF1MUgF4PDL3pbwYqJbzVTqmm-U,14607
 adaptive_sdk/patch.py,sha256=mF9iw2VnxirxA7_T7TdatpCTwk3EXfXml9elYAvmy7U,223
@@ -9,24 +9,24 @@ adaptive_sdk/external/__init__.py,sha256=-C_MT56OM6xlNhzkwnQ-ke-VV8fJdZIsOzdbNYm
 adaptive_sdk/external/base_model.py,sha256=_WSTDyJdV61Pno63gCXsaL0r9dOahw_JoZ7e7Q6z2QE,295
 adaptive_sdk/external/constants.py,sha256=HaEbIoGG6u0a_tr8_NdNmM-8ApTG1Q0n2G5aGF4Z_nA,280
 adaptive_sdk/external/requests_journal.py,sha256=nJG2CIhl85dkWHibOyvsmcNh_YDVtoyj0PbyLuOYFT4,4686
-adaptive_sdk/external/reward_client.py,sha256=u_kAiASqbOZL7ZIW8xjoSCeZuVEQnWfdeWBNqdSvcM0,6638
+adaptive_sdk/external/reward_client.py,sha256=TSVdi437McUkMModDQV6ZswduaDmmG1Bvkz_jWYTTj4,7324
 adaptive_sdk/external/reward_server.py,sha256=yT8vqAEIoaq8nFZYaku5IoK0_7fX9uX_dfF1CxcDKnA,7488
 adaptive_sdk/external/reward_types.py,sha256=aZmTolT0OjgObo-64zJkarUKOY4RdxHnsQt8AKAaq1w,1710
 adaptive_sdk/external/websocket_utils.py,sha256=QN-K1IRbqe1LdQTz6vXhVgn-R2SjOB80NjGD2w_LaAo,1284
-adaptive_sdk/graphql_client/__init__.py,sha256=0Hiaxqe0xTLNz1UAoDGLC4Y5-SXd6a7wZobuSDzpF-Y,20001
+adaptive_sdk/graphql_client/__init__.py,sha256=mlVK5fr6Sazx2vMUsX73SU1cd2-kAFgg1fKCiliYirk,20995
 adaptive_sdk/graphql_client/add_external_model.py,sha256=9VpQHlJMm5t_Ja_SX2MkYcSG1IQpem6mrdXMfNTa1oo,352
 adaptive_sdk/graphql_client/add_hf_model.py,sha256=aC2IcYftepe28Hi01Kg0w3A7bjKrStWasx6aXiqghJU,312
 adaptive_sdk/graphql_client/add_model.py,sha256=Uyhpxkziz1Pv2S7Q0wOaKGK4krjmEf2PeLK0yEs89Hs,461
 adaptive_sdk/graphql_client/add_remote_env.py,sha256=X94F94tnMzuX9TC_Ieowngw35gJxnKL2YST2sqEJ7CY,328
-adaptive_sdk/graphql_client/async_base_client_open_telemetry.py,sha256=FA5yIdgoq2Geq8fzQaG7UNXCwKXPPGi3bf7xVRhU5t0,19773
-adaptive_sdk/graphql_client/async_client.py,sha256=MEI7X7vLB_-j6v03CtT2uOnO65JnwnDdhGwz0DVvUoo,112309
+adaptive_sdk/graphql_client/async_base_client_open_telemetry.py,sha256=XSRJGEcyfWnFjrDz4Un6xfQWOxr7jWto1vYjbYTQeo0,19761
+adaptive_sdk/graphql_client/async_client.py,sha256=LvX0-wOVVh7lhyQUzbY9jbeEuyjir8RNharyLXpXmi4,115656
 adaptive_sdk/graphql_client/attach_model_to_use_case.py,sha256=WyERM4wxKrUS1u9VZ3FUasZu5AVSbRswzy9jmxssTFM,361
-adaptive_sdk/graphql_client/base_client_open_telemetry.py,sha256=j_JEzPlgRdqyHgjwWqo85KkpfdaPGUgiyjkSqQRHyFk,9410
+adaptive_sdk/graphql_client/base_client_open_telemetry.py,sha256=IV96gRr5FuH-dWMU5PBvQhTwEsV7udsXU-Dyh-Mx-4k,9398
 adaptive_sdk/graphql_client/base_model.py,sha256=2xxKA4sIdlGPIezw06XP9bseSxBURU7nivgt_gL44iA,621
-adaptive_sdk/graphql_client/base_operation.py,sha256=2DuKKDMdshl-t6mgvN6RYjZMdZ3sVzl5IlZw7xThSn4,4641
+adaptive_sdk/graphql_client/base_operation.py,sha256=aooq1M4r79cvMoganZ2RvQ3-v0io22fGLOVfl3UBqPQ,4635
 adaptive_sdk/graphql_client/cancel_ab_campaign.py,sha256=JAEpmedz0jOu90U3KR0PYCMAhf2_E6h6WOT30HSE91k,176
 adaptive_sdk/graphql_client/cancel_job.py,sha256=3ZLoqrULi1mp5A5i4rD-gXliKhD8038IPfvCNBg0dPs,291
-adaptive_sdk/graphql_client/client.py,sha256=j4w8k6hZ4YH9A8HA0AJcpQMiV0j4mZ3hUwKVwj1zYtk,111400
+adaptive_sdk/graphql_client/client.py,sha256=P9BgiDKnfU5GOTAQ0MoZpnqLWPzxWnvfvlIzA8OKVV4,114747
 adaptive_sdk/graphql_client/create_ab_campaign.py,sha256=___2iFSGnWKkjXz_MfxKUxi1EoQcSBv9AX8S7YoUeVw,374
 adaptive_sdk/graphql_client/create_custom_recipe.py,sha256=eIVLDHbdFG2qWjoZBLC3Xs40Fjy6L-u6wrABV5ibUxo,382
 adaptive_sdk/graphql_client/create_grader.py,sha256=H6xdf7Zn7JkOehINWMkULH3ObASrtvPLKaX-BjW-3zw,321
@@ -38,10 +38,10 @@ adaptive_sdk/graphql_client/create_role.py,sha256=6aTdNOZxavMyjkH-g01uYOZgpjYWcA
 adaptive_sdk/graphql_client/create_team.py,sha256=6Alt1ralE1-Xvp2wrEaLUHMW5RtiFqz2fIsUYE_2LbM,370
 adaptive_sdk/graphql_client/create_use_case.py,sha256=sekD76jWCo3zNCfMsBGhVYfNSIK4JPPBz9066BOt49g,332
 adaptive_sdk/graphql_client/create_user.py,sha256=gurD0kZgncXt1HBr7Oo5AkK5ubqFKpJvaR1rn506gHo,301
-adaptive_sdk/graphql_client/custom_fields.py,sha256=XpyLzICGJQWahfIhZtLHpSH8GgU-4XVYB286SbIq3qg,95761
-adaptive_sdk/graphql_client/custom_mutations.py,sha256=Hefsjk0UqSRYPrWg7tLNExV7wa0qQSR-nNZAFNndrU0,24945
-adaptive_sdk/graphql_client/custom_queries.py,sha256=E3j7r3qrA37qPOoTDYSNOHVj2gOWYvfXSE7FAozmr84,16850
-adaptive_sdk/graphql_client/custom_typing_fields.py,sha256=cU1PgxbzDQiM1lBJyB4C1IzNirxJS_NJRbHqi_PLM50,18935
+adaptive_sdk/graphql_client/custom_fields.py,sha256=T5QG7zjX6URKy1lLkmbfjCNCkTU3pbqHn_3MzuYE3AA,98874
+adaptive_sdk/graphql_client/custom_mutations.py,sha256=meo96-odIhk_NoFL_uDj78xguLOD-_DIYW2bj6ilvII,25341
+adaptive_sdk/graphql_client/custom_queries.py,sha256=QsFX6ovy1W9nUuCD0-wo9GhiJACl9yQGCifHnKIs6Bo,17359
+adaptive_sdk/graphql_client/custom_typing_fields.py,sha256=aS7Oj2z-oLcIqMZrVNNM9wVxXQK6rYSTF1GaXAnFVr4,20608
 adaptive_sdk/graphql_client/delete_custom_recipe.py,sha256=tsWGqherJyVe5OBFG8QCG5GUdlX8s-nd8MSiMAXfSqk,183
 adaptive_sdk/graphql_client/delete_dataset.py,sha256=k5enUd6zO89RmA349eVzYajtZigWjOdDdPTcUzTFahI,167
 adaptive_sdk/graphql_client/delete_grader.py,sha256=U9r26BtvOaThzyf0VGouvkuEaJ1wJGPGjbHluSDWBsc,350
@@ -57,13 +57,13 @@ adaptive_sdk/graphql_client/describe_metric_admin.py,sha256=_SKKwnFhZnbOTT97elEr
 adaptive_sdk/graphql_client/describe_model.py,sha256=UnsOnAyBjNsnkJaS4q5uwkSSvInHwRqUj3XqAoO0yO4,434
 adaptive_sdk/graphql_client/describe_model_admin.py,sha256=XUt_CBSMw1HRleUEWZn2snYt2BNSux_siqrVlwtqH-w,484
 adaptive_sdk/graphql_client/describe_use_case.py,sha256=WW0QkTmdfggN8YBUNGi8ShrP_fr2jXPR6Fer6jlQxu0,353
-adaptive_sdk/graphql_client/enums.py,sha256=lEBc54SWx__kz20yhh8CpAWAL_-BaB4ez1iQU53vYW4,4794
+adaptive_sdk/graphql_client/enums.py,sha256=NykoTVLryXuy6pbHDRLfAN3m7EUi8B2Evbgp41Em7sQ,5168
 adaptive_sdk/graphql_client/exceptions.py,sha256=NiC6v-5S7aRDlvQTcHH3K5KvxWvk-c-PkIQQHkipTB8,2268
-adaptive_sdk/graphql_client/fragments.py,sha256=zkGLGnbMdoc9vO5PJL-iDnMtIKetNx-8vRbSl16ExRk,20431
+adaptive_sdk/graphql_client/fragments.py,sha256=_F3IgC3lJN4erM5Ory1bnvnLbzjZ9Z1Xfv4jdw8dJEo,22046
 adaptive_sdk/graphql_client/get_custom_recipe.py,sha256=7qxBZGQTqpc69k-NwzgFctaHWaRz0tHl7YlVSsEad6U,383
 adaptive_sdk/graphql_client/get_grader.py,sha256=kubHDBtUcq6mZtUR5_Of0QbjnGUPSYuavF3_xwmwbY8,233
 adaptive_sdk/graphql_client/get_judge.py,sha256=urEnHW3XfURi5GAFBPfbqzOZGQDxgsGRA6nZmUKmoMA,224
-adaptive_sdk/graphql_client/input_types.py,sha256=IjFiu3NZeDpcMXcLbXgEgpo-wpOydKyJ5gWQqMnRmQQ,19822
+adaptive_sdk/graphql_client/input_types.py,sha256=FLa_O8WmkLbTUma7BEaLg5Epe39Ap5cnGBhc6Y_jAuw,20411
 adaptive_sdk/graphql_client/link_metric.py,sha256=EDH67ckBzzc6MYIGfsmgZRBnjqxLsCGwFUaFMXPEsBY,327
 adaptive_sdk/graphql_client/list_ab_campaigns.py,sha256=SIbU6I2OQkNHt0Gw6YStoiiwJHUk2rfXnpoGLzrFjxc,379
 adaptive_sdk/graphql_client/list_compute_pools.py,sha256=4Qli5Foxm3jztbUAL5gbwqtcrElwwlC4LGJMOMBI6Cc,782
@@ -101,17 +101,19 @@ adaptive_sdk/graphql_client/update_model_compute_config.py,sha256=0RtrzzfuMrt-i6
 adaptive_sdk/graphql_client/update_user.py,sha256=E6vtj177-Hx1uUUo8MiSrynULXrnxtJma3Kq69FPC6M,897
 adaptive_sdk/input_types/__init__.py,sha256=SVy4Ks4pm1M32LZ_yHN2itlzNW0g2Kg3b3_ibNLI-7s,806
 adaptive_sdk/input_types/typed_dicts.py,sha256=0HWbJ6nY4zrNCsMg6MynkAO2VUr_7Iu9u9F1Y-71gWE,4942
+adaptive_sdk/output_types/__init__.py,sha256=gIfALHcp-hOt8J7bFO7KAYbJc0JXtp7d_vVBqD0zPLE,136
+adaptive_sdk/output_types/job_wrapper.py,sha256=dXdS33TKLHuTRAVq5LFvA7bsUidsWuhE-NDbq5k59g0,4247
 adaptive_sdk/resources/__init__.py,sha256=-cHNzI-hg1u-GGe-fVvV_L4JOgat2zJuKnC3E2In-iw,1326
 adaptive_sdk/resources/abtests.py,sha256=9PCPjxuWwY9ec88ewnq54gkoELq5U5iaBmHhzLCAsFU,7698
 adaptive_sdk/resources/base_resource.py,sha256=D9adWSFxiDB7chVstDuBu1jcuXkE71UQJexnWENpC4A,1497
 adaptive_sdk/resources/chat.py,sha256=k-jh_IZfKwqoWNN2P5BhAM3P_f5HLybVZp35pUadwso,11954
 adaptive_sdk/resources/compute_pools.py,sha256=4eHP8FMkZOsGPjZ-qBvda2PunA6GMyvvJTHnhAEGTo4,938
-adaptive_sdk/resources/datasets.py,sha256=G7aq9aJmNu-ZPB3d2_bDaIo_Na3mjw2Bqn6SMweMRWY,14258
+adaptive_sdk/resources/datasets.py,sha256=TjYkU4eKY-Si5ffQhMkX3sIEj5Ybz1U3ksOJs4Fa6BA,14622
 adaptive_sdk/resources/embeddings.py,sha256=-ov_EChHU6PJJOJRtDlCo4sYyr9hwyvRjnBhub8QNFg,3922
 adaptive_sdk/resources/feedback.py,sha256=lujqwFIhxi6iovL8JWL05Kr-gkzR4QEwUXZbTx33raA,14116
 adaptive_sdk/resources/graders.py,sha256=ekQQ5fqmLZpZHeLr6iUm6m45wDevoDJdj3mG-axR-m8,29014
 adaptive_sdk/resources/interactions.py,sha256=9A0aKyfE5dhMj-rj6NOiF7kxAl89SXksFsRJXXjPGK8,10810
-adaptive_sdk/resources/jobs.py,sha256=WpgB7S5EFKFt3ktVfGaRj_MrtC5Wh_VZHtNjSgDRvOo,4429
+adaptive_sdk/resources/jobs.py,sha256=dN19HgOvGEq9O5hhgDZlWhi15noXo4qHLT_fd-qjrAU,5318
 adaptive_sdk/resources/models.py,sha256=krQbfMnVkjNqXfPG-8irH_xlloDpFpQiqYsbED3-8z8,18591
 adaptive_sdk/resources/permissions.py,sha256=ckO-oacWkvgcwXBK5iW_8qaK-g0SHXpEEy1qZy5lrB0,1053
 adaptive_sdk/resources/recipes.py,sha256=f4Q_noEAjx-LGkgzLOU6UtgFTK75cQu0cYBakNjUCv0,13224
@@ -122,6 +124,6 @@ adaptive_sdk/resources/users.py,sha256=SoGWwdDCdhK4KjYOcAws-ZWlW7Edii7D3Vxfdu-NZ
 adaptive_sdk/rest/__init__.py,sha256=P9uhkOoc9cgUkJ5MBoV5soPgQWSkvPrTwHzPGX7i5tY,610
 adaptive_sdk/rest/base_model.py,sha256=P06TNhnqXa6JEje_B_94vAa5zqPYIVxMZAp6aZ4d80U,516
 adaptive_sdk/rest/rest_types.py,sha256=aSN7901_1goByBEl7Ka6onU2ia-_RhlmtH9suFkWkic,8859
-adaptive_sdk-0.1.6.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-adaptive_sdk-0.1.6.dist-info/METADATA,sha256=bpycg-FYRxcPfaEUiZADXeW24AWtYKqCZvtCWmxNJdo,1446
-adaptive_sdk-0.1.6.dist-info/RECORD,,
+adaptive_sdk-0.1.8.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+adaptive_sdk-0.1.8.dist-info/METADATA,sha256=stv3nHM_GV-XMi8mtWvzxBw9Of8_fMLFU71wmBoTB-E,1446
+adaptive_sdk-0.1.8.dist-info/RECORD,,