vellum-ai 0.14.45__py3-none-any.whl → 0.14.47__py3-none-any.whl

vellum/client/README.md

@@ -132,10 +132,10 @@ for chunk in response:
 ### Retries
 
 The SDK is instrumented with automatic retries with exponential backoff. A request will be retried as long
-as the request is deemed retriable and the number of retry attempts has not grown larger than the configured
+as the request is deemed retryable and the number of retry attempts has not grown larger than the configured
 retry limit (default: 2).
 
-A request is deemed retriable when any of the following HTTP status codes is returned:
+A request is deemed retryable when any of the following HTTP status codes is returned:
 
 - [408](https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/408) (Timeout)
 - [429](https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/429) (Too Many Requests)

vellum/client/__init__.py

@@ -219,6 +219,9 @@ class Vellum:
                     object_=bearer_token, annotation=ExecuteApiRequestBearerToken, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -307,6 +310,9 @@ class Vellum:
                 ),
                 "output_type": output_type,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -425,6 +431,9 @@ class Vellum:
                 "expand_raw": expand_raw,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -575,6 +584,9 @@ class Vellum:
                 "expand_raw": expand_raw,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         ) as _response:
@@ -591,7 +603,7 @@ class Vellum:
                                     object_=json.loads(_text),
                                 ),
                             )
-                        except:
+                        except Exception:
                             pass
                     return
                 _response.read()
@@ -719,6 +731,9 @@ class Vellum:
                 "external_id": external_id,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -852,6 +867,9 @@ class Vellum:
                 "event_types": event_types,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         ) as _response:
@@ -868,7 +886,7 @@ class Vellum:
                                     object_=json.loads(_text),
                                 ),
                             )
-                        except:
+                        except Exception:
                             pass
                     return
                 _response.read()
@@ -973,6 +991,9 @@ class Vellum:
                     object_=options, annotation=GenerateOptionsRequest, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1098,6 +1119,9 @@ class Vellum:
                     object_=options, annotation=GenerateOptionsRequest, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         ) as _response:
@@ -1114,7 +1138,7 @@ class Vellum:
                                     object_=json.loads(_text),
                                 ),
                             )
-                        except:
+                        except Exception:
                             pass
                     return
                 _response.read()
@@ -1225,6 +1249,9 @@ class Vellum:
                 ),
                 "document_index": document_index,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1323,6 +1350,9 @@ class Vellum:
                     object_=actuals, annotation=typing.Sequence[SubmitCompletionActualRequest], direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1417,6 +1447,9 @@ class Vellum:
                 "execution_id": execution_id,
                 "external_id": external_id,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1570,6 +1603,9 @@ class AsyncVellum:
                     object_=bearer_token, annotation=ExecuteApiRequestBearerToken, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1666,6 +1702,9 @@ class AsyncVellum:
                 ),
                 "output_type": output_type,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1792,6 +1831,9 @@ class AsyncVellum:
                 "expand_raw": expand_raw,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1950,6 +1992,9 @@ class AsyncVellum:
                 "expand_raw": expand_raw,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         ) as _response:
@@ -1966,7 +2011,7 @@ class AsyncVellum:
                                     object_=json.loads(_text),
                                 ),
                             )
-                        except:
+                        except Exception:
                             pass
                     return
                 await _response.aread()
@@ -2102,6 +2147,9 @@ class AsyncVellum:
                 "external_id": external_id,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2243,6 +2291,9 @@ class AsyncVellum:
                 "event_types": event_types,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         ) as _response:
@@ -2259,7 +2310,7 @@ class AsyncVellum:
                                     object_=json.loads(_text),
                                 ),
                             )
-                        except:
+                        except Exception:
                             pass
                     return
                 await _response.aread()
@@ -2372,6 +2423,9 @@ class AsyncVellum:
                     object_=options, annotation=GenerateOptionsRequest, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2505,6 +2559,9 @@ class AsyncVellum:
                     object_=options, annotation=GenerateOptionsRequest, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         ) as _response:
