vellum-ai 0.9.12__tar.gz → 0.9.14__tar.gz

{vellum_ai-0.9.12 → vellum_ai-0.9.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.9.12
+Version: 0.9.14
 Summary: 
 License: MIT
 Requires-Python: >=3.8,<4.0

{vellum_ai-0.9.12 → vellum_ai-0.9.14}/pyproject.toml

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

{vellum_ai-0.9.12 → vellum_ai-0.9.14}/src/vellum/__init__.py

@@ -135,7 +135,7 @@ from .types import (
     FunctionCallVariableValue,
     FunctionCallVellumValue,
     FunctionCallVellumValueRequest,
-    FunctionDefinitionPromptBlock,
+    FunctionDefinition,
     GenerateOptionsRequest,
     GenerateRequest,
     GenerateResponse,
@@ -650,7 +650,7 @@ __all__ = [
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
-    "FunctionDefinitionPromptBlock",
+    "FunctionDefinition",
     "GenerateOptionsRequest",
     "GenerateRequest",
     "GenerateResponse",

{vellum_ai-0.9.12 → vellum_ai-0.9.14}/src/vellum/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.9.12",
+            "X-Fern-SDK-Version": "0.9.14",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.9.12 → vellum_ai-0.9.14}/src/vellum/resources/ad_hoc/client.py

@@ -7,6 +7,7 @@ from ...types.vellum_variable import VellumVariable
 from ...types.prompt_parameters import PromptParameters
 from ...types.prompt_block import PromptBlock
 from ...types.prompt_settings import PromptSettings
+from ...types.function_definition import FunctionDefinition
 from ...types.ad_hoc_expand_meta import AdHocExpandMeta
 from ...core.request_options import RequestOptions
 from ...types.ad_hoc_execute_prompt_event import AdHocExecutePromptEvent
@@ -37,6 +38,7 @@ class AdHocClient:
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
         settings: typing.Optional[PromptSettings] = OMIT,
+        functions: typing.Optional[typing.Sequence[FunctionDefinition]] = OMIT,
         expand_meta: typing.Optional[AdHocExpandMeta] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[AdHocExecutePromptEvent]:
@@ -57,6 +59,8 @@ class AdHocClient:
 
         settings : typing.Optional[PromptSettings]
 
+        functions : typing.Optional[typing.Sequence[FunctionDefinition]]
+
         expand_meta : typing.Optional[AdHocExpandMeta]
 
         request_options : typing.Optional[RequestOptions]
@@ -72,6 +76,7 @@ class AdHocClient:
         from vellum import (
             AdHocExpandMeta,
             EphemeralPromptCacheConfig,
+            FunctionDefinition,
             JinjaPromptBlock,
             PromptParameters,
             PromptRequestStringInput,
@@ -128,6 +133,17 @@ class AdHocClient:
                     template="string",
                 )
             ],
+            functions=[
+                FunctionDefinition(
+                    state="ENABLED",
+                    cache_config=EphemeralPromptCacheConfig(),
+                    name="string",
+                    description="string",
+                    parameters={"string": {"key": "value"}},
+                    forced=True,
+                    strict=True,
+                )
+            ],
             expand_meta=AdHocExpandMeta(
                 cost=True,
                 model_name=True,
@@ -159,6 +175,9 @@ class AdHocClient:
                 "blocks": convert_and_respect_annotation_metadata(
                     object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
                 ),
+                "functions": convert_and_respect_annotation_metadata(
+                    object_=functions, annotation=typing.Sequence[FunctionDefinition], direction="write"
+                ),
                 "expand_meta": convert_and_respect_annotation_metadata(
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),
@@ -232,6 +251,7 @@ class AsyncAdHocClient:
         parameters: PromptParameters,
         blocks: typing.Sequence[PromptBlock],
         settings: typing.Optional[PromptSettings] = OMIT,
+        functions: typing.Optional[typing.Sequence[FunctionDefinition]] = OMIT,
         expand_meta: typing.Optional[AdHocExpandMeta] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[AdHocExecutePromptEvent]:
@@ -252,6 +272,8 @@ class AsyncAdHocClient:
 
         settings : typing.Optional[PromptSettings]
 
+        functions : typing.Optional[typing.Sequence[FunctionDefinition]]
+
         expand_meta : typing.Optional[AdHocExpandMeta]
 
         request_options : typing.Optional[RequestOptions]
@@ -270,6 +292,7 @@ class AsyncAdHocClient:
             AdHocExpandMeta,
             AsyncVellum,
             EphemeralPromptCacheConfig,
+            FunctionDefinition,
             JinjaPromptBlock,
             PromptParameters,
             PromptRequestStringInput,
@@ -328,6 +351,17 @@ class AsyncAdHocClient:
                         template="string",
                     )
                 ],
