vellum-ai 0.3.22__py3-none-any.whl → 0.5.0__py3-none-any.whl

vellum/types/logical_operator.py

@@ -1,177 +1,27 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class LogicalOperator(str, enum.Enum):
-    """
-    - `=` - EQUALS
-    - `!=` - DOES_NOT_EQUAL
-    - `<` - LESS_THAN
-    - `>` - GREATER_THAN
-    - `<=` - LESS_THAN_OR_EQUAL_TO
-    - `>=` - GREATER_THAN_OR_EQUAL_TO
-    - `contains` - CONTAINS
-    - `beginsWith` - BEGINS_WITH
-    - `endsWith` - ENDS_WITH
-    - `doesNotContain` - DOES_NOT_CONTAIN
-    - `doesNotBeginWith` - DOES_NOT_BEGIN_WITH
-    - `doesNotEndWith` - DOES_NOT_END_WITH
-    - `null` - NULL
-    - `notNull` - NOT_NULL
-    - `in` - IN
-    - `notIn` - NOT_IN
-    - `between` - BETWEEN
-    - `notBetween` - NOT_BETWEEN
-    """
-
-    EQUALS = "="
-    """
-    Equals
-    """
-
-    DOES_NOT_EQUAL = "!="
-    """
-    Does not equal
-    """
-
-    LESS_THAN = "<"
-    """
-    Less than
-    """
-
-    GREATER_THAN = ">"
-    """
-    Greater than
-    """
-
-    LESS_THAN_OR_EQUAL_TO = "<="
-    """
-    Less than or equal to
-    """
-
-    GREATER_THAN_OR_EQUAL_TO = ">="
-    """
-    Greater than or equal to
-    """
-
-    CONTAINS = "contains"
-    """
-    Contains
-    """
-
-    BEGINS_WITH = "beginsWith"
-    """
-    Begins with
-    """
-
-    ENDS_WITH = "endsWith"
-    """
-    Ends with
-    """
-
-    DOES_NOT_CONTAIN = "doesNotContain"
-    """
-    Does not contain
-    """
-
-    DOES_NOT_BEGIN_WITH = "doesNotBeginWith"
-    """
-    Does not begin with
-    """
-
-    DOES_NOT_END_WITH = "doesNotEndWith"
-    """
-    Does not end with
-    """
-
-    NULL = "null"
-    """
-    Null
-    """
-
-    NOT_NULL = "notNull"
-    """
-    Not null
-    """
-
-    IN = "in"
-    """
-    In
-    """
-
-    NOT_IN = "notIn"
-    """
-    Not in
-    """
-
-    BETWEEN = "between"
-    """
-    Between
-    """
-
-    NOT_BETWEEN = "notBetween"
-    """
-    Not between
-    """
-
-    def visit(
-        self,
-        equals: typing.Callable[[], T_Result],
-        does_not_equal: typing.Callable[[], T_Result],
-        less_than: typing.Callable[[], T_Result],
-        greater_than: typing.Callable[[], T_Result],
-        less_than_or_equal_to: typing.Callable[[], T_Result],
-        greater_than_or_equal_to: typing.Callable[[], T_Result],
-        contains: typing.Callable[[], T_Result],
-        begins_with: typing.Callable[[], T_Result],
-        ends_with: typing.Callable[[], T_Result],
-        does_not_contain: typing.Callable[[], T_Result],
-        does_not_begin_with: typing.Callable[[], T_Result],
-        does_not_end_with: typing.Callable[[], T_Result],
-        null: typing.Callable[[], T_Result],
-        not_null: typing.Callable[[], T_Result],
-        in_: typing.Callable[[], T_Result],
-        not_in: typing.Callable[[], T_Result],
-        between: typing.Callable[[], T_Result],
-        not_between: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is LogicalOperator.EQUALS:
-            return equals()
-        if self is LogicalOperator.DOES_NOT_EQUAL:
-            return does_not_equal()
-        if self is LogicalOperator.LESS_THAN:
-            return less_than()
-        if self is LogicalOperator.GREATER_THAN:
-            return greater_than()
-        if self is LogicalOperator.LESS_THAN_OR_EQUAL_TO:
-            return less_than_or_equal_to()
-        if self is LogicalOperator.GREATER_THAN_OR_EQUAL_TO:
-            return greater_than_or_equal_to()
-        if self is LogicalOperator.CONTAINS:
-            return contains()
-        if self is LogicalOperator.BEGINS_WITH:
-            return begins_with()
-        if self is LogicalOperator.ENDS_WITH:
-            return ends_with()
-        if self is LogicalOperator.DOES_NOT_CONTAIN:
-            return does_not_contain()
-        if self is LogicalOperator.DOES_NOT_BEGIN_WITH:
-            return does_not_begin_with()
-        if self is LogicalOperator.DOES_NOT_END_WITH:
-            return does_not_end_with()
-        if self is LogicalOperator.NULL:
-            return null()
-        if self is LogicalOperator.NOT_NULL:
-            return not_null()
-        if self is LogicalOperator.IN:
-            return in_()
-        if self is LogicalOperator.NOT_IN:
-            return not_in()
-        if self is LogicalOperator.BETWEEN:
-            return between()
-        if self is LogicalOperator.NOT_BETWEEN:
-            return not_between()
+LogicalOperator = typing.Union[
+    typing.Literal[
+        "=",
+        "!=",
+        "<",
+        ">",
+        "<=",
+        ">=",
+        "contains",
+        "beginsWith",
+        "endsWith",
+        "doesNotContain",
+        "doesNotBeginWith",
+        "doesNotEndWith",
+        "null",
+        "notNull",
+        "in",
+        "notIn",
+        "between",
+        "notBetween",
+    ],
+    typing.Any,
+]

vellum/types/logprobs_enum.py

@@ -1,22 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class LogprobsEnum(str, enum.Enum):
-    """
-    - `ALL` - ALL
-    - `NONE` - NONE
-    """
-
-    ALL = "ALL"
-    NONE = "NONE"
-
-    def visit(self, all_: typing.Callable[[], T_Result], none: typing.Callable[[], T_Result]) -> T_Result:
-        if self is LogprobsEnum.ALL:
-            return all_()
-        if self is LogprobsEnum.NONE:
-            return none()
+LogprobsEnum = typing.Union[typing.Literal["ALL", "NONE"], typing.Any]

vellum/types/metadata_filter_config_request.py

@@ -4,17 +4,13 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .logical_operator import LogicalOperator
 from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
 from .metadata_filter_rule_request import MetadataFilterRuleRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class MetadataFilterConfigRequest(pydantic.BaseModel):
+class MetadataFilterConfigRequest(pydantic_v1.BaseModel):
     combinator: typing.Optional[MetadataFilterRuleCombinator] = None
     negated: typing.Optional[bool] = None
     rules: typing.Optional[typing.List[MetadataFilterRuleRequest]] = None
@@ -33,5 +29,5 @@ class MetadataFilterConfigRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/metadata_filter_rule_combinator.py

@@ -1,22 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class MetadataFilterRuleCombinator(str, enum.Enum):
-    """
-    - `and` - AND
-    - `or` - OR
-    """
-
-    AND = "and"
-    OR = "or"
-
-    def visit(self, and_: typing.Callable[[], T_Result], or_: typing.Callable[[], T_Result]) -> T_Result:
-        if self is MetadataFilterRuleCombinator.AND:
-            return and_()
-        if self is MetadataFilterRuleCombinator.OR:
-            return or_()
+MetadataFilterRuleCombinator = typing.Union[typing.Literal["and", "or"], typing.Any]

vellum/types/metadata_filter_rule_request.py

@@ -6,16 +6,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .logical_operator import LogicalOperator
 from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class MetadataFilterRuleRequest(pydantic.BaseModel):
+class MetadataFilterRuleRequest(pydantic_v1.BaseModel):
     combinator: typing.Optional[MetadataFilterRuleCombinator] = None
     negated: typing.Optional[bool] = None
     rules: typing.Optional[typing.List[MetadataFilterRuleRequest]] = None
@@ -34,7 +30,7 @@ class MetadataFilterRuleRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}
 
 

