vellum-ai 0.8.23__py3-none-any.whl → 0.8.26__py3-none-any.whl

vellum/resources/workspace_secrets/client.py

@@ -0,0 +1,127 @@
+# 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_secret_read import WorkspaceSecretRead
+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 ...core.client_wrapper import AsyncClientWrapper
+
+
+class WorkspaceSecretsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> WorkspaceSecretRead:
+        """
+        Used to retrieve a Workspace Secret given its ID or name.
+
+        Parameters
+        ----------
+        id : str
+            Either the Workspace Secret's ID or its unique name
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceSecretRead
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.workspace_secrets.retrieve(
+            id="id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/workspace-secrets/{jsonable_encoder(id)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceSecretRead,
+                    parse_obj_as(
+                        type_=WorkspaceSecretRead,  # 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 AsyncWorkspaceSecretsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def retrieve(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> WorkspaceSecretRead:
+        """
+        Used to retrieve a Workspace Secret given its ID or name.
+
+        Parameters
+        ----------
+        id : str
+            Either the Workspace Secret's ID or its unique name
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceSecretRead
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.workspace_secrets.retrieve(
+                id="id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/workspace-secrets/{jsonable_encoder(id)}",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceSecretRead,
+                    parse_obj_as(
+                        type_=WorkspaceSecretRead,  # 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/types/__init__.py

@@ -106,6 +106,17 @@ from .execution_vellum_value import ExecutionVellumValue
 from .external_test_case_execution import ExternalTestCaseExecution
 from .external_test_case_execution_request import ExternalTestCaseExecutionRequest
 from .finish_reason_enum import FinishReasonEnum
+from .folder_entity import FolderEntity
+from .folder_entity_document_index import FolderEntityDocumentIndex
+from .folder_entity_document_index_data import FolderEntityDocumentIndexData
+from .folder_entity_folder import FolderEntityFolder
+from .folder_entity_folder_data import FolderEntityFolderData
+from .folder_entity_prompt_sandbox import FolderEntityPromptSandbox
+from .folder_entity_prompt_sandbox_data import FolderEntityPromptSandboxData
+from .folder_entity_test_suite import FolderEntityTestSuite
+from .folder_entity_test_suite_data import FolderEntityTestSuiteData
+from .folder_entity_workflow_sandbox import FolderEntityWorkflowSandbox
+from .folder_entity_workflow_sandbox_data import FolderEntityWorkflowSandboxData
 from .fulfilled_ad_hoc_execute_prompt_event import FulfilledAdHocExecutePromptEvent
 from .fulfilled_enum import FulfilledEnum
 from .fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
@@ -236,6 +247,7 @@ from .open_ai_vectorizer_text_embedding_3_small_request import OpenAiVectorizerT
 from .open_ai_vectorizer_text_embedding_ada_002 import OpenAiVectorizerTextEmbeddingAda002
 from .open_ai_vectorizer_text_embedding_ada_002_request import OpenAiVectorizerTextEmbeddingAda002Request
 from .paginated_document_index_read_list import PaginatedDocumentIndexReadList
+from .paginated_folder_entity_list import PaginatedFolderEntityList
 from .paginated_slim_deployment_read_list import PaginatedSlimDeploymentReadList
 from .paginated_slim_document_list import PaginatedSlimDocumentList
 from .paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploymentList
@@ -261,6 +273,7 @@ from .prompt_request_chat_history_input_request import PromptRequestChatHistoryI
 from .prompt_request_input_request import PromptRequestInputRequest
 from .prompt_request_json_input_request import PromptRequestJsonInputRequest
 from .prompt_request_string_input_request import PromptRequestStringInputRequest
+from .prompt_settings_request import PromptSettingsRequest
 from .raw_prompt_execution_overrides_request import RawPromptExecutionOverridesRequest
 from .reducto_chunker_config import ReductoChunkerConfig
 from .reducto_chunker_config_request import ReductoChunkerConfigRequest
@@ -298,6 +311,7 @@ from .search_results_variable_value import SearchResultsVariableValue
 from .search_results_vellum_value import SearchResultsVellumValue
 from .search_results_vellum_value_request import SearchResultsVellumValueRequest
 from .search_weights_request import SearchWeightsRequest
+from .secret_type_enum import SecretTypeEnum
 from .sentence_chunker_config import SentenceChunkerConfig
 from .sentence_chunker_config_request import SentenceChunkerConfigRequest
 from .sentence_chunking import SentenceChunking
@@ -467,6 +481,7 @@ from .workflow_result_event_output_data_number import WorkflowResultEventOutputD
 from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
 from .workflow_result_event_output_data_string import WorkflowResultEventOutputDataString
 from .workflow_stream_event import WorkflowStreamEvent
+from .workspace_secret_read import WorkspaceSecretRead
 
 __all__ = [
     "AdHocExecutePromptEvent",
@@ -567,6 +582,17 @@ __all__ = [
     "ExternalTestCaseExecution",
     "ExternalTestCaseExecutionRequest",
     "FinishReasonEnum",
+    "FolderEntity",
+    "FolderEntityDocumentIndex",
+    "FolderEntityDocumentIndexData",
+    "FolderEntityFolder",
+    "FolderEntityFolderData",
+    "FolderEntityPromptSandbox",
+    "FolderEntityPromptSandboxData",
+    "FolderEntityTestSuite",
+    "FolderEntityTestSuiteData",
+    "FolderEntityWorkflowSandbox",
+    "FolderEntityWorkflowSandboxData",
     "FulfilledAdHocExecutePromptEvent",
     "FulfilledEnum",
     "FulfilledExecutePromptEvent",
@@ -693,6 +719,7 @@ __all__ = [
     "OpenAiVectorizerTextEmbeddingAda002",
     "OpenAiVectorizerTextEmbeddingAda002Request",
     "PaginatedDocumentIndexReadList",
+    "PaginatedFolderEntityList",
     "PaginatedSlimDeploymentReadList",
     "PaginatedSlimDocumentList",
     "PaginatedSlimWorkflowDeploymentList",
@@ -718,6 +745,7 @@ __all__ = [
     "PromptRequestInputRequest",
     "PromptRequestJsonInputRequest",
     "PromptRequestStringInputRequest",
+    "PromptSettingsRequest",
     "RawPromptExecutionOverridesRequest",
     "ReductoChunkerConfig",
     "ReductoChunkerConfigRequest",
@@ -755,6 +783,7 @@ __all__ = [
     "SearchResultsVellumValue",
     "SearchResultsVellumValueRequest",
     "SearchWeightsRequest",
+    "SecretTypeEnum",
     "SentenceChunkerConfig",
     "SentenceChunkerConfigRequest",
     "SentenceChunking",
@@ -920,4 +949,5 @@ __all__ = [
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
     "WorkflowStreamEvent",
+    "WorkspaceSecretRead",
 ]

vellum/types/api_node_result_data.py

@@ -1,20 +1,20 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.pydantic_utilities import UniversalBaseModel
-import typing
 import typing_extensions
+import typing
 from ..core.serialization import FieldMetadata
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
 class ApiNodeResultData(UniversalBaseModel):
-    text_output_id: str
-    text: typing.Optional[str] = None
-    json_output_id: str
     json_: typing_extensions.Annotated[
         typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]], FieldMetadata(alias="json")
     ] = None
+    text_output_id: str
+    text: typing.Optional[str] = None
+    json_output_id: str
     status_code_output_id: str
     status_code: int
 

vellum/types/folder_entity.py

@@ -0,0 +1,16 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from .folder_entity_folder import FolderEntityFolder
+from .folder_entity_prompt_sandbox import FolderEntityPromptSandbox
+from .folder_entity_workflow_sandbox import FolderEntityWorkflowSandbox
+from .folder_entity_document_index import FolderEntityDocumentIndex
+from .folder_entity_test_suite import FolderEntityTestSuite
+
+FolderEntity = typing.Union[
+    FolderEntityFolder,
+    FolderEntityPromptSandbox,
+    FolderEntityWorkflowSandbox,
+    FolderEntityDocumentIndex,
+    FolderEntityTestSuite,
+]

vellum/types/folder_entity_document_index.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .folder_entity_document_index_data import FolderEntityDocumentIndexData
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FolderEntityDocumentIndex(UniversalBaseModel):
+    """
+    A slim representation of a Document Index, as it exists within a Folder.
+    """
+
+    id: str
+    type: typing.Literal["DOCUMENT_INDEX"] = "DOCUMENT_INDEX"
+    data: FolderEntityDocumentIndexData
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/folder_entity_document_index_data.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+from .entity_status import EntityStatus
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class FolderEntityDocumentIndexData(UniversalBaseModel):
+    id: str
+    label: str
+    created: dt.datetime
+    modified: dt.datetime
+    status: EntityStatus
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/folder_entity_folder.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .folder_entity_folder_data import FolderEntityFolderData
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FolderEntityFolder(UniversalBaseModel):
+    """
+    A slim representation of a Folder, as it exists within another Folder.
+    """
+
+    id: str
+    type: typing.Literal["FOLDER"] = "FOLDER"
+    data: FolderEntityFolderData
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/folder_entity_folder_data.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class FolderEntityFolderData(UniversalBaseModel):
+    id: str
+    label: str
+    created: dt.datetime
+    modified: dt.datetime
+    has_contents: bool
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/folder_entity_prompt_sandbox.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .folder_entity_prompt_sandbox_data import FolderEntityPromptSandboxData
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FolderEntityPromptSandbox(UniversalBaseModel):
+    """
+    A slim representation of a Prompt Sandbox, as it exists within a Folder.
+    """
+
+    id: str
+    type: typing.Literal["PROMPT_SANDBOX"] = "PROMPT_SANDBOX"
+    data: FolderEntityPromptSandboxData
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/folder_entity_prompt_sandbox_data.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+from .entity_status import EntityStatus
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FolderEntityPromptSandboxData(UniversalBaseModel):
+    id: str
+    label: str
+    created: dt.datetime
+    modified: dt.datetime
+    status: EntityStatus
+    last_deployed_on: typing.Optional[dt.datetime] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/folder_entity_test_suite.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .folder_entity_test_suite_data import FolderEntityTestSuiteData
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FolderEntityTestSuite(UniversalBaseModel):
+    """
+    A slim representation of a Test Suite, as it exists within a Folder.
+    """
+
+    id: str
+    type: typing.Literal["TEST_SUITE"] = "TEST_SUITE"
+    data: FolderEntityTestSuiteData
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/folder_entity_test_suite_data.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+from .entity_status import EntityStatus
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class FolderEntityTestSuiteData(UniversalBaseModel):
+    id: str
+    label: str
+    created: dt.datetime
+    modified: dt.datetime
+    status: EntityStatus
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/folder_entity_workflow_sandbox.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .folder_entity_workflow_sandbox_data import FolderEntityWorkflowSandboxData
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FolderEntityWorkflowSandbox(UniversalBaseModel):
+    """
+    A slim representation of a Workflow Sandbox, as it exists within a Folder.
+    """
+
+    id: str
+    type: typing.Literal["WORKFLOW_SANDBOX"] = "WORKFLOW_SANDBOX"
+    data: FolderEntityWorkflowSandboxData
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/folder_entity_workflow_sandbox_data.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+from .entity_status import EntityStatus
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FolderEntityWorkflowSandboxData(UniversalBaseModel):
+    id: str
+    label: str
+    created: dt.datetime
+    modified: dt.datetime
+    status: EntityStatus
+    last_deployed_on: typing.Optional[dt.datetime] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/paginated_folder_entity_list.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .folder_entity import FolderEntity
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class PaginatedFolderEntityList(UniversalBaseModel):
+    count: typing.Optional[int] = None
+    next: typing.Optional[str] = None
+    previous: typing.Optional[str] = None
+    results: typing.Optional[typing.List[FolderEntity]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/prompt_settings_request.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class PromptSettingsRequest(UniversalBaseModel):
+    timeout: typing.Optional[float] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/types/secret_type_enum.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+SecretTypeEnum = typing.Union[typing.Literal["USER_DEFINED", "HMAC", "INTERNAL_API_KEY"], typing.Any]

vellum/types/vellum_error_code_enum.py

@@ -3,5 +3,8 @@
 import typing
 
 VellumErrorCodeEnum = typing.Union[
-    typing.Literal["INVALID_REQUEST", "PROVIDER_ERROR", "INTERNAL_SERVER_ERROR", "USER_DEFINED_ERROR"], typing.Any
+    typing.Literal[
+        "INVALID_REQUEST", "PROVIDER_ERROR", "REQUEST_TIMEOUT", "INTERNAL_SERVER_ERROR", "USER_DEFINED_ERROR"
+    ],
+    typing.Any,
 ]

vellum/types/workspace_secret_read.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+from .secret_type_enum import SecretTypeEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class WorkspaceSecretRead(UniversalBaseModel):
+    id: str
+    modified: dt.datetime
+    name: str
+    label: str
+    secret_type: SecretTypeEnum
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

{vellum_ai-0.8.23.dist-info → vellum_ai-0.8.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.8.23
+Version: 0.8.26
 Summary: 
 License: MIT
 Requires-Python: >=3.8,<4.0