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

vellum/types/function_call_variable_value.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 .function_call import FunctionCall
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class FunctionCallVariableValue(pydantic.BaseModel):
+class FunctionCallVariableValue(pydantic_v1.BaseModel):
     value: FunctionCall
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -26,5 +22,5 @@ class FunctionCallVariableValue(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/generate_error_response.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
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class GenerateErrorResponse(pydantic.BaseModel):
-    detail: str = pydantic.Field()
+class GenerateErrorResponse(pydantic_v1.BaseModel):
+    detail: str = pydantic_v1.Field()
     """
     Details about why the request failed.
     """
@@ -28,5 +24,5 @@ class GenerateErrorResponse(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/generate_options_request.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .logprobs_enum import LogprobsEnum
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class GenerateOptionsRequest(pydantic.BaseModel):
-    logprobs: typing.Optional[LogprobsEnum] = pydantic.Field(default=None)
+class GenerateOptionsRequest(pydantic_v1.BaseModel):
+    logprobs: typing.Optional[LogprobsEnum] = pydantic_v1.Field(default=None)
     """
     Which logprobs to include, if any. Defaults to NONE.
     
@@ -32,5 +28,5 @@ class GenerateOptionsRequest(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/generate_request.py

@@ -4,26 +4,22 @@ 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 GenerateRequest(pydantic.BaseModel):
-    input_values: typing.Dict[str, typing.Any] = pydantic.Field()
+class GenerateRequest(pydantic_v1.BaseModel):
+    input_values: typing.Dict[str, typing.Any] = pydantic_v1.Field()
     """
     Key/value pairs for each template variable defined in the deployment's prompt.
     """
 
-    chat_history: typing.Optional[typing.List[ChatMessageRequest]] = pydantic.Field(default=None)
+    chat_history: typing.Optional[typing.List[ChatMessageRequest]] = pydantic_v1.Field(default=None)
     """
     Optionally provide a list of chat messages that'll be used in place of the special chat_history variable, if included in the prompt.
     """
 
-    external_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    external_ids: typing.Optional[typing.List[str]] = pydantic_v1.Field(default=None)
     """
     Optionally include a unique identifier for each generation, as represented outside of Vellum. Note that this should generally be a list of length one.
     """
@@ -39,5 +35,5 @@ class GenerateRequest(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/generate_result.py

@@ -4,22 +4,18 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .generate_result_data import GenerateResultData
 from .generate_result_error import GenerateResultError
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class GenerateResult(pydantic.BaseModel):
-    data: typing.Optional[GenerateResultData] = pydantic.Field(default=None)
+class GenerateResult(pydantic_v1.BaseModel):
+    data: typing.Optional[GenerateResultData] = pydantic_v1.Field(default=None)
     """
     An object containing the resulting generation. This key will be absent if the LLM provider experienced an error.
     """
 
-    error: typing.Optional[GenerateResultError] = pydantic.Field(default=None)
+    error: typing.Optional[GenerateResultError] = pydantic_v1.Field(default=None)
     """
     An object containing details about the error that occurred. This key will be absent if the LLM provider did not experience an error.
     """
@@ -35,5 +31,5 @@ class GenerateResult(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/generate_result_data.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .enriched_normalized_completion import EnrichedNormalizedCompletion
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class GenerateResultData(pydantic.BaseModel):
-    completions: typing.List[EnrichedNormalizedCompletion] = pydantic.Field()
+class GenerateResultData(pydantic_v1.BaseModel):
+    completions: typing.List[EnrichedNormalizedCompletion] = pydantic_v1.Field()
     """
     The generated completions. This will generally be a list of length one.
     """
@@ -29,5 +25,5 @@ class GenerateResultData(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/generate_result_error.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
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class GenerateResultError(pydantic.BaseModel):
-    message: str = pydantic.Field()
+class GenerateResultError(pydantic_v1.BaseModel):
+    message: str = pydantic_v1.Field()
     """
     The error message returned by the LLM provider.
     """
@@ -28,5 +24,5 @@ class GenerateResultError(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/generate_stream_response.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 .generate_stream_result import GenerateStreamResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class GenerateStreamResponse(pydantic.BaseModel):
+class GenerateStreamResponse(pydantic_v1.BaseModel):
     delta: GenerateStreamResult
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -26,5 +22,5 @@ class GenerateStreamResponse(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/generate_stream_result.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .generate_result_error import GenerateResultError
 from .generate_stream_result_data import GenerateStreamResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class GenerateStreamResult(pydantic.BaseModel):
+class GenerateStreamResult(pydantic_v1.BaseModel):
     request_index: int
     data: typing.Optional[GenerateStreamResultData] = None
     error: typing.Optional[GenerateResultError] = None
@@ -29,5 +25,5 @@ class GenerateStreamResult(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/generate_stream_result_data.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 .enriched_normalized_completion import EnrichedNormalizedCompletion
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class GenerateStreamResultData(pydantic.BaseModel):
+class GenerateStreamResultData(pydantic_v1.BaseModel):
     completion_index: int
     completion: EnrichedNormalizedCompletion
 
@@ -27,5 +23,5 @@ class GenerateStreamResultData(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/image_chat_message_content.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_image import VellumImage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ImageChatMessageContent(pydantic.BaseModel):
+class ImageChatMessageContent(pydantic_v1.BaseModel):
     """
     An image value that is used in a chat message.
     """
@@ -30,5 +26,5 @@ class ImageChatMessageContent(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/image_chat_message_content_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_image_request import VellumImageRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ImageChatMessageContentRequest(pydantic.BaseModel):
+class ImageChatMessageContentRequest(pydantic_v1.BaseModel):
     """
     An image value that is used in a chat message.
     """
@@ -30,5 +26,5 @@ class ImageChatMessageContentRequest(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/image_variable_value.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_image import VellumImage
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class ImageVariableValue(pydantic.BaseModel):
+class ImageVariableValue(pydantic_v1.BaseModel):
     """
     A base Vellum primitive value representing an image.
     """
@@ -30,5 +26,5 @@ class ImageVariableValue(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/indexing_state_enum.py

@@ -1,41 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class IndexingStateEnum(str, enum.Enum):
-    """
-    - `AWAITING_PROCESSING` - Awaiting Processing
-    - `QUEUED` - Queued
-    - `INDEXING` - Indexing
-    - `INDEXED` - Indexed
-    - `FAILED` - Failed
-    """
-
-    AWAITING_PROCESSING = "AWAITING_PROCESSING"
-    QUEUED = "QUEUED"
-    INDEXING = "INDEXING"
-    INDEXED = "INDEXED"
-    FAILED = "FAILED"
-
-    def visit(
-        self,
-        awaiting_processing: typing.Callable[[], T_Result],
-        queued: typing.Callable[[], T_Result],
-        indexing: typing.Callable[[], T_Result],
-        indexed: typing.Callable[[], T_Result],
-        failed: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is IndexingStateEnum.AWAITING_PROCESSING:
-            return awaiting_processing()
-        if self is IndexingStateEnum.QUEUED:
-            return queued()
-        if self is IndexingStateEnum.INDEXING:
-            return indexing()
-        if self is IndexingStateEnum.INDEXED:
-            return indexed()
-        if self is IndexingStateEnum.FAILED:
-            return failed()
+IndexingStateEnum = typing.Union[
+    typing.Literal["AWAITING_PROCESSING", "QUEUED", "INDEXING", "INDEXED", "FAILED"], typing.Any
+]

vellum/types/initiated_execute_prompt_event.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 .initiated_prompt_execution_meta import InitiatedPromptExecutionMeta
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class InitiatedExecutePromptEvent(pydantic.BaseModel):
+class InitiatedExecutePromptEvent(pydantic_v1.BaseModel):
     """
     The initial data returned indicating that the response from the model has returned and begun streaming.
     """
@@ -31,5 +27,5 @@ class InitiatedExecutePromptEvent(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/initiated_prompt_execution_meta.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 InitiatedPromptExecutionMeta(pydantic.BaseModel):
+class InitiatedPromptExecutionMeta(pydantic_v1.BaseModel):
     """
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
@@ -32,5 +28,5 @@ class InitiatedPromptExecutionMeta(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/initiated_workflow_node_result_event.py

@@ -4,16 +4,12 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .node_input_variable_compiled_value import NodeInputVariableCompiledValue
 from .workflow_node_result_data import WorkflowNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class InitiatedWorkflowNodeResultEvent(pydantic.BaseModel):
+class InitiatedWorkflowNodeResultEvent(pydantic_v1.BaseModel):
     """
     An event that indicates that the node has initiated its execution.
     """
@@ -37,5 +33,5 @@ class InitiatedWorkflowNodeResultEvent(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/json_input_request.py

@@ -4,19 +4,15 @@ 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 JsonInputRequest(pydantic.BaseModel):
+class JsonInputRequest(pydantic_v1.BaseModel):
     """
     A user input representing a JSON object
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the deployment.
     """
@@ -34,5 +30,5 @@ class JsonInputRequest(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/json_variable_value.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 JsonVariableValue(pydantic.BaseModel):
+class JsonVariableValue(pydantic_v1.BaseModel):
     value: typing.Optional[typing.Dict[str, typing.Any]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class JsonVariableValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}