vellum-ai 0.8.7__py3-none-any.whl → 0.8.9__py3-none-any.whl

vellum/__init__.py

@@ -209,6 +209,7 @@ from .types import (
     PdfSearchResultMetaSource,
     PdfSearchResultMetaSourceRequest,
     PlainTextPromptBlockRequest,
+    Price,
     ProcessingFailureReasonEnum,
     ProcessingStateEnum,
     PromptBlockRequest,
@@ -333,6 +334,7 @@ from .types import (
     TestSuiteRunExternalExecConfigDataRequest,
     TestSuiteRunExternalExecConfigRequest,
     TestSuiteRunMetricErrorOutput,
+    TestSuiteRunMetricJsonOutput,
     TestSuiteRunMetricNumberOutput,
     TestSuiteRunMetricOutput,
     TestSuiteRunMetricStringOutput,
@@ -362,6 +364,7 @@ from .types import (
     TokenOverlappingWindowChunkerConfigRequest,
     TokenOverlappingWindowChunking,
     TokenOverlappingWindowChunkingRequest,
+    UnitEnum,
     UploadDocumentResponse,
     UpsertTestSuiteTestCaseRequest,
     VariablePromptBlockRequest,
@@ -651,6 +654,7 @@ __all__ = [
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
     "PlainTextPromptBlockRequest",
+    "Price",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
     "PromptBlockRequest",
@@ -775,6 +779,7 @@ __all__ = [
     "TestSuiteRunExternalExecConfigDataRequest",
     "TestSuiteRunExternalExecConfigRequest",
     "TestSuiteRunMetricErrorOutput",
+    "TestSuiteRunMetricJsonOutput",
     "TestSuiteRunMetricNumberOutput",
     "TestSuiteRunMetricOutput",
     "TestSuiteRunMetricStringOutput",
@@ -804,6 +809,7 @@ __all__ = [
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",
     "TokenOverlappingWindowChunkingRequest",
+    "UnitEnum",
     "UploadDocumentResponse",
     "UpsertTestSuiteTestCaseRequest",
     "VariablePromptBlockRequest",

vellum/client.py

@@ -344,6 +344,7 @@ class Vellum:
             expand_meta=PromptDeploymentExpandMetaRequest(
                 model_name=True,
                 usage=True,
+                cost=True,
                 finish_reason=True,
                 latency=True,
                 deployment_release_tag=True,
@@ -1535,6 +1536,7 @@ class AsyncVellum:
                 expand_meta=PromptDeploymentExpandMetaRequest(
                     model_name=True,
                     usage=True,
+                    cost=True,
                     finish_reason=True,
                     latency=True,
                     deployment_release_tag=True,

vellum/core/client_wrapper.py

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

vellum/types/__init__.py

@@ -216,6 +216,7 @@ from .paginated_test_suite_test_case_list import PaginatedTestSuiteTestCaseList
 from .pdf_search_result_meta_source import PdfSearchResultMetaSource
 from .pdf_search_result_meta_source_request import PdfSearchResultMetaSourceRequest
 from .plain_text_prompt_block_request import PlainTextPromptBlockRequest
+from .price import Price
 from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .processing_state_enum import ProcessingStateEnum
 from .prompt_block_request import PromptBlockRequest
@@ -342,6 +343,7 @@ from .test_suite_run_external_exec_config_data import TestSuiteRunExternalExecCo
 from .test_suite_run_external_exec_config_data_request import TestSuiteRunExternalExecConfigDataRequest
 from .test_suite_run_external_exec_config_request import TestSuiteRunExternalExecConfigRequest
 from .test_suite_run_metric_error_output import TestSuiteRunMetricErrorOutput
+from .test_suite_run_metric_json_output import TestSuiteRunMetricJsonOutput
 from .test_suite_run_metric_number_output import TestSuiteRunMetricNumberOutput
 from .test_suite_run_metric_output import TestSuiteRunMetricOutput
 from .test_suite_run_metric_string_output import TestSuiteRunMetricStringOutput
@@ -373,6 +375,7 @@ from .token_overlapping_window_chunker_config import TokenOverlappingWindowChunk
 from .token_overlapping_window_chunker_config_request import TokenOverlappingWindowChunkerConfigRequest
 from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
 from .token_overlapping_window_chunking_request import TokenOverlappingWindowChunkingRequest
+from .unit_enum import UnitEnum
 from .upload_document_response import UploadDocumentResponse
 from .upsert_test_suite_test_case_request import UpsertTestSuiteTestCaseRequest
 from .variable_prompt_block_request import VariablePromptBlockRequest
@@ -635,6 +638,7 @@ __all__ = [
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
     "PlainTextPromptBlockRequest",
+    "Price",
     "ProcessingFailureReasonEnum",
     "ProcessingStateEnum",
     "PromptBlockRequest",
@@ -759,6 +763,7 @@ __all__ = [
     "TestSuiteRunExternalExecConfigDataRequest",
     "TestSuiteRunExternalExecConfigRequest",
     "TestSuiteRunMetricErrorOutput",
+    "TestSuiteRunMetricJsonOutput",
     "TestSuiteRunMetricNumberOutput",
     "TestSuiteRunMetricOutput",
     "TestSuiteRunMetricStringOutput",
@@ -788,6 +793,7 @@ __all__ = [
     "TokenOverlappingWindowChunkerConfigRequest",
     "TokenOverlappingWindowChunking",
     "TokenOverlappingWindowChunkingRequest",
+    "UnitEnum",
     "UploadDocumentResponse",
     "UpsertTestSuiteTestCaseRequest",
     "VariablePromptBlockRequest",

vellum/types/ad_hoc_expand_meta_request.py

@@ -7,7 +7,11 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class AdHocExpandMetaRequest(UniversalBaseModel):
-    cost: typing.Optional[bool] = None
+    cost: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    If enabled, the response will include model host cost tracking. This may increase latency for some model hosts.
+    """
+
     model_name: typing.Optional[bool] = pydantic.Field(default=None)
     """
     If enabled, the response will include the model identifier representing the ML Model invoked by the Prompt.

vellum/types/price.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .unit_enum import UnitEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class Price(UniversalBaseModel):
+    value: float
+    unit: UnitEnum = "USD"
+
+    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_deployment_expand_meta_request.py

@@ -17,6 +17,11 @@ class PromptDeploymentExpandMetaRequest(UniversalBaseModel):
     If enabled, the response will include model host usage tracking. This may increase latency for some model hosts.
     """
 
+    cost: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    If enabled, the response will include model host cost tracking. This may increase latency for some model hosts.
+    """
+
     finish_reason: typing.Optional[bool] = pydantic.Field(default=None)
     """
     If enabled, the response will include the reason provided by the model for why the execution finished.

vellum/types/prompt_execution_meta.py

@@ -4,6 +4,7 @@ from ..core.pydantic_utilities import UniversalBaseModel
 import typing
 from .finish_reason_enum import FinishReasonEnum
 from .ml_model_usage import MlModelUsage
+from .price import Price
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
@@ -19,6 +20,7 @@ class PromptExecutionMeta(UniversalBaseModel):
     prompt_version_id: typing.Optional[str] = None
     finish_reason: typing.Optional[FinishReasonEnum] = None
     usage: typing.Optional[MlModelUsage] = None
+    cost: typing.Optional[Price] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/types/test_suite_run_metric_json_output.py

@@ -0,0 +1,25 @@
+# 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 TestSuiteRunMetricJsonOutput(UniversalBaseModel):
+    """
+    Output for a test suite run metric that is of type NUMBER
+    """
+
+    value: typing.Optional[typing.Any] = None
+    type: typing.Literal["JSON"] = "JSON"
+    name: str
+
+    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/test_suite_run_metric_output.py

@@ -3,8 +3,12 @@
 import typing
 from .test_suite_run_metric_string_output import TestSuiteRunMetricStringOutput
 from .test_suite_run_metric_number_output import TestSuiteRunMetricNumberOutput
+from .test_suite_run_metric_json_output import TestSuiteRunMetricJsonOutput
 from .test_suite_run_metric_error_output import TestSuiteRunMetricErrorOutput
 
 TestSuiteRunMetricOutput = typing.Union[
-    TestSuiteRunMetricStringOutput, TestSuiteRunMetricNumberOutput, TestSuiteRunMetricErrorOutput
+    TestSuiteRunMetricStringOutput,
+    TestSuiteRunMetricNumberOutput,
+    TestSuiteRunMetricJsonOutput,
+    TestSuiteRunMetricErrorOutput,
 ]

vellum/types/unit_enum.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+UnitEnum = typing.Literal["USD"]

{vellum_ai-0.8.7.dist-info → vellum_ai-0.8.9.dist-info}/METADATA

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

{vellum_ai-0.8.7.dist-info → vellum_ai-0.8.9.dist-info}/RECORD

@@ -1,8 +1,8 @@
-vellum/__init__.py,sha256=sYfFTNke8uuFbEl3rRMDj2DIsxjSxRicUW_uxEx9ftI,30024
-vellum/client.py,sha256=kUF5BvbURnGFgxkJC-ql8GZ6lRX7_i3h3EQNCC_SRIo,101036
+vellum/__init__.py,sha256=ZjILSZ1gKX9YUZjjOzr0wngRMV7aYaCvuCXgoxt7n2o,30148
+vellum/client.py,sha256=FzaKpt5tvigJpdckmhhtIZemZnt_qM-50WfXENgyA74,101094
 vellum/core/__init__.py,sha256=FzSvKbXjuM18Hdk3iGK-jsGY_DfouyRS659thZV5c1Y,1394
 vellum/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/core/client_wrapper.py,sha256=RpTIosTnX7hFWiqTFmJqDkLpt_WlOMc8YTxVl366zNY,1897
+vellum/core/client_wrapper.py,sha256=ufeiagHJMhTy-ivorHYOOgDbHND76t2jMmfgX9VcJN8,1897
 vellum/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/core/file.py,sha256=vliNmlB7PbDfi4EKiVPNq5QaGXJ4zlDBGupv7Qciy7g,1520
 vellum/core/http_client.py,sha256=rZSidd9LazCjduvdBhZ7GDs4iGfn-OAfRGsDXB1z8f4,18882
@@ -66,9 +66,9 @@ vellum/terraform/ml_model/__init__.py,sha256=I8h1Ru-Rb-Hi_HusK6G7nJQZEKQGsAAHMmw
 vellum/terraform/provider/__init__.py,sha256=-06xKmAmknpohVzw5TD-t1bnUHta8OrQYqvMd04XM-U,12684
 vellum/terraform/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 vellum/terraform/versions.json,sha256=45c7jjRD5i4w9DJQHs5ZqLLVXRnQwP9Rirq3mWY-xEo,56
-vellum/types/__init__.py,sha256=ny17OcNpC81ttCOqeLChuV89DykV0hzB_artvlSK1nk,45850
+vellum/types/__init__.py,sha256=_v6i_DtXNY5yiWC8FnzvVP2T2hpd_G9m2Rgsupfzj3c,46048
 vellum/types/ad_hoc_execute_prompt_event.py,sha256=bCjujA2XsOgyF3bRZbcEqV2rOIymRgsLoIRtZpB14xg,607
-vellum/types/ad_hoc_expand_meta_request.py,sha256=MzfXqeR8jxVjO1hrdTpPTWD7cg6lkaWcQbcRh1b6tH8,1181
+vellum/types/ad_hoc_expand_meta_request.py,sha256=hS8PC3hC--OKvRKi2ZFj_RJPQ1bxo2GXno8qJq1kk28,1338
 vellum/types/ad_hoc_fulfilled_prompt_execution_meta.py,sha256=Bfvf1d_dkmshxRACVM5vcxbH_7AQY23RmrrnPc0ytYY,939
 vellum/types/ad_hoc_initiated_prompt_execution_meta.py,sha256=gOHP4kmnVCJJkTpfRPOgkFB70mhz5ySg_c2qRffDGWQ,785
 vellum/types/ad_hoc_rejected_prompt_execution_meta.py,sha256=Yo3TNjemHUfxU9R3EMzAGh-ey9pSClpEX7uGgGZiXDg,849
@@ -275,13 +275,14 @@ vellum/types/paginated_test_suite_test_case_list.py,sha256=SANqBECiz9yu6_JxOcNdk
 vellum/types/pdf_search_result_meta_source.py,sha256=EMVhqdN1bwE6Ujdx4VhlmKQtJvitN-57kY8oZPxh9dI,1126
 vellum/types/pdf_search_result_meta_source_request.py,sha256=nUhaD2Kw1paGC6O_ICVNu3R0e1SVgTshRTkGNgmcjXo,1133
 vellum/types/plain_text_prompt_block_request.py,sha256=ljdh2OD-xUOAhEF8raHc6WnQDnWdtEY8aSNsEiuC5nY,971
+vellum/types/price.py,sha256=ewzXDBVLaleuXMVQ-gQ3G1Nl5J2OWOVEMEFfnQIpiTk,610
 vellum/types/processing_failure_reason_enum.py,sha256=R_KIW7TcQejhc-vLhtNf9SdkYADgoZCn4ch4_RRIvsI,195
 vellum/types/processing_state_enum.py,sha256=lIEunnCpgYQExm2bGyTb12KyjQ3O7XOx636aWXb_Iwo,190
 vellum/types/prompt_block_request.py,sha256=A5I5J6bDjmauk5L4CexRXk_VMrGDZRApYZyevMHBPpI,740
 vellum/types/prompt_block_state.py,sha256=BRAzTYARoSU36IVZGWMeeqhl5fgFMXCyhJ8rCbfB-f0,163
-vellum/types/prompt_deployment_expand_meta_request.py,sha256=BO06BkRCjzeBNeUnDNBbx-cjQ8UFhX2RCq8FngEC6HY,1714
+vellum/types/prompt_deployment_expand_meta_request.py,sha256=agsiAaHB6lDoZPlnfJ2nmhB4Ud4EiJJTX05YmduyCPo,1910
 vellum/types/prompt_deployment_input_request.py,sha256=KrT4-Ew2VvTWXEkYQz2oyHn5EDOgrMW7FzRFaPH3ARg,353
-vellum/types/prompt_execution_meta.py,sha256=OEkOFFXTJNwWtH_ecF32GqIqDMyDeuBos_ThRlXe9lg,1076
+vellum/types/prompt_execution_meta.py,sha256=3hhMZgdAR5mKfnh2e_eVN3oKfT0E9w26khVPrpjn7jk,1141
 vellum/types/prompt_node_execution_meta.py,sha256=k8zPFTsXl0iuHi-CNXEQcWP5Adbr8xDv1G1EWZ4KJNM,779
 vellum/types/prompt_node_result.py,sha256=3jewO-nPodoXTq_5RxgwhKfDZrvoPjRZ_vUXLeqiuHY,749
 vellum/types/prompt_node_result_data.py,sha256=fNOxBfK3ablDBxkUWVVstJMYaGdHGgu27WxP87E6UQ4,872
@@ -399,8 +400,9 @@ vellum/types/test_suite_run_external_exec_config_data.py,sha256=tCbioorIkNJYmvaa
 vellum/types/test_suite_run_external_exec_config_data_request.py,sha256=KuNf7pnIVkcvJYFzyDs4TCBU5lWeIweAe9czD3jREzw,847
 vellum/types/test_suite_run_external_exec_config_request.py,sha256=CwWiz7I1Yx4ETIRW_tr5ZwAnTGKyLlzuefqAExK4_LQ,1114
 vellum/types/test_suite_run_metric_error_output.py,sha256=trW1KTH4SJafRfDFRyBS7TpzcT88tXkpWEoysij6IXQ,755
+vellum/types/test_suite_run_metric_json_output.py,sha256=DI3mJR5kpi8Hm2n6tx6buySHPwTkkbBbr8Lkk4k6HH4,738
 vellum/types/test_suite_run_metric_number_output.py,sha256=8pddeSds6Rrn0xGqyvgPsG1hr1tu6eOiQAp8kkM_aBk,739
-vellum/types/test_suite_run_metric_output.py,sha256=ZrDEp3_1nao9KdBHR3eqgDkHWu1SvFP_JSmsU83RT1s,459
+vellum/types/test_suite_run_metric_output.py,sha256=qxEJ6Ii4XOhJpM3tgn6ctRnldysduEIe8tQFnWBEOvQ,578
 vellum/types/test_suite_run_metric_string_output.py,sha256=YXerGfpvJdBtKrzgutSqEfG-N6cZoeOL59qZ5k6DwQA,737
 vellum/types/test_suite_run_read.py,sha256=oBA5_N6tUgc9wTM2I06d_1F7B5ZQcndPFumyZH4J1PM,1222
 vellum/types/test_suite_run_state.py,sha256=E4f_AfzXBnxhObLLZ12dBzdoYlRm-gaTqkzrZQ_KfCo,197
@@ -428,6 +430,7 @@ vellum/types/token_overlapping_window_chunker_config.py,sha256=_8vR9AiZQmb5OA3Oo
 vellum/types/token_overlapping_window_chunker_config_request.py,sha256=O58w5om6EsCgZeqM7n3KSzwo1PqINyHWln46EFW4Inc,738
 vellum/types/token_overlapping_window_chunking.py,sha256=TghiPKWZg3Eg_UzGI9VmjQgVPZFABrnhfsz4iPLEem8,889
 vellum/types/token_overlapping_window_chunking_request.py,sha256=IjCs9UDrwBT6tnffdInd3278Lg2L46jXgn0V-sEBwb0,918
+vellum/types/unit_enum.py,sha256=BKWRVp2WfHtGK4D6TsolhNJHGHfExzrRHkFn8H8QkwQ,113
 vellum/types/upload_document_response.py,sha256=6_5Cm4yBPq5nD-rEql6GsmrAtSVVtNRczOL5YwsBVMI,649
 vellum/types/upsert_test_suite_test_case_request.py,sha256=nBL0DR9SksQCLA9MhmwHNEOsmdY7bklXsrtge-ywANI,1812
 vellum/types/variable_prompt_block_request.py,sha256=XYiA3R_jaMZ2Mq37Lbh7CfBqqj93Yv8ZMVYGheodBdY,990
@@ -481,7 +484,7 @@ vellum/types/workflow_result_event_output_data_search_results.py,sha256=U34IK7Zv
 vellum/types/workflow_result_event_output_data_string.py,sha256=tM3kgh6tEhD0dFEb_7UU0-UspeN4pUdINCcCrD64W74,1228
 vellum/types/workflow_stream_event.py,sha256=Wn3Yzuy9MqWAeo8tEaXDTKDEbJoA8DdYdMVq8EKuhu8,361
 vellum/version.py,sha256=jq-1PlAYxN9AXuaZqbYk9ak27SgE2lw9Ia5gx1b1gVI,76
-vellum_ai-0.8.7.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
-vellum_ai-0.8.7.dist-info/METADATA,sha256=Ch6_D1ekSO1Mv1wB8Zp4OHo0oQ6lvCaR5nmCiHqPX_A,4394
-vellum_ai-0.8.7.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-vellum_ai-0.8.7.dist-info/RECORD,,
+vellum_ai-0.8.9.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
+vellum_ai-0.8.9.dist-info/METADATA,sha256=ApCmRTG3GYNsYwkGWltiBbBqNx03p4HsJzr9ikoVJtA,4394
+vellum_ai-0.8.9.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+vellum_ai-0.8.9.dist-info/RECORD,,