vellum-ai 0.0.41__py3-none-any.whl → 0.0.43__py3-none-any.whl

vellum/types/search_result_merging_request.py

@@ -3,13 +3,16 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchResultMergingRequest(pydantic.BaseModel):
-    enabled: bool = pydantic.Field(description="Whether to enable merging results")
+    enabled: typing.Optional[bool] = pydantic.Field(description="Whether to enable merging results")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/search_weights_request.py

@@ -3,14 +3,19 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SearchWeightsRequest(pydantic.BaseModel):
-    semantic_similarity: float = pydantic.Field(description="The relative weight to give to semantic similarity")
-    keywords: float = pydantic.Field(description="The relative weight to give to keyword matches")
+    semantic_similarity: typing.Optional[float] = pydantic.Field(
+        description="The relative weight to give to semantic similarity"
+    )
+    keywords: typing.Optional[float] = pydantic.Field(description="The relative weight to give to keywords")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

vellum/types/slim_document.py

@@ -3,14 +3,17 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .document_document_to_document_index import DocumentDocumentToDocumentIndex
 from .document_status import DocumentStatus
 from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .processing_state_enum import ProcessingStateEnum
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SlimDocument(pydantic.BaseModel):
     id: str = pydantic.Field(description="Vellum-generated ID that uniquely identifies this document.")

