vellum-ai 0.3.23__py3-none-any.whl → 0.5.1__py3-none-any.whl

vellum/types/test_suite_run_external_exec_config_data_request.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .external_test_case_execution_request import ExternalTestCaseExecutionRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunExternalExecConfigDataRequest(pydantic.BaseModel):
-    executions: typing.List[ExternalTestCaseExecutionRequest] = pydantic.Field()
+class TestSuiteRunExternalExecConfigDataRequest(pydantic_v1.BaseModel):
+    executions: typing.List[ExternalTestCaseExecutionRequest] = pydantic_v1.Field()
     """
     The executions of some callable external to Vellum whose outputs you would like to evaluate.
     """
@@ -29,5 +25,5 @@ class TestSuiteRunExternalExecConfigDataRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_external_exec_config_request.py

@@ -4,21 +4,17 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .test_suite_run_external_exec_config_data_request import TestSuiteRunExternalExecConfigDataRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunExternalExecConfigRequest(pydantic.BaseModel):
+class TestSuiteRunExternalExecConfigRequest(pydantic_v1.BaseModel):
     """
     Execution configuration for running a Vellum Test Suite against an external callable
     """
 
     data: TestSuiteRunExternalExecConfigDataRequest
-    test_case_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    test_case_ids: typing.Optional[typing.List[str]] = pydantic_v1.Field(default=None)
     """
     Optionally specify a subset of test case ids to run. If not provided, all test cases within the test suite will be run by default.
     """
@@ -34,5 +30,5 @@ class TestSuiteRunExternalExecConfigRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_metric_error_output.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error import VellumError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunMetricErrorOutput(pydantic.BaseModel):
+class TestSuiteRunMetricErrorOutput(pydantic_v1.BaseModel):
     """
     Output for a test suite run metric that is of type ERROR
     """
