vellum-ai 0.1.4__tar.gz → 0.1.5__tar.gz

{vellum_ai-0.1.4 → vellum_ai-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.1.4
+Version: 0.1.5
 Summary: 
 Requires-Python: >=3.7,<4.0
 Classifier: Programming Language :: Python :: 3

{vellum_ai-0.1.4 → vellum_ai-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "v0.1.4"
+version = "v0.1.5"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.1.4 → vellum_ai-0.1.5}/src/vellum/__init__.py

@@ -4,6 +4,7 @@ from .types import (
     ApiNodeResult,
     ApiNodeResultData,
     BlockTypeEnum,
+    ChatHistoryInputRequest,
     ChatMessage,
     ChatMessageRequest,
     ChatMessageRole,
@@ -18,8 +19,14 @@ from .types import (
     DocumentStatus,
     EnrichedNormalizedCompletion,
     EnvironmentEnum,
+    ErrorExecutePromptResponse,
     EvaluationParams,
     EvaluationParamsRequest,
+    ExecutePromptApiErrorResponse,
+    ExecutePromptResponse,
+    ExecutePromptResponse_Error,
+    ExecutePromptResponse_Json,
+    ExecutePromptResponse_String,
     ExecuteWorkflowStreamErrorResponse,
     FinishReasonEnum,
     GenerateErrorResponse,
@@ -33,6 +40,8 @@ from .types import (
     GenerateStreamResult,
     GenerateStreamResultData,
     IndexingStateEnum,
+    JsonExecutePromptResponse,
+    JsonInputRequest,
     LogicalOperator,
     LogprobsEnum,
     MetadataFilterConfigRequest,
@@ -62,6 +71,10 @@ from .types import (
     PaginatedSlimDocumentList,
     ProcessingFailureReasonEnum,
     ProcessingStateEnum,
+    PromptDeploymentInputRequest,
+    PromptDeploymentInputRequest_ChatHistory,
+    PromptDeploymentInputRequest_Json,
+    PromptDeploymentInputRequest_String,
     PromptNodeResult,
     PromptNodeResultData,
     PromptTemplateBlock,
@@ -100,6 +113,8 @@ from .types import (
     SearchResultRequest,
     SearchWeightsRequest,
     SlimDocument,
+    StringExecutePromptResponse,
+    StringInputRequest,
     SubmitCompletionActualRequest,
     SubmitCompletionActualsErrorResponse,
     SubmitWorkflowExecutionActualRequest,
@@ -229,6 +244,7 @@ __all__ = [
     "ApiNodeResultData",
     "BadRequestError",
     "BlockTypeEnum",
+    "ChatHistoryInputRequest",
     "ChatMessage",
     "ChatMessageRequest",
     "ChatMessageRole",
@@ -244,8 +260,14 @@ __all__ = [
     "DocumentStatus",
     "EnrichedNormalizedCompletion",
     "EnvironmentEnum",
+    "ErrorExecutePromptResponse",
     "EvaluationParams",
     "EvaluationParamsRequest",
+    "ExecutePromptApiErrorResponse",
+    "ExecutePromptResponse",
+    "ExecutePromptResponse_Error",
+    "ExecutePromptResponse_Json",
+    "ExecutePromptResponse_String",
     "ExecuteWorkflowStreamErrorResponse",
     "FinishReasonEnum",
     "ForbiddenError",
@@ -261,6 +283,8 @@ __all__ = [
     "GenerateStreamResultData",
     "IndexingStateEnum",
     "InternalServerError",
+    "JsonExecutePromptResponse",
+    "JsonInputRequest",
     "LogicalOperator",
     "LogprobsEnum",
     "MetadataFilterConfigRequest",
@@ -291,6 +315,10 @@ __all__ = [
     "PaginatedSlimDocumentList",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
+    "PromptDeploymentInputRequest",
+    "PromptDeploymentInputRequest_ChatHistory",
+    "PromptDeploymentInputRequest_Json",
+    "PromptDeploymentInputRequest_String",
     "PromptNodeResult",
     "PromptNodeResultData",
     "PromptTemplateBlock",
@@ -329,6 +357,8 @@ __all__ = [
     "SearchResultRequest",
     "SearchWeightsRequest",
     "SlimDocument",
+    "StringExecutePromptResponse",
+    "StringInputRequest",
     "SubmitCompletionActualRequest",
     "SubmitCompletionActualsErrorResponse",
     "SubmitWorkflowExecutionActualRequest",

{vellum_ai-0.1.4 → vellum_ai-0.1.5}/src/vellum/client.py

@@ -22,11 +22,12 @@ from .resources.model_versions.client import AsyncModelVersionsClient, ModelVers
 from .resources.registered_prompts.client import AsyncRegisteredPromptsClient, RegisteredPromptsClient
 from .resources.sandboxes.client import AsyncSandboxesClient, SandboxesClient
 from .resources.test_suites.client import AsyncTestSuitesClient, TestSuitesClient
-from .types.generate_error_response import GenerateErrorResponse
+from .types.execute_prompt_response import ExecutePromptResponse
 from .types.generate_options_request import GenerateOptionsRequest
 from .types.generate_request import GenerateRequest
 from .types.generate_response import GenerateResponse
 from .types.generate_stream_response import GenerateStreamResponse
+from .types.prompt_deployment_input_request import PromptDeploymentInputRequest
 from .types.search_request_options_request import SearchRequestOptionsRequest
 from .types.search_response import SearchResponse
 from .types.submit_completion_actual_request import SubmitCompletionActualRequest
@@ -66,6 +67,61 @@ class Vellum:
         self.sandboxes = SandboxesClient(client_wrapper=self._client_wrapper)
         self.test_suites = TestSuitesClient(client_wrapper=self._client_wrapper)
 
+    def execute_prompt(
+        self,
+        *,
+        inputs: typing.List[PromptDeploymentInputRequest],
+        prompt_deployment_id: typing.Optional[str] = OMIT,
+        prompt_deployment_name: typing.Optional[str] = OMIT,
+        release_tag: typing.Optional[str] = OMIT,
+        external_id: typing.Optional[str] = OMIT,
+    ) -> ExecutePromptResponse:
+        """
+        Executes a deployed Prompt and returns the result.
+
+        Parameters:
+            - inputs: typing.List[PromptDeploymentInputRequest].
+
+            - prompt_deployment_id: typing.Optional[str]. The ID of the Prompt Deployment. Must provide either this or prompt_deployment_name.
+
+            - prompt_deployment_name: typing.Optional[str]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+
+            - release_tag: typing.Optional[str]. Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
+
+            - external_id: typing.Optional[str].
+        """
+        _request: typing.Dict[str, typing.Any] = {"inputs": inputs}
+        if prompt_deployment_id is not OMIT:
+            _request["prompt_deployment_id"] = prompt_deployment_id
+        if prompt_deployment_name is not OMIT:
+            _request["prompt_deployment_name"] = prompt_deployment_name
+        if release_tag is not OMIT:
+            _request["release_tag"] = release_tag
+        if external_id is not OMIT:
+            _request["external_id"] = external_id
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/execute-prompt"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=None,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ExecutePromptResponse, _response.json())  # type: ignore
+        if _response.status_code == 400:
+            raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 403:
+            raise ForbiddenError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 500:
+            raise InternalServerError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        try:
+            _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)
+
     def execute_workflow_stream(
         self,
         *,
@@ -187,7 +243,7 @@ class Vellum:
         if _response.status_code == 400:
             raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 403:
-            raise ForbiddenError(pydantic.parse_obj_as(GenerateErrorResponse, _response.json()))  # type: ignore
+            raise ForbiddenError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 404:
             raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:
@@ -244,7 +300,7 @@ class Vellum:
             if _response.status_code == 400:
                 raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 403:
-                raise ForbiddenError(pydantic.parse_obj_as(GenerateErrorResponse, _response.json()))  # type: ignore
+                raise ForbiddenError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 404:
                 raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 500:
@@ -432,6 +488,61 @@ class AsyncVellum:
         self.sandboxes = AsyncSandboxesClient(client_wrapper=self._client_wrapper)
         self.test_suites = AsyncTestSuitesClient(client_wrapper=self._client_wrapper)
 
+    async def execute_prompt(
+        self,
+        *,
+        inputs: typing.List[PromptDeploymentInputRequest],
+        prompt_deployment_id: typing.Optional[str] = OMIT,
+        prompt_deployment_name: typing.Optional[str] = OMIT,
+        release_tag: typing.Optional[str] = OMIT,
+        external_id: typing.Optional[str] = OMIT,
+    ) -> ExecutePromptResponse:
+        """
+        Executes a deployed Prompt and returns the result.
+
+        Parameters:
+            - inputs: typing.List[PromptDeploymentInputRequest].
+
+            - prompt_deployment_id: typing.Optional[str]. The ID of the Prompt Deployment. Must provide either this or prompt_deployment_name.
+
+            - prompt_deployment_name: typing.Optional[str]. The name of the Prompt Deployment. Must provide either this or prompt_deployment_id.
+
+            - release_tag: typing.Optional[str]. Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
+
+            - external_id: typing.Optional[str].
+        """
+        _request: typing.Dict[str, typing.Any] = {"inputs": inputs}
+        if prompt_deployment_id is not OMIT:
+            _request["prompt_deployment_id"] = prompt_deployment_id
+        if prompt_deployment_name is not OMIT:
+            _request["prompt_deployment_name"] = prompt_deployment_name
+        if release_tag is not OMIT:
+            _request["release_tag"] = release_tag
+        if external_id is not OMIT:
+            _request["external_id"] = external_id
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/execute-prompt"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=None,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ExecutePromptResponse, _response.json())  # type: ignore
+        if _response.status_code == 400:
+            raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 403:
+            raise ForbiddenError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        if _response.status_code == 500:
+            raise InternalServerError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+        try:
+            _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)
+
     async def execute_workflow_stream(
         self,
         *,
@@ -553,7 +664,7 @@ class AsyncVellum:
         if _response.status_code == 400:
             raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 403:
-            raise ForbiddenError(pydantic.parse_obj_as(GenerateErrorResponse, _response.json()))  # type: ignore
+            raise ForbiddenError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 404:
             raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:
@@ -610,7 +721,7 @@ class AsyncVellum:
             if _response.status_code == 400:
                 raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 403:
-                raise ForbiddenError(pydantic.parse_obj_as(GenerateErrorResponse, _response.json()))  # type: ignore
+                raise ForbiddenError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 404:
                 raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
             if _response.status_code == 500:

{vellum_ai-0.1.4 → vellum_ai-0.1.5}/src/vellum/core/client_wrapper.py

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

{vellum_ai-0.1.4 → vellum_ai-0.1.5}/src/vellum/errors/forbidden_error.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import typing
+
 from ..core.api_error import ApiError
-from ..types.generate_error_response import GenerateErrorResponse
 
 
 class ForbiddenError(ApiError):
-    def __init__(self, body: GenerateErrorResponse):
+    def __init__(self, body: typing.Any):
         super().__init__(status_code=403, body=body)

{vellum_ai-0.1.4 → vellum_ai-0.1.5}/src/vellum/types/__init__.py

@@ -3,6 +3,7 @@
 from .api_node_result import ApiNodeResult
 from .api_node_result_data import ApiNodeResultData
 from .block_type_enum import BlockTypeEnum
+from .chat_history_input_request import ChatHistoryInputRequest
 from .chat_message import ChatMessage
 from .chat_message_request import ChatMessageRequest
 from .chat_message_role import ChatMessageRole
@@ -17,8 +18,16 @@ from .document_read import DocumentRead
 from .document_status import DocumentStatus
 from .enriched_normalized_completion import EnrichedNormalizedCompletion
 from .environment_enum import EnvironmentEnum
+from .error_execute_prompt_response import ErrorExecutePromptResponse
 from .evaluation_params import EvaluationParams
 from .evaluation_params_request import EvaluationParamsRequest
+from .execute_prompt_api_error_response import ExecutePromptApiErrorResponse
+from .execute_prompt_response import (
+    ExecutePromptResponse,
+    ExecutePromptResponse_Error,
+    ExecutePromptResponse_Json,
+    ExecutePromptResponse_String,
+)
 from .execute_workflow_stream_error_response import ExecuteWorkflowStreamErrorResponse
 from .finish_reason_enum import FinishReasonEnum
 from .generate_error_response import GenerateErrorResponse
@@ -32,6 +41,8 @@ from .generate_stream_response import GenerateStreamResponse
 from .generate_stream_result import GenerateStreamResult
 from .generate_stream_result_data import GenerateStreamResultData
 from .indexing_state_enum import IndexingStateEnum
+from .json_execute_prompt_response import JsonExecutePromptResponse
+from .json_input_request import JsonInputRequest
 from .logical_operator import LogicalOperator
 from .logprobs_enum import LogprobsEnum
 from .metadata_filter_config_request import MetadataFilterConfigRequest
@@ -63,6 +74,12 @@ from .normalized_token_log_probs import NormalizedTokenLogProbs
 from .paginated_slim_document_list import PaginatedSlimDocumentList
 from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .processing_state_enum import ProcessingStateEnum
+from .prompt_deployment_input_request import (
+    PromptDeploymentInputRequest,
+    PromptDeploymentInputRequest_ChatHistory,
+    PromptDeploymentInputRequest_Json,
+    PromptDeploymentInputRequest_String,
+)
 from .prompt_node_result import PromptNodeResult
 from .prompt_node_result_data import PromptNodeResultData
 from .prompt_template_block import PromptTemplateBlock
@@ -101,6 +118,8 @@ from .search_result_merging_request import SearchResultMergingRequest
 from .search_result_request import SearchResultRequest
 from .search_weights_request import SearchWeightsRequest
 from .slim_document import SlimDocument
+from .string_execute_prompt_response import StringExecutePromptResponse
+from .string_input_request import StringInputRequest
 from .submit_completion_actual_request import SubmitCompletionActualRequest
 from .submit_completion_actuals_error_response import SubmitCompletionActualsErrorResponse
 from .submit_workflow_execution_actual_request import (
@@ -231,6 +250,7 @@ __all__ = [
     "ApiNodeResult",
     "ApiNodeResultData",
     "BlockTypeEnum",
+    "ChatHistoryInputRequest",
     "ChatMessage",
     "ChatMessageRequest",
     "ChatMessageRole",
@@ -245,8 +265,14 @@ __all__ = [
     "DocumentStatus",
     "EnrichedNormalizedCompletion",
     "EnvironmentEnum",
+    "ErrorExecutePromptResponse",
     "EvaluationParams",
     "EvaluationParamsRequest",
+    "ExecutePromptApiErrorResponse",
+    "ExecutePromptResponse",
+    "ExecutePromptResponse_Error",
+    "ExecutePromptResponse_Json",
+    "ExecutePromptResponse_String",
     "ExecuteWorkflowStreamErrorResponse",
     "FinishReasonEnum",
     "GenerateErrorResponse",
@@ -260,6 +286,8 @@ __all__ = [
     "GenerateStreamResult",
     "GenerateStreamResultData",
     "IndexingStateEnum",
+    "JsonExecutePromptResponse",
+    "JsonInputRequest",
     "LogicalOperator",
     "LogprobsEnum",
     "MetadataFilterConfigRequest",
@@ -289,6 +317,10 @@ __all__ = [
     "PaginatedSlimDocumentList",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
+    "PromptDeploymentInputRequest",
+    "PromptDeploymentInputRequest_ChatHistory",
+    "PromptDeploymentInputRequest_Json",
+    "PromptDeploymentInputRequest_String",
     "PromptNodeResult",
     "PromptNodeResultData",
     "PromptTemplateBlock",
@@ -327,6 +359,8 @@ __all__ = [
     "SearchResultRequest",
     "SearchWeightsRequest",
     "SlimDocument",
+    "StringExecutePromptResponse",
+    "StringInputRequest",
     "SubmitCompletionActualRequest",
     "SubmitCompletionActualsErrorResponse",
     "SubmitWorkflowExecutionActualRequest",

vellum_ai-0.1.5/src/vellum/types/chat_history_input_request.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .chat_message_request import ChatMessageRequest
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ChatHistoryInputRequest(pydantic.BaseModel):
+    name: str = pydantic.Field(description="The variable's name, as defined in the deployment.")
+    value: typing.List[ChatMessageRequest]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

vellum_ai-0.1.5/src/vellum/types/error_execute_prompt_response.py

@@ -0,0 +1,30 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .vellum_error import VellumError
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ErrorExecutePromptResponse(pydantic.BaseModel):
+    value: VellumError
+    execution_id: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

vellum_ai-0.1.5/src/vellum/types/execute_prompt_api_error_response.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExecutePromptApiErrorResponse(pydantic.BaseModel):
+    detail: str = pydantic.Field(description="Details about why the request failed.")
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

vellum_ai-0.1.5/src/vellum/types/execute_prompt_response.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from .error_execute_prompt_response import ErrorExecutePromptResponse
+from .json_execute_prompt_response import JsonExecutePromptResponse
+from .string_execute_prompt_response import StringExecutePromptResponse
+
+
+class ExecutePromptResponse_Error(ErrorExecutePromptResponse):
+    type: typing_extensions.Literal["ERROR"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ExecutePromptResponse_Json(JsonExecutePromptResponse):
+    type: typing_extensions.Literal["JSON"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ExecutePromptResponse_String(StringExecutePromptResponse):
+    type: typing_extensions.Literal["STRING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+ExecutePromptResponse = typing.Union[
+    ExecutePromptResponse_Error, ExecutePromptResponse_Json, ExecutePromptResponse_String
+]

vellum_ai-0.1.5/src/vellum/types/json_execute_prompt_response.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class JsonExecutePromptResponse(pydantic.BaseModel):
+    value: typing.Dict[str, typing.Any]
+    execution_id: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

vellum_ai-0.1.5/src/vellum/types/json_input_request.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class JsonInputRequest(pydantic.BaseModel):
+    name: str = pydantic.Field(description="The variable's name, as defined in the deployment.")
+    value: typing.Dict[str, typing.Any]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

vellum_ai-0.1.5/src/vellum/types/prompt_deployment_input_request.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from .chat_history_input_request import ChatHistoryInputRequest
+from .json_input_request import JsonInputRequest
+from .string_input_request import StringInputRequest
+
+
+class PromptDeploymentInputRequest_String(StringInputRequest):
+    type: typing_extensions.Literal["STRING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PromptDeploymentInputRequest_Json(JsonInputRequest):
+    type: typing_extensions.Literal["JSON"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PromptDeploymentInputRequest_ChatHistory(ChatHistoryInputRequest):
+    type: typing_extensions.Literal["CHAT_HISTORY"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+PromptDeploymentInputRequest = typing.Union[
+    PromptDeploymentInputRequest_String, PromptDeploymentInputRequest_Json, PromptDeploymentInputRequest_ChatHistory
+]