vellum-ai 0.12.17__py3-none-any.whl → 0.13.0__py3-none-any.whl

vellum/client/types/__init__.py

@@ -14,14 +14,11 @@ from .array_chat_message_content_item import ArrayChatMessageContentItem
 from .array_chat_message_content_item_request import ArrayChatMessageContentItemRequest
 from .array_chat_message_content_request import ArrayChatMessageContentRequest
 from .array_input import ArrayInput
-from .array_variable_value import ArrayVariableValue
-from .array_variable_value_item import ArrayVariableValueItem
 from .array_vellum_value import ArrayVellumValue
 from .array_vellum_value_request import ArrayVellumValueRequest
 from .audio_chat_message_content import AudioChatMessageContent
 from .audio_chat_message_content_request import AudioChatMessageContentRequest
 from .audio_prompt_block import AudioPromptBlock
-from .audio_variable_value import AudioVariableValue
 from .audio_vellum_value import AudioVellumValue
 from .audio_vellum_value_request import AudioVellumValueRequest
 from .basic_vectorizer_intfloat_multilingual_e_5_large import BasicVectorizerIntfloatMultilingualE5Large
@@ -40,7 +37,6 @@ from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request
 )
 from .chat_history_input import ChatHistoryInput
 from .chat_history_input_request import ChatHistoryInputRequest
-from .chat_history_variable_value import ChatHistoryVariableValue
 from .chat_history_vellum_value import ChatHistoryVellumValue
 from .chat_history_vellum_value_request import ChatHistoryVellumValueRequest
 from .chat_message import ChatMessage
@@ -97,7 +93,6 @@ from .environment_enum import EnvironmentEnum
 from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
 from .ephemeral_prompt_cache_config_type_enum import EphemeralPromptCacheConfigTypeEnum
 from .error_input import ErrorInput
-from .error_variable_value import ErrorVariableValue
 from .error_vellum_value import ErrorVellumValue
 from .error_vellum_value_request import ErrorVellumValueRequest
 from .execute_prompt_event import ExecutePromptEvent
@@ -141,7 +136,6 @@ from .function_call_chat_message_content_value_request import FunctionCallChatMe
 from .function_call_input import FunctionCallInput
 from .function_call_prompt_block import FunctionCallPromptBlock
 from .function_call_request import FunctionCallRequest
-from .function_call_variable_value import FunctionCallVariableValue
 from .function_call_vellum_value import FunctionCallVellumValue
 from .function_call_vellum_value_request import FunctionCallVellumValueRequest
 from .function_definition import FunctionDefinition
@@ -169,7 +163,6 @@ from .hkunlp_instructor_xl_vectorizer_request import HkunlpInstructorXlVectorize
 from .image_chat_message_content import ImageChatMessageContent
 from .image_chat_message_content_request import ImageChatMessageContentRequest
 from .image_prompt_block import ImagePromptBlock
-from .image_variable_value import ImageVariableValue
 from .image_vellum_value import ImageVellumValue
 from .image_vellum_value_request import ImageVellumValueRequest
 from .indexing_config_vectorizer import IndexingConfigVectorizer
@@ -185,7 +178,6 @@ from .iteration_state_enum import IterationStateEnum
 from .jinja_prompt_block import JinjaPromptBlock
 from .json_input import JsonInput
 from .json_input_request import JsonInputRequest
-from .json_variable_value import JsonVariableValue
 from .json_vellum_value import JsonVellumValue
 from .json_vellum_value_request import JsonVellumValueRequest
 from .logical_operator import LogicalOperator
@@ -226,6 +218,7 @@ from .named_test_case_string_variable_value import NamedTestCaseStringVariableVa
 from .named_test_case_string_variable_value_request import NamedTestCaseStringVariableValueRequest
 from .named_test_case_variable_value import NamedTestCaseVariableValue
 from .named_test_case_variable_value_request import NamedTestCaseVariableValueRequest
+from .new_member_join_behavior_enum import NewMemberJoinBehaviorEnum
 from .node_input_compiled_array_value import NodeInputCompiledArrayValue
 from .node_input_compiled_chat_history_value import NodeInputCompiledChatHistoryValue
 from .node_input_compiled_error_value import NodeInputCompiledErrorValue