@@ -2521,7 +2578,7 @@ class AsyncVellum:
                                     object_=json.loads(_text),
                                 ),
                             )
-                        except:
+                        except Exception:
                             pass
                     return
                 await _response.aread()
@@ -2640,6 +2697,9 @@ class AsyncVellum:
                 ),
                 "document_index": document_index,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2746,6 +2806,9 @@ class AsyncVellum:
                     object_=actuals, annotation=typing.Sequence[SubmitCompletionActualRequest], direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -2848,6 +2911,9 @@ class AsyncVellum:
                 "execution_id": execution_id,
                 "external_id": external_id,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )

vellum/client/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.14.45",
+            "X-Fern-SDK-Version": "0.14.47",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

vellum/client/core/file.py

@@ -43,20 +43,25 @@ def convert_file_dict_to_httpx_tuples(
     return httpx_tuples
 
 
-def with_content_type(*, file: File, content_type: str) -> File:
-    """ """
+def with_content_type(*, file: File, default_content_type: str) -> File:
+    """
+    This function resolves to the file's content type, if provided, and defaults
+    to the default_content_type value if not.
+    """
     if isinstance(file, tuple):
         if len(file) == 2:
             filename, content = cast(Tuple[Optional[str], FileContent], file)  # type: ignore
-            return (filename, content, content_type)
+            return (filename, content, default_content_type)
         elif len(file) == 3:
-            filename, content, _ = cast(Tuple[Optional[str], FileContent, Optional[str]], file)  # type: ignore
-            return (filename, content, content_type)
+            filename, content, file_content_type = cast(Tuple[Optional[str], FileContent, Optional[str]], file)  # type: ignore
+            out_content_type = file_content_type or default_content_type
+            return (filename, content, out_content_type)
         elif len(file) == 4:
-            filename, content, _, headers = cast(  # type: ignore
+            filename, content, file_content_type, headers = cast(  # type: ignore
                 Tuple[Optional[str], FileContent, Optional[str], Mapping[str, str]], file
             )
-            return (filename, content, content_type, headers)
+            out_content_type = file_content_type or default_content_type
+            return (filename, content, out_content_type, headers)
         else:
             raise ValueError(f"Unexpected tuple length: {len(file)}")
-    return (None, file, content_type)
+    return (None, file, default_content_type)

vellum/client/core/http_client.py

@@ -85,8 +85,8 @@ def _retry_timeout(response: httpx.Response, retries: int) -> float:
 
 
 def _should_retry(response: httpx.Response) -> bool:
-    retriable_400s = [429, 408, 409]
-    return response.status_code >= 500 or response.status_code in retriable_400s
+    retryable_400s = [429, 408, 409]
+    return response.status_code >= 500 or response.status_code in retryable_400s
 
 
 def remove_omit_from_dict(
@@ -183,7 +183,7 @@ class HttpClient:
         files: typing.Optional[typing.Dict[str, typing.Optional[typing.Union[File, typing.List[File]]]]] = None,
         headers: typing.Optional[typing.Dict[str, typing.Any]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-        retries: int = 0,
+        retries: int = 2,
         omit: typing.Optional[typing.Any] = None,
     ) -> httpx.Response:
         base_url = self.get_base_url(base_url)
@@ -227,9 +227,11 @@ class HttpClient:
             json=json_body,
             data=data_body,
             content=content,
-            files=convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
-            if (files is not None and files is not omit)
-            else None,
+            files=(
+                convert_file_dict_to_httpx_tuples(remove_omit_from_dict(remove_none_from_dict(files), omit))
+                if (files is not None and files is not omit)
+                else None
+            ),
             timeout=timeout,
         )
 
@@ -267,7 +269,7 @@ class HttpClient:
         files: typing.Optional[typing.Dict[str, typing.Optional[typing.Union[File, typing.List[File]]]]] = None,
         headers: typing.Optional[typing.Dict[str, typing.Any]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-        retries: int = 0,
+        retries: int = 2,
         omit: typing.Optional[typing.Any] = None,
     ) -> typing.Iterator[httpx.Response]:
         base_url = self.get_base_url(base_url)
@@ -311,9 +313,11 @@ class HttpClient:
             json=json_body,
             data=data_body,
             content=content,
-            files=convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
-            if (files is not None and files is not omit)
-            else None,
+            files=(
+                convert_file_dict_to_httpx_tuples(remove_omit_from_dict(remove_none_from_dict(files), omit))
+                if (files is not None and files is not omit)
+                else None
+            ),
             timeout=timeout,
         ) as stream:
             yield stream
@@ -355,7 +359,7 @@ class AsyncHttpClient:
         files: typing.Optional[typing.Dict[str, typing.Optional[typing.Union[File, typing.List[File]]]]] = None,
         headers: typing.Optional[typing.Dict[str, typing.Any]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-        retries: int = 0,
+        retries: int = 2,
         omit: typing.Optional[typing.Any] = None,
     ) -> httpx.Response:
         base_url = self.get_base_url(base_url)
@@ -400,7 +404,11 @@ class AsyncHttpClient:
             json=json_body,
             data=data_body,
             content=content,
-            files=convert_file_dict_to_httpx_tuples(remove_none_from_dict(files)) if files is not None else None,
+            files=(
+                convert_file_dict_to_httpx_tuples(remove_omit_from_dict(remove_none_from_dict(files), omit))
+                if files is not None
+                else None
+            ),
             timeout=timeout,
         )
 
