vellum-ai 0.3.23__py3-none-any.whl → 0.5.1__py3-none-any.whl

vellum/types/search_response.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 .search_result import SearchResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class SearchResponse(pydantic.BaseModel):
-    results: typing.List[SearchResult] = pydantic.Field()
+class SearchResponse(pydantic_v1.BaseModel):
+    results: typing.List[SearchResult] = pydantic_v1.Field()
     """
     The results of the search. Each result represents a chunk that matches the search query.
     """
@@ -29,5 +25,5 @@ class SearchResponse(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/search_result.py

@@ -4,27 +4,23 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .search_result_document import SearchResultDocument
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class SearchResult(pydantic.BaseModel):
-    text: str = pydantic.Field()
+class SearchResult(pydantic_v1.BaseModel):
+    text: str = pydantic_v1.Field()
     """
     The text of the chunk that matched the search query.
     """
 
-    score: float = pydantic.Field()
+    score: float = pydantic_v1.Field()
     """
     A score representing how well the chunk matches the search query.
     """
 
     keywords: typing.List[str]
-    document: SearchResultDocument = pydantic.Field()
+    document: SearchResultDocument = pydantic_v1.Field()
     """
     The document that contains the chunk that matched the search query.
     """
@@ -40,5 +36,5 @@ class SearchResult(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/search_result_document.py

@@ -4,30 +4,26 @@ 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 SearchResultDocument(pydantic.BaseModel):
-    id: str = pydantic.Field()
+class SearchResultDocument(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     The ID of the document.
     """
 
-    label: str = pydantic.Field()
+    label: str = pydantic_v1.Field()
     """
     The human-readable name for the document.
     """
 
-    external_id: typing.Optional[str] = pydantic.Field(default=None)
+    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The unique ID of the document as represented in an external system and specified when it was originally uploaded.
     """
 
-    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
     """
     A previously supplied JSON object containing metadata that can be filtered on when searching.
     """
@@ -43,5 +39,5 @@ class SearchResultDocument(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/search_result_document_request.py

@@ -4,25 +4,21 @@ 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 SearchResultDocumentRequest(pydantic.BaseModel):
-    label: str = pydantic.Field()
+class SearchResultDocumentRequest(pydantic_v1.BaseModel):
+    label: str = pydantic_v1.Field()
     """
     The human-readable name for the document.
     """
 
-    external_id: typing.Optional[str] = pydantic.Field(default=None)
+    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The unique ID of the document as represented in an external system and specified when it was originally uploaded.
     """
 
-    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
     """
     A previously supplied JSON object containing metadata that can be filtered on when searching.
     """
@@ -38,5 +34,5 @@ class SearchResultDocumentRequest(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/search_result_merging_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
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class SearchResultMergingRequest(pydantic.BaseModel):
-    enabled: typing.Optional[bool] = pydantic.Field(default=None)
+class SearchResultMergingRequest(pydantic_v1.BaseModel):
+    enabled: typing.Optional[bool] = pydantic_v1.Field(default=None)
     """
     Whether to enable merging results
     """
@@ -28,5 +24,5 @@ class SearchResultMergingRequest(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/search_result_request.py

@@ -4,27 +4,23 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .search_result_document_request import SearchResultDocumentRequest
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class SearchResultRequest(pydantic.BaseModel):
-    text: str = pydantic.Field()
+class SearchResultRequest(pydantic_v1.BaseModel):
+    text: str = pydantic_v1.Field()
     """
     The text of the chunk that matched the search query.
     """
 
-    score: float = pydantic.Field()
+    score: float = pydantic_v1.Field()
     """
     A score representing how well the chunk matches the search query.
     """
 
     keywords: typing.List[str]
-    document: SearchResultDocumentRequest = pydantic.Field()
+    document: SearchResultDocumentRequest = pydantic_v1.Field()
     """
     The document that contains the chunk that matched the search query.
     """
@@ -40,5 +36,5 @@ class SearchResultRequest(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/search_results_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 .search_result import SearchResult
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class SearchResultsVariableValue(pydantic.BaseModel):
+class SearchResultsVariableValue(pydantic_v1.BaseModel):
     value: typing.Optional[typing.List[SearchResult]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -26,5 +22,5 @@ class SearchResultsVariableValue(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/search_weights_request.py

@@ -4,20 +4,16 @@ 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 SearchWeightsRequest(pydantic.BaseModel):
-    semantic_similarity: typing.Optional[float] = pydantic.Field(default=None)
+class SearchWeightsRequest(pydantic_v1.BaseModel):
+    semantic_similarity: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     The relative weight to give to semantic similarity
     """
 
-    keywords: typing.Optional[float] = pydantic.Field(default=None)
+    keywords: typing.Optional[float] = pydantic_v1.Field(default=None)
     """
     The relative weight to give to keywords
     """
@@ -33,5 +29,5 @@ class SearchWeightsRequest(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/slim_deployment_read.py

@@ -4,30 +4,26 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .vellum_variable import VellumVariable
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class SlimDeploymentRead(pydantic.BaseModel):
+class SlimDeploymentRead(pydantic_v1.BaseModel):
     id: str
     created: dt.datetime
-    label: str = pydantic.Field()
+    label: str = pydantic_v1.Field()
     """
     A human-readable label for the deployment
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     A name that uniquely identifies this deployment within its workspace
     """
 
-    status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
+    status: typing.Optional[EntityStatus] = pydantic_v1.Field(default=None)
     """
     The current status of the deployment
     
@@ -35,7 +31,7 @@ class SlimDeploymentRead(pydantic.BaseModel):
     - `ARCHIVED` - Archived
     """
 
-    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
+    environment: typing.Optional[EnvironmentEnum] = pydantic_v1.Field(default=None)
     """
     The environment this deployment is used in
     
@@ -58,5 +54,5 @@ class SlimDeploymentRead(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/slim_document.py

@@ -4,39 +4,35 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 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()
+class SlimDocument(pydantic_v1.BaseModel):
+    id: str = pydantic_v1.Field()
     """
     Vellum-generated ID that uniquely identifies this document.
     """
 
-    external_id: typing.Optional[str] = pydantic.Field(default=None)
+    external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
     """
     The external ID that was originally provided when uploading the document.
     """
 
-    last_uploaded_at: dt.datetime = pydantic.Field()
+    last_uploaded_at: dt.datetime = pydantic_v1.Field()
     """
     A timestamp representing when this document was most recently uploaded.
     """
 
-    label: str = pydantic.Field()
+    label: str = pydantic_v1.Field()
     """
     Human-friendly name for this document.
     """
 
-    processing_state: typing.Optional[ProcessingStateEnum] = pydantic.Field(default=None)
+    processing_state: typing.Optional[ProcessingStateEnum] = pydantic_v1.Field(default=None)
     """
     An enum value representing where this document is along its processing lifecycle. Note that this is different than its indexing lifecycle.
     
@@ -46,7 +42,7 @@ class SlimDocument(pydantic.BaseModel):
     - `FAILED` - Failed
     """
 
-    processing_failure_reason: typing.Optional[ProcessingFailureReasonEnum] = pydantic.Field(default=None)
+    processing_failure_reason: typing.Optional[ProcessingFailureReasonEnum] = pydantic_v1.Field(default=None)
     """
     An enum value representing why the document could not be processed. Is null unless processing_state is FAILED.
     
@@ -54,19 +50,19 @@ class SlimDocument(pydantic.BaseModel):
     - `INVALID_FILE` - Invalid File
     """
 
-    status: typing.Optional[DocumentStatus] = pydantic.Field(default=None)
+    status: typing.Optional[DocumentStatus] = pydantic_v1.Field(default=None)
     """
     The document's current status.
     
     - `ACTIVE` - Active
     """
 
-    keywords: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    keywords: typing.Optional[typing.List[str]] = pydantic_v1.Field(default=None)
     """
     A list of keywords associated with this document. Originally provided when uploading the document.
     """
 
-    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
+    metadata: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
     """
     A previously supplied JSON object containing metadata that can be filtered on when searching.
     """
@@ -84,5 +80,5 @@ class SlimDocument(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/slim_workflow_deployment.py

@@ -4,29 +4,25 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
 from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .vellum_variable import VellumVariable
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class SlimWorkflowDeployment(pydantic.BaseModel):
+class SlimWorkflowDeployment(pydantic_v1.BaseModel):
     id: str
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     A name that uniquely identifies this workflow deployment within its workspace
     """
 
-    label: str = pydantic.Field()
+    label: str = pydantic_v1.Field()
     """
     A human-readable label for the workflow deployment
     """
 
-    status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
+    status: typing.Optional[EntityStatus] = pydantic_v1.Field(default=None)
     """
     The current status of the workflow deployment
     
@@ -34,7 +30,7 @@ class SlimWorkflowDeployment(pydantic.BaseModel):
     - `ARCHIVED` - Archived
     """
 
-    environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
+    environment: typing.Optional[EnvironmentEnum] = pydantic_v1.Field(default=None)
     """
     The environment this workflow deployment is used in
     
@@ -45,12 +41,12 @@ class SlimWorkflowDeployment(pydantic.BaseModel):
 
     created: dt.datetime
     last_deployed_on: dt.datetime
-    input_variables: typing.List[VellumVariable] = pydantic.Field()
+    input_variables: typing.List[VellumVariable] = pydantic_v1.Field()
     """
     The input variables this Workflow Deployment expects to receive values for when it is executed.
     """
 
-    output_variables: typing.List[VellumVariable] = pydantic.Field()
+    output_variables: typing.List[VellumVariable] = pydantic_v1.Field()
     """
     The output variables this Workflow Deployment will produce when it is executed.
     """
@@ -66,5 +62,5 @@ class SlimWorkflowDeployment(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/streaming_execute_prompt_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 .prompt_output import PromptOutput
 from .streaming_prompt_execution_meta import StreamingPromptExecutionMeta
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class StreamingExecutePromptEvent(pydantic.BaseModel):
+class StreamingExecutePromptEvent(pydantic_v1.BaseModel):
     """
     The data returned for each delta during the prompt execution stream.
     """
@@ -22,7 +18,7 @@ class StreamingExecutePromptEvent(pydantic.BaseModel):
     output_index: int
     execution_id: str
     meta: typing.Optional[StreamingPromptExecutionMeta] = None
-    raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
+    raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(default=None)
     """
     The subset of the raw response from the model that the request opted into with `expand_raw`.
     """
@@ -38,5 +34,5 @@ class StreamingExecutePromptEvent(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/streaming_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 StreamingPromptExecutionMeta(pydantic.BaseModel):
+class StreamingPromptExecutionMeta(pydantic_v1.BaseModel):
     """
     The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
     """
@@ -29,5 +25,5 @@ class StreamingPromptExecutionMeta(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/streaming_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_output_compiled_value import NodeOutputCompiledValue
 from .workflow_node_result_data import WorkflowNodeResultData
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
 
-
-class StreamingWorkflowNodeResultEvent(pydantic.BaseModel):
+class StreamingWorkflowNodeResultEvent(pydantic_v1.BaseModel):
     """
     An event that indicates that the node has execution is in progress.
     """
@@ -38,5 +34,5 @@ class StreamingWorkflowNodeResultEvent(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/string_chat_message_content.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 StringChatMessageContent(pydantic.BaseModel):
+class StringChatMessageContent(pydantic_v1.BaseModel):
     """
     A string value that is used in a chat message.
     """
@@ -29,5 +25,5 @@ class StringChatMessageContent(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/string_chat_message_content_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 StringChatMessageContentRequest(pydantic.BaseModel):
+class StringChatMessageContentRequest(pydantic_v1.BaseModel):
     """
     A string value that is used in a chat message.
     """
@@ -29,5 +25,5 @@ class StringChatMessageContentRequest(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/string_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 StringInputRequest(pydantic.BaseModel):
+class StringInputRequest(pydantic_v1.BaseModel):
     """
     A user input representing a string value
     """
 
-    name: str = pydantic.Field()
+    name: str = pydantic_v1.Field()
     """
     The variable's name, as defined in the deployment.
     """
@@ -34,5 +30,5 @@ class StringInputRequest(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/string_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 StringVariableValue(pydantic.BaseModel):
+class StringVariableValue(pydantic_v1.BaseModel):
     value: typing.Optional[str] = None
 
     def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class StringVariableValue(pydantic.BaseModel):
     class Config:
         frozen = True
         smart_union = True
-        extra = pydantic.Extra.allow
+        extra = pydantic_v1.Extra.allow
         json_encoders = {dt.datetime: serialize_datetime}