vellum/types/ml_model_usage.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class MlModelUsage(pydantic.BaseModel):
+class MlModelUsage(pydantic_v1.BaseModel):
     output_token_count: typing.Optional[int] = None
     input_token_count: typing.Optional[int] = None
     input_char_count: typing.Optional[int] = None
@@ -29,5 +25,5 @@ class MlModelUsage(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_scenario_input_chat_history_variable_value_request.py

@@ -0,0 +1,31 @@
+# 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 ..core.pydantic_utilities import pydantic_v1
+from .chat_message_request import ChatMessageRequest
+
+
+class NamedScenarioInputChatHistoryVariableValueRequest(pydantic_v1.BaseModel):
+    """
+    Named Prompt Sandbox Scenario input value that is of type CHAT_HISTORY
+    """
+
+    value: typing.Optional[typing.List[ChatMessageRequest]] = None
+    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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_scenario_input_request.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
+from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
+
+
+class NamedScenarioInputRequest_String(NamedScenarioInputStringVariableValueRequest):
+    type: typing.Literal["STRING"] = "STRING"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+class NamedScenarioInputRequest_ChatHistory(NamedScenarioInputChatHistoryVariableValueRequest):
+    type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        populate_by_name = True
+
+
+NamedScenarioInputRequest = typing.Union[NamedScenarioInputRequest_String, NamedScenarioInputRequest_ChatHistory]

vellum/types/{prompt_template_block_data.py → named_scenario_input_string_variable_value_request.py}

@@ -4,17 +4,16 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .prompt_template_block import PromptTemplateBlock
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
+class NamedScenarioInputStringVariableValueRequest(pydantic_v1.BaseModel):
+    """
+    Named Prompt Sandbox Scenario input value that is of type STRING
+    """
 
-class PromptTemplateBlockData(pydantic.BaseModel):
-    version: int
-    blocks: typing.List[PromptTemplateBlock]
+    value: typing.Optional[str] = None
+    name: str
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,5 +26,5 @@ class PromptTemplateBlockData(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/{prompt_template_block_data_request.py → named_test_case_chat_history_variable_value.py}

@@ -4,17 +4,17 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .prompt_template_block_request import PromptTemplateBlockRequest
+from ..core.pydantic_utilities import pydantic_v1
+from .chat_message import ChatMessage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
+class NamedTestCaseChatHistoryVariableValue(pydantic_v1.BaseModel):
+    """
+    Named Test Case value that is of type CHAT_HISTORY
+    """
 
-class PromptTemplateBlockDataRequest(pydantic.BaseModel):
-    version: int
-    blocks: typing.List[PromptTemplateBlockRequest]
+    value: typing.Optional[typing.List[ChatMessage]] = None
+    name: str
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -27,5 +27,5 @@ class PromptTemplateBlockDataRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_chat_history_variable_value_request.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .chat_message_request import ChatMessageRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NamedTestCaseChatHistoryVariableValueRequest(pydantic.BaseModel):
+class NamedTestCaseChatHistoryVariableValueRequest(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type CHAT_HISTORY
     """
@@ -31,5 +27,5 @@ class NamedTestCaseChatHistoryVariableValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_error_variable_value.py

@@ -0,0 +1,31 @@
+# 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 ..core.pydantic_utilities import pydantic_v1
+from .vellum_error import VellumError
+
+
+class NamedTestCaseErrorVariableValue(pydantic_v1.BaseModel):
+    """
+    Named Test Case value that is of type ERROR
+    """
+
+    value: typing.Optional[VellumError] = None
+    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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_error_variable_value_request.py

@@ -4,15 +4,11 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .vellum_error_request import VellumErrorRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NamedTestCaseErrorVariableValueRequest(pydantic.BaseModel):
+class NamedTestCaseErrorVariableValueRequest(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type ERROR
     """
@@ -31,5 +27,5 @@ class NamedTestCaseErrorVariableValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/{register_prompt_prompt.py → named_test_case_function_call_variable_value.py}

@@ -4,23 +4,17 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
+from .fulfilled_function_call import FulfilledFunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class RegisterPromptPrompt(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class NamedTestCaseFunctionCallVariableValue(pydantic_v1.BaseModel):
     """
-    The ID of the generated prompt.
+    Named Test Case value that is of type FUNCTION_CALL
     """
 
-    label: str = pydantic.Field()
-    """
-    A human-friendly label for the generated prompt.
-    """
+    value: typing.Optional[FulfilledFunctionCall] = None
+    name: str
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
@@ -33,5 +27,5 @@ class RegisterPromptPrompt(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_function_call_variable_value_request.py

@@ -0,0 +1,31 @@
+# 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 ..core.pydantic_utilities import pydantic_v1
+from .fulfilled_function_call_request import FulfilledFunctionCallRequest
+
+
+class NamedTestCaseFunctionCallVariableValueRequest(pydantic_v1.BaseModel):
+    """
+    Named Test Case value that is of type FUNCTION_CALL
+    """
+
+    value: typing.Optional[FulfilledFunctionCallRequest] = None
+    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
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

vellum/types/{register_prompt_error_response.py → named_test_case_json_variable_value.py}

@@ -4,19 +4,17 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class RegisterPromptErrorResponse(pydantic.BaseModel):
-    detail: str = pydantic.Field()
+class NamedTestCaseJsonVariableValue(pydantic_v1.BaseModel):
     """
-    Details about why the request failed.
+    Named Test Case value that is of type JSON
     """
 
+    value: typing.Optional[typing.Dict[str, typing.Any]] = None
+    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)
@@ -28,5 +26,5 @@ class RegisterPromptErrorResponse(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}

vellum/types/named_test_case_json_variable_value_request.py

@@ -4,14 +4,10 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class NamedTestCaseJsonVariableValueRequest(pydantic.BaseModel):
+class NamedTestCaseJsonVariableValueRequest(pydantic_v1.BaseModel):
     """
     Named Test Case value that is of type JSON
     """
@@ -30,5 +26,5 @@ class NamedTestCaseJsonVariableValueRequest(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}