vellum-ai 0.7.11__py3-none-any.whl → 0.8.5__py3-none-any.whl

vellum/resources/ml_models/client.py

@@ -1,22 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+from ...core.client_wrapper import SyncClientWrapper
+from ...core.request_options import RequestOptions
+from ...types.paginated_ml_model_read_list import PaginatedMlModelReadList
+from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
-
 from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import pydantic_v1
-from ...core.request_options import RequestOptions
+from ...types.ml_model_family import MlModelFamily
 from ...types.hosted_by_enum import HostedByEnum
 from ...types.ml_model_developer import MlModelDeveloper
-from ...types.ml_model_display_config_request import MlModelDisplayConfigRequest
 from ...types.ml_model_exec_config_request import MlModelExecConfigRequest
-from ...types.ml_model_family import MlModelFamily
 from ...types.ml_model_parameter_config_request import MlModelParameterConfigRequest
-from ...types.ml_model_read import MlModelRead
-from ...types.paginated_ml_model_read_list import PaginatedMlModelReadList
+from ...types.ml_model_display_config_request import MlModelDisplayConfigRequest
 from ...types.visibility_enum import VisibilityEnum
+from ...types.ml_model_read import MlModelRead
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -58,7 +58,7 @@ class MlModelsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -69,12 +69,22 @@ class MlModelsClient:
             "v1/ml-models",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
