vellum-ai 0.12.16__py3-none-any.whl → 0.13.0__py3-none-any.whl

vellum/__init__.py

@@ -18,14 +18,11 @@ from .types import (
     ArrayChatMessageContentItemRequest,
     ArrayChatMessageContentRequest,
     ArrayInput,
-    ArrayVariableValue,
-    ArrayVariableValueItem,
     ArrayVellumValue,
     ArrayVellumValueRequest,
     AudioChatMessageContent,
     AudioChatMessageContentRequest,
     AudioPromptBlock,
-    AudioVariableValue,
     AudioVellumValue,
     AudioVellumValueRequest,
     BasicVectorizerIntfloatMultilingualE5Large,
@@ -36,7 +33,6 @@ from .types import (
     BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
     ChatHistoryInput,
     ChatHistoryInputRequest,
-    ChatHistoryVariableValue,
     ChatHistoryVellumValue,
     ChatHistoryVellumValueRequest,
     ChatMessage,
@@ -93,7 +89,6 @@ from .types import (
     EphemeralPromptCacheConfig,
     EphemeralPromptCacheConfigTypeEnum,
     ErrorInput,
-    ErrorVariableValue,
     ErrorVellumValue,
     ErrorVellumValueRequest,
     ExecutePromptEvent,
@@ -137,7 +132,6 @@ from .types import (
     FunctionCallInput,
     FunctionCallPromptBlock,
     FunctionCallRequest,
-    FunctionCallVariableValue,
     FunctionCallVellumValue,
     FunctionCallVellumValueRequest,
     FunctionDefinition,
@@ -161,7 +155,6 @@ from .types import (
     ImageChatMessageContent,
     ImageChatMessageContentRequest,
     ImagePromptBlock,
-    ImageVariableValue,
     ImageVellumValue,
     ImageVellumValueRequest,
     IndexingConfigVectorizer,
@@ -177,7 +170,6 @@ from .types import (
     JinjaPromptBlock,
     JsonInput,
     JsonInputRequest,
-    JsonVariableValue,
     JsonVellumValue,
     JsonVellumValueRequest,
     LogicalOperator,
@@ -218,6 +210,7 @@ from .types import (
     NamedTestCaseStringVariableValueRequest,
     NamedTestCaseVariableValue,
     NamedTestCaseVariableValueRequest,
+    NewMemberJoinBehaviorEnum,
     NodeInputCompiledArrayValue,
     NodeInputCompiledChatHistoryValue,
     NodeInputCompiledErrorValue,
@@ -239,7 +232,6 @@ from .types import (
     NormalizedLogProbs,
     NormalizedTokenLogProbs,
     NumberInput,
-    NumberVariableValue,
     NumberVellumValue,
     NumberVellumValueRequest,
     OpenAiVectorizerConfig,
@@ -250,6 +242,7 @@ from .types import (
     OpenAiVectorizerTextEmbedding3SmallRequest,
     OpenAiVectorizerTextEmbeddingAda002,
     OpenAiVectorizerTextEmbeddingAda002Request,
+    OrganizationRead,
     PaginatedContainerImageReadList,
     PaginatedDeploymentReleaseTagReadList,
     PaginatedDocumentIndexReadList,
@@ -313,7 +306,6 @@ from .types import (
     SearchResultMetaRequest,
     SearchResultRequest,
     SearchResultsInput,
-    SearchResultsVariableValue,
     SearchResultsVellumValue,
     SearchResultsVellumValueRequest,
     SearchWeightsRequest,
@@ -334,7 +326,6 @@ from .types import (
     StringChatMessageContentRequest,
     StringInput,
     StringInputRequest,
-    StringVariableValue,
     StringVellumValue,
     StringVellumValueRequest,
     SubmitCompletionActualRequest,
@@ -498,6 +489,7 @@ from .types import (
     WorkflowResultEventOutputDataSearchResults,
     WorkflowResultEventOutputDataString,
     WorkflowStreamEvent,
+    WorkspaceRead,
     WorkspaceSecretRead,
 )
 from .errors import BadRequestError, ForbiddenError, InternalServerError, NotFoundError
@@ -517,6 +509,7 @@ from .resources import (
     folder_entities,
     metric_definitions,
     ml_models,
+    organizations,
     sandboxes,
     test_suite_runs,
     test_suites,
@@ -524,6 +517,7 @@ from .resources import (
     workflow_sandboxes,
     workflows,
     workspace_secrets,
+    workspaces,
 )
 from .client import AsyncVellum, Vellum
 from .environment import VellumEnvironment
@@ -544,15 +538,12 @@ __all__ = [
     "ArrayChatMessageContentItemRequest",
     "ArrayChatMessageContentRequest",
     "ArrayInput",
-    "ArrayVariableValue",
-    "ArrayVariableValueItem",
     "ArrayVellumValue",
     "ArrayVellumValueRequest",
     "AsyncVellum",
     "AudioChatMessageContent",
     "AudioChatMessageContentRequest",
     "AudioPromptBlock",
-    "AudioVariableValue",
     "AudioVellumValue",
     "AudioVellumValueRequest",
     "BadRequestError",
@@ -564,7 +555,6 @@ __all__ = [
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
     "ChatHistoryInput",
     "ChatHistoryInputRequest",
-    "ChatHistoryVariableValue",
     "ChatHistoryVellumValue",
     "ChatHistoryVellumValueRequest",
     "ChatMessage",
@@ -623,7 +613,6 @@ __all__ = [
     "EphemeralPromptCacheConfig",
     "EphemeralPromptCacheConfigTypeEnum",
     "ErrorInput",
-    "ErrorVariableValue",
     "ErrorVellumValue",
     "ErrorVellumValueRequest",
     "ExecutePromptEvent",
@@ -669,7 +658,6 @@ __all__ = [
     "FunctionCallInput",
     "FunctionCallPromptBlock",
     "FunctionCallRequest",
-    "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
     "FunctionDefinition",
@@ -693,7 +681,6 @@ __all__ = [
     "ImageChatMessageContent",
     "ImageChatMessageContentRequest",
     "ImagePromptBlock",
-    "ImageVariableValue",
     "ImageVellumValue",
     "ImageVellumValueRequest",
     "IndexingConfigVectorizer",
@@ -710,7 +697,6 @@ __all__ = [
     "JinjaPromptBlock",
     "JsonInput",
     "JsonInputRequest",
-    "JsonVariableValue",
     "JsonVellumValue",
     "JsonVellumValueRequest",
     "ListDeploymentReleaseTagsRequestSource",
@@ -753,6 +739,7 @@ __all__ = [
     "NamedTestCaseStringVariableValueRequest",
     "NamedTestCaseVariableValue",
     "NamedTestCaseVariableValueRequest",
+    "NewMemberJoinBehaviorEnum",
     "NodeInputCompiledArrayValue",
     "NodeInputCompiledChatHistoryValue",
     "NodeInputCompiledErrorValue",
@@ -775,7 +762,6 @@ __all__ = [
     "NormalizedTokenLogProbs",
     "NotFoundError",
     "NumberInput",
-    "NumberVariableValue",
     "NumberVellumValue",
     "NumberVellumValueRequest",
     "OpenAiVectorizerConfig",
@@ -786,6 +772,7 @@ __all__ = [
     "OpenAiVectorizerTextEmbedding3SmallRequest",
     "OpenAiVectorizerTextEmbeddingAda002",
     "OpenAiVectorizerTextEmbeddingAda002Request",
+    "OrganizationRead",
     "PaginatedContainerImageReadList",
     "PaginatedDeploymentReleaseTagReadList",
     "PaginatedDocumentIndexReadList",
@@ -849,7 +836,6 @@ __all__ = [
     "SearchResultMetaRequest",
     "SearchResultRequest",
     "SearchResultsInput",
-    "SearchResultsVariableValue",
     "SearchResultsVellumValue",
     "SearchResultsVellumValueRequest",
     "SearchWeightsRequest",
@@ -870,7 +856,6 @@ __all__ = [
     "StringChatMessageContentRequest",
     "StringInput",
     "StringInputRequest",
-    "StringVariableValue",
     "StringVellumValue",
     "StringVellumValueRequest",
     "SubmitCompletionActualRequest",
@@ -1038,6 +1023,7 @@ __all__ = [
     "WorkflowResultEventOutputDataString",
     "WorkflowStreamEvent",
     "WorkflowsPullRequestFormat",
+    "WorkspaceRead",
     "WorkspaceSecretRead",
     "__version__",
     "ad_hoc",
@@ -1048,6 +1034,7 @@ __all__ = [
     "folder_entities",
     "metric_definitions",
     "ml_models",
+    "organizations",
     "sandboxes",
     "test_suite_runs",
     "test_suites",
@@ -1055,4 +1042,5 @@ __all__ = [
     "workflow_sandboxes",
     "workflows",
     "workspace_secrets",
+    "workspaces",
 ]

vellum/client/__init__.py

@@ -12,6 +12,7 @@ from .resources.documents.client import DocumentsClient
 from .resources.folder_entities.client import FolderEntitiesClient
 from .resources.metric_definitions.client import MetricDefinitionsClient
 from .resources.ml_models.client import MlModelsClient
+from .resources.organizations.client import OrganizationsClient
 from .resources.sandboxes.client import SandboxesClient
 from .resources.test_suite_runs.client import TestSuiteRunsClient
 from .resources.test_suites.client import TestSuitesClient
@@ -19,6 +20,7 @@ from .resources.workflow_deployments.client import WorkflowDeploymentsClient
 from .resources.workflow_sandboxes.client import WorkflowSandboxesClient
 from .resources.workflows.client import WorkflowsClient
 from .resources.workspace_secrets.client import WorkspaceSecretsClient
+from .resources.workspaces.client import WorkspacesClient
 from .types.code_execution_runtime import CodeExecutionRuntime
 from .types.code_executor_input import CodeExecutorInput
 from .types.code_execution_package import CodeExecutionPackage
@@ -61,6 +63,7 @@ from .resources.documents.client import AsyncDocumentsClient
 from .resources.folder_entities.client import AsyncFolderEntitiesClient
 from .resources.metric_definitions.client import AsyncMetricDefinitionsClient
 from .resources.ml_models.client import AsyncMlModelsClient
+from .resources.organizations.client import AsyncOrganizationsClient
 from .resources.sandboxes.client import AsyncSandboxesClient
 from .resources.test_suite_runs.client import AsyncTestSuiteRunsClient
 from .resources.test_suites.client import AsyncTestSuitesClient
@@ -68,6 +71,7 @@ from .resources.workflow_deployments.client import AsyncWorkflowDeploymentsClien
 from .resources.workflow_sandboxes.client import AsyncWorkflowSandboxesClient
 from .resources.workflows.client import AsyncWorkflowsClient
 from .resources.workspace_secrets.client import AsyncWorkspaceSecretsClient
+from .resources.workspaces.client import AsyncWorkspacesClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -135,6 +139,7 @@ class Vellum:
         self.folder_entities = FolderEntitiesClient(client_wrapper=self._client_wrapper)
         self.metric_definitions = MetricDefinitionsClient(client_wrapper=self._client_wrapper)
         self.ml_models = MlModelsClient(client_wrapper=self._client_wrapper)
+        self.organizations = OrganizationsClient(client_wrapper=self._client_wrapper)
         self.sandboxes = SandboxesClient(client_wrapper=self._client_wrapper)
         self.test_suite_runs = TestSuiteRunsClient(client_wrapper=self._client_wrapper)
         self.test_suites = TestSuitesClient(client_wrapper=self._client_wrapper)
@@ -142,6 +147,7 @@ class Vellum:
         self.workflow_sandboxes = WorkflowSandboxesClient(client_wrapper=self._client_wrapper)
         self.workflows = WorkflowsClient(client_wrapper=self._client_wrapper)
         self.workspace_secrets = WorkspaceSecretsClient(client_wrapper=self._client_wrapper)
+        self.workspaces = WorkspacesClient(client_wrapper=self._client_wrapper)
 
     def execute_code(
         self,
@@ -1398,6 +1404,7 @@ class AsyncVellum:
         self.folder_entities = AsyncFolderEntitiesClient(client_wrapper=self._client_wrapper)
         self.metric_definitions = AsyncMetricDefinitionsClient(client_wrapper=self._client_wrapper)
         self.ml_models = AsyncMlModelsClient(client_wrapper=self._client_wrapper)
+        self.organizations = AsyncOrganizationsClient(client_wrapper=self._client_wrapper)
         self.sandboxes = AsyncSandboxesClient(client_wrapper=self._client_wrapper)
         self.test_suite_runs = AsyncTestSuiteRunsClient(client_wrapper=self._client_wrapper)
         self.test_suites = AsyncTestSuitesClient(client_wrapper=self._client_wrapper)
@@ -1405,6 +1412,7 @@ class AsyncVellum:
         self.workflow_sandboxes = AsyncWorkflowSandboxesClient(client_wrapper=self._client_wrapper)
         self.workflows = AsyncWorkflowsClient(client_wrapper=self._client_wrapper)
         self.workspace_secrets = AsyncWorkspaceSecretsClient(client_wrapper=self._client_wrapper)
+        self.workspaces = AsyncWorkspacesClient(client_wrapper=self._client_wrapper)
 
     async def execute_code(
         self,

vellum/client/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.12.16",
+            "X-Fern-SDK-Version": "0.13.0",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/client/resources/__init__.py

@@ -9,6 +9,7 @@ from . import (
     folder_entities,
     metric_definitions,
     ml_models,
+    organizations,
     sandboxes,
     test_suite_runs,
     test_suites,
@@ -16,6 +17,7 @@ from . import (
     workflow_sandboxes,
     workflows,
     workspace_secrets,
+    workspaces,
 )
 from .deployments import DeploymentsListRequestStatus, ListDeploymentReleaseTagsRequestSource
 from .document_indexes import DocumentIndexesListRequestStatus
@@ -39,6 +41,7 @@ __all__ = [
     "folder_entities",
     "metric_definitions",
     "ml_models",
+    "organizations",
     "sandboxes",
     "test_suite_runs",
     "test_suites",
@@ -46,4 +49,5 @@ __all__ = [
     "workflow_sandboxes",
     "workflows",
     "workspace_secrets",
+    "workspaces",
 ]

vellum/client/resources/organizations/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

vellum/client/resources/organizations/client.py

@@ -0,0 +1,116 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ...core.client_wrapper import SyncClientWrapper
+import typing
+from ...core.request_options import RequestOptions
+from ...types.organization_read import OrganizationRead
+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
+
+
+class OrganizationsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def organization_identity(self, *, request_options: typing.Optional[RequestOptions] = None) -> OrganizationRead:
+        """
+        Retrieves information about the active Organization
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OrganizationRead
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.organizations.organization_identity()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/organizations/identity",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    OrganizationRead,
+                    parse_obj_as(
+                        type_=OrganizationRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncOrganizationsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def organization_identity(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> OrganizationRead:
+        """
+        Retrieves information about the active Organization
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OrganizationRead
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.organizations.organization_identity()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/organizations/identity",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    OrganizationRead,
+                    parse_obj_as(
+                        type_=OrganizationRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

vellum/client/resources/workspaces/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

vellum/client/resources/workspaces/client.py

@@ -0,0 +1,114 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ...core.client_wrapper import SyncClientWrapper
+import typing
+from ...core.request_options import RequestOptions
+from ...types.workspace_read import WorkspaceRead
+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
+
+
+class WorkspacesClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def workspace_identity(self, *, request_options: typing.Optional[RequestOptions] = None) -> WorkspaceRead:
+        """
+        Retrieves information about the active Workspace
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceRead
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.workspaces.workspace_identity()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/workspaces/identity",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceRead,
+                    parse_obj_as(
+                        type_=WorkspaceRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncWorkspacesClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def workspace_identity(self, *, request_options: typing.Optional[RequestOptions] = None) -> WorkspaceRead:
+        """
+        Retrieves information about the active Workspace
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceRead
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.workspaces.workspace_identity()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/workspaces/identity",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceRead,
+                    parse_obj_as(
+                        type_=WorkspaceRead,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)