vellum-ai 0.9.12__py3-none-any.whl → 0.9.13__py3-none-any.whl

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/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.13",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

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/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/types/{function_definition_prompt_block.py → function_definition.py}

@@ -8,19 +8,18 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
-class FunctionDefinitionPromptBlock(UniversalBaseModel):
+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
-    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
+    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

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.12.dist-info → vellum_ai-0.9.13.dist-info}/METADATA

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

{vellum_ai-0.9.12.dist-info → vellum_ai-0.9.13.dist-info}/RECORD

@@ -1,8 +1,8 @@
-vellum/__init__.py,sha256=jgwx_FeayL9btCxwf_azwPxMqwBIGyJ4oPdiKjLKmdo,34242
+vellum/__init__.py,sha256=m0Vfy02ACEtGf7LlRkVWFCAVF19SDF47PWJrCKIDsgo,34220
 vellum/client.py,sha256=GtF36JNWxnFRV348M0-YcGd9N6ES-l0Q5b32rwHCR5E,115031
 vellum/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
 vellum/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/core/client_wrapper.py,sha256=yB4sWmj_3iPctZKEdAasBft2xWadUsYYb9r9OguEmcs,1890
+vellum/core/client_wrapper.py,sha256=NxjpjNNf5AJNEHW0u_g8NqGubMRojQRS0LPjOo3EbKE,1890
 vellum/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/core/file.py,sha256=X9IbmkZmB2bB_DpmZAO3crWdXagOakAyn6UCOCImCPg,2322
 vellum/core/http_client.py,sha256=R0pQpCppnEtxccGvXl4uJ76s7ro_65Fo_erlNNLp_AI,19228
@@ -32,7 +32,7 @@ vellum/lib/utils/uuid.py,sha256=nedyhTNQDS2YvrU5gL3PtvG9cgGH87yKOcpGDJAe44E,214
 vellum/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/resources/__init__.py,sha256=6tqe3AwLJGLW38iua0Tje0n3uz3a4vkqMFxbUJGRs98,1346
 vellum/resources/ad_hoc/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-vellum/resources/ad_hoc/client.py,sha256=bTNVk1CzPUHhegK_oVI1aN49J8Unin2V9nHGYXfaF3k,16091
+vellum/resources/ad_hoc/client.py,sha256=6rsCw-oaq35-DVrb7NTyptc2WU9L6Eka99H3fNB0jA8,17749
 vellum/resources/container_images/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/container_images/client.py,sha256=jK1n-NFsdBKCeEKh-EIqvw7R8AG9PP4GxxcoH9F0GYs,15463
 vellum/resources/deployments/__init__.py,sha256=m64MNuPx3qVazOnTNwOY8oEeDrAkNwMJvUEe5xoMDvs,239
@@ -82,7 +82,7 @@ vellum/terraform/ml_model/__init__.py,sha256=I8h1Ru-Rb-Hi_HusK6G7nJQZEKQGsAAHMmw
 vellum/terraform/provider/__init__.py,sha256=-06xKmAmknpohVzw5TD-t1bnUHta8OrQYqvMd04XM-U,12684
 vellum/terraform/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 vellum/terraform/versions.json,sha256=45c7jjRD5i4w9DJQHs5ZqLLVXRnQwP9Rirq3mWY-xEo,56
-vellum/types/__init__.py,sha256=8zqfmyAI_sKFL7ugWSV9dlmIlWo1Ah5xzppj1qqvDCY,51735
+vellum/types/__init__.py,sha256=bkZN7uoG71R0R34OWEp-fjl4XQUiHkUQsp-TIFco_dQ,51700
 vellum/types/ad_hoc_execute_prompt_event.py,sha256=bCjujA2XsOgyF3bRZbcEqV2rOIymRgsLoIRtZpB14xg,607
 vellum/types/ad_hoc_expand_meta.py,sha256=1gv-NCsy_6xBYupLvZH979yf2VMdxAU-l0y0ynMKZaw,1331
 vellum/types/ad_hoc_fulfilled_prompt_execution_meta.py,sha256=Bfvf1d_dkmshxRACVM5vcxbH_7AQY23RmrrnPc0ytYY,939