@@ -247,7 +240,6 @@ from .node_output_compiled_value import NodeOutputCompiledValue
 from .normalized_log_probs import NormalizedLogProbs
 from .normalized_token_log_probs import NormalizedTokenLogProbs
 from .number_input import NumberInput
-from .number_variable_value import NumberVariableValue
 from .number_vellum_value import NumberVellumValue
 from .number_vellum_value_request import NumberVellumValueRequest
 from .open_ai_vectorizer_config import OpenAiVectorizerConfig
@@ -258,6 +250,7 @@ from .open_ai_vectorizer_text_embedding_3_small import OpenAiVectorizerTextEmbed
 from .open_ai_vectorizer_text_embedding_3_small_request import OpenAiVectorizerTextEmbedding3SmallRequest
 from .open_ai_vectorizer_text_embedding_ada_002 import OpenAiVectorizerTextEmbeddingAda002
 from .open_ai_vectorizer_text_embedding_ada_002_request import OpenAiVectorizerTextEmbeddingAda002Request
+from .organization_read import OrganizationRead
 from .paginated_container_image_read_list import PaginatedContainerImageReadList
 from .paginated_deployment_release_tag_read_list import PaginatedDeploymentReleaseTagReadList
 from .paginated_document_index_read_list import PaginatedDocumentIndexReadList
@@ -321,7 +314,6 @@ from .search_result_meta import SearchResultMeta
 from .search_result_meta_request import SearchResultMetaRequest
 from .search_result_request import SearchResultRequest
 from .search_results_input import SearchResultsInput
-from .search_results_variable_value import SearchResultsVariableValue
 from .search_results_vellum_value import SearchResultsVellumValue
 from .search_results_vellum_value_request import SearchResultsVellumValueRequest
 from .search_weights_request import SearchWeightsRequest
@@ -342,7 +334,6 @@ from .string_chat_message_content import StringChatMessageContent
 from .string_chat_message_content_request import StringChatMessageContentRequest
 from .string_input import StringInput
 from .string_input_request import StringInputRequest
-from .string_variable_value import StringVariableValue
 from .string_vellum_value import StringVellumValue
 from .string_vellum_value_request import StringVellumValueRequest
 from .submit_completion_actual_request import SubmitCompletionActualRequest
@@ -522,6 +513,7 @@ from .workflow_result_event_output_data_number import WorkflowResultEventOutputD
 from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
 from .workflow_result_event_output_data_string import WorkflowResultEventOutputDataString
 from .workflow_stream_event import WorkflowStreamEvent
