vellum-ai 0.13.12__tar.gz → 0.13.14__tar.gz

{vellum_ai-0.13.12 → vellum_ai-0.13.14}/PKG-INFO

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

{vellum_ai-0.13.12 → vellum_ai-0.13.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.13.12"
+version = "0.13.14"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.13.12 → vellum_ai-0.13.14}/src/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.13.12",
+            "X-Fern-SDK-Version": "0.13.13",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.13.12 → vellum_ai-0.13.14}/src/vellum/client/resources/documents/client.py

@@ -32,6 +32,7 @@ class DocumentsClient:
         limit: typing.Optional[int] = None,
         offset: typing.Optional[int] = None,
         ordering: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> PaginatedSlimDocumentList:
         """
@@ -51,6 +52,9 @@ class DocumentsClient:
         ordering : typing.Optional[str]
             Which field to use when ordering the results.
 
+        search : typing.Optional[str]
+            A search term.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -77,6 +81,7 @@ class DocumentsClient:
                 "limit": limit,
                 "offset": offset,
                 "ordering": ordering,
+                "search": search,
             },
             request_options=request_options,
         )
@@ -392,6 +397,7 @@ class AsyncDocumentsClient:
         limit: typing.Optional[int] = None,
         offset: typing.Optional[int] = None,
         ordering: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> PaginatedSlimDocumentList:
         """
@@ -411,6 +417,9 @@ class AsyncDocumentsClient:
         ordering : typing.Optional[str]
             Which field to use when ordering the results.
 
+        search : typing.Optional[str]
+            A search term.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -445,6 +454,7 @@ class AsyncDocumentsClient:
                 "limit": limit,
                 "offset": offset,
                 "ordering": ordering,
+                "search": search,
             },
             request_options=request_options,
         )

vellum_ai-0.13.14/src/vellum/workflows/environment/environment.py

@@ -0,0 +1,9 @@
+from typing import Optional
+
+from vellum.workflows.references import EnvironmentVariableReference
+
+
+class Environment:
+    @staticmethod
+    def get(name: str, default: Optional[str] = None) -> EnvironmentVariableReference:
+        return EnvironmentVariableReference(name=name, default=default)

{vellum_ai-0.13.12 → vellum_ai-0.13.14}/src/vellum/workflows/events/tests/test_event.py

@@ -27,7 +27,9 @@ from vellum.workflows.events.workflow import (
 from vellum.workflows.inputs.base import BaseInputs
 from vellum.workflows.nodes.bases.base import BaseNode
 from vellum.workflows.outputs.base import BaseOutput
+from vellum.workflows.references.vellum_secret import VellumSecretReference
 from vellum.workflows.state.base import BaseState
+from vellum.workflows.types.core import VellumSecret
 from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum.workflows.workflows.base import BaseWorkflow
 
@@ -38,6 +40,7 @@ class MockInputs(BaseInputs):
 
 class MockNode(BaseNode):
     node_foo = MockInputs.foo
+    node_secret = VellumSecretReference("secret")
 
     class Outputs(BaseNode.Outputs):
         example: str
@@ -97,6 +100,7 @@ mock_node_uuid = str(uuid4_from_hash(MockNode.__qualname__))
                     node_definition=MockNode,
                     inputs={
                         MockNode.node_foo: "bar",
+                        MockNode.node_secret: VellumSecret(name="secret"),
                     },
                 ),
                 parent=NodeParentContext(
@@ -123,6 +127,9 @@ mock_node_uuid = str(uuid4_from_hash(MockNode.__qualname__))
                     },
                     "inputs": {
                         "node_foo": "bar",
+                        "node_secret": {
+                            "name": "secret",
+                        },
                     },
                 },
                 "parent": {

{vellum_ai-0.13.12 → vellum_ai-0.13.14}/src/vellum/workflows/references/environment_variable.py

@@ -1,5 +1,5 @@
 import os
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Optional
 
 from vellum.workflows.descriptors.base import BaseDescriptor
 
@@ -8,13 +8,17 @@ if TYPE_CHECKING:
 
 
 class EnvironmentVariableReference(BaseDescriptor[str]):
-    def __init__(self, *, name: str) -> None:
+    def __init__(self, *, name: str, default: Optional[str]):
         super().__init__(name=name, types=(str,))
+        self._default = default
 
     def resolve(self, state: "BaseState") -> str:
         env_value = os.environ.get(self.name)
         if env_value is not None:
             return env_value
 
+        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")

{vellum_ai-0.13.12 → vellum_ai-0.13.14}/src/vellum/workflows/types/core.py

@@ -9,6 +9,8 @@ from typing import (  # type: ignore[attr-defined]
     _UnionGenericAlias,
 )
 
+from vellum.client.core.pydantic_utilities import UniversalBaseModel
+
 JsonArray = List["Json"]
 JsonObject = Dict[str, "Json"]
 Json = Union[None, bool, int, float, str, JsonArray, JsonObject]
@@ -20,12 +22,9 @@ SpecialGenericAlias = _SpecialGenericAlias
 UnionGenericAlias = _UnionGenericAlias
 
 
-class VellumSecret:
+class VellumSecret(UniversalBaseModel):
     name: str
 
-    def __init__(self, name: str):
-        self.name = name
-
 
 EntityInputsInterface = Dict[str, Any]
 

vellum_ai-0.13.12/src/vellum/workflows/environment/environment.py

@@ -1,7 +0,0 @@
-from vellum.workflows.references import EnvironmentVariableReference
-
-
-class Environment:
-    @staticmethod
-    def get(name: str) -> EnvironmentVariableReference:
-        return EnvironmentVariableReference(name=name)