vellum-ai 0.9.16__py3-none-any.whl → 0.9.16rc4__py3-none-any.whl

vellum/types/test_case_json_variable_value.py

@@ -1,26 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestCaseJsonVariableValue(UniversalBaseModel):
-    """
-    A JSON value for a variable in a Test Case.
-    """
-
-    variable_id: str
-    name: str
-    type: typing.Literal["JSON"] = "JSON"
-    value: typing.Optional[typing.Any] = 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
+from vellum.client.types.test_case_json_variable_value import *

vellum/types/test_case_number_variable_value.py

@@ -1,26 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestCaseNumberVariableValue(UniversalBaseModel):
-    """
-    A numerical value for a variable in a Test Case.
-    """
-
-    variable_id: str
-    name: str
-    type: typing.Literal["NUMBER"] = "NUMBER"
-    value: 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
+from vellum.client.types.test_case_number_variable_value import *

vellum/types/test_case_search_results_variable_value.py

@@ -1,27 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .search_result import SearchResult
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestCaseSearchResultsVariableValue(UniversalBaseModel):
-    """
-    A search results value for a variable in a Test Case.
-    """
-
-    variable_id: str
-    name: str
-    type: typing.Literal["SEARCH_RESULTS"] = "SEARCH_RESULTS"
-    value: typing.Optional[typing.List[SearchResult]] = 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
+from vellum.client.types.test_case_search_results_variable_value import *

vellum/types/test_case_string_variable_value.py

@@ -1,26 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestCaseStringVariableValue(UniversalBaseModel):
-    """
-    A string value for a variable in a Test Case.
-    """
-
-    variable_id: str
-    name: str
-    type: typing.Literal["STRING"] = "STRING"
-    value: typing.Optional[str] = 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
+from vellum.client.types.test_case_string_variable_value import *

vellum/types/test_case_variable_value.py

@@ -1,22 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-import typing
-from .test_case_string_variable_value import TestCaseStringVariableValue
-from .test_case_number_variable_value import TestCaseNumberVariableValue
-from .test_case_json_variable_value import TestCaseJsonVariableValue
-from .test_case_chat_history_variable_value import TestCaseChatHistoryVariableValue
-from .test_case_search_results_variable_value import TestCaseSearchResultsVariableValue
-from .test_case_error_variable_value import TestCaseErrorVariableValue
-from .test_case_function_call_variable_value import TestCaseFunctionCallVariableValue
-from .test_case_array_variable_value import TestCaseArrayVariableValue
-
-TestCaseVariableValue = typing.Union[
-    TestCaseStringVariableValue,
-    TestCaseNumberVariableValue,
-    TestCaseJsonVariableValue,
-    TestCaseChatHistoryVariableValue,
-    TestCaseSearchResultsVariableValue,
-    TestCaseErrorVariableValue,
-    TestCaseFunctionCallVariableValue,
-    TestCaseArrayVariableValue,
-]
+from vellum.client.types.test_case_variable_value import *

vellum/types/test_suite_run_deployment_release_tag_exec_config.py

@@ -1,29 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .test_suite_run_deployment_release_tag_exec_config_data import TestSuiteRunDeploymentReleaseTagExecConfigData
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-
-
-class TestSuiteRunDeploymentReleaseTagExecConfig(UniversalBaseModel):
-    """
-    Execution configuration for running a Test Suite against a Prompt Deployment
-    """
-
-    type: typing.Literal["DEPLOYMENT_RELEASE_TAG"] = "DEPLOYMENT_RELEASE_TAG"
-    data: TestSuiteRunDeploymentReleaseTagExecConfigData
-    test_case_ids: typing.Optional[typing.List[str]] = pydantic.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.
-    """
-
-    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
+from vellum.client.types.test_suite_run_deployment_release_tag_exec_config import *

vellum/types/test_suite_run_deployment_release_tag_exec_config_data.py

@@ -1,27 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-
-
-class TestSuiteRunDeploymentReleaseTagExecConfigData(UniversalBaseModel):
-    deployment_id: str = pydantic.Field()
-    """
-    The ID of the Prompt Deployment to run the Test Suite against.
-    """
-
-    tag: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    A tag identifying which release of the Prompt Deployment to run the Test Suite against. Useful for testing past versions of the Prompt Deployment
-    """
-
-    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
+from vellum.client.types.test_suite_run_deployment_release_tag_exec_config_data import *

vellum/types/test_suite_run_deployment_release_tag_exec_config_data_request.py

@@ -1,27 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-
-
-class TestSuiteRunDeploymentReleaseTagExecConfigDataRequest(UniversalBaseModel):
-    deployment_id: str = pydantic.Field()
-    """
-    The ID of the Prompt Deployment to run the Test Suite against.
-    """
-
-    tag: typing.Optional[str] = pydantic.Field(default=None)
-    """
-    A tag identifying which release of the Prompt Deployment to run the Test Suite against. Useful for testing past versions of the Prompt Deployment
-    """
-
-    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
+from vellum.client.types.test_suite_run_deployment_release_tag_exec_config_data_request import *