-            params={"limit": limit, "offset": offset, "ordering": ordering},
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PaginatedMlModelReadList, _response.json())  # type: ignore
+                return typing.cast(
+                    PaginatedMlModelReadList,
+                    parse_obj_as(
+                        type_=PaginatedMlModelReadList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -195,8 +205,7 @@ class MlModelsClient:
 
         Examples
         --------
-        from vellum import MlModelExecConfigRequest
-        from vellum.client import Vellum
+        from vellum import MlModelExecConfigRequest, Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -233,7 +242,13 @@ class MlModelsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlModelRead, _response.json())  # type: ignore
+                return typing.cast(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -258,7 +273,7 @@ class MlModelsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -275,7 +290,13 @@ class MlModelsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlModelRead, _response.json())  # type: ignore
+                return typing.cast(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -326,7 +347,7 @@ class MlModelsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -350,7 +371,13 @@ class MlModelsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlModelRead, _response.json())  # type: ignore
+                return typing.cast(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -401,7 +428,7 @@ class MlModelsClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -425,7 +452,13 @@ class MlModelsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlModelRead, _response.json())  # type: ignore
+                return typing.cast(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -470,7 +503,7 @@ class AsyncMlModelsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -487,12 +520,22 @@ class AsyncMlModelsClient:
             "v1/ml-models",
             base_url=self._client_wrapper.get_environment().default,
             method="GET",
-            params={"limit": limit, "offset": offset, "ordering": ordering},
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(PaginatedMlModelReadList, _response.json())  # type: ignore
+                return typing.cast(
+                    PaginatedMlModelReadList,
+                    parse_obj_as(
+                        type_=PaginatedMlModelReadList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -615,8 +658,7 @@ class AsyncMlModelsClient:
         --------
         import asyncio
 
-        from vellum import MlModelExecConfigRequest
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum, MlModelExecConfigRequest
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -659,7 +701,13 @@ class AsyncMlModelsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlModelRead, _response.json())  # type: ignore
+                return typing.cast(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -686,7 +734,7 @@ class AsyncMlModelsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -709,7 +757,13 @@ class AsyncMlModelsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlModelRead, _response.json())  # type: ignore
+                return typing.cast(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -762,7 +816,7 @@ class AsyncMlModelsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -792,7 +846,13 @@ class AsyncMlModelsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlModelRead, _response.json())  # type: ignore
+                return typing.cast(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -845,7 +905,7 @@ class AsyncMlModelsClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -875,7 +935,13 @@ class AsyncMlModelsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(MlModelRead, _response.json())  # type: ignore
+                return typing.cast(
+                    MlModelRead,
+                    parse_obj_as(
+                        type_=MlModelRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)

vellum/resources/sandboxes/client.py

@@ -1,16 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from json.decoder import JSONDecodeError
-
-from ...core.api_error import ApiError
-from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ...core.jsonable_encoder import jsonable_encoder
-from ...core.pydantic_utilities import pydantic_v1
+from ...core.client_wrapper import SyncClientWrapper
 from ...core.request_options import RequestOptions
 from ...types.deployment_read import DeploymentRead
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ...core.api_error import ApiError
 from ...types.named_scenario_input_request import NamedScenarioInputRequest
 from ...types.sandbox_scenario import SandboxScenario
+from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -62,7 +62,7 @@ class SandboxesClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -87,7 +87,13 @@ class SandboxesClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(DeploymentRead, _response.json())  # type: ignore
+                return typing.cast(
+                    DeploymentRead,
+                    parse_obj_as(
+                        type_=DeploymentRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -134,7 +140,7 @@ class SandboxesClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import NamedScenarioInputStringVariableValueRequest, Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -142,20 +148,39 @@ class SandboxesClient:
         client.sandboxes.upsert_sandbox_scenario(
             id="id",
             label="Scenario 1",
-            inputs=[],
+            inputs=[
+                NamedScenarioInputStringVariableValueRequest(
+                    value="Hello, world!",
+                    name="var_1",
+                ),
+                NamedScenarioInputStringVariableValueRequest(
+                    value="Why hello, there!",
+                    name="var_2",
+                ),
+            ],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
             f"v1/sandboxes/{jsonable_encoder(id)}/scenarios",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json={"label": label, "inputs": inputs, "scenario_id": scenario_id},
+            json={
+                "label": label,
+                "inputs": inputs,
+                "scenario_id": scenario_id,
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(SandboxScenario, _response.json())  # type: ignore
+                return typing.cast(
+                    SandboxScenario,
+                    parse_obj_as(
+                        type_=SandboxScenario,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -184,7 +209,7 @@ class SandboxesClient:
 
         Examples
         --------
-        from vellum.client import Vellum
+        from vellum import Vellum
 
         client = Vellum(
             api_key="YOUR_API_KEY",
@@ -257,7 +282,7 @@ class AsyncSandboxesClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -288,7 +313,13 @@ class AsyncSandboxesClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(DeploymentRead, _response.json())  # type: ignore
+                return typing.cast(
+                    DeploymentRead,
+                    parse_obj_as(
+                        type_=DeploymentRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -337,7 +368,7 @@ class AsyncSandboxesClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum, NamedScenarioInputStringVariableValueRequest
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
@@ -348,7 +379,16 @@ class AsyncSandboxesClient:
             await client.sandboxes.upsert_sandbox_scenario(
                 id="id",
                 label="Scenario 1",
-                inputs=[],
+                inputs=[
+                    NamedScenarioInputStringVariableValueRequest(
+                        value="Hello, world!",
+                        name="var_1",
+                    ),
+                    NamedScenarioInputStringVariableValueRequest(
+                        value="Why hello, there!",
+                        name="var_2",
+                    ),
+                ],
             )
 
 
@@ -358,13 +398,23 @@ class AsyncSandboxesClient:
             f"v1/sandboxes/{jsonable_encoder(id)}/scenarios",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json={"label": label, "inputs": inputs, "scenario_id": scenario_id},
+            json={
+                "label": label,
+                "inputs": inputs,
+                "scenario_id": scenario_id,
+            },
             request_options=request_options,
             omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(SandboxScenario, _response.json())  # type: ignore
+                return typing.cast(
+                    SandboxScenario,
+                    parse_obj_as(
+                        type_=SandboxScenario,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
@@ -395,7 +445,7 @@ class AsyncSandboxesClient:
         --------
         import asyncio
 
-        from vellum.client import AsyncVellum
+        from vellum import AsyncVellum
 
         client = AsyncVellum(
             api_key="YOUR_API_KEY",