@@ -217,7 +217,7 @@ vellum/types/function_call_request.py,sha256=eJBIN-wLkkkDUIwAy1nMeWHu3MZ5aJpOXyW
 vellum/types/function_call_variable_value.py,sha256=VQKCiEtJsmIK3i7CtFV_2ZpxeX70rqpUViXIvAci8L0,702
 vellum/types/function_call_vellum_value.py,sha256=lLJb-S_-S_UXm6una1BMyCbqLpMhbbMcaVIYNO45h5o,759
 vellum/types/function_call_vellum_value_request.py,sha256=oUteuCfWcj7UJbSE_Vywmmva9kyTaeL9iv5WJHabDVs,788
-vellum/types/function_definition_prompt_block.py,sha256=SB84MTRAXXaXaCMGA2O47tuXIk4wvDAOs5pUD5SJ6qg,1261
+vellum/types/function_definition.py,sha256=D6uZTIndSeTT4O6FQjU7jVB6Tn1VWOqlAjCa92qX5cE,1127
 vellum/types/generate_options_request.py,sha256=TUDqsH0tiPWDZH4T-p5gsvKvwVHEVZ_k6oI3qsjlsk4,782
 vellum/types/generate_request.py,sha256=gL6ywAJe6YCJ5oKbtYwL2H_TMdC_6PJZAI7-P3UOF3I,1286
 vellum/types/generate_response.py,sha256=QJmSRsYhZhtDmk2xpE9ueQEkHyXmYsaEQqqlKl9-bS4,699
@@ -339,7 +339,7 @@ vellum/types/pdf_search_result_meta_source_request.py,sha256=nUhaD2Kw1paGC6O_ICV
 vellum/types/plain_text_prompt_block.py,sha256=GlzovdRbT6mhunMvIpF94_0apjh_0SZtkdlx5HFrE0k,930
 vellum/types/price.py,sha256=ewzXDBVLaleuXMVQ-gQ3G1Nl5J2OWOVEMEFfnQIpiTk,610
 vellum/types/processing_failure_reason_enum.py,sha256=R_KIW7TcQejhc-vLhtNf9SdkYADgoZCn4ch4_RRIvsI,195
-vellum/types/prompt_block.py,sha256=qJISp3lSZ0EVrTuqs9306KO14mZPQmj-yeghBqqfBao,606
+vellum/types/prompt_block.py,sha256=MIsxxmAmuT0thkkG12xm3THO5dlRLbFeMZBVTSvb788,493
 vellum/types/prompt_block_state.py,sha256=BRAzTYARoSU36IVZGWMeeqhl5fgFMXCyhJ8rCbfB-f0,163
 vellum/types/prompt_deployment_expand_meta_request.py,sha256=agsiAaHB6lDoZPlnfJ2nmhB4Ud4EiJJTX05YmduyCPo,1910
 vellum/types/prompt_deployment_input_request.py,sha256=KrT4-Ew2VvTWXEkYQz2oyHn5EDOgrMW7FzRFaPH3ARg,353
@@ -563,7 +563,7 @@ vellum/types/workflow_result_event_output_data_string.py,sha256=tM3kgh6tEhD0dFEb
 vellum/types/workflow_stream_event.py,sha256=Wn3Yzuy9MqWAeo8tEaXDTKDEbJoA8DdYdMVq8EKuhu8,361
 vellum/types/workspace_secret_read.py,sha256=3CnHDG72IAY0KRNvc31F0xLmhnpwjQHnDYCfQJzCxI0,714
 vellum/version.py,sha256=jq-1PlAYxN9AXuaZqbYk9ak27SgE2lw9Ia5gx1b1gVI,76
-vellum_ai-0.9.12.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
-vellum_ai-0.9.12.dist-info/METADATA,sha256=x24G6l3nAPZze0IEi_D4vwitY6b5lpYrlqGnGgxk8lE,4395
-vellum_ai-0.9.12.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-vellum_ai-0.9.12.dist-info/RECORD,,
+vellum_ai-0.9.13.dist-info/LICENSE,sha256=CcaljEIoOBaU-wItPH4PmM_mDCGpyuUY0Er1BGu5Ti8,1073
+vellum_ai-0.9.13.dist-info/METADATA,sha256=iYy5x4J3YCS9_eHg_3GjymUtO-Li8PBCxOOuGjvbFqs,4395
+vellum_ai-0.9.13.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+vellum_ai-0.9.13.dist-info/RECORD,,