@@ -31,5 +27,5 @@ class TestSuiteRunMetricErrorOutput(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_metric_number_output.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunMetricNumberOutput(pydantic.BaseModel):
+class TestSuiteRunMetricNumberOutput(pydantic_v1.BaseModel):
     """
     Output for a test suite run metric that is of type NUMBER
     """
@@ -30,5 +26,5 @@ class TestSuiteRunMetricNumberOutput(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_metric_string_output.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunMetricStringOutput(pydantic.BaseModel):
+class TestSuiteRunMetricStringOutput(pydantic_v1.BaseModel):
     """
     Output for a test suite run metric that is of type STRING
     """
@@ -30,5 +26,5 @@ class TestSuiteRunMetricStringOutput(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_read.py

@@ -4,21 +4,17 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .test_suite_run_exec_config import TestSuiteRunExecConfig
 from .test_suite_run_state import TestSuiteRunState
 from .test_suite_run_test_suite import TestSuiteRunTestSuite
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunRead(pydantic.BaseModel):
+class TestSuiteRunRead(pydantic_v1.BaseModel):
     id: str
     created: dt.datetime
     test_suite: TestSuiteRunTestSuite
-    state: TestSuiteRunState = pydantic.Field()
+    state: TestSuiteRunState = pydantic_v1.Field()
     """
     The current state of this run
     
@@ -29,7 +25,7 @@ class TestSuiteRunRead(pydantic.BaseModel):
     - `CANCELLED` - Cancelled
     """
 
-    exec_config: typing.Optional[TestSuiteRunExecConfig] = pydantic.Field(default=None)
+    exec_config: typing.Optional[TestSuiteRunExecConfig] = pydantic_v1.Field(default=None)
     """
     Configuration that defines how the Test Suite should be run
     """
@@ -45,5 +41,5 @@ class TestSuiteRunRead(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_state.py

@@ -1,41 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class TestSuiteRunState(str, enum.Enum):
-    """
-    - `QUEUED` - Queued
-    - `RUNNING` - Running
-    - `COMPLETE` - Complete
-    - `FAILED` - Failed
-    - `CANCELLED` - Cancelled
-    """
-
-    QUEUED = "QUEUED"
-    RUNNING = "RUNNING"
-    COMPLETE = "COMPLETE"
-    FAILED = "FAILED"
-    CANCELLED = "CANCELLED"
-
-    def visit(
-        self,
-        queued: typing.Callable[[], T_Result],
-        running: typing.Callable[[], T_Result],
-        complete: typing.Callable[[], T_Result],
-        failed: typing.Callable[[], T_Result],
-        cancelled: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is TestSuiteRunState.QUEUED:
-            return queued()
-        if self is TestSuiteRunState.RUNNING:
-            return running()
-        if self is TestSuiteRunState.COMPLETE:
-            return complete()
-        if self is TestSuiteRunState.FAILED:
-            return failed()
-        if self is TestSuiteRunState.CANCELLED:
-            return cancelled()
+TestSuiteRunState = typing.Union[typing.Literal["QUEUED", "RUNNING", "COMPLETE", "FAILED", "CANCELLED"], typing.Any]

vellum/types/test_suite_run_test_suite.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunTestSuite(pydantic.BaseModel):
+class TestSuiteRunTestSuite(pydantic_v1.BaseModel):
     id: str
     history_item_id: str
     label: str
@@ -27,5 +23,5 @@ class TestSuiteRunTestSuite(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_workflow_release_tag_exec_config.py

@@ -4,21 +4,17 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .test_suite_run_workflow_release_tag_exec_config_data import TestSuiteRunWorkflowReleaseTagExecConfigData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunWorkflowReleaseTagExecConfig(pydantic.BaseModel):
+class TestSuiteRunWorkflowReleaseTagExecConfig(pydantic_v1.BaseModel):
     """
     Execution configuration for running a Test Suite against a Workflow Deployment
     """
 
     data: TestSuiteRunWorkflowReleaseTagExecConfigData
-    test_case_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    test_case_ids: typing.Optional[typing.List[str]] = pydantic_v1.Field(default=None)
     """
     Optionally specify a subset of test case ids to run. If not provided, all test cases within the test suite will be run by default.
     """
@@ -34,5 +30,5 @@ class TestSuiteRunWorkflowReleaseTagExecConfig(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_workflow_release_tag_exec_config_data.py

@@ -4,20 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunWorkflowReleaseTagExecConfigData(pydantic.BaseModel):
-    workflow_deployment_id: str = pydantic.Field()
+class TestSuiteRunWorkflowReleaseTagExecConfigData(pydantic_v1.BaseModel):
+    workflow_deployment_id: str = pydantic_v1.Field()
     """
     The ID of the Workflow Deployment to run the Test Suite against.
     """
 
-    tag: typing.Optional[str] = pydantic.Field(default=None)
+    tag: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     A tag identifying which release of the Workflow Deployment to run the Test Suite against. Useful for testing past versions of the Workflow Deployment
     """
@@ -33,5 +29,5 @@ class TestSuiteRunWorkflowReleaseTagExecConfigData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_workflow_release_tag_exec_config_data_request.py

@@ -4,20 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunWorkflowReleaseTagExecConfigDataRequest(pydantic.BaseModel):
-    workflow_deployment_id: str = pydantic.Field()
+class TestSuiteRunWorkflowReleaseTagExecConfigDataRequest(pydantic_v1.BaseModel):
+    workflow_deployment_id: str = pydantic_v1.Field()
     """
     The ID of the Workflow Deployment to run the Test Suite against.
     """
 
-    tag: typing.Optional[str] = pydantic.Field(default=None)
+    tag: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     A tag identifying which release of the Workflow Deployment to run the Test Suite against. Useful for testing past versions of the Workflow Deployment
     """
@@ -33,5 +29,5 @@ class TestSuiteRunWorkflowReleaseTagExecConfigDataRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_run_workflow_release_tag_exec_config_request.py

@@ -4,23 +4,19 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .test_suite_run_workflow_release_tag_exec_config_data_request import (
     TestSuiteRunWorkflowReleaseTagExecConfigDataRequest,
 )
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteRunWorkflowReleaseTagExecConfigRequest(pydantic.BaseModel):
+class TestSuiteRunWorkflowReleaseTagExecConfigRequest(pydantic_v1.BaseModel):
     """
     Execution configuration for running a Test Suite against a Workflow Deployment
     """
 
     data: TestSuiteRunWorkflowReleaseTagExecConfigDataRequest
-    test_case_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    test_case_ids: typing.Optional[typing.List[str]] = pydantic_v1.Field(default=None)
     """
     Optionally specify a subset of test case ids to run. If not provided, all test cases within the test suite will be run by default.
     """
@@ -36,5 +32,5 @@ class TestSuiteRunWorkflowReleaseTagExecConfigRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/test_suite_test_case.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .test_case_variable_value import TestCaseVariableValue
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class TestSuiteTestCase(pydantic.BaseModel):
+class TestSuiteTestCase(pydantic_v1.BaseModel):
     id: typing.Optional[str] = None
     label: typing.Optional[str] = None
     input_values: typing.List[TestCaseVariableValue]
@@ -29,5 +25,5 @@ class TestSuiteTestCase(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/upload_document_error_response.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class UploadDocumentErrorResponse(pydantic.BaseModel):
+class UploadDocumentErrorResponse(pydantic_v1.BaseModel):
     detail: str
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class UploadDocumentErrorResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/upload_document_response.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class UploadDocumentResponse(pydantic.BaseModel):
-    document_id: str = pydantic.Field()
+class UploadDocumentResponse(pydantic_v1.BaseModel):
+    document_id: str = pydantic_v1.Field()
     """
     The ID of the newly created document.
     """
@@ -28,5 +24,5 @@ class UploadDocumentResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/vellum_error.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error_code_enum import VellumErrorCodeEnum
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class VellumError(pydantic.BaseModel):
+class VellumError(pydantic_v1.BaseModel):
     message: str
     code: VellumErrorCodeEnum
 
@@ -27,5 +23,5 @@ class VellumError(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/vellum_error_code_enum.py

@@ -1,36 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class VellumErrorCodeEnum(str, enum.Enum):
-    """
-    - `INVALID_REQUEST` - INVALID_REQUEST
-    - `PROVIDER_ERROR` - PROVIDER_ERROR
-    - `INTERNAL_SERVER_ERROR` - INTERNAL_SERVER_ERROR
-    - `USER_DEFINED_ERROR` - USER_DEFINED_ERROR
-    """
-
-    INVALID_REQUEST = "INVALID_REQUEST"
-    PROVIDER_ERROR = "PROVIDER_ERROR"
-    INTERNAL_SERVER_ERROR = "INTERNAL_SERVER_ERROR"
-    USER_DEFINED_ERROR = "USER_DEFINED_ERROR"
-
-    def visit(
-        self,
-        invalid_request: typing.Callable[[], T_Result],
-        provider_error: typing.Callable[[], T_Result],
-        internal_server_error: typing.Callable[[], T_Result],
-        user_defined_error: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is VellumErrorCodeEnum.INVALID_REQUEST:
-            return invalid_request()
-        if self is VellumErrorCodeEnum.PROVIDER_ERROR:
-            return provider_error()
-        if self is VellumErrorCodeEnum.INTERNAL_SERVER_ERROR:
-            return internal_server_error()
-        if self is VellumErrorCodeEnum.USER_DEFINED_ERROR:
-            return user_defined_error()
+VellumErrorCodeEnum = typing.Union[
+    typing.Literal["INVALID_REQUEST", "PROVIDER_ERROR", "INTERNAL_SERVER_ERROR", "USER_DEFINED_ERROR"], typing.Any
+]

vellum/types/vellum_error_request.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error_code_enum import VellumErrorCodeEnum
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class VellumErrorRequest(pydantic.BaseModel):
+class VellumErrorRequest(pydantic_v1.BaseModel):
     message: str
     code: VellumErrorCodeEnum
 
@@ -27,5 +23,5 @@ class VellumErrorRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/vellum_image.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class VellumImage(pydantic.BaseModel):
+class VellumImage(pydantic_v1.BaseModel):
     src: str
     metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
 
@@ -26,5 +22,5 @@ class VellumImage(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/vellum_image_request.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class VellumImageRequest(pydantic.BaseModel):
+class VellumImageRequest(pydantic_v1.BaseModel):
     src: str
     metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
 
@@ -26,5 +22,5 @@ class VellumImageRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/vellum_variable.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_variable_type import VellumVariableType
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class VellumVariable(pydantic.BaseModel):
+class VellumVariable(pydantic_v1.BaseModel):
     id: str
     key: str
     type: VellumVariableType
@@ -28,5 +24,5 @@ class VellumVariable(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}