+                functions=[
+                    FunctionDefinition(
+                        state="ENABLED",
+                        cache_config=EphemeralPromptCacheConfig(),
+                        name="string",
+                        description="string",
+                        parameters={"string": {"key": "value"}},
+                        forced=True,
+                        strict=True,
+                    )
+                ],
                 expand_meta=AdHocExpandMeta(
                     cost=True,
                     model_name=True,
@@ -362,6 +396,9 @@ class AsyncAdHocClient:
                 "blocks": convert_and_respect_annotation_metadata(
                     object_=blocks, annotation=typing.Sequence[PromptBlock], direction="write"
                 ),
+                "functions": convert_and_respect_annotation_metadata(
+                    object_=functions, annotation=typing.Sequence[FunctionDefinition], direction="write"
+                ),
                 "expand_meta": convert_and_respect_annotation_metadata(
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),

{vellum_ai-0.9.12 → vellum_ai-0.9.14}/src/vellum/resources/workflows/client.py

@@ -11,8 +11,8 @@ from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
 from ...types.workflow_push_exec_config import WorkflowPushExecConfig
 from ...types.workflow_push_deployment_config_request import WorkflowPushDeploymentConfigRequest
+from ... import core
 from ...types.workflow_push_response import WorkflowPushResponse
-from ...core.serialization import convert_and_respect_annotation_metadata
 from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -97,6 +97,7 @@ class WorkflowsClient:
         label: str,
         workflow_sandbox_id: typing.Optional[str] = OMIT,
         deployment_config: typing.Optional[WorkflowPushDeploymentConfigRequest] = OMIT,
+        artifact: typing.Optional[core.File] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> WorkflowPushResponse:
         """
@@ -105,6 +106,7 @@ class WorkflowsClient:
         Parameters
         ----------
         exec_config : WorkflowPushExecConfig
+            The execution configuration of the workflow.
 
         label : str
 
@@ -112,6 +114,9 @@ class WorkflowsClient:
 
         deployment_config : typing.Optional[WorkflowPushDeploymentConfigRequest]
 
+        artifact : typing.Optional[core.File]
+            See core.File for more documentation
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -128,7 +133,7 @@ class WorkflowsClient:
             api_key="YOUR_API_KEY",
         )
         client.workflows.push(
-            exec_config={"key": "value"},
+            exec_config="exec_config",
             label="label",
         )
         """
@@ -136,13 +141,14 @@ class WorkflowsClient:
             "v1/workflows/push",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json={
+            data={
                 "exec_config": exec_config,
                 "label": label,
                 "workflow_sandbox_id": workflow_sandbox_id,
-                "deployment_config": convert_and_respect_annotation_metadata(
-                    object_=deployment_config, annotation=WorkflowPushDeploymentConfigRequest, direction="write"
-                ),
+                "deployment_config": deployment_config,
+            },
+            files={
+                "artifact": artifact,
             },
             request_options=request_options,
             omit=OMIT,
@@ -248,6 +254,7 @@ class AsyncWorkflowsClient:
         label: str,
         workflow_sandbox_id: typing.Optional[str] = OMIT,
         deployment_config: typing.Optional[WorkflowPushDeploymentConfigRequest] = OMIT,
+        artifact: typing.Optional[core.File] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> WorkflowPushResponse:
         """
@@ -256,6 +263,7 @@ class AsyncWorkflowsClient:
         Parameters
         ----------
         exec_config : WorkflowPushExecConfig
+            The execution configuration of the workflow.
 
         label : str
 
@@ -263,6 +271,9 @@ class AsyncWorkflowsClient:
 
         deployment_config : typing.Optional[WorkflowPushDeploymentConfigRequest]
 
+        artifact : typing.Optional[core.File]
+            See core.File for more documentation
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -284,7 +295,7 @@ class AsyncWorkflowsClient:
 
         async def main() -> None:
             await client.workflows.push(
-                exec_config={"key": "value"},
+                exec_config="exec_config",
                 label="label",
             )
 
@@ -295,13 +306,14 @@ class AsyncWorkflowsClient:
             "v1/workflows/push",
             base_url=self._client_wrapper.get_environment().default,
             method="POST",
-            json={
+            data={
                 "exec_config": exec_config,
                 "label": label,
                 "workflow_sandbox_id": workflow_sandbox_id,
-                "deployment_config": convert_and_respect_annotation_metadata(
-                    object_=deployment_config, annotation=WorkflowPushDeploymentConfigRequest, direction="write"
-                ),
+                "deployment_config": deployment_config,
+            },
+            files={
+                "artifact": artifact,
             },
             request_options=request_options,
             omit=OMIT,

{vellum_ai-0.9.12 → vellum_ai-0.9.14}/src/vellum/types/__init__.py

@@ -142,7 +142,7 @@ 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_prompt_block import FunctionDefinitionPromptBlock
+from .function_definition import FunctionDefinition
 from .generate_options_request import GenerateOptionsRequest
 from .generate_request import GenerateRequest
 from .generate_response import GenerateResponse
@@ -631,7 +631,7 @@ __all__ = [
     "FunctionCallVariableValue",
     "FunctionCallVellumValue",
     "FunctionCallVellumValueRequest",
-    "FunctionDefinitionPromptBlock",
+    "FunctionDefinition",
     "GenerateOptionsRequest",
     "GenerateRequest",
     "GenerateResponse",

vellum_ai-0.9.14/src/vellum/types/function_definition.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 .prompt_block_state import PromptBlockState
+from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class FunctionDefinition(UniversalBaseModel):
+    """
+    A block that represents a function definition in a prompt template.
+    """
+
+    state: typing.Optional[PromptBlockState] = None
+    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
+    name: typing.Optional[str] = None
+    description: typing.Optional[str] = None
+    parameters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    forced: typing.Optional[bool] = None
+    strict: typing.Optional[bool] = 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

{vellum_ai-0.9.12 → vellum_ai-0.9.14}/src/vellum/types/prompt_block.py

@@ -3,13 +3,10 @@
 from __future__ import annotations
 import typing
 from .jinja_prompt_block import JinjaPromptBlock
-from .function_definition_prompt_block import FunctionDefinitionPromptBlock
 from .variable_prompt_block import VariablePromptBlock
 from .rich_text_prompt_block import RichTextPromptBlock
 import typing
 
 if typing.TYPE_CHECKING:
     from .chat_message_prompt_block import ChatMessagePromptBlock
-PromptBlock = typing.Union[
-    JinjaPromptBlock, "ChatMessagePromptBlock", FunctionDefinitionPromptBlock, VariablePromptBlock, RichTextPromptBlock
-]
+PromptBlock = typing.Union[JinjaPromptBlock, "ChatMessagePromptBlock", VariablePromptBlock, RichTextPromptBlock]

vellum_ai-0.9.14/src/vellum/types/workflow_push_exec_config.py

@@ -0,0 +1,3 @@
+# This file was auto-generated by Fern from our API Definition.
+
+WorkflowPushExecConfig = str

vellum_ai-0.9.12/src/vellum/types/function_definition_prompt_block.py

@@ -1,32 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .prompt_block_state import PromptBlockState
-from .ephemeral_prompt_cache_config import EphemeralPromptCacheConfig
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class FunctionDefinitionPromptBlock(UniversalBaseModel):
-    """
-    A block that represents a function definition in a prompt template.
-    """
-
-    state: typing.Optional[PromptBlockState] = None
-    cache_config: typing.Optional[EphemeralPromptCacheConfig] = None
-    block_type: typing.Literal["FUNCTION_DEFINITION"] = "FUNCTION_DEFINITION"
-    function_name: typing.Optional[str] = None
-    function_description: typing.Optional[str] = None
-    function_parameters: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
-    function_forced: typing.Optional[bool] = None
-    function_strict: typing.Optional[bool] = 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

vellum_ai-0.9.12/src/vellum/types/workflow_push_exec_config.py

@@ -1,5 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-WorkflowPushExecConfig = typing.Dict[str, typing.Optional[typing.Any]]