@@ -437,7 +445,7 @@ class AsyncHttpClient:
         files: typing.Optional[typing.Dict[str, typing.Optional[typing.Union[File, typing.List[File]]]]] = None,
         headers: typing.Optional[typing.Dict[str, typing.Any]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-        retries: int = 0,
+        retries: int = 2,
         omit: typing.Optional[typing.Any] = None,
     ) -> typing.AsyncIterator[httpx.Response]:
         base_url = self.get_base_url(base_url)
@@ -481,7 +489,11 @@ class AsyncHttpClient:
             json=json_body,
             data=data_body,
             content=content,
-            files=convert_file_dict_to_httpx_tuples(remove_none_from_dict(files)) if files is not None else None,
+            files=(
+                convert_file_dict_to_httpx_tuples(remove_omit_from_dict(remove_none_from_dict(files), omit))
+                if files is not None
+                else None
+            ),
             timeout=timeout,
         ) as stream:
             yield stream

vellum/client/core/pydantic_utilities.py

@@ -82,7 +82,7 @@ def to_jsonable_with_fallback(
 class UniversalBaseModel(pydantic.BaseModel):
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            # Allow fields begining with `model_` to be used in the model
+            # Allow fields beginning with `model_` to be used in the model
             protected_namespaces=(),
         )  # type: ignore # Pydantic v2
 
@@ -131,7 +131,7 @@ class UniversalBaseModel(pydantic.BaseModel):
         Override the default dict method to `exclude_unset` by default. This function patches
         `exclude_unset` to work include fields within non-None default values.
         """
-        # Note: the logic here is multi-plexed given the levers exposed in Pydantic V1 vs V2
+        # Note: the logic here is multiplexed given the levers exposed in Pydantic V1 vs V2
         # Pydantic V1's .dict can be extremely slow, so we do not want to call it twice.
         #
         # We'd ideally do the same for Pydantic V2, but it shells out to a library to serialize models

vellum/client/core/request_options.py

@@ -23,6 +23,8 @@ class RequestOptions(typing.TypedDict, total=False):
         - additional_query_parameters: typing.Dict[str, typing.Any]. A dictionary containing additional parameters to spread into the request's query parameters dict
 
         - additional_body_parameters: typing.Dict[str, typing.Any]. A dictionary containing additional parameters to spread into the request's body parameters dict
+
+        - chunk_size: int. The size, in bytes, to process each chunk of data being streamed back within the response. This equates to leveraging `chunk_size` within `requests` or `httpx`, and is only leveraged for file downloads.
     """
 
     timeout_in_seconds: NotRequired[int]
