vellum-ai 0.7.2__tar.gz → 0.7.3__tar.gz

{vellum_ai-0.7.2 → vellum_ai-0.7.3}/PKG-INFO

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

{vellum_ai-0.7.2 → vellum_ai-0.7.3}/pyproject.toml

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

{vellum_ai-0.7.2 → vellum_ai-0.7.3}/src/vellum/__init__.py

@@ -76,6 +76,8 @@ from .types import (
     CodeExecutionNodeResultOutput_String,
     CodeExecutionNodeSearchResultsResult,
     CodeExecutionNodeStringResult,
+    CompilePromptDeploymentExpandMetaRequest,
+    CompilePromptMeta,
     ConditionalNodeResult,
     ConditionalNodeResultData,
     CreateEnum,
@@ -734,6 +736,8 @@ __all__ = [
     "CodeExecutionNodeResultOutput_String",
     "CodeExecutionNodeSearchResultsResult",
     "CodeExecutionNodeStringResult",
+    "CompilePromptDeploymentExpandMetaRequest",
+    "CompilePromptMeta",
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
     "CreateEnum",

{vellum_ai-0.7.2 → vellum_ai-0.7.3}/src/vellum/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.7.2",
+            "X-Fern-SDK-Version": "0.7.3",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.7.2 → vellum_ai-0.7.3}/src/vellum/resources/deployments/client.py

@@ -14,6 +14,7 @@ from ...errors.bad_request_error import BadRequestError
 from ...errors.forbidden_error import ForbiddenError
 from ...errors.internal_server_error import InternalServerError
 from ...errors.not_found_error import NotFoundError
+from ...types.compile_prompt_deployment_expand_meta_request import CompilePromptDeploymentExpandMetaRequest
 from ...types.deployment_provider_payload_response import DeploymentProviderPayloadResponse
 from ...types.deployment_read import DeploymentRead
 from ...types.deployment_release_tag_read import DeploymentReleaseTagRead
@@ -276,6 +277,7 @@ class DeploymentsClient:
         deployment_name: typing.Optional[str] = OMIT,
         inputs: typing.Sequence[PromptDeploymentInputRequest],
         release_tag: typing.Optional[str] = OMIT,
+        expand_meta: typing.Optional[CompilePromptDeploymentExpandMetaRequest] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DeploymentProviderPayloadResponse:
         """
@@ -288,6 +290,8 @@ class DeploymentsClient:
 
             - release_tag: typing.Optional[str]. Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
+            - expand_meta: typing.Optional[CompilePromptDeploymentExpandMetaRequest].
+
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from vellum.client import Vellum
@@ -306,6 +310,8 @@ class DeploymentsClient:
             _request["deployment_name"] = deployment_name
         if release_tag is not OMIT:
             _request["release_tag"] = release_tag
+        if expand_meta is not OMIT:
+            _request["expand_meta"] = expand_meta
         _response = self._client_wrapper.httpx_client.request(
             method="POST",
             url=urllib.parse.urljoin(
@@ -602,6 +608,7 @@ class AsyncDeploymentsClient:
         deployment_name: typing.Optional[str] = OMIT,
         inputs: typing.Sequence[PromptDeploymentInputRequest],
         release_tag: typing.Optional[str] = OMIT,
+        expand_meta: typing.Optional[CompilePromptDeploymentExpandMetaRequest] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DeploymentProviderPayloadResponse:
         """
@@ -614,6 +621,8 @@ class AsyncDeploymentsClient:
 
             - release_tag: typing.Optional[str]. Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
+            - expand_meta: typing.Optional[CompilePromptDeploymentExpandMetaRequest].
+
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from vellum.client import AsyncVellum
@@ -632,6 +641,8 @@ class AsyncDeploymentsClient:
             _request["deployment_name"] = deployment_name
         if release_tag is not OMIT:
             _request["release_tag"] = release_tag
+        if expand_meta is not OMIT:
+            _request["expand_meta"] = expand_meta
         _response = await self._client_wrapper.httpx_client.request(
             method="POST",
             url=urllib.parse.urljoin(

{vellum_ai-0.7.2 → vellum_ai-0.7.3}/src/vellum/types/__init__.py

@@ -99,6 +99,8 @@ from .code_execution_node_result_output import (
 )
 from .code_execution_node_search_results_result import CodeExecutionNodeSearchResultsResult
 from .code_execution_node_string_result import CodeExecutionNodeStringResult
+from .compile_prompt_deployment_expand_meta_request import CompilePromptDeploymentExpandMetaRequest
+from .compile_prompt_meta import CompilePromptMeta
 from .conditional_node_result import ConditionalNodeResult
 from .conditional_node_result_data import ConditionalNodeResultData
 from .create_enum import CreateEnum
@@ -798,6 +800,8 @@ __all__ = [
     "CodeExecutionNodeResultOutput_String",
     "CodeExecutionNodeSearchResultsResult",
     "CodeExecutionNodeStringResult",
+    "CompilePromptDeploymentExpandMetaRequest",
+    "CompilePromptMeta",
     "ConditionalNodeResult",
     "ConditionalNodeResultData",
     "CreateEnum",

vellum_ai-0.7.3/src/vellum/types/compile_prompt_deployment_expand_meta_request.py

@@ -0,0 +1,38 @@
+# 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 ..core.pydantic_utilities import pydantic_v1
+
+
+class CompilePromptDeploymentExpandMetaRequest(pydantic_v1.BaseModel):
+    model_name: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    """
+    If enabled, the response will include the model identifier representing the ML Model invoked by the Prompt.
+    """
+
+    deployment_release_tag: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    """
+    If enabled, the response will include the release tag of the Prompt Deployment.
+    """
+
+    prompt_version_id: typing.Optional[bool] = pydantic_v1.Field(default=None)
+    """
+    If enabled, the response will include the ID of the Prompt Version backing the deployment.
+    """
+
+    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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum_ai-0.7.3/src/vellum/types/compile_prompt_meta.py

@@ -0,0 +1,31 @@
+# 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 ..core.pydantic_utilities import pydantic_v1
+
+
+class CompilePromptMeta(pydantic_v1.BaseModel):
+    """
+    The subset of the metadata tracked by Vellum during Prompt Deployment compilation that the request opted into with `expand_meta`.
+    """
+
+    model_name: typing.Optional[str] = None
+    deployment_release_tag: typing.Optional[str] = None
+    prompt_version_id: typing.Optional[str] = None
+
+    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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

{vellum_ai-0.7.2 → vellum_ai-0.7.3}/src/vellum/types/deployment_provider_payload_response.py

@@ -5,11 +5,13 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
+from .compile_prompt_meta import CompilePromptMeta
 from .deployment_provider_payload_response_payload import DeploymentProviderPayloadResponsePayload
 
 
 class DeploymentProviderPayloadResponse(pydantic_v1.BaseModel):
     payload: DeploymentProviderPayloadResponsePayload
+    meta: typing.Optional[CompilePromptMeta] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}