vellum-ai 0.11.0__py3-none-any.whl → 0.11.3__py3-none-any.whl

vellum/__init__.py

@@ -319,6 +319,7 @@ from .types import (
     SentenceChunkingRequest,
     SlimDeploymentRead,
     SlimDocument,
+    SlimDocumentDocumentToDocumentIndex,
     SlimWorkflowDeployment,
     StreamingAdHocExecutePromptEvent,
     StreamingExecutePromptEvent,
@@ -393,6 +394,10 @@ from .types import (
     TestSuiteRunMetricNumberOutput,
     TestSuiteRunMetricOutput,
     TestSuiteRunMetricStringOutput,
+    TestSuiteRunPromptSandboxHistoryItemExecConfig,
+    TestSuiteRunPromptSandboxHistoryItemExecConfigData,
+    TestSuiteRunPromptSandboxHistoryItemExecConfigDataRequest,
+    TestSuiteRunPromptSandboxHistoryItemExecConfigRequest,
     TestSuiteRunRead,
     TestSuiteRunState,
     TestSuiteRunTestSuite,
@@ -400,6 +405,10 @@ from .types import (
     TestSuiteRunWorkflowReleaseTagExecConfigData,
     TestSuiteRunWorkflowReleaseTagExecConfigDataRequest,
     TestSuiteRunWorkflowReleaseTagExecConfigRequest,
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfig,
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigData,
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigDataRequest,
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigRequest,
     TestSuiteTestCase,
     TestSuiteTestCaseBulkOperationRequest,
     TestSuiteTestCaseBulkResult,
@@ -843,6 +852,7 @@ __all__ = [
     "SentenceChunkingRequest",
     "SlimDeploymentRead",
     "SlimDocument",
+    "SlimDocumentDocumentToDocumentIndex",
     "SlimWorkflowDeployment",
     "StreamingAdHocExecutePromptEvent",
     "StreamingExecutePromptEvent",
@@ -917,6 +927,10 @@ __all__ = [
     "TestSuiteRunMetricNumberOutput",
     "TestSuiteRunMetricOutput",
     "TestSuiteRunMetricStringOutput",
+    "TestSuiteRunPromptSandboxHistoryItemExecConfig",
+    "TestSuiteRunPromptSandboxHistoryItemExecConfigData",
+    "TestSuiteRunPromptSandboxHistoryItemExecConfigDataRequest",
+    "TestSuiteRunPromptSandboxHistoryItemExecConfigRequest",
     "TestSuiteRunRead",
     "TestSuiteRunState",
     "TestSuiteRunTestSuite",
@@ -924,6 +938,10 @@ __all__ = [
     "TestSuiteRunWorkflowReleaseTagExecConfigData",
     "TestSuiteRunWorkflowReleaseTagExecConfigDataRequest",
     "TestSuiteRunWorkflowReleaseTagExecConfigRequest",
+    "TestSuiteRunWorkflowSandboxHistoryItemExecConfig",
+    "TestSuiteRunWorkflowSandboxHistoryItemExecConfigData",
+    "TestSuiteRunWorkflowSandboxHistoryItemExecConfigDataRequest",
+    "TestSuiteRunWorkflowSandboxHistoryItemExecConfigRequest",
     "TestSuiteTestCase",
     "TestSuiteTestCaseBulkOperationRequest",
     "TestSuiteTestCaseBulkResult",

vellum/client/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.11.0",
+            "X-Fern-SDK-Version": "0.11.2",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/client/types/__init__.py

@@ -330,6 +330,7 @@ from .sentence_chunking import SentenceChunking
 from .sentence_chunking_request import SentenceChunkingRequest
 from .slim_deployment_read import SlimDeploymentRead
 from .slim_document import SlimDocument
+from .slim_document_document_to_document_index import SlimDocumentDocumentToDocumentIndex
 from .slim_workflow_deployment import SlimWorkflowDeployment
 from .streaming_ad_hoc_execute_prompt_event import StreamingAdHocExecutePromptEvent
 from .streaming_execute_prompt_event import StreamingExecutePromptEvent
@@ -406,6 +407,16 @@ 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
+from .test_suite_run_prompt_sandbox_history_item_exec_config import TestSuiteRunPromptSandboxHistoryItemExecConfig
+from .test_suite_run_prompt_sandbox_history_item_exec_config_data import (
+    TestSuiteRunPromptSandboxHistoryItemExecConfigData,
+)
+from .test_suite_run_prompt_sandbox_history_item_exec_config_data_request import (
+    TestSuiteRunPromptSandboxHistoryItemExecConfigDataRequest,
+)
+from .test_suite_run_prompt_sandbox_history_item_exec_config_request import (
+    TestSuiteRunPromptSandboxHistoryItemExecConfigRequest,
+)
 from .test_suite_run_read import TestSuiteRunRead
 from .test_suite_run_state import TestSuiteRunState
 from .test_suite_run_test_suite import TestSuiteRunTestSuite
@@ -415,6 +426,16 @@ from .test_suite_run_workflow_release_tag_exec_config_data_request import (
     TestSuiteRunWorkflowReleaseTagExecConfigDataRequest,
 )
 from .test_suite_run_workflow_release_tag_exec_config_request import TestSuiteRunWorkflowReleaseTagExecConfigRequest
+from .test_suite_run_workflow_sandbox_history_item_exec_config import TestSuiteRunWorkflowSandboxHistoryItemExecConfig
+from .test_suite_run_workflow_sandbox_history_item_exec_config_data import (
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigData,
+)
+from .test_suite_run_workflow_sandbox_history_item_exec_config_data_request import (
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigDataRequest,
+)
+from .test_suite_run_workflow_sandbox_history_item_exec_config_request import (
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigRequest,
+)
 from .test_suite_test_case import TestSuiteTestCase
 from .test_suite_test_case_bulk_operation_request import TestSuiteTestCaseBulkOperationRequest
 from .test_suite_test_case_bulk_result import TestSuiteTestCaseBulkResult
@@ -819,6 +840,7 @@ __all__ = [
     "SentenceChunkingRequest",
     "SlimDeploymentRead",
     "SlimDocument",
+    "SlimDocumentDocumentToDocumentIndex",
     "SlimWorkflowDeployment",
     "StreamingAdHocExecutePromptEvent",
     "StreamingExecutePromptEvent",
@@ -893,6 +915,10 @@ __all__ = [
     "TestSuiteRunMetricNumberOutput",
     "TestSuiteRunMetricOutput",
     "TestSuiteRunMetricStringOutput",
+    "TestSuiteRunPromptSandboxHistoryItemExecConfig",
+    "TestSuiteRunPromptSandboxHistoryItemExecConfigData",
+    "TestSuiteRunPromptSandboxHistoryItemExecConfigDataRequest",
+    "TestSuiteRunPromptSandboxHistoryItemExecConfigRequest",
     "TestSuiteRunRead",
     "TestSuiteRunState",
     "TestSuiteRunTestSuite",
@@ -900,6 +926,10 @@ __all__ = [
     "TestSuiteRunWorkflowReleaseTagExecConfigData",
     "TestSuiteRunWorkflowReleaseTagExecConfigDataRequest",
     "TestSuiteRunWorkflowReleaseTagExecConfigRequest",
+    "TestSuiteRunWorkflowSandboxHistoryItemExecConfig",
+    "TestSuiteRunWorkflowSandboxHistoryItemExecConfigData",
+    "TestSuiteRunWorkflowSandboxHistoryItemExecConfigDataRequest",
+    "TestSuiteRunWorkflowSandboxHistoryItemExecConfigRequest",
     "TestSuiteTestCase",
     "TestSuiteTestCaseBulkOperationRequest",
     "TestSuiteTestCaseBulkResult",

vellum/client/types/document_document_to_document_index.py

@@ -8,6 +8,10 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class DocumentDocumentToDocumentIndex(UniversalBaseModel):
+    """
+    A detailed representation of the link between a Document and a Document Index it's a member of.
+    """
+
     id: str = pydantic.Field()
     """
     Vellum-generated ID that uniquely identifies this link.
@@ -29,6 +33,8 @@ class DocumentDocumentToDocumentIndex(UniversalBaseModel):
     - `FAILED` - Failed
     """
 
+    extracted_text_file_url: 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:

vellum/client/types/slim_document.py

@@ -7,7 +7,7 @@ import datetime as dt
 from .document_processing_state import DocumentProcessingState
 from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .document_status import DocumentStatus
-from .document_document_to_document_index import DocumentDocumentToDocumentIndex
+from .slim_document_document_to_document_index import SlimDocumentDocumentToDocumentIndex
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
@@ -58,7 +58,7 @@ class SlimDocument(UniversalBaseModel):
     A previously supplied JSON object containing metadata that can be filtered on when searching.
     """
 
-    document_to_document_indexes: typing.List[DocumentDocumentToDocumentIndex]
+    document_to_document_indexes: typing.List[SlimDocumentDocumentToDocumentIndex]
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/slim_document_document_to_document_index.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+import typing
+from .indexing_state_enum import IndexingStateEnum
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class SlimDocumentDocumentToDocumentIndex(UniversalBaseModel):
+    """
+    A slim representation of the link between a Document and a Document Index it's a member of.
+    """
+
+    id: str = pydantic.Field()
+    """
+    Vellum-generated ID that uniquely identifies this link.
+    """
+
+    document_index_id: str = pydantic.Field()
+    """
+    Vellum-generated ID that uniquely identifies the index this document is included in.
+    """
+
+    indexing_state: typing.Optional[IndexingStateEnum] = pydantic.Field(default=None)
+    """
+    An enum value representing where this document is along its indexing lifecycle for this index.
+    
+    - `AWAITING_PROCESSING` - Awaiting Processing
+    - `QUEUED` - Queued
+    - `INDEXING` - Indexing
+    - `INDEXED` - Indexed
+    - `FAILED` - Failed
+    """
+
+    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/client/types/test_suite_run_exec_config.py

@@ -2,9 +2,15 @@
 
 import typing
 from .test_suite_run_deployment_release_tag_exec_config import TestSuiteRunDeploymentReleaseTagExecConfig
+from .test_suite_run_prompt_sandbox_history_item_exec_config import TestSuiteRunPromptSandboxHistoryItemExecConfig
 from .test_suite_run_workflow_release_tag_exec_config import TestSuiteRunWorkflowReleaseTagExecConfig
+from .test_suite_run_workflow_sandbox_history_item_exec_config import TestSuiteRunWorkflowSandboxHistoryItemExecConfig
 from .test_suite_run_external_exec_config import TestSuiteRunExternalExecConfig
 
 TestSuiteRunExecConfig = typing.Union[
-    TestSuiteRunDeploymentReleaseTagExecConfig, TestSuiteRunWorkflowReleaseTagExecConfig, TestSuiteRunExternalExecConfig
+    TestSuiteRunDeploymentReleaseTagExecConfig,
+    TestSuiteRunPromptSandboxHistoryItemExecConfig,
+    TestSuiteRunWorkflowReleaseTagExecConfig,
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfig,
+    TestSuiteRunExternalExecConfig,
 ]

vellum/client/types/test_suite_run_exec_config_request.py

@@ -2,11 +2,19 @@
 
 import typing
 from .test_suite_run_deployment_release_tag_exec_config_request import TestSuiteRunDeploymentReleaseTagExecConfigRequest
+from .test_suite_run_prompt_sandbox_history_item_exec_config_request import (
+    TestSuiteRunPromptSandboxHistoryItemExecConfigRequest,
+)
 from .test_suite_run_workflow_release_tag_exec_config_request import TestSuiteRunWorkflowReleaseTagExecConfigRequest
+from .test_suite_run_workflow_sandbox_history_item_exec_config_request import (
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigRequest,
+)
 from .test_suite_run_external_exec_config_request import TestSuiteRunExternalExecConfigRequest
 
 TestSuiteRunExecConfigRequest = typing.Union[
     TestSuiteRunDeploymentReleaseTagExecConfigRequest,
+    TestSuiteRunPromptSandboxHistoryItemExecConfigRequest,
     TestSuiteRunWorkflowReleaseTagExecConfigRequest,
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigRequest,
     TestSuiteRunExternalExecConfigRequest,
 ]

vellum/client/types/test_suite_run_prompt_sandbox_history_item_exec_config.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .test_suite_run_prompt_sandbox_history_item_exec_config_data import (
+    TestSuiteRunPromptSandboxHistoryItemExecConfigData,
+)
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class TestSuiteRunPromptSandboxHistoryItemExecConfig(UniversalBaseModel):
+    """
+    Execution configuration for running a Test Suite against a Prompt Sandbox History Item
+    """
+
+    type: typing.Literal["PROMPT_SANDBOX_HISTORY_ITEM"] = "PROMPT_SANDBOX_HISTORY_ITEM"
+    data: TestSuiteRunPromptSandboxHistoryItemExecConfigData
+    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

vellum/client/types/test_suite_run_prompt_sandbox_history_item_exec_config_data.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
+
+class TestSuiteRunPromptSandboxHistoryItemExecConfigData(UniversalBaseModel):
+    history_item_id: str = pydantic.Field()
+    """
+    The ID of the Prompt Sandbox History Item that the Test Suite will run against.
+    """
+
+    prompt_variant_id: str = pydantic.Field()
+    """
+    The ID of the Prompt Variant within the Prompt Sandbox History Item that you'd like to run the Test Suite against.
+    """
+
+    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/client/types/test_suite_run_prompt_sandbox_history_item_exec_config_data_request.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
+
+class TestSuiteRunPromptSandboxHistoryItemExecConfigDataRequest(UniversalBaseModel):
+    history_item_id: str = pydantic.Field()
+    """
+    The ID of the Prompt Sandbox History Item that the Test Suite will run against.
+    """
+
+    prompt_variant_id: str = pydantic.Field()
+    """
+    The ID of the Prompt Variant within the Prompt Sandbox History Item that you'd like to run the Test Suite against.
+    """
+
+    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/client/types/test_suite_run_prompt_sandbox_history_item_exec_config_request.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .test_suite_run_prompt_sandbox_history_item_exec_config_data_request import (
+    TestSuiteRunPromptSandboxHistoryItemExecConfigDataRequest,
+)
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class TestSuiteRunPromptSandboxHistoryItemExecConfigRequest(UniversalBaseModel):
+    """
+    Execution configuration for running a Test Suite against a Prompt Sandbox History Item
+    """
+
+    type: typing.Literal["PROMPT_SANDBOX_HISTORY_ITEM"] = "PROMPT_SANDBOX_HISTORY_ITEM"
+    data: TestSuiteRunPromptSandboxHistoryItemExecConfigDataRequest
+    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

vellum/client/types/test_suite_run_workflow_sandbox_history_item_exec_config.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .test_suite_run_workflow_sandbox_history_item_exec_config_data import (
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigData,
+)
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class TestSuiteRunWorkflowSandboxHistoryItemExecConfig(UniversalBaseModel):
+    """
+    Execution configuration for running a Test Suite against a Workflow Sandbox History Item
+    """
+
+    type: typing.Literal["WORKFLOW_SANDBOX_HISTORY_ITEM"] = "WORKFLOW_SANDBOX_HISTORY_ITEM"
+    data: TestSuiteRunWorkflowSandboxHistoryItemExecConfigData
+    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

vellum/client/types/test_suite_run_workflow_sandbox_history_item_exec_config_data.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
+
+class TestSuiteRunWorkflowSandboxHistoryItemExecConfigData(UniversalBaseModel):
+    history_item_id: str = pydantic.Field()
+    """
+    The ID of the Workflow Sandbox History Item that the Test Suite will run against.
+    """
+
+    workflow_variant_id: str = pydantic.Field()
+    """
+    The ID of the Workflow Variant within the Workflow Sandbox History Item that you'd like to run the Test Suite against.
+    """
+
+    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/client/types/test_suite_run_workflow_sandbox_history_item_exec_config_data_request.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
+
+class TestSuiteRunWorkflowSandboxHistoryItemExecConfigDataRequest(UniversalBaseModel):
+    history_item_id: str = pydantic.Field()
+    """
+    The ID of the Workflow Sandbox History Item that the Test Suite will run against.
+    """
+
+    workflow_variant_id: str = pydantic.Field()
+    """
+    The ID of the Workflow Variant within the Workflow Sandbox History Item that you'd like to run the Test Suite against.
+    """
+
+    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/client/types/test_suite_run_workflow_sandbox_history_item_exec_config_request.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .test_suite_run_workflow_sandbox_history_item_exec_config_data_request import (
+    TestSuiteRunWorkflowSandboxHistoryItemExecConfigDataRequest,
+)
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
+
+class TestSuiteRunWorkflowSandboxHistoryItemExecConfigRequest(UniversalBaseModel):
+    """
+    Execution configuration for running a Test Suite against a Workflow Sandbox History Item
+    """
+
+    type: typing.Literal["WORKFLOW_SANDBOX_HISTORY_ITEM"] = "WORKFLOW_SANDBOX_HISTORY_ITEM"
+    data: TestSuiteRunWorkflowSandboxHistoryItemExecConfigDataRequest
+    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

vellum/types/slim_document_document_to_document_index.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.slim_document_document_to_document_index import *

vellum/types/test_suite_run_prompt_sandbox_history_item_exec_config.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_suite_run_prompt_sandbox_history_item_exec_config import *

vellum/types/test_suite_run_prompt_sandbox_history_item_exec_config_data.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_suite_run_prompt_sandbox_history_item_exec_config_data import *

vellum/types/test_suite_run_prompt_sandbox_history_item_exec_config_data_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_suite_run_prompt_sandbox_history_item_exec_config_data_request import *

vellum/types/test_suite_run_prompt_sandbox_history_item_exec_config_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_suite_run_prompt_sandbox_history_item_exec_config_request import *

vellum/types/test_suite_run_workflow_sandbox_history_item_exec_config.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_suite_run_workflow_sandbox_history_item_exec_config import *

vellum/types/test_suite_run_workflow_sandbox_history_item_exec_config_data.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_suite_run_workflow_sandbox_history_item_exec_config_data import *

vellum/types/test_suite_run_workflow_sandbox_history_item_exec_config_data_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_suite_run_workflow_sandbox_history_item_exec_config_data_request import *

vellum/types/test_suite_run_workflow_sandbox_history_item_exec_config_request.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.types.test_suite_run_workflow_sandbox_history_item_exec_config_request import *

vellum/workflows/context.py

@@ -0,0 +1,42 @@
+from contextlib import contextmanager
+import threading
+from typing import Iterator, Optional, cast
+
+from vellum.client.core import UniversalBaseModel
+from vellum.workflows.events.types import ParentContext
+
+
+class ExecutionContext(UniversalBaseModel):
+    parent_context: Optional[ParentContext] = None
+
+
+_CONTEXT_KEY = "_execution_context"
+
+local = threading.local()
+
+
+def get_execution_context() -> ExecutionContext:
+    """Retrieve the current execution context."""
+    return getattr(local, _CONTEXT_KEY, ExecutionContext())
+
+
+def set_execution_context(context: ExecutionContext) -> None:
+    """Set the current execution context."""
+    setattr(local, _CONTEXT_KEY, context)
+
+
+def get_parent_context() -> ParentContext:
+    return cast(ParentContext, get_execution_context().parent_context)
+
+
+@contextmanager
+def execution_context(parent_context: Optional[ParentContext] = None) -> Iterator[None]:
+    """Context manager for handling execution context."""
+    prev_context = get_execution_context()
+    set_context = ExecutionContext(parent_context=parent_context) if parent_context else prev_context
+
+    try:
+        set_execution_context(set_context)
+        yield
+    finally:
+        set_execution_context(prev_context)

vellum/workflows/descriptors/base.py

@@ -30,7 +30,7 @@ if TYPE_CHECKING:
     from vellum.workflows.nodes.bases import BaseNode
     from vellum.workflows.state.base import BaseState
 
-_T = TypeVar("_T")
+_T = TypeVar("_T", covariant=True)
 _O = TypeVar("_O")
 _O2 = TypeVar("_O2")
 

vellum/workflows/nodes/bases/tests/test_base_node.py

@@ -25,7 +25,7 @@ def test_base_node__node_resolution__unset_pydantic_fields():
     assert node.data.dict() == my_data.dict()
 
 
-def test_base_node__node_resolution__descriptor_in_dict():
+def test_base_node__node_resolution__descriptors_in_dict():
     # GIVEN an Input and State class
     class Inputs(BaseInputs):
         hello: str

vellum/workflows/nodes/core/inline_subworkflow_node/node.py

@@ -1,10 +1,12 @@
 from typing import TYPE_CHECKING, Generic, Iterator, Optional, Set, Type, TypeVar
 
+from vellum.workflows.context import execution_context, get_parent_context
 from vellum.workflows.errors.types import VellumErrorCode
 from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.bases.base_subworkflow_node import BaseSubworkflowNode
 from vellum.workflows.outputs.base import BaseOutput, BaseOutputs
 from vellum.workflows.state.base import BaseState
+from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.types.generics import StateType, WorkflowInputsType
 
 if TYPE_CHECKING:
@@ -24,18 +26,22 @@ class InlineSubworkflowNode(BaseSubworkflowNode[StateType], Generic[StateType, W
     subworkflow: Type["BaseWorkflow[WorkflowInputsType, InnerStateType]"]
 
     def run(self) -> Iterator[BaseOutput]:
-        subworkflow = self.subworkflow(
-            parent_state=self.state,
-            context=self._context,
-        )
-        subworkflow_stream = subworkflow.stream(
-            inputs=self._compile_subworkflow_inputs(),
-        )
+        with execution_context(parent_context=get_parent_context() or self._context.parent_context):
+            subworkflow = self.subworkflow(
+                parent_state=self.state,
+                context=WorkflowContext(
+                    _vellum_client=self._context._vellum_client,
+                ),
+            )
+            subworkflow_stream = subworkflow.stream(
+                inputs=self._compile_subworkflow_inputs(),
+            )
 
         outputs: Optional[BaseOutputs] = None
         fulfilled_output_names: Set[str] = set()
 
         for event in subworkflow_stream:
+            self._context._emit_subworkflow_event(event)
             if event.name == "workflow.execution.streaming":
                 if event.output.is_fulfilled:
                     fulfilled_output_names.add(event.output.name)