+from .workspace_read import WorkspaceRead
 from .workspace_secret_read import WorkspaceSecretRead
 
 __all__ = [
@@ -539,14 +531,11 @@ __all__ = [
     "ArrayChatMessageContentItemRequest",
     "ArrayChatMessageContentRequest",
     "ArrayInput",
-    "ArrayVariableValue",
-    "ArrayVariableValueItem",
     "ArrayVellumValue",
     "ArrayVellumValueRequest",
     "AudioChatMessageContent",
     "AudioChatMessageContentRequest",
     "AudioPromptBlock",
-    "AudioVariableValue",
     "AudioVellumValue",
     "AudioVellumValueRequest",
     "BasicVectorizerIntfloatMultilingualE5Large",
@@ -557,7 +546,6 @@ __all__ = [
     "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
     "ChatHistoryInput",
     "ChatHistoryInputRequest",
-    "ChatHistoryVariableValue",
     "ChatHistoryVellumValue",
     "ChatHistoryVellumValueRequest",
     "ChatMessage",
@@ -614,7 +602,6 @@ __all__ = [
     "EphemeralPromptCacheConfig",
     "EphemeralPromptCacheConfigTypeEnum",
     "ErrorInput",
-    "ErrorVariableValue",
     "ErrorVellumValue",
     "ErrorVellumValueRequest",
     "ExecutePromptEvent",
@@ -658,7 +645,6 @@ __all__ = [
     "FunctionCallInput",
     "FunctionCallPromptBlock",
     "FunctionCallRequest",
-    "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
     "FunctionDefinition",
@@ -682,7 +668,6 @@ __all__ = [
     "ImageChatMessageContent",
     "ImageChatMessageContentRequest",
     "ImagePromptBlock",
-    "ImageVariableValue",
     "ImageVellumValue",
     "ImageVellumValueRequest",
     "IndexingConfigVectorizer",
@@ -698,7 +683,6 @@ __all__ = [
     "JinjaPromptBlock",
     "JsonInput",
     "JsonInputRequest",
-    "JsonVariableValue",
     "JsonVellumValue",
     "JsonVellumValueRequest",
     "LogicalOperator",
@@ -739,6 +723,7 @@ __all__ = [
     "NamedTestCaseStringVariableValueRequest",
     "NamedTestCaseVariableValue",
     "NamedTestCaseVariableValueRequest",
+    "NewMemberJoinBehaviorEnum",
     "NodeInputCompiledArrayValue",
     "NodeInputCompiledChatHistoryValue",
     "NodeInputCompiledErrorValue",
@@ -760,7 +745,6 @@ __all__ = [
     "NormalizedLogProbs",
     "NormalizedTokenLogProbs",
     "NumberInput",
-    "NumberVariableValue",
     "NumberVellumValue",
     "NumberVellumValueRequest",
     "OpenAiVectorizerConfig",
@@ -771,6 +755,7 @@ __all__ = [
     "OpenAiVectorizerTextEmbedding3SmallRequest",
     "OpenAiVectorizerTextEmbeddingAda002",
     "OpenAiVectorizerTextEmbeddingAda002Request",
+    "OrganizationRead",
     "PaginatedContainerImageReadList",
     "PaginatedDeploymentReleaseTagReadList",
     "PaginatedDocumentIndexReadList",
@@ -834,7 +819,6 @@ __all__ = [
     "SearchResultMetaRequest",
     "SearchResultRequest",
     "SearchResultsInput",
-    "SearchResultsVariableValue",
     "SearchResultsVellumValue",
     "SearchResultsVellumValueRequest",
     "SearchWeightsRequest",
@@ -855,7 +839,6 @@ __all__ = [
     "StringChatMessageContentRequest",
     "StringInput",
     "StringInputRequest",
-    "StringVariableValue",
     "StringVellumValue",
     "StringVellumValueRequest",
     "SubmitCompletionActualRequest",
@@ -1019,5 +1002,6 @@ __all__ = [
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
     "WorkflowStreamEvent",
+    "WorkspaceRead",
     "WorkspaceSecretRead",
 ]

vellum/client/types/new_member_join_behavior_enum.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+NewMemberJoinBehaviorEnum = typing.Union[
+    typing.Literal["AUTO_ACCEPT_FROM_SHARED_DOMAIN", "ALLOW_REQUESTS_FROM_SHARED_DOMAIN", "REQUIRE_EXPLICIT_INVITE"],
+    typing.Any,
+]

vellum/client/types/{function_call_variable_value.py → organization_read.py}

@@ -2,14 +2,16 @@
 
 from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .function_call import FunctionCall
+from .new_member_join_behavior_enum import NewMemberJoinBehaviorEnum
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class FunctionCallVariableValue(UniversalBaseModel):
-    type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
-    value: typing.Optional[FunctionCall] = None
+class OrganizationRead(UniversalBaseModel):
+    id: str
+    name: str
+    allow_staff_access: typing.Optional[bool] = None
+    new_member_join_behavior: NewMemberJoinBehaviorEnum
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/workflow_execution_workflow_result_event.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
-from .array_variable_value import ArrayVariableValue
 from .array_vellum_value import ArrayVellumValue
 import typing
 from .workflow_result_event import WorkflowResultEvent
@@ -32,5 +31,4 @@ class WorkflowExecutionWorkflowResultEvent(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVariableValue, WorkflowExecutionWorkflowResultEvent=WorkflowExecutionWorkflowResultEvent)
 update_forward_refs(ArrayVellumValue, WorkflowExecutionWorkflowResultEvent=WorkflowExecutionWorkflowResultEvent)

vellum/client/types/workflow_result_event.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
-from .array_variable_value import ArrayVariableValue
 from .array_vellum_value import ArrayVellumValue
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 import datetime as dt
@@ -35,5 +34,4 @@ class WorkflowResultEvent(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVariableValue, WorkflowResultEvent=WorkflowResultEvent)
 update_forward_refs(ArrayVellumValue, WorkflowResultEvent=WorkflowResultEvent)

vellum/client/types/workflow_result_event_output_data_array.py

@@ -2,11 +2,11 @@
 
 from __future__ import annotations
 from ..core.pydantic_utilities import UniversalBaseModel
-from .array_variable_value import ArrayVariableValue
+from .array_vellum_value import ArrayVellumValue
 import typing
 from .workflow_node_result_event_state import WorkflowNodeResultEventState
 import pydantic
-from .array_variable_value_item import ArrayVariableValueItem
+from .vellum_value import VellumValue
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 from ..core.pydantic_utilities import update_forward_refs
 
@@ -26,7 +26,7 @@ class WorkflowResultEventOutputDataArray(UniversalBaseModel):
     """
 
     type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
+    value: typing.Optional[typing.List[VellumValue]] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -38,4 +38,4 @@ class WorkflowResultEventOutputDataArray(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-update_forward_refs(ArrayVariableValue, WorkflowResultEventOutputDataArray=WorkflowResultEventOutputDataArray)
+update_forward_refs(ArrayVellumValue, WorkflowResultEventOutputDataArray=WorkflowResultEventOutputDataArray)

vellum/client/types/{string_variable_value.py → workspace_read.py}

@@ -1,14 +1,21 @@
 # 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
+import datetime as dt
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
 
+class WorkspaceRead(UniversalBaseModel):
+    id: str
+    name: str = pydantic.Field()
+    """
+    The name of the Workspace.
+    """
 
-class StringVariableValue(UniversalBaseModel):
-    type: typing.Literal["STRING"] = "STRING"
-    value: typing.Optional[str] = None
+    label: str
+    created: dt.datetime
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/{types/json_variable_value.py → resources/organizations/__init__.py}

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

vellum/resources/organizations/client.py

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

vellum/{types/image_variable_value.py → resources/workspaces/__init__.py}

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

vellum/{types/array_variable_value.py → resources/workspaces/client.py}

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

vellum/types/{array_variable_value_item.py → new_member_join_behavior_enum.py}

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

vellum/types/{audio_variable_value.py → organization_read.py}

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

vellum/types/{error_variable_value.py → workspace_read.py}

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

vellum/workflows/workflows/base.py

@@ -437,40 +437,8 @@ class BaseWorkflow(Generic[WorkflowInputsType, StateType], metaclass=_BaseWorkfl
             raise ValueError(f"No workflows found in {module_path}")
         elif len(workflows) > 1:
             raise ValueError(f"Multiple workflows found in {module_path}")
-        try:
-            BaseWorkflow.import_node_display(module_path)
-        except ModuleNotFoundError:
-            pass
-
         return workflows[0]
 
-    @staticmethod
-    def import_node_display(module_path):
-        # Import the nodes package
-        nodes_package = importlib.import_module(f"{module_path}.display.nodes")
-        # Use the loader to get the code
-        if hasattr(nodes_package, "__spec__") and nodes_package.__spec__ and nodes_package.__spec__.loader:
-            loader = nodes_package.__spec__.loader
-
-            # Check if the loader has a code attribute
-            if hasattr(loader, "code"):
-                code = loader.code
-
-                # Parse the code to find import statements
-                import_lines = [line.strip() for line in code.splitlines() if line.startswith("from ")]
-
-                # Import each module specified in the code
-                for line in import_lines:
-                    try:
-                        # Extract module name from the import line
-                        module_name = line.split(" ")[1]
-                        full_module_path = f"{module_path}.display.nodes{module_name}"
-                        importlib.import_module(full_module_path)
-                    except Exception:
-                        continue
-        # Also import from workflow.py
-        importlib.import_module(f"{module_path}.display.workflow")
-
 
 WorkflowExecutionInitiatedBody.model_rebuild()
 WorkflowExecutionFulfilledBody.model_rebuild()

{vellum_ai-0.12.17.dist-info → vellum_ai-0.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.12.17
+Version: 0.13.0
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0