vellum-ai 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

vellum/__init__.py

@@ -52,7 +52,12 @@ from .types import (
     FulfilledEnum,
     FulfilledExecutePromptEvent,
     FulfilledExecutePromptResponse,
+    FulfilledFunctionCall,
     FulfilledPromptExecutionMeta,
+    FunctionCall,
+    FunctionCallVariableValue,
+    FunctionCall_Fulfilled,
+    FunctionCall_Rejected,
     GenerateErrorResponse,
     GenerateOptionsRequest,
     GenerateRequest,
@@ -121,6 +126,7 @@ from .types import (
     PromptNodeResultData,
     PromptOutput,
     PromptOutput_Error,
+    PromptOutput_FunctionCall,
     PromptOutput_Json,
     PromptOutput_String,
     PromptTemplateBlock,
@@ -144,6 +150,7 @@ from .types import (
     RejectedEnum,
     RejectedExecutePromptEvent,
     RejectedExecutePromptResponse,
+    RejectedFunctionCall,
     RejectedPromptExecutionMeta,
     SandboxMetricInputParams,
     SandboxMetricInputParamsRequest,
@@ -338,7 +345,12 @@ __all__ = [
     "FulfilledEnum",
     "FulfilledExecutePromptEvent",
     "FulfilledExecutePromptResponse",
+    "FulfilledFunctionCall",
     "FulfilledPromptExecutionMeta",
+    "FunctionCall",
+    "FunctionCallVariableValue",
+    "FunctionCall_Fulfilled",
+    "FunctionCall_Rejected",
     "GenerateErrorResponse",
     "GenerateOptionsRequest",
     "GenerateRequest",
@@ -409,6 +421,7 @@ __all__ = [
     "PromptNodeResultData",
     "PromptOutput",
     "PromptOutput_Error",
+    "PromptOutput_FunctionCall",
     "PromptOutput_Json",
     "PromptOutput_String",
     "PromptTemplateBlock",
@@ -432,6 +445,7 @@ __all__ = [
     "RejectedEnum",
     "RejectedExecutePromptEvent",
     "RejectedExecutePromptResponse",
+    "RejectedFunctionCall",
     "RejectedPromptExecutionMeta",
     "SandboxMetricInputParams",
     "SandboxMetricInputParamsRequest",

vellum/core/client_wrapper.py

@@ -16,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "v0.1.10",
+            "X-Fern-SDK-Version": "0.1.12",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/types/__init__.py

@@ -57,7 +57,10 @@ from .finish_reason_enum import FinishReasonEnum
 from .fulfilled_enum import FulfilledEnum
 from .fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
 from .fulfilled_execute_prompt_response import FulfilledExecutePromptResponse
+from .fulfilled_function_call import FulfilledFunctionCall
 from .fulfilled_prompt_execution_meta import FulfilledPromptExecutionMeta
+from .function_call import FunctionCall, FunctionCall_Fulfilled, FunctionCall_Rejected
+from .function_call_variable_value import FunctionCallVariableValue
 from .generate_error_response import GenerateErrorResponse
 from .generate_options_request import GenerateOptionsRequest
 from .generate_request import GenerateRequest
@@ -130,7 +133,13 @@ from .prompt_deployment_input_request import (
 from .prompt_execution_meta import PromptExecutionMeta
 from .prompt_node_result import PromptNodeResult
 from .prompt_node_result_data import PromptNodeResultData
-from .prompt_output import PromptOutput, PromptOutput_Error, PromptOutput_Json, PromptOutput_String
+from .prompt_output import (
+    PromptOutput,
+    PromptOutput_Error,
+    PromptOutput_FunctionCall,
+    PromptOutput_Json,
+    PromptOutput_String,
+)
 from .prompt_template_block import PromptTemplateBlock
 from .prompt_template_block_data import PromptTemplateBlockData
 from .prompt_template_block_data_request import PromptTemplateBlockDataRequest
@@ -152,6 +161,7 @@ from .registered_prompt_sandbox_snapshot import RegisteredPromptSandboxSnapshot
 from .rejected_enum import RejectedEnum
 from .rejected_execute_prompt_event import RejectedExecutePromptEvent
 from .rejected_execute_prompt_response import RejectedExecutePromptResponse
+from .rejected_function_call import RejectedFunctionCall
 from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .sandbox_metric_input_params import SandboxMetricInputParams
 from .sandbox_metric_input_params_request import SandboxMetricInputParamsRequest
@@ -343,7 +353,12 @@ __all__ = [
     "FulfilledEnum",
     "FulfilledExecutePromptEvent",
     "FulfilledExecutePromptResponse",
+    "FulfilledFunctionCall",
     "FulfilledPromptExecutionMeta",
+    "FunctionCall",
+    "FunctionCallVariableValue",
+    "FunctionCall_Fulfilled",
+    "FunctionCall_Rejected",
     "GenerateErrorResponse",
     "GenerateOptionsRequest",
     "GenerateRequest",
@@ -412,6 +427,7 @@ __all__ = [
     "PromptNodeResultData",
     "PromptOutput",
     "PromptOutput_Error",
+    "PromptOutput_FunctionCall",
     "PromptOutput_Json",
     "PromptOutput_String",
     "PromptTemplateBlock",
@@ -435,6 +451,7 @@ __all__ = [
     "RejectedEnum",
     "RejectedExecutePromptEvent",
     "RejectedExecutePromptResponse",
+    "RejectedFunctionCall",
     "RejectedPromptExecutionMeta",
     "SandboxMetricInputParams",
     "SandboxMetricInputParamsRequest",

vellum/types/fulfilled_function_call.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class FulfilledFunctionCall(pydantic.BaseModel):
+    """
+    The final resolved function call value.
+    """
+
+    arguments: typing.Dict[str, typing.Any]
+    id: typing.Optional[str]
+    name: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/function_call.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from .fulfilled_function_call import FulfilledFunctionCall
+from .rejected_function_call import RejectedFunctionCall
+
+
+class FunctionCall_Fulfilled(FulfilledFunctionCall):
+    state: typing_extensions.Literal["FULFILLED"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class FunctionCall_Rejected(RejectedFunctionCall):
+    state: typing_extensions.Literal["REJECTED"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+FunctionCall = typing.Union[FunctionCall_Fulfilled, FunctionCall_Rejected]

vellum/types/function_call_variable_value.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .function_call import FunctionCall
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class FunctionCallVariableValue(pydantic.BaseModel):
+    value: FunctionCall
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/prompt_output.py

@@ -7,6 +7,7 @@ import typing
 import typing_extensions
 
 from .error_variable_value import ErrorVariableValue
+from .function_call_variable_value import FunctionCallVariableValue
 from .json_variable_value import JsonVariableValue
 from .string_variable_value import StringVariableValue
 
@@ -38,4 +39,13 @@ class PromptOutput_Error(ErrorVariableValue):
         allow_population_by_field_name = True
 
 
-PromptOutput = typing.Union[PromptOutput_String, PromptOutput_Json, PromptOutput_Error]
+class PromptOutput_FunctionCall(FunctionCallVariableValue):
+    type: typing_extensions.Literal["FUNCTION_CALL"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+PromptOutput = typing.Union[PromptOutput_String, PromptOutput_Json, PromptOutput_Error, PromptOutput_FunctionCall]

vellum/types/rejected_function_call.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .vellum_error import VellumError
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class RejectedFunctionCall(pydantic.BaseModel):
+    """
+    Returned if the function call failed to parse for some reason.
+    """
+
+    error: VellumError
+    id: typing.Optional[str]
+    name: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/vellum_variable_type.py

@@ -15,6 +15,7 @@ class VellumVariableType(str, enum.Enum):
     - `SEARCH_RESULTS` - SEARCH_RESULTS
     - `ERROR` - ERROR
     - `ARRAY` - ARRAY
+    - `FUNCTION_CALL` - FUNCTION_CALL
     """
 
     STRING = "STRING"
@@ -24,6 +25,7 @@ class VellumVariableType(str, enum.Enum):
     SEARCH_RESULTS = "SEARCH_RESULTS"
     ERROR = "ERROR"
     ARRAY = "ARRAY"
+    FUNCTION_CALL = "FUNCTION_CALL"
 
     def visit(
         self,
@@ -34,6 +36,7 @@ class VellumVariableType(str, enum.Enum):
         search_results: typing.Callable[[], T_Result],
         error: typing.Callable[[], T_Result],
         array: typing.Callable[[], T_Result],
+        function_call: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is VellumVariableType.STRING:
             return string()
@@ -49,3 +52,5 @@ class VellumVariableType(str, enum.Enum):
             return error()
         if self is VellumVariableType.ARRAY:
             return array()
+        if self is VellumVariableType.FUNCTION_CALL:
+            return function_call()

{vellum_ai-0.1.10.dist-info → vellum_ai-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.1.10
+Version: 0.1.12
 Summary: 
 Requires-Python: >=3.7,<4.0
 Classifier: Programming Language :: Python :: 3

{vellum_ai-0.1.10.dist-info → vellum_ai-0.1.12.dist-info}/RECORD

@@ -1,8 +1,8 @@
-vellum/__init__.py,sha256=p7CWPTlrwlGlE9tu6qL8gfoI4-amEuZ-mH3Rhe9U0gU,19047
+vellum/__init__.py,sha256=u_BoPxw9hJ2eFHB1spRk6IVZRk6rnU7XUIDJW7gPbm4,19437
 vellum/client.py,sha256=oz0l_KYFAQq3VI2uGHdAluuk3iac0c45XCdyRg055Bw,55245
 vellum/core/__init__.py,sha256=QJS3CJ2TYP2E1Tge0CS6Z7r8LTNzJHQVX1hD3558eP0,519
 vellum/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/core/client_wrapper.py,sha256=rhGN6OMXQDU8yR-8_i7MymJI4s65NwQw5O3RonH3pC4,1214
+vellum/core/client_wrapper.py,sha256=DuSoycZHqWhXzgKqtztIxlIl5FVaMx13J250DIYhb-k,1213
 vellum/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/core/jsonable_encoder.py,sha256=MTYkDov2EryHgee4QM46uZiBOuOXK9KTHlBdBwU-CpU,3799
 vellum/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
@@ -29,7 +29,7 @@ vellum/resources/sandboxes/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roe
 vellum/resources/sandboxes/client.py,sha256=RkMHXuNGblYEMDdbUeFHzJDGEIxJd9HGiThXUbR1MNg,8201
 vellum/resources/test_suites/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/resources/test_suites/client.py,sha256=FuO3i0gkjdCddUfygXiGvqoB1BxbS-GvS_Qpf2dt1Kk,8693
-vellum/types/__init__.py,sha256=uE4GCtzivR9fD2j1bznJim3UPbYJisY5AeIe0J6hpW8,25833
+vellum/types/__init__.py,sha256=X0xTGtWGJM0VH_6jA-JRzMnLK3nok1oR8ChQp4On2fc,26358
 vellum/types/api_node_result.py,sha256=1zAcNznjyVkTQM2V--UE1lYFzRWbYEogoChZBGcsliE,936
 vellum/types/api_node_result_data.py,sha256=HvpZaAKYXsoBOnobACIYCmIdxbRc7Zp-ibIohiz_Nzc,1125
 vellum/types/block_type_enum.py,sha256=w6z5D3RTe75YQBG6HpX8NCbO_kSZxIL0Xy1xss2RKS8,1051
@@ -69,7 +69,10 @@ vellum/types/finish_reason_enum.py,sha256=zAvOLiq2ngmDILEftziwCGYb4z-wlih0XHRNOo
 vellum/types/fulfilled_enum.py,sha256=mMP8i4YCkFeJXZ5SqOLGNercErechFzpEs8Gs_sIWto,146
 vellum/types/fulfilled_execute_prompt_event.py,sha256=VFPXczMrjnwXi5oEG3r_jVnK2wM-pmlclddvn5GaRwk,1248
 vellum/types/fulfilled_execute_prompt_response.py,sha256=YAvwKdFuo_uT8G2LqOOjtkwu9zVYtXBYm_RsxYgIN5M,1450
+vellum/types/fulfilled_function_call.py,sha256=fZrTh53XN3dO5bvdtVWGtQ91VgLPKzeMY-b-Z53-uKo,1012
 vellum/types/fulfilled_prompt_execution_meta.py,sha256=FDCQoqisV2SzUVD_Yw0lMv9tM7gW_6wHTbobUBz9Jwk,1145
+vellum/types/function_call.py,sha256=-eKu1NTK3WevQDZVVwaMzYkctl8KErZq9bh1qOxc4MA,770
+vellum/types/function_call_variable_value.py,sha256=tS-nHSwuue4EWOLXEuLI4xSjSNGVoHBK15Dd0C4hu30,932
 vellum/types/generate_error_response.py,sha256=auz4NPbyceabCe0eN89lavEwotWbqsTSbkJtI_lgXJs,950
 vellum/types/generate_options_request.py,sha256=iqmaim-vkUgVsPMmDJ7sMmSl1WOGB4Psgfg-GvNe0ik,1117
 vellum/types/generate_request.py,sha256=Wf5j8TIhLqPwwJJz63TiMNty2cvnJI9xVmUKtWlJIpQ,1581
@@ -121,7 +124,7 @@ vellum/types/prompt_deployment_input_request.py,sha256=pwI4mGEz88s8gJr1j_5-4t501
 vellum/types/prompt_execution_meta.py,sha256=tF9eMOh_CAMvtq2lXVRfVOaWp12jDhuoEUY2S2aENsM,1266
 vellum/types/prompt_node_result.py,sha256=TxFRE6-2LFrG1Mo9uA7dOfNs8_Jk6u7bOWhmOzYQiJY,948
 vellum/types/prompt_node_result_data.py,sha256=LyNkZz-9lYejVtIaej1m7u5CiWmXVFG8IAr6hVQBgZg,945
-vellum/types/prompt_output.py,sha256=cyp7UWyYPy-pgmdNJqc1N15gE2du2WJTd72cqI_Ltp4,1012
+vellum/types/prompt_output.py,sha256=Bz8hTTGUdg0GAmvTw0_gSCoB3jzs_zhF7kPbpKDMSD0,1336
 vellum/types/prompt_template_block.py,sha256=2UfBd8ktRrvVp_gZFLWevrV5D1UjXCji4YbeHPAVl0c,1164
 vellum/types/prompt_template_block_data.py,sha256=fEveYsxf-CAPNhxjirVRNCcqTtmNBxFdjMks-W9-02c,983
 vellum/types/prompt_template_block_data_request.py,sha256=HgrcEIgO8OSseYsqqNX9VKpjutfrK6LEYofg3Ifu8mU,1012
@@ -143,6 +146,7 @@ vellum/types/registered_prompt_sandbox_snapshot.py,sha256=Dv4sY70rvdLtyKi8oN2mWw
 vellum/types/rejected_enum.py,sha256=FKiuB_gp3EQ5caugx-eiFwRO6iv5lVDS_JBUjPTeRhQ,144
 vellum/types/rejected_execute_prompt_event.py,sha256=56LDUfgEkDEkaqtloD4Yg6CUQci69KJYoYWj-zr5d1M,1172
 vellum/types/rejected_execute_prompt_response.py,sha256=JOf1cYJ3TVR8aWLuL1R-sj7DFYi2igaOs0yaPmc2-Aw,1410
+vellum/types/rejected_function_call.py,sha256=c14z0BPFs48ifoccraOShFaQDz6D6g2kpktJs1TE39E,1051
 vellum/types/rejected_prompt_execution_meta.py,sha256=rJsnrz0E0W6Mkq9gBLN4c3sXrr1Mg14feUtj4bKZgvY,1144
 vellum/types/sandbox_metric_input_params.py,sha256=WPrqAWggrSvu9syP2Fk4SGj08YvV55oF9dD6Cb40gjE,961
 vellum/types/sandbox_metric_input_params_request.py,sha256=UbTt4QimyaoYjzhWNggpFL7SlYLgKqZ4i6_skWzAo2I,990
@@ -203,7 +207,7 @@ vellum/types/vellum_error.py,sha256=rO2oP5qbRRpk17X5D-Gf8Ea2oSkqkZpp7n71Dz-Qivw,
 vellum/types/vellum_error_code_enum.py,sha256=X7aUaKZR1jiYnaa9OxPu-bze3Tua31Gh22HJAxw6fyI,969
 vellum/types/vellum_error_request.py,sha256=kMDn9F-Cg4IWQeaX_nCdvadu7JlaIvtf6Rmm3tdVkD0,964
 vellum/types/vellum_variable.py,sha256=mVchGYwlkdti5QKb4nTqalA_rotmwhI0fLlT-dmrweQ,964
-vellum/types/vellum_variable_type.py,sha256=c6AU50N-v2_AoIcUMzczCHxbxJh9BsEd49DAb8UlC04,1479
+vellum/types/vellum_variable_type.py,sha256=eedC83HixsZLDvv29EQ-NYrmWPEQ-cEL_CMm7H0MS_g,1695
 vellum/types/workflow_event_error.py,sha256=X1jEuhnsZoh75FN6XpTvtbO67UD_1SiACsBk5bhn3wQ,1001
 vellum/types/workflow_execution_actual_chat_history_request.py,sha256=skI-SuvBfJc8wxH2EDMqyKCGUgjfFFBUPkCLyncHgmM,2014
 vellum/types/workflow_execution_actual_json_request.py,sha256=hVKpBukLehwDUopO7iiKv-wTvxDDg4SXMYmmexZDMo0,1951
@@ -229,6 +233,6 @@ vellum/types/workflow_result_event_output_data_number.py,sha256=zWtQor27iaDDHmY5
 vellum/types/workflow_result_event_output_data_search_results.py,sha256=frCaJ5kWrIqCeV-waBNfd7rO4fqWe5aYpSI8PM4-oRw,1323
 vellum/types/workflow_result_event_output_data_string.py,sha256=TByZxyQh9ci4UIdEmoEi_JK1U_JwYCnVZeB_4kGuXKM,1405
 vellum/types/workflow_stream_event.py,sha256=OQUSzwoM-OCfWxNzeOVVLsjCue_WWqin3tGMtwvp_rc,873
-vellum_ai-0.1.10.dist-info/METADATA,sha256=FMHtByGy9cDhishaDiKO3lgp_dhTAaE74Loyb1vUlks,3487
-vellum_ai-0.1.10.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-vellum_ai-0.1.10.dist-info/RECORD,,
+vellum_ai-0.1.12.dist-info/METADATA,sha256=wJvQglTcH4jgg0iyNQ5Q0Fh0F_50MIH_IEPnsneMAWA,3487
+vellum_ai-0.1.12.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+vellum_ai-0.1.12.dist-info/RECORD,,