vellum-ai 0.14.68__py3-none-any.whl → 0.14.70__py3-none-any.whl

vellum/__init__.py

@@ -123,6 +123,8 @@ from .types import (
     ExternalInputDescriptor,
     ExternalTestCaseExecution,
     ExternalTestCaseExecutionRequest,
+    FastEmbedVectorizerBaaiBgeSmallEnV15,
+    FastEmbedVectorizerBaaiBgeSmallEnV15Request,
     FinishReasonEnum,
     FolderEntity,
     FolderEntityDocumentIndex,
@@ -750,6 +752,8 @@ __all__ = [
     "ExternalInputDescriptor",
     "ExternalTestCaseExecution",
     "ExternalTestCaseExecutionRequest",
+    "FastEmbedVectorizerBaaiBgeSmallEnV15",
+    "FastEmbedVectorizerBaaiBgeSmallEnV15Request",
     "FinishReasonEnum",
     "FolderEntitiesListRequestEntityStatus",
     "FolderEntity",

vellum/client/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.14.68",
+            "X-Fern-SDK-Version": "0.14.70",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

vellum/client/types/__init__.py

@@ -127,6 +127,8 @@ from .execution_vellum_value import ExecutionVellumValue
 from .external_input_descriptor import ExternalInputDescriptor
 from .external_test_case_execution import ExternalTestCaseExecution
 from .external_test_case_execution_request import ExternalTestCaseExecutionRequest
+from .fast_embed_vectorizer_baai_bge_small_en_v_15 import FastEmbedVectorizerBaaiBgeSmallEnV15
+from .fast_embed_vectorizer_baai_bge_small_en_v_15_request import FastEmbedVectorizerBaaiBgeSmallEnV15Request
 from .finish_reason_enum import FinishReasonEnum
 from .folder_entity import FolderEntity
 from .folder_entity_document_index import FolderEntityDocumentIndex
@@ -737,6 +739,8 @@ __all__ = [
     "ExternalInputDescriptor",
     "ExternalTestCaseExecution",
     "ExternalTestCaseExecutionRequest",
+    "FastEmbedVectorizerBaaiBgeSmallEnV15",
+    "FastEmbedVectorizerBaaiBgeSmallEnV15Request",
     "FinishReasonEnum",
     "FolderEntity",
     "FolderEntityDocumentIndex",

vellum/client/types/fast_embed_vectorizer_baai_bge_small_en_v_15.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FastEmbedVectorizerBaaiBgeSmallEnV15(UniversalBaseModel):
+    """
+    FastEmbed vectorizer for BAAI/bge-small-en-v1.5.
+    """
+
+    model_name: typing.Literal["BAAI/bge-small-en-v1.5"] = "BAAI/bge-small-en-v1.5"
+
+    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/fast_embed_vectorizer_baai_bge_small_en_v_15_request.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FastEmbedVectorizerBaaiBgeSmallEnV15Request(UniversalBaseModel):
+    """
+    FastEmbed vectorizer for BAAI/bge-small-en-v1.5.
+    """
+
+    model_name: typing.Literal["BAAI/bge-small-en-v1.5"] = "BAAI/bge-small-en-v1.5"
+
+    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/folder_entity_document_index_data.py

@@ -3,6 +3,7 @@
 from ..core.pydantic_utilities import UniversalBaseModel
 import datetime as dt
 from .entity_status import EntityStatus
+from .document_index_indexing_config import DocumentIndexIndexingConfig
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 import pydantic
@@ -14,6 +15,7 @@ class FolderEntityDocumentIndexData(UniversalBaseModel):
     created: dt.datetime
     modified: dt.datetime
     status: EntityStatus
+    indexing_config: DocumentIndexIndexingConfig
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/indexing_config_vectorizer.py

@@ -16,6 +16,7 @@ from .google_vertex_ai_vectorizer_text_embedding_004 import GoogleVertexAiVector
 from .google_vertex_ai_vectorizer_text_multilingual_embedding_002 import (
     GoogleVertexAiVectorizerTextMultilingualEmbedding002,
 )
+from .fast_embed_vectorizer_baai_bge_small_en_v_15 import FastEmbedVectorizerBaaiBgeSmallEnV15
 
 IndexingConfigVectorizer = typing.Union[
     OpenAiVectorizerTextEmbedding3Small,
@@ -27,4 +28,5 @@ IndexingConfigVectorizer = typing.Union[
     HkunlpInstructorXlVectorizer,
     GoogleVertexAiVectorizerTextEmbedding004,
     GoogleVertexAiVectorizerTextMultilingualEmbedding002,
+    FastEmbedVectorizerBaaiBgeSmallEnV15,
 ]

vellum/client/types/indexing_config_vectorizer_request.py

@@ -16,6 +16,7 @@ from .google_vertex_ai_vectorizer_text_embedding_004_request import GoogleVertex
 from .google_vertex_ai_vectorizer_text_multilingual_embedding_002_request import (
     GoogleVertexAiVectorizerTextMultilingualEmbedding002Request,
 )
+from .fast_embed_vectorizer_baai_bge_small_en_v_15_request import FastEmbedVectorizerBaaiBgeSmallEnV15Request
 
 IndexingConfigVectorizerRequest = typing.Union[
     OpenAiVectorizerTextEmbedding3SmallRequest,
@@ -27,4 +28,5 @@ IndexingConfigVectorizerRequest = typing.Union[
     HkunlpInstructorXlVectorizerRequest,
     GoogleVertexAiVectorizerTextEmbedding004Request,
     GoogleVertexAiVectorizerTextMultilingualEmbedding002Request,
+    FastEmbedVectorizerBaaiBgeSmallEnV15Request,
 ]

vellum/types/fast_embed_vectorizer_baai_bge_small_en_v_15.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.fast_embed_vectorizer_baai_bge_small_en_v_15 import *

vellum/types/fast_embed_vectorizer_baai_bge_small_en_v_15_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.fast_embed_vectorizer_baai_bge_small_en_v_15_request import *

vellum/workflows/environment/__init__.py

@@ -1,5 +1,6 @@
-from .environment import Environment
+from .environment import Environment, EnvironmentVariables
 
 __all__ = [
+    "EnvironmentVariables",
     "Environment",
 ]

vellum/workflows/environment/environment.py

@@ -3,7 +3,11 @@ from typing import Optional
 from vellum.workflows.references import EnvironmentVariableReference
 
 
-class Environment:
+class EnvironmentVariables:
     @staticmethod
     def get(name: str, default: Optional[str] = None) -> EnvironmentVariableReference:
         return EnvironmentVariableReference(name=name, default=default)
+
+
+# Deprecated: Use EnvironmentVariables instead. Will be removed in v0.15.0
+Environment = EnvironmentVariables

vellum/workflows/nodes/displayable/bases/search_node.py

@@ -106,11 +106,23 @@ class BaseSearchNode(BaseNode[StateType], Generic[StateType]):
                 message=f"Document Index '{self.document_index}' not found",
                 code=WorkflowErrorCode.INVALID_INPUTS,
             )
-        except ApiError:
+        except ApiError as e:
+            if e.status_code and e.status_code == 403 and isinstance(e.body, dict):
+                raise NodeException(
+                    message=e.body.get("detail", "Provider credentials is missing or unavailable"),
+                    code=WorkflowErrorCode.PROVIDER_CREDENTIALS_UNAVAILABLE,
+                )
+            elif e.status_code and e.status_code >= 400 and e.status_code < 500 and isinstance(e.body, dict):
+                raise NodeException(
+                    message=e.body.get(
+                        "detail", f"An error occurred while searching against Document Index '{self.document_index}'"
+                    ),
+                    code=WorkflowErrorCode.INVALID_INPUTS,
+                ) from e
             raise NodeException(
-                message=f"An error occurred while searching against Document Index '{self.document_index}'",  # noqa: E501
+                message=f"An error occurred while searching against Document Index '{self.document_index}'",
                 code=WorkflowErrorCode.INTERNAL_ERROR,
-            )
+            ) from e
 
     def _get_options_request(self) -> SearchRequestOptionsRequest:
         return SearchRequestOptionsRequest(

vellum/workflows/nodes/displayable/tests/test_search_node_error_handling.py

@@ -0,0 +1,215 @@
+import pytest
+
+from vellum.client.core.api_error import ApiError
+from vellum.workflows.errors.types import WorkflowErrorCode
+from vellum.workflows.exceptions import NodeException
+from vellum.workflows.inputs import BaseInputs
+from vellum.workflows.nodes.displayable.search_node import SearchNode as BaseSearchNode
+from vellum.workflows.state import BaseState
+from vellum.workflows.state.base import StateMeta
+
+
+def test_search_node_handles_403_error(vellum_client):
+    """Test that SearchNode properly handles 403 API errors with user-facing messages."""
+
+    class Inputs(BaseInputs):
+        query: str
+        document_index: str
+
+    class State(BaseState):
+        pass
+
+    class SearchNode(BaseSearchNode):
+        query = Inputs.query
+        document_index = Inputs.document_index
+
+    vellum_client.search.side_effect = ApiError(
+        status_code=403, body={"detail": "Provider credentials is missing or unavailable"}
+    )
+
+    node = SearchNode(
+        state=State(
+            meta=StateMeta(
+                workflow_inputs=Inputs(
+                    query="test query",
+                    document_index="test-index",
+                )
+            ),
+        )
+    )
+
+    with pytest.raises(NodeException) as exc_info:
+        node.run()
+
+    assert exc_info.value.code == WorkflowErrorCode.PROVIDER_CREDENTIALS_UNAVAILABLE
+    assert exc_info.value.message == "Provider credentials is missing or unavailable"
+
+
+def test_search_node_handles_403_error_with_custom_detail(vellum_client):
+    """Test that SearchNode properly handles 403 API errors with custom detail message."""
+
+    class Inputs(BaseInputs):
+        query: str
+        document_index: str
+
+    class State(BaseState):
+        pass
+
+    class SearchNode(BaseSearchNode):
+        query = Inputs.query
+        document_index = Inputs.document_index
+
+    vellum_client.search.side_effect = ApiError(status_code=403, body={"detail": "Access denied to document index"})
+
+    node = SearchNode(
+        state=State(
+            meta=StateMeta(
+                workflow_inputs=Inputs(
+                    query="test query",
+                    document_index="test-index",
+                )
+            ),
+        )
+    )
+
+    with pytest.raises(NodeException) as exc_info:
+        node.run()
+
+    assert exc_info.value.code == WorkflowErrorCode.PROVIDER_CREDENTIALS_UNAVAILABLE
+    assert exc_info.value.message == "Access denied to document index"
+
+
+def test_search_node_handles_403_error_without_detail(vellum_client):
+    """Test that SearchNode properly handles 403 API errors without detail in body."""
+
+    class Inputs(BaseInputs):
+        query: str
+        document_index: str
+
+    class State(BaseState):
+        pass
+
+    class SearchNode(BaseSearchNode):
+        query = Inputs.query
+        document_index = Inputs.document_index
+
+    vellum_client.search.side_effect = ApiError(status_code=403, body={})
+
+    node = SearchNode(
+        state=State(
+            meta=StateMeta(
+                workflow_inputs=Inputs(
+                    query="test query",
+                    document_index="test-index",
+                )
+            ),
+        )
+    )
+
+    with pytest.raises(NodeException) as exc_info:
+        node.run()
+
+    assert exc_info.value.code == WorkflowErrorCode.PROVIDER_CREDENTIALS_UNAVAILABLE
+    assert exc_info.value.message == "Provider credentials is missing or unavailable"
+
+
+def test_search_node_handles_other_4xx_errors(vellum_client):
+    """Test that SearchNode properly handles other 4xx API errors."""
+
+    class Inputs(BaseInputs):
+        query: str
+        document_index: str
+
+    class State(BaseState):
+        pass
+
+    class SearchNode(BaseSearchNode):
+        query = Inputs.query
+        document_index = Inputs.document_index
+
+    vellum_client.search.side_effect = ApiError(status_code=400, body={"detail": "Invalid request parameters"})
+
+    node = SearchNode(
+        state=State(
+            meta=StateMeta(
+                workflow_inputs=Inputs(
+                    query="test query",
+                    document_index="test-index",
+                )
+            ),
+        )
+    )
+
+    with pytest.raises(NodeException) as exc_info:
+        node.run()
+
+    assert exc_info.value.code == WorkflowErrorCode.INVALID_INPUTS
+    assert "Invalid request parameters" in exc_info.value.message
+
+
+def test_search_node_handles_5xx_errors(vellum_client):
+    """Test that SearchNode properly handles 5xx API errors as internal errors."""
+
+    class Inputs(BaseInputs):
+        query: str
+        document_index: str
+
+    class State(BaseState):
+        pass
+
+    class SearchNode(BaseSearchNode):
+        query = Inputs.query
+        document_index = Inputs.document_index
+
+    vellum_client.search.side_effect = ApiError(status_code=500, body={"detail": "Internal server error"})
+
+    node = SearchNode(
+        state=State(
+            meta=StateMeta(
+                workflow_inputs=Inputs(
+                    query="test query",
+                    document_index="test-index",
+                )
+            ),
+        )
+    )
+
+    with pytest.raises(NodeException) as exc_info:
+        node.run()
+
+    assert exc_info.value.code == WorkflowErrorCode.INTERNAL_ERROR
+    assert "An error occurred while searching against Document Index 'test-index'" in exc_info.value.message
+
+
+def test_search_node_handles_api_error_without_status_code(vellum_client):
+    """Test that SearchNode properly handles API errors without status code."""
+
+    class Inputs(BaseInputs):
+        query: str
+        document_index: str
+
+    class State(BaseState):
+        pass
+
+    class SearchNode(BaseSearchNode):
+        query = Inputs.query
+        document_index = Inputs.document_index
+
+    vellum_client.search.side_effect = ApiError(status_code=None, body={"detail": "Unknown error"})
+
+    node = SearchNode(
+        state=State(
+            meta=StateMeta(
+                workflow_inputs=Inputs(
+                    query="test query",
+                    document_index="test-index",
+                )
+            ),
+        )
+    )
+
+    with pytest.raises(NodeException) as exc_info:
+        node.run()
+
+    assert exc_info.value.code == WorkflowErrorCode.INTERNAL_ERROR
+    assert "An error occurred while searching against Document Index 'test-index'" in exc_info.value.message

vellum/workflows/nodes/experimental/tool_calling_node/tests/test_node.py

@@ -1,7 +1,17 @@
+import json
+from typing import Any, List
+
+from vellum import ChatMessage
 from vellum.client.types.function_call import FunctionCall
 from vellum.client.types.function_call_vellum_value import FunctionCallVellumValue
-from vellum.workflows.nodes.experimental.tool_calling_node.utils import create_tool_router_node
+from vellum.client.types.string_chat_message_content import StringChatMessageContent
+from vellum.workflows import BaseWorkflow
+from vellum.workflows.inputs.base import BaseInputs
+from vellum.workflows.nodes.bases import BaseNode
+from vellum.workflows.nodes.experimental.tool_calling_node.utils import create_function_node, create_tool_router_node
+from vellum.workflows.outputs.base import BaseOutputs
 from vellum.workflows.state.base import BaseState, StateMeta
+from vellum.workflows.state.context import WorkflowContext
 
 
 def first_function() -> str:
@@ -51,3 +61,69 @@ def test_port_condition_match_function_name():
     # AND the default port should be false
     default_port = getattr(router_node.Ports, "default")
     assert default_port.resolve_condition(state) is False
+
+
+def test_tool_calling_node_inline_workflow_context():
+    """
+    Test that the tool calling node correctly passes the context to the inline workflow.
+    This specifically tests that inline workflows receive the correct context.
+    """
+
+    # GIVEN a test workflow that captures its context
+    class MyNode(BaseNode):
+        class Outputs(BaseOutputs):
+            generated_files: Any
+
+        def run(self) -> Outputs:
+            return self.Outputs(generated_files=self._context.generated_files)
+
+    class MyWorkflow(BaseWorkflow[BaseInputs, BaseState]):
+        graph = MyNode
+
+        class Outputs(BaseOutputs):
+            generated_files = MyNode.Outputs.generated_files
+
+    # GIVEN a tool router node
+    tool_router_node = create_tool_router_node(
+        ml_model="test-model",
+        blocks=[],
+        functions=[MyWorkflow],
+        prompt_inputs=None,
+    )
+
+    # WHEN we create a function node for the workflow
+    function_node_class = create_function_node(
+        function=MyWorkflow,
+        tool_router_node=tool_router_node,
+    )
+
+    # AND we create an instance with a context containing generated_files
+    function_node = function_node_class()
+
+    # Create a parent context with test data
+    parent_context = WorkflowContext(
+        generated_files={"script.py": "print('hello world')"},
+    )
+    function_node._context = parent_context
+
+    # Create a state with chat_history for the function node
+    class TestState(BaseState):
+        chat_history: List[ChatMessage] = []
+
+    function_node.state = TestState(meta=StateMeta(node_outputs={tool_router_node.Outputs.text: '{"arguments": {}}'}))
+
+    # WHEN the function node runs
+    outputs = function_node.run()
+
+    # THEN the workflow should have run successfully
+    assert outputs is not None
+
+    # AND the chat history should contain a function response
+    assert len(function_node.state.chat_history) == 1
+    function_response = function_node.state.chat_history[0]
+    assert function_response.role == "FUNCTION"
+
+    # AND the response should contain the generated files
+    assert isinstance(function_response.content, StringChatMessageContent)
+    data = json.loads(function_response.content.value)
+    assert data["generated_files"] == {"script.py": "print('hello world')"}

vellum/workflows/nodes/experimental/tool_calling_node/utils.py

@@ -22,6 +22,7 @@ from vellum.workflows.outputs.base import BaseOutput
 from vellum.workflows.ports.port import Port
 from vellum.workflows.references.lazy import LazyReference
 from vellum.workflows.state.base import BaseState
+from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.state.encoder import DefaultStateEncoder
 from vellum.workflows.types.core import EntityInputsInterface, MergeBehavior
 from vellum.workflows.types.generics import is_workflow_class
@@ -152,8 +153,7 @@ def create_function_node(
 
             # Call the function based on its type
             inputs_instance = function.get_inputs_class()(**arguments)
-
-            workflow = function()
+            workflow = function(context=WorkflowContext.create_from(self._context))
             terminal_event = workflow.run(
                 inputs=inputs_instance,
             )

vellum/workflows/references/environment_variable.py

@@ -8,7 +8,7 @@ if TYPE_CHECKING:
 
 
 class EnvironmentVariableReference(BaseDescriptor[str]):
-    def __init__(self, *, name: str, default: Optional[str]):
+    def __init__(self, *, name: str, default: Optional[str] = None):
         super().__init__(name=name, types=(str,))
         self._default = default
 
@@ -20,5 +20,4 @@ class EnvironmentVariableReference(BaseDescriptor[str]):
         if self._default is not None:
             return self._default
 
-        # Fetch Vellum Environment Variable named `self.name` once that project is done
-        raise ValueError(f"No environment variable named '{self.name}' found")
+        return ""

{vellum_ai-0.14.68.dist-info → vellum_ai-0.14.70.dist-info}/METADATA

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