vellum-ai 0.1.10__tar.gz → 0.1.12__tar.gz

{vellum_ai-0.1.10 → vellum_ai-0.1.12}/PKG-INFO

@@ -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 → vellum_ai-0.1.12}/pyproject.toml

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

{vellum_ai-0.1.10 → vellum_ai-0.1.12}/src/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_ai-0.1.10 → vellum_ai-0.1.12}/src/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_ai-0.1.10 → vellum_ai-0.1.12}/src/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_ai-0.1.12/src/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_ai-0.1.12/src/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_ai-0.1.12/src/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_ai-0.1.10 → vellum_ai-0.1.12}/src/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_ai-0.1.12/src/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_ai-0.1.10 → vellum_ai-0.1.12}/src/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()