adaptive-sdk 0.1.13__py3-none-any.whl → 0.1.14__py3-none-any.whl

adaptive_sdk/resources/interactions.py

@@ -1,26 +1,27 @@
 from __future__ import annotations
-from copy import deepcopy
-import humps
-from typing import Dict, List, Literal, TYPE_CHECKING
+
+from typing import TYPE_CHECKING, Any, Dict, List, Literal
 from uuid import UUID
 
+import humps
+from typing_extensions import override
+
 from adaptive_sdk import input_types
 from adaptive_sdk.error_handling import rest_error_handler
 from adaptive_sdk.graphql_client import (
+    CompletionData,
     CompletionGroupBy,
     CursorPageInput,
     ListCompletionsFilterInput,
-    OrderPair,
-    ListInteractionsCompletions,
     ListGroupedInteractionsCompletionsGrouped,
-    CompletionData,
+    ListInteractionsCompletions,
+    OrderPair,
 )
 from adaptive_sdk.graphql_client.base_model import UNSET
 from adaptive_sdk.rest import rest_types
 from adaptive_sdk.utils import convert_optional_UUID
-from typing_extensions import override
 
-from .base_resource import SyncAPIResource, AsyncAPIResource, UseCaseResource
+from .base_resource import AsyncAPIResource, SyncAPIResource, UseCaseResource
 
 if TYPE_CHECKING:
     from adaptive_sdk.client import Adaptive, AsyncAdaptive
@@ -32,9 +33,7 @@ def _prepare_add_interactions_inputs(
     messages: list[input_types.ChatMessage],
     feedbacks: list[input_types.InteractionFeedbackDict] | None,
 ):