vellum/types/test_suite_run_deployment_release_tag_exec_config_request.py

@@ -1,31 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .test_suite_run_deployment_release_tag_exec_config_data_request import (
-    TestSuiteRunDeploymentReleaseTagExecConfigDataRequest,
-)
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-
-
-class TestSuiteRunDeploymentReleaseTagExecConfigRequest(UniversalBaseModel):
-    """
-    Execution configuration for running a Test Suite against a Prompt Deployment
-    """
-
-    type: typing.Literal["DEPLOYMENT_RELEASE_TAG"] = "DEPLOYMENT_RELEASE_TAG"
-    data: TestSuiteRunDeploymentReleaseTagExecConfigDataRequest
-    test_case_ids: typing.Optional[typing.List[str]] = pydantic.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.
-    """
-
-    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
+from vellum.client.types.test_suite_run_deployment_release_tag_exec_config_request import *

vellum/types/test_suite_run_exec_config.py

@@ -1,10 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-import typing
-from .test_suite_run_deployment_release_tag_exec_config import TestSuiteRunDeploymentReleaseTagExecConfig
-from .test_suite_run_workflow_release_tag_exec_config import TestSuiteRunWorkflowReleaseTagExecConfig
-from .test_suite_run_external_exec_config import TestSuiteRunExternalExecConfig
-
-TestSuiteRunExecConfig = typing.Union[
-    TestSuiteRunDeploymentReleaseTagExecConfig, TestSuiteRunWorkflowReleaseTagExecConfig, TestSuiteRunExternalExecConfig
-]
+from vellum.client.types.test_suite_run_exec_config import *

vellum/types/test_suite_run_exec_config_request.py

@@ -1,12 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-import typing
-from .test_suite_run_deployment_release_tag_exec_config_request import TestSuiteRunDeploymentReleaseTagExecConfigRequest
-from .test_suite_run_workflow_release_tag_exec_config_request import TestSuiteRunWorkflowReleaseTagExecConfigRequest
-from .test_suite_run_external_exec_config_request import TestSuiteRunExternalExecConfigRequest
-
-TestSuiteRunExecConfigRequest = typing.Union[
-    TestSuiteRunDeploymentReleaseTagExecConfigRequest,
-    TestSuiteRunWorkflowReleaseTagExecConfigRequest,
-    TestSuiteRunExternalExecConfigRequest,
-]
+from vellum.client.types.test_suite_run_exec_config_request import *

vellum/types/test_suite_run_execution.py

@@ -1,30 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
-import typing
-from .test_suite_run_execution_output import TestSuiteRunExecutionOutput
-from .test_suite_run_execution_metric_result import TestSuiteRunExecutionMetricResult
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ..core.pydantic_utilities import update_forward_refs
-
-
-class TestSuiteRunExecution(UniversalBaseModel):
-    id: str
-    test_case_id: str
-    outputs: typing.List[TestSuiteRunExecutionOutput]
-    metric_results: typing.List[TestSuiteRunExecutionMetricResult]
-
-    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
-
-
-update_forward_refs(ArrayVellumValue, TestSuiteRunExecution=TestSuiteRunExecution)
+from vellum.client.types.test_suite_run_execution import *

vellum/types/test_suite_run_execution_array_output.py

@@ -1,33 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
-import typing
-from .vellum_value import VellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ..core.pydantic_utilities import update_forward_refs
-
-
-class TestSuiteRunExecutionArrayOutput(UniversalBaseModel):
-    """
-    Execution output of an entity evaluated during a Test Suite Run that is of type ARRAY
-    """
-
-    name: str
-    type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.Optional[typing.List[VellumValue]] = None
-    output_variable_id: 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
-
-
-update_forward_refs(ArrayVellumValue, TestSuiteRunExecutionArrayOutput=TestSuiteRunExecutionArrayOutput)
+from vellum.client.types.test_suite_run_execution_array_output import *

vellum/types/test_suite_run_execution_chat_history_output.py

@@ -1,27 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .chat_message import ChatMessage
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestSuiteRunExecutionChatHistoryOutput(UniversalBaseModel):
-    """
-    Execution output of an entity evaluated during a Test Suite Run that is of type CHAT_HISTORY
-    """
-
-    name: str
-    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
-    value: typing.Optional[typing.List[ChatMessage]] = None
-    output_variable_id: 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
+from vellum.client.types.test_suite_run_execution_chat_history_output import *

vellum/types/test_suite_run_execution_error_output.py

@@ -1,27 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .vellum_error import VellumError
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestSuiteRunExecutionErrorOutput(UniversalBaseModel):
-    """
-    Execution output of an entity evaluated during a Test Suite Run that is of type ERROR
-    """
-
-    name: str
-    type: typing.Literal["ERROR"] = "ERROR"
-    value: typing.Optional[VellumError] = None
-    output_variable_id: 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
+from vellum.client.types.test_suite_run_execution_error_output import *

vellum/types/test_suite_run_execution_function_call_output.py

@@ -1,27 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .function_call import FunctionCall
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestSuiteRunExecutionFunctionCallOutput(UniversalBaseModel):
-    """
-    Execution output of an entity evaluated during a Test Suite Run that is of type FUNCTION_CALL
-    """
-
-    name: str
-    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
-    value: typing.Optional[FunctionCall] = None
-    output_variable_id: 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
+from vellum.client.types.test_suite_run_execution_function_call_output import *

vellum/types/test_suite_run_execution_json_output.py

@@ -1,26 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestSuiteRunExecutionJsonOutput(UniversalBaseModel):
-    """
-    Execution output of an entity evaluated during a Test Suite Run that is of type JSON
-    """
-
-    name: str
-    type: typing.Literal["JSON"] = "JSON"
-    value: typing.Optional[typing.Any] = None
-    output_variable_id: 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
+from vellum.client.types.test_suite_run_execution_json_output import *

vellum/types/test_suite_run_execution_metric_definition.py

@@ -1,21 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestSuiteRunExecutionMetricDefinition(UniversalBaseModel):
-    id: typing.Optional[str] = None
-    label: typing.Optional[str] = None
-    name: typing.Optional[str] = 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
+from vellum.client.types.test_suite_run_execution_metric_definition import *

vellum/types/test_suite_run_execution_metric_result.py

@@ -1,24 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .test_suite_run_metric_output import TestSuiteRunMetricOutput
-from .test_suite_run_execution_metric_definition import TestSuiteRunExecutionMetricDefinition
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestSuiteRunExecutionMetricResult(UniversalBaseModel):
-    metric_id: str
-    outputs: typing.List[TestSuiteRunMetricOutput]
-    metric_label: typing.Optional[str] = None
-    metric_definition: typing.Optional[TestSuiteRunExecutionMetricDefinition] = 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
+from vellum.client.types.test_suite_run_execution_metric_result import *

vellum/types/test_suite_run_execution_number_output.py

@@ -1,26 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestSuiteRunExecutionNumberOutput(UniversalBaseModel):
-    """
-    Execution output of an entity evaluated during a Test Suite Run that is of type NUMBER
-    """
-
-    name: str
-    type: typing.Literal["NUMBER"] = "NUMBER"
-    value: typing.Optional[float] = None
-    output_variable_id: 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
+from vellum.client.types.test_suite_run_execution_number_output import *

vellum/types/test_suite_run_execution_output.py

@@ -1,22 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-import typing
-from .test_suite_run_execution_string_output import TestSuiteRunExecutionStringOutput
-from .test_suite_run_execution_number_output import TestSuiteRunExecutionNumberOutput
-from .test_suite_run_execution_json_output import TestSuiteRunExecutionJsonOutput
-from .test_suite_run_execution_chat_history_output import TestSuiteRunExecutionChatHistoryOutput
-from .test_suite_run_execution_search_results_output import TestSuiteRunExecutionSearchResultsOutput
-from .test_suite_run_execution_error_output import TestSuiteRunExecutionErrorOutput
-from .test_suite_run_execution_function_call_output import TestSuiteRunExecutionFunctionCallOutput
-from .test_suite_run_execution_array_output import TestSuiteRunExecutionArrayOutput
-
-TestSuiteRunExecutionOutput = typing.Union[
-    TestSuiteRunExecutionStringOutput,
-    TestSuiteRunExecutionNumberOutput,
-    TestSuiteRunExecutionJsonOutput,
-    TestSuiteRunExecutionChatHistoryOutput,
-    TestSuiteRunExecutionSearchResultsOutput,
-    TestSuiteRunExecutionErrorOutput,
-    TestSuiteRunExecutionFunctionCallOutput,
-    TestSuiteRunExecutionArrayOutput,
-]
+from vellum.client.types.test_suite_run_execution_output import *

vellum/types/test_suite_run_execution_search_results_output.py

@@ -1,27 +1,3 @@
-# This file was auto-generated by Fern from our API Definition.
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .search_result import SearchResult
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class TestSuiteRunExecutionSearchResultsOutput(UniversalBaseModel):
-    """
-    Execution output of an entity evaluated during a Test Suite Run that is of type SEARCH_RESULTS
-    """
-
-    name: str
-    type: typing.Literal["SEARCH_RESULTS"] = "SEARCH_RESULTS"
-    value: typing.Optional[typing.List[SearchResult]] = None
-    output_variable_id: 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
+from vellum.client.types.test_suite_run_execution_search_results_output import *