vellum/types/submit_completion_actual_request.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SubmitCompletionActualRequest(pydantic.BaseModel):
     id: typing.Optional[str] = pydantic.Field(

vellum/types/submit_completion_actuals_error_response.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class SubmitCompletionActualsErrorResponse(pydantic.BaseModel):
     detail: str

vellum/types/templating_node_chat_history_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message import ChatMessage
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TemplatingNodeChatHistoryResult(pydantic.BaseModel):
     id: str

vellum/types/templating_node_json_result.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TemplatingNodeJsonResult(pydantic.BaseModel):
     id: str

vellum/types/templating_node_number_result.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
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class TemplatingNodeNumberResult(pydantic.BaseModel):
+    id: str
+    value: typing.Optional[int]
+
+    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/templating_node_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .templating_node_result_data import TemplatingNodeResultData
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TemplatingNodeResult(pydantic.BaseModel):
     data: TemplatingNodeResultData

vellum/types/templating_node_result_data.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .templating_node_result_output import TemplatingNodeResultOutput
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TemplatingNodeResultData(pydantic.BaseModel):
     output: TemplatingNodeResultOutput

vellum/types/templating_node_result_output.py

@@ -8,6 +8,7 @@ import typing_extensions
 
 from .templating_node_chat_history_result import TemplatingNodeChatHistoryResult
 from .templating_node_json_result import TemplatingNodeJsonResult
+from .templating_node_number_result import TemplatingNodeNumberResult
 from .templating_node_search_results_result import TemplatingNodeSearchResultsResult
 from .templating_node_string_result import TemplatingNodeStringResult
 
@@ -21,6 +22,15 @@ class TemplatingNodeResultOutput_String(TemplatingNodeStringResult):
         allow_population_by_field_name = True
 
 
+class TemplatingNodeResultOutput_Number(TemplatingNodeNumberResult):
+    type: typing_extensions.Literal["NUMBER"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
 class TemplatingNodeResultOutput_Json(TemplatingNodeJsonResult):
     type: typing_extensions.Literal["JSON"]
 
@@ -50,6 +60,7 @@ class TemplatingNodeResultOutput_SearchResults(TemplatingNodeSearchResultsResult
 
 TemplatingNodeResultOutput = typing.Union[
     TemplatingNodeResultOutput_String,
+    TemplatingNodeResultOutput_Number,
     TemplatingNodeResultOutput_Json,
     TemplatingNodeResultOutput_ChatHistory,
     TemplatingNodeResultOutput_SearchResults,

vellum/types/templating_node_search_results_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .search_result import SearchResult
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TemplatingNodeSearchResultsResult(pydantic.BaseModel):
     id: str

vellum/types/templating_node_string_result.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TemplatingNodeStringResult(pydantic.BaseModel):
     id: str

vellum/types/terminal_node_chat_history_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message import ChatMessage
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TerminalNodeChatHistoryResult(pydantic.BaseModel):
     id: typing.Optional[str]

vellum/types/terminal_node_json_result.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TerminalNodeJsonResult(pydantic.BaseModel):
     id: typing.Optional[str]

vellum/types/terminal_node_number_result.py

@@ -0,0 +1,30 @@
+# 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 TerminalNodeNumberResult(pydantic.BaseModel):
+    id: typing.Optional[str]
+    name: str = pydantic.Field(description="The unique name given to the terminal node that produced this output.")
+    value: typing.Optional[int]
+
+    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/terminal_node_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .terminal_node_result_data import TerminalNodeResultData
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TerminalNodeResult(pydantic.BaseModel):
     data: TerminalNodeResultData

vellum/types/terminal_node_result_data.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .terminal_node_result_output import TerminalNodeResultOutput
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TerminalNodeResultData(pydantic.BaseModel):
     output: TerminalNodeResultOutput

vellum/types/terminal_node_result_output.py

@@ -8,6 +8,7 @@ import typing_extensions
 
 from .terminal_node_chat_history_result import TerminalNodeChatHistoryResult
 from .terminal_node_json_result import TerminalNodeJsonResult
+from .terminal_node_number_result import TerminalNodeNumberResult
 from .terminal_node_search_results_result import TerminalNodeSearchResultsResult
 from .terminal_node_string_result import TerminalNodeStringResult
 
@@ -21,6 +22,15 @@ class TerminalNodeResultOutput_String(TerminalNodeStringResult):
         allow_population_by_field_name = True
 
 
+class TerminalNodeResultOutput_Number(TerminalNodeNumberResult):
+    type: typing_extensions.Literal["NUMBER"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
 class TerminalNodeResultOutput_Json(TerminalNodeJsonResult):
     type: typing_extensions.Literal["JSON"]
 
@@ -50,6 +60,7 @@ class TerminalNodeResultOutput_SearchResults(TerminalNodeSearchResultsResult):
 
 TerminalNodeResultOutput = typing.Union[
     TerminalNodeResultOutput_String,
+    TerminalNodeResultOutput_Number,
     TerminalNodeResultOutput_Json,
     TerminalNodeResultOutput_ChatHistory,
     TerminalNodeResultOutput_SearchResults,

vellum/types/terminal_node_search_results_result.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .search_result import SearchResult
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TerminalNodeSearchResultsResult(pydantic.BaseModel):
     id: typing.Optional[str]

vellum/types/terminal_node_string_result.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TerminalNodeStringResult(pydantic.BaseModel):
     id: typing.Optional[str]

vellum/types/test_suite_test_case.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .evaluation_params import EvaluationParams
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class TestSuiteTestCase(pydantic.BaseModel):
     test_case_id: typing.Optional[str] = pydantic.Field(

vellum/types/upload_document_error_response.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class UploadDocumentErrorResponse(pydantic.BaseModel):
     detail: str

vellum/types/upload_document_response.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class UploadDocumentResponse(pydantic.BaseModel):
     document_id: str = pydantic.Field(description="The ID of the newly created document.")

vellum/types/vellum_variable.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .vellum_variable_type import VellumVariableType
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class VellumVariable(pydantic.BaseModel):
     id: str

vellum/types/vellum_variable_type.py

@@ -8,13 +8,15 @@ T_Result = typing.TypeVar("T_Result")
 
 class VellumVariableType(str, enum.Enum):
     """
-    * `STRING` - String
+    * `STRING` - STRING
+    * `NUMBER` - NUMBER
     * `JSON` - JSON
-    * `CHAT_HISTORY` - Chat History
-    * `SEARCH_RESULTS` - Search Results
+    * `CHAT_HISTORY` - CHAT_HISTORY
+    * `SEARCH_RESULTS` - SEARCH_RESULTS
     """
 
     STRING = "STRING"
+    NUMBER = "NUMBER"
     JSON = "JSON"
     CHAT_HISTORY = "CHAT_HISTORY"
     SEARCH_RESULTS = "SEARCH_RESULTS"
@@ -22,12 +24,15 @@ class VellumVariableType(str, enum.Enum):
     def visit(
         self,
         string: typing.Callable[[], T_Result],
+        number: typing.Callable[[], T_Result],
         json: typing.Callable[[], T_Result],
         chat_history: typing.Callable[[], T_Result],
         search_results: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is VellumVariableType.STRING:
             return string()
+        if self is VellumVariableType.NUMBER:
+            return number()
         if self is VellumVariableType.JSON:
             return json()
         if self is VellumVariableType.CHAT_HISTORY:

vellum/types/workflow_event_error.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .workflow_execution_event_error_code import WorkflowExecutionEventErrorCode
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowEventError(pydantic.BaseModel):
     message: str

vellum/types/workflow_execution_actual_chat_history_request.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .chat_message_request import ChatMessageRequest
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowExecutionActualChatHistoryRequest(pydantic.BaseModel):
     output_id: typing.Optional[str] = pydantic.Field(

vellum/types/workflow_execution_actual_json_request.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowExecutionActualJsonRequest(pydantic.BaseModel):
     output_id: typing.Optional[str] = pydantic.Field(

vellum/types/workflow_execution_actual_string_request.py

@@ -3,10 +3,13 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowExecutionActualStringRequest(pydantic.BaseModel):
     output_id: typing.Optional[str] = pydantic.Field(

vellum/types/workflow_execution_node_result_event.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .workflow_node_result_event import WorkflowNodeResultEvent
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowExecutionNodeResultEvent(pydantic.BaseModel):
     execution_id: str

vellum/types/workflow_execution_workflow_result_event.py

@@ -3,11 +3,14 @@
 import datetime as dt
 import typing
 
-import pydantic
-
 from ..core.datetime_utils import serialize_datetime
 from .workflow_result_event import WorkflowResultEvent
 
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
 
 class WorkflowExecutionWorkflowResultEvent(pydantic.BaseModel):
     execution_id: str

vellum/types/workflow_node_result_data.py

@@ -6,6 +6,7 @@ import typing
 
 import typing_extensions
 
+from .api_node_result import ApiNodeResult
 from .conditional_node_result import ConditionalNodeResult
 from .prompt_node_result import PromptNodeResult
 from .search_node_result import SearchNodeResult
@@ -49,6 +50,15 @@ class WorkflowNodeResultData_Conditional(ConditionalNodeResult):
         allow_population_by_field_name = True
 
 
+class WorkflowNodeResultData_Api(ApiNodeResult):
+    type: typing_extensions.Literal["API"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
 class WorkflowNodeResultData_Terminal(TerminalNodeResult):
     type: typing_extensions.Literal["TERMINAL"]
 
@@ -63,5 +73,6 @@ WorkflowNodeResultData = typing.Union[
     WorkflowNodeResultData_Search,
     WorkflowNodeResultData_Templating,
     WorkflowNodeResultData_Conditional,
+    WorkflowNodeResultData_Api,
     WorkflowNodeResultData_Terminal,
 ]