-    input_messages = (
-        [rest_types.ChatMessage(role=m["role"], content=m["content"]) for m in messages]
-    )
+    input_messages = [rest_types.ChatMessage(role=m["role"], content=m["content"]) for m in messages]
     input_feedbacks = (
         [
             rest_types.InteractionFeedback(
@@ -123,20 +122,18 @@ class Interactions(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
             group_by: Retrieve interactions grouped by selected dimension.
 
         """
-        new_filters = {} if filters is None else deepcopy(filters)
         order = [] if order is None else order
         new_page = {} if page is None else page
 
-        new_filters = humps.camelize(new_filters)
+        filter_list = _build_filter_list(filters)
         new_order = humps.camelize(order)
         new_page = humps.camelize(new_page)
 
-        if new_filters.get("timerange"):
-            new_filters["timerange"]["from"] = new_filters["timerange"]["from_"]  # type: ignore
-            del new_filters["timerange"]["from_"]  # type: ignore
-
-        new_filters.update({"useCase": self.use_case_key(use_case)})  # type: ignore
         order_inputs = [OrderPair.model_validate(o) for o in new_order] if new_order else UNSET
+        new_filters = {
+            "useCase": self.use_case_key(use_case),
+            "advancedFilters": {"and": [filter_list]},
+        }
         if group_by:
             return self._gql_client.list_grouped_interactions(
                 filter=ListCompletionsFilterInput.model_validate(new_filters),
@@ -234,20 +231,18 @@ class AsyncInteractions(AsyncAPIResource, UseCaseResource):  # type: ignore[misc
             group_by: Retrieve interactions grouped by selected dimension.
 
         """
-        new_filters = {} if filters is None else deepcopy(filters)
         order = [] if order is None else order
         new_page = {} if page is None else page
 
-        new_filters = humps.camelize(new_filters)
+        filter_list = _build_filter_list(filters)
         new_order = humps.camelize(order)
         new_page = humps.camelize(new_page)
 
-        if new_filters.get("timerange"):
-            new_filters["timerange"]["from"] = new_filters["timerange"]["from_"]  # type: ignore
-            del new_filters["timerange"]["from_"]  # type: ignore
-
-        new_filters.update({"useCase": self.use_case_key(use_case)})  # type: ignore
         order_inputs = [OrderPair.model_validate(o) for o in new_order] if new_order else UNSET
+        new_filters = {
+            "useCase": self.use_case_key(use_case),
+            "advancedFilters": {"and": [filter_list]},
+        }
         if group_by:
             result = await self._gql_client.list_grouped_interactions(
                 filter=ListCompletionsFilterInput.model_validate(new_filters),
@@ -277,3 +272,40 @@ class AsyncInteractions(AsyncAPIResource, UseCaseResource):  # type: ignore[misc
         """
         result = await self._gql_client.describe_interaction(use_case=self.use_case_key(use_case), id=completion_id)
         return result.completion
+
+
+def _build_filter_list(
+    filters: input_types.ListCompletionsFilterInput | None = None,
+) -> List[Any]:
+    filter_list: List[Any] = []
+    if filters is not None:
+        if "models" in filters:
+            filter_list.append({"model": {"in": filters["models"]}})
+        if "timerange" in filters:
+            filter_list.append(
+                {
+                    "timerange": {
+                        "from": filters["timerange"]["from_"],
+                        "to": filters["timerange"]["to"],
+                    }
+                }
+            )
+        if "session_id" in filters:
+            filter_list.append({"sessionId": filters["session_id"]})
+        if "completion_id" in filters:
+            filter_list.append({"completionId": filters["completion_id"]})
+        if "user_id" in filters:
+            filter_list.append({"userId": filters["user_id"]})
+        if "feedbacks" in filters:
+            filter_list.append({"feedbacks": filters["feedbacks"]})
+        if "comparisons" in filters:
+            filter_list.append({"comparisons": filters["comparisons"]})
+        if "labels" in filters:
+            for label_cond in filters["labels"]:
+                filter_list.append({"labels": label_cond})
+        if "prompt_hash" in filters:
+            filter_list.append({"promptHash": {"eq": filters["prompt_hash"]}})
+        if "source" in filters:
+            for source in filters["source"]:
+                filter_list.append({"source": source})
+    return filter_list

adaptive_sdk/resources/models.py

@@ -1,27 +1,31 @@
 from __future__ import annotations
-from typing import get_args, Callable, Literal, Sequence, TYPE_CHECKING
 
+from typing import TYPE_CHECKING, Callable, Literal, Sequence, get_args
+
+from adaptive_sdk import input_types
 from adaptive_sdk.graphql_client import (
-    OpenAIModel,
-    OpenAIProviderDataInput,
-    GoogleProviderDataInput,
-    ModelProviderDataInput,
     AddExternalModelInput,
+    AddHFModelInput,
     ExternalModelProviderName,
-    AttachModel,
-    UpdateModelService,
-    ModelData,
+    GoogleProviderDataInput,
     JobData,
-    ModelServiceData,
     ListModelsModels,
-    AddHFModelInput,
+    ModelComputeConfigInput,
+    ModelData,
     ModelFilter,
     ModelPlacementInput,
-    ModelComputeConfigInput,
+    ModelProviderDataInput,
+    ModelServiceData,
+    OpenAIProviderDataInput,
+    UpdateModelService,
+)
+from adaptive_sdk.graphql_client.input_types import (
+    AddModelToUseCaseInput,
+    DeployModelInput,
+    RemoveModelFromUseCaseInput,
 )
-from adaptive_sdk import input_types
 
-from .base_resource import SyncAPIResource, AsyncAPIResource, UseCaseResource
+from .base_resource import AsyncAPIResource, SyncAPIResource, UseCaseResource
 
 if TYPE_CHECKING:
     from adaptive_sdk.client import Adaptive, AsyncAdaptive
@@ -29,9 +33,7 @@ if TYPE_CHECKING:
 provider_config = {
     "open_ai": {
         "provider_data": lambda api_key, model_id: ModelProviderDataInput(
-            openAI=OpenAIProviderDataInput(
-                apiKey=api_key, externalModelId=OpenAIModel(model_id)
-            )
+            openAI=OpenAIProviderDataInput(apiKey=api_key, externalModelId=model_id)
         ),
     },
     "google": {
@@ -88,9 +90,7 @@ def is_supported_model(model_id: str):
     supported_models = get_args(SupportedHFModels)
     if model_id not in supported_models:
         supported_models_str = "\n".join(supported_models)
-        raise ValueError(
-            f"Model {model_id} is not supported.\n\nChoose from:\n{supported_models_str}"
-        )
+        raise ValueError(f"Model {model_id} is not supported.\n\nChoose from:\n{supported_models_str}")
 
 
 class Models(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
@@ -155,22 +155,16 @@ class Models(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
                 provider_data = provider_data_fn(api_key, external_model_id)
             case "azure":
                 if not endpoint:
-                    raise ValueError(
-                        "`endpoint` is required to connect Azure external model."
-                    )
+                    raise ValueError("`endpoint` is required to connect Azure external model.")
                 provider_data = provider_data_fn(api_key, external_model_id, endpoint)
             case _:
                 raise ValueError(f"Provider {provider} is not supported")
 
         provider_enum = ExternalModelProviderName(provider.upper())
-        input = AddExternalModelInput(
-            name=name, provider=provider_enum, providerData=provider_data
-        )
+        input = AddExternalModelInput(name=name, provider=provider_enum, providerData=provider_data)
         return self._gql_client.add_external_model(input).add_external_model
 
-    def list(
-        self, filter: input_types.ModelFilter | None = None
-    ) -> Sequence[ListModelsModels]:
+    def list(self, filter: input_types.ModelFilter | None = None) -> Sequence[ListModelsModels]:
         """
         List all models in Adaptive model registry.
         """
@@ -203,32 +197,79 @@ class Models(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
                 If `True`, this call blocks until model is `Online`.
             make_default: Make the model the use case's default on attachment.
         """
-        input = AttachModel(
+
+        input = AddModelToUseCaseInput(
+            model=model,
+            useCase=self.use_case_key(use_case),
+        )
+        self._gql_client.add_model_to_use_case(input)
+        input = DeployModelInput(
             model=model,
             useCase=self.use_case_key(use_case),
-            attached=True,
+            placement=(ModelPlacementInput.model_validate(placement) if placement else None),
             wait=wait,
-            placement=(
-                ModelPlacementInput.model_validate(placement) if placement else None
-            ),
         )
-        result = self._gql_client.attach_model_to_use_case(input).attach_model
+        result: ModelServiceData = self._gql_client.deploy_model(input).deploy_model
         if make_default:
-            result = self.update(model=model, is_default=make_default)  # type: ignore[assignment]
+            result = self.update(model=model, is_default=make_default)
         return result
 
-    def detach(
+    def add_to_use_case(
+        self,
+        model: str,
+        use_case: str | None = None,
+    ) -> bool:
+        """
+        Attach a model to the client's use case.
+
+        Args:
+            model: Model key.
+            wait: If the model is not deployed already, attaching it to the use case will automatically deploy it.
+                If `True`, this call blocks until model is `Online`.
+            make_default: Make the model the use case's default on attachment.
+        """
+
+        input = AddModelToUseCaseInput(
+            model=model,
+            useCase=self.use_case_key(use_case),
+        )
+        return self._gql_client.add_model_to_use_case(input).add_model_to_use_case
+
+    def deploy(
         self,
         model: str,
+        wait: bool = False,
+        make_default: bool = False,
         use_case: str | None = None,
+        placement: input_types.ModelPlacementInput | None = None,
     ) -> ModelServiceData:
+        input = DeployModelInput(
+            model=model,
+            useCase=self.use_case_key(use_case),
+            placement=(ModelPlacementInput.model_validate(placement) if placement else None),
+            wait=wait,
+        )
+        result: ModelServiceData = self._gql_client.deploy_model(input).deploy_model
+        if make_default:
+            result = self.update(model=model, is_default=make_default)
+        return result
+
+    def detach(
+        self,
+        model: str,
+        use_case: str,
+    ) -> bool:
         """
         Detach model from client's use case.
 
         Args:
             model: Model key.
         """
-        return self.update(model=model, attached=False, use_case=use_case)
+        input = RemoveModelFromUseCaseInput(
+            model=model,
+            useCase=use_case,
+        )
+        return self._gql_client.remove_model_from_use_case(input).remove_model_from_use_case
 
     def update_compute_config(
         self,
@@ -247,7 +288,6 @@ class Models(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
         self,
         model: str,
         is_default: bool | None = None,
-        attached: bool | None = None,
         desired_online: bool | None = None,
         use_case: str | None = None,
         placement: input_types.ModelPlacementInput | None = None,
@@ -269,11 +309,8 @@ class Models(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
             useCase=self.use_case_key(use_case),
             modelService=model,
             isDefault=is_default,
-            attached=attached,
             desiredOnline=desired_online,
-            placement=(
-                ModelPlacementInput.model_validate(placement) if placement else None
-            ),
+            placement=(ModelPlacementInput.model_validate(placement) if placement else None),
         )
         return self._gql_client.update_model(input).update_model_service
 
@@ -286,9 +323,7 @@ class Models(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
             force: If model is attached to several use cases, `force` must equal `True` in order
                 for the model to be terminated.
         """
-        return self._gql_client.terminate_model(
-            id_or_key=model, force=force
-        ).terminate_model
+        return self._gql_client.terminate_model(id_or_key=model, force=force).terminate_model
 
 
 class AsyncModels(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
@@ -353,23 +388,17 @@ class AsyncModels(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
                 provider_data = provider_data_fn(api_key, external_model_id)
             case "azure":
                 if not endpoint:
-                    raise ValueError(
-                        "`endpoint` is required to connect Azure external model."
-                    )
+                    raise ValueError("`endpoint` is required to connect Azure external model.")
                 provider_data = provider_data_fn(api_key, external_model_id, endpoint)
             case _:
                 raise ValueError(f"Provider {provider} is not supported")
 
         provider_enum = ExternalModelProviderName(provider.upper())
-        input = AddExternalModelInput(
-            name=name, provider=provider_enum, providerData=provider_data
-        )
+        input = AddExternalModelInput(name=name, provider=provider_enum, providerData=provider_data)
         result = await self._gql_client.add_external_model(input)
         return result.add_external_model
 
-    async def list(
-        self, filter: input_types.ModelFilter | None = None
-    ) -> Sequence[ListModelsModels]:
+    async def list(self, filter: input_types.ModelFilter | None = None) -> Sequence[ListModelsModels]:
         """
         List all models in Adaptive model registry.
         """
@@ -385,38 +414,6 @@ class AsyncModels(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
         """
         return (await self._gql_client.describe_model(input=model)).model
 
-    async def attach(
-        self,
-        model: str,
-        wait: bool = True,
-        make_default: bool = False,
-        use_case: str | None = None,
-        placement: input_types.ModelPlacementInput | None = None,
-    ) -> ModelServiceData:
-        """
-        Attach a model to the client's use case.
-
-        Args:
-            model: Model key.
-            wait: If the model is not deployed already, attaching it to the use case will automatically deploy it.
-                If `True`, this call blocks until model is `Online`.
-            make_default: Make the model the use case's default on attachment.
-        """
-        input = AttachModel(
-            model=model,
-            useCase=self.use_case_key(use_case),
-            attached=True,
-            wait=wait,
-            placement=(
-                ModelPlacementInput.model_validate(placement) if placement else None
-            ),
-        )
-        result = await self._gql_client.attach_model_to_use_case(input)
-        result = result.attach_model
-        if make_default:
-            result = await self.update(model=model, is_default=make_default)  # type: ignore[assignment]
-        return result
-
     async def detach(
         self,
         model: str,
@@ -428,7 +425,7 @@ class AsyncModels(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
         Args:
             model: Model key.
         """
-        return await self.update(model=model, attached=False, use_case=use_case)
+        return await self.update(model=model, use_case=use_case)
 
     async def update_compute_config(
         self,
@@ -449,7 +446,6 @@ class AsyncModels(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
         self,
         model: str,
         is_default: bool | None = None,
-        attached: bool | None = None,
         desired_online: bool | None = None,
         use_case: str | None = None,
         placement: input_types.ModelPlacementInput | None = None,
@@ -471,11 +467,8 @@ class AsyncModels(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
             useCase=self.use_case_key(use_case),
             modelService=model,
             isDefault=is_default,
-            attached=attached,
             desiredOnline=desired_online,
-            placement=(
-                ModelPlacementInput.model_validate(placement) if placement else None
-            ),
+            placement=(ModelPlacementInput.model_validate(placement) if placement else None),
         )
         result = await self._gql_client.update_model(input)
         return result.update_model_service
@@ -489,6 +482,4 @@ class AsyncModels(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
             force: If model is attached to several use cases, `force` must equal `True` in order
                 for the model to be terminated.
         """
-        return (
-            await self._gql_client.terminate_model(id_or_key=model, force=force)
-        ).terminate_model
+        return (await self._gql_client.terminate_model(id_or_key=model, force=force)).terminate_model

adaptive_sdk/rest/rest_types.py

@@ -256,4 +256,5 @@ class ChatInput(GenerateParameters):
     system_prompt_args: Annotated[Optional[Dict[str, Any]], Field(description='Will be used to render system prompt template')] = None
     tags: Optional[List[str]] = None
     use_tools: Optional[bool] = None
-    tools: Annotated[Optional[List[ToolOverride]], Field(description='Override tool configuration for this request - enables/disables specific tools')] = None
+    tools: Annotated[Optional[List[ToolOverride]], Field(description='Override tool configuration for this request - enables/disables specific tools')] = None
+    store: Optional[bool] = None

{adaptive_sdk-0.1.13.dist-info → adaptive_sdk-0.1.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: adaptive-sdk
-Version: 0.1.13
+Version: 0.1.14
 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
@@ -27,8 +27,11 @@ Requires-Dist: toml>=0.10.2 ; extra == "dev"
 Requires-Dist: black==25.1.0 ; extra == "dev"
 Requires-Dist: pytest==8.4.1 ; extra == "dev"
 Requires-Dist: pytest-asyncio==0.26.0 ; extra == "dev"
+Requires-Dist: pytest-dependency==0.6.0 ; extra == "dev"
 Requires-Dist: mypy==1.16.0 ; extra == "dev"
 Requires-Dist: tenacity==9.1.2 ; extra == "dev"
+Requires-Dist: pyright >= 1.1 ; extra == "dev"
+Requires-Dist: adaptive-harmony ; extra == "dev"
 Provides-Extra: dev
 
 # adaptive-sdk

{adaptive_sdk-0.1.13.dist-info → adaptive_sdk-0.1.14.dist-info}/RECORD

@@ -1,6 +1,6 @@
 adaptive_sdk/__init__.py,sha256=RQWUHUyEWFSqvTKQpM7Q98nRPnisXRgIZlYd65aJucs,308
 adaptive_sdk/base_client.py,sha256=kDeRwwsoue3MqKSXT-k-aUbUdAUb1Abow8oc_AULTnU,4935
-adaptive_sdk/client.py,sha256=p-kMH1fboIF6fpHx3sSyRqmlN3gbx9mxjEkDEfrgs_U,5509
+adaptive_sdk/client.py,sha256=tTQ793NF_hjOf5yZiVVFLEm5f5RAQtbjcG-W8d0zifU,5663
 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
@@ -13,20 +13,20 @@ adaptive_sdk/external/reward_client.py,sha256=TSVdi437McUkMModDQV6ZswduaDmmG1Bvk
 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=XWJK6FD8OKrmFuNn1RC3ofVC5xqivMLSULP0rEamURM,21523
+adaptive_sdk/graphql_client/__init__.py,sha256=2T0Zv8HkVhQ0EwawugXeoqWrvcwfbbAqSJ8Zy47hjJM,22094
 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_model_to_use_case.py,sha256=PQ6_hdCiiCRbbW9ylbfzjr5aAMgHXG16Fdtm_UjvJiE,182
 adaptive_sdk/graphql_client/add_remote_env.py,sha256=X94F94tnMzuX9TC_Ieowngw35gJxnKL2YST2sqEJ7CY,328
 adaptive_sdk/graphql_client/async_base_client_open_telemetry.py,sha256=XSRJGEcyfWnFjrDz4Un6xfQWOxr7jWto1vYjbYTQeo0,19761
-adaptive_sdk/graphql_client/async_client.py,sha256=Q2nOtA-t6lOgQ-S3jC7cQYlFArDiZAPHMcL8gMBX7iU,118289
-adaptive_sdk/graphql_client/attach_model_to_use_case.py,sha256=WyERM4wxKrUS1u9VZ3FUasZu5AVSbRswzy9jmxssTFM,361
+adaptive_sdk/graphql_client/async_client.py,sha256=RhNbCIsBPl8te-7c2QfkY4VMCBpLP2tU1DLbViSSyh8,119538
 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=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=6EFDa1fUpX1wwdvXgm929xE1pcNMEClhjQ19PC-4i6k,117356
+adaptive_sdk/graphql_client/client.py,sha256=-eJ4ok5_N4CkNGVXtlntvGzXzGNAVQC2x65DqPgF90E,118581
 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_dataset_from_multipart_upload.py,sha256=eoqOfEviCFww5ElTknknV0qOpFTJQmQ1BeGLsRQ5iHc,730
@@ -39,8 +39,8 @@ 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=s77Stl8__yrPDAimCp2zoKefZdT6YxAwwN5atsfMw5Y,99937
-adaptive_sdk/graphql_client/custom_mutations.py,sha256=htvXPIHOpHhWgdoHQNB29IhgdkaRH9yQSMfyth-1umQ,24779
+adaptive_sdk/graphql_client/custom_fields.py,sha256=lRdkGYIa2gsbAf4_akOSfqHOFgwpIx67GeIcDYnWHYY,100895
+adaptive_sdk/graphql_client/custom_mutations.py,sha256=cKtpF9qmuwj8QqSesUtW58tW5TmDqnLFpjOjRMxPb3U,26335
 adaptive_sdk/graphql_client/custom_queries.py,sha256=dVMj82YFJg9wjcnZdmnCbEKEaLiL8QUgpCjr5ONuQAU,18242
 adaptive_sdk/graphql_client/custom_typing_fields.py,sha256=yT_a4kZOMIccqIm-sAG0zu70c-plQtQrZ6ynmfxjlDU,20786
 adaptive_sdk/graphql_client/dataset_upload_processing_status.py,sha256=Xwj9bxmRf0RVxMT5kf30yX0vQaCYEuTI5-alCiqedqI,705
@@ -49,6 +49,7 @@ adaptive_sdk/graphql_client/delete_dataset.py,sha256=k5enUd6zO89RmA349eVzYajtZig
 adaptive_sdk/graphql_client/delete_grader.py,sha256=U9r26BtvOaThzyf0VGouvkuEaJ1wJGPGjbHluSDWBsc,350
 adaptive_sdk/graphql_client/delete_judge.py,sha256=rdiA12Q3Q0JEZTOBjTK3Tz33f2aurj5G9vqSoJUjw3I,342
 adaptive_sdk/graphql_client/delete_user.py,sha256=zMlvEaxGQAfDn-vMc0LWuhgzv1DJb69DuFwJcKGdF5o,301
+adaptive_sdk/graphql_client/deploy_model.py,sha256=UI6__BkSWUjQWusPRcukl-6te0dv2CFKt73QqfC3mz8,325
 adaptive_sdk/graphql_client/describe_ab_campaign.py,sha256=1-z2U5aPHx3EBk0JawW1MO0n1hUpuFNk_oEgzn_6hhQ,594
 adaptive_sdk/graphql_client/describe_dataset.py,sha256=3E4vhnIsTXAjBz51TpfNhw8ldItoePumGn5hw5B3EkM,300
 adaptive_sdk/graphql_client/describe_interaction.py,sha256=1gJ2b7ssrrcOujEKzzj7oH18h--V3DSk8HL84hlwuUc,331
@@ -58,13 +59,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=dneXg7eIX_UzVNfdImhs_Z1JPulePP-zl_XqKk7Q1Qw,5257
+adaptive_sdk/graphql_client/enums.py,sha256=tIHf3CU79ldu_UOq9hu0P1bLCvFmtHyafUnZvISXdFo,4902
 adaptive_sdk/graphql_client/exceptions.py,sha256=NiC6v-5S7aRDlvQTcHH3K5KvxWvk-c-PkIQQHkipTB8,2268
-adaptive_sdk/graphql_client/fragments.py,sha256=NMata4RWlGeR5rYJouo-PqRFGB2tN_9VBh3bBW6K8w4,22485
+adaptive_sdk/graphql_client/fragments.py,sha256=mZgJhxudB3x4u7vs5RnCp3rycUhEuP09-A91N-xClbk,22515
 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=BLT_xkluyu6O8w0wZGBgpZjgPsA_II02WAnk63UNpAA,20215
+adaptive_sdk/graphql_client/input_types.py,sha256=6bFD-TnDFWFzaEVKNpRVTZKr782RRmmPCnW8SG8Kg-E,25302
 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
@@ -88,6 +89,7 @@ adaptive_sdk/graphql_client/list_users.py,sha256=9LCNz49jqxrUapHyOdnzs-ZtU1xsejq
 adaptive_sdk/graphql_client/load_dataset.py,sha256=P_h3wPTT4E5mbgJoR5jX1_5GhDLIJsA6lmYPT27VxY8,323
 adaptive_sdk/graphql_client/lock_grader.py,sha256=cCISIKjFaSakf-tr4oEWebieT5yrpvGxdS_R_E4iww8,305
 adaptive_sdk/graphql_client/me.py,sha256=oR_m5QRalokWb9lZYSNgDzOBcNLuAgBIXZh7zfBerqw,227
+adaptive_sdk/graphql_client/remove_model_from_use_case.py,sha256=izNvNE0JDc54r9Nsqk5sWa1ez3rEywZ7XkUs822_h4k,197
 adaptive_sdk/graphql_client/remove_remote_env.py,sha256=lmEwudWc6bkU1ev2g9_QEOgkJT_n7n9P5wGm5AV2MY8,173
 adaptive_sdk/graphql_client/remove_team_member.py,sha256=H75WLcB4bM0yctfmmzztxSAzxvbOwWmkVZhO5DUuh1g,350
 adaptive_sdk/graphql_client/resize_inference_partition.py,sha256=osrAGQGYCFjAyfkftkxYW4Wv2VzMq_oW8i0x3_B8aKg,200
@@ -102,21 +104,22 @@ adaptive_sdk/graphql_client/update_model.py,sha256=w4-6gHg5k28k9SQwZ9qQZKbs3LQhn
 adaptive_sdk/graphql_client/update_model_compute_config.py,sha256=0RtrzzfuMrt-i6lABANRhem8jcDNkXORBEDFdSVpxfs,417
 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/input_types/typed_dicts.py,sha256=3g5raKSGUOcdc9DBAACj604kiyydNZC9YUeCS-K1t_Q,4911
 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/__init__.py,sha256=rl49jl2fu5KSSye2LMSm9GdV-89iyKI-UaNX6PmahHY,1414
 adaptive_sdk/resources/abtests.py,sha256=9PCPjxuWwY9ec88ewnq54gkoELq5U5iaBmHhzLCAsFU,7698
+adaptive_sdk/resources/artifacts.py,sha256=2A9FZYkFb8nKQwgVVewQC6InOtSg86oRDk-W9v5-41M,1947
 adaptive_sdk/resources/base_resource.py,sha256=D9adWSFxiDB7chVstDuBu1jcuXkE71UQJexnWENpC4A,1497
-adaptive_sdk/resources/chat.py,sha256=k-jh_IZfKwqoWNN2P5BhAM3P_f5HLybVZp35pUadwso,11954
+adaptive_sdk/resources/chat.py,sha256=5Kwcsdd6tDSdp28Yv7OxsYyB4D4K4NS5iFHdfn2FbgA,12164
 adaptive_sdk/resources/compute_pools.py,sha256=_c_eDkXyqKm53ZcfL91EtcuitYqd7ds_3Uz4PUY-TLw,3441
 adaptive_sdk/resources/datasets.py,sha256=sgGP2BwhaezaGei8xXoH0aKHyZFc64ZvIllxFUKNTd8,13648
 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/interactions.py,sha256=MJEhNHjgyNFZWqW0b6VHeQMETw26GqqdMBGDEWLSdiY,11846
 adaptive_sdk/resources/jobs.py,sha256=3ueoHp_58ELIAWfRfiAp51Jt7hazMmbfngaep2wnPn8,5398
-adaptive_sdk/resources/models.py,sha256=oaKEagYvw-lkPRgStDSgWEGSNaBvdA9FiWXes--mnsw,17536
+adaptive_sdk/resources/models.py,sha256=tPjb3cNIB66unjctGOXTfmLaH97Xwb8NI5T8qowezyY,17567
 adaptive_sdk/resources/permissions.py,sha256=ckO-oacWkvgcwXBK5iW_8qaK-g0SHXpEEy1qZy5lrB0,1053
 adaptive_sdk/resources/recipes.py,sha256=RdxpI6YT2GoitA_UfLp-jvQSlmF1dMSg4slDgzgB-Pk,16115
 adaptive_sdk/resources/roles.py,sha256=fD1F5Gd3ddkATsU2aFj7japTJVZngVgqt6TXskBQEOA,2218
@@ -125,7 +128,7 @@ adaptive_sdk/resources/use_cases.py,sha256=a_Ls3kbYgByJMlOYM_UGi4MTiW4wU7m057bud
 adaptive_sdk/resources/users.py,sha256=SoGWwdDCdhK4KjYOcAws-ZWlW7Edii7D3Vxfdu-NZY4,4406
 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.13.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-adaptive_sdk-0.1.13.dist-info/METADATA,sha256=JWMPrFY4S7HXbM5ZeNkMxNlK1zqcsY8jx_HFsH1Wz8c,1447
-adaptive_sdk-0.1.13.dist-info/RECORD,,
+adaptive_sdk/rest/rest_types.py,sha256=D7uH0iLZP8rDwUblVQiFb04208Bf5oPbqFiaq1YAnD8,8892
+adaptive_sdk-0.1.14.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+adaptive_sdk-0.1.14.dist-info/METADATA,sha256=OE7kVUoFtTEn-EVtDKYNG27G5U-ECpJMaPDWK0TtRXU,1600
+adaptive_sdk-0.1.14.dist-info/RECORD,,

adaptive_sdk/graphql_client/attach_model_to_use_case.py

@@ -1,12 +0,0 @@
-from pydantic import Field
-from .base_model import BaseModel
-from .fragments import ModelServiceData
-
-class AttachModelToUseCase(BaseModel):
-    """@public"""
-    attach_model: 'AttachModelToUseCaseAttachModel' = Field(alias='attachModel')
-
-class AttachModelToUseCaseAttachModel(ModelServiceData):
-    """@public"""
-    pass
-AttachModelToUseCase.model_rebuild()