@@ -30,3 +32,4 @@ class RequestOptions(typing.TypedDict, total=False):
     additional_headers: NotRequired[typing.Dict[str, typing.Any]]
     additional_query_parameters: NotRequired[typing.Dict[str, typing.Any]]
     additional_body_parameters: NotRequired[typing.Dict[str, typing.Any]]
+    chunk_size: NotRequired[int]

vellum/client/resources/ad_hoc/client.py

@@ -133,6 +133,9 @@ class AdHocClient:
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -286,6 +289,9 @@ class AdHocClient:
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         ) as _response:
@@ -302,7 +308,7 @@ class AdHocClient:
                                     object_=json.loads(_text),
                                 ),
                             )
-                        except:
+                        except Exception:
                             pass
                     return
                 _response.read()
@@ -458,6 +464,9 @@ class AsyncAdHocClient:
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -619,6 +628,9 @@ class AsyncAdHocClient:
                     object_=expand_meta, annotation=AdHocExpandMeta, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         ) as _response:
@@ -635,7 +647,7 @@ class AsyncAdHocClient:
                                     object_=json.loads(_text),
                                 ),
                             )
-                        except:
+                        except Exception:
                             pass
                     return
                 await _response.aread()

vellum/client/resources/container_images/client.py

@@ -220,6 +220,9 @@ class ContainerImagesClient:
                 "sha": sha,
                 "tags": tags,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -476,6 +479,9 @@ class AsyncContainerImagesClient:
                 "sha": sha,
                 "tags": tags,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )

vellum/client/resources/deployments/client.py

@@ -380,6 +380,9 @@ class DeploymentsClient:
             json={
                 "history_item_id": history_item_id,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -474,6 +477,9 @@ class DeploymentsClient:
                     object_=expand_meta, annotation=CompilePromptDeploymentExpandMetaRequest, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -932,6 +938,9 @@ class AsyncDeploymentsClient:
             json={
                 "history_item_id": history_item_id,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -1034,6 +1043,9 @@ class AsyncDeploymentsClient:
                     object_=expand_meta, annotation=CompilePromptDeploymentExpandMetaRequest, direction="write"
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )

vellum/client/resources/document_indexes/client.py

@@ -187,6 +187,9 @@ class DocumentIndexesClient:
                 ),
                 "copy_documents_from_index_id": copy_documents_from_index_id,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -305,6 +308,9 @@ class DocumentIndexesClient:
                 "label": label,
                 "status": status,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -415,6 +421,9 @@ class DocumentIndexesClient:
                 "label": label,
                 "status": status,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -711,6 +720,9 @@ class AsyncDocumentIndexesClient:
                 ),
                 "copy_documents_from_index_id": copy_documents_from_index_id,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -845,6 +857,9 @@ class AsyncDocumentIndexesClient:
                 "label": label,
                 "status": status,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -971,6 +986,9 @@ class AsyncDocumentIndexesClient:
                 "label": label,
                 "status": status,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )

vellum/client/resources/documents/client.py

@@ -241,6 +241,9 @@ class DocumentsClient:
                 "status": status,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -627,6 +630,9 @@ class AsyncDocumentsClient:
                 "status": status,
                 "metadata": metadata,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )

vellum/client/resources/folder_entities/client.py

@@ -152,6 +152,9 @@ class FolderEntitiesClient:
             json={
                 "entity_id": entity_id,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -317,6 +320,9 @@ class AsyncFolderEntitiesClient:
             json={
                 "entity_id": entity_id,
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )