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

vellum/resources/document_indexes/client.py

@@ -7,6 +7,7 @@ from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import pydantic_v1
 from ...core.remove_none_from_dict import remove_none_from_dict
 from ...core.request_options import RequestOptions
 from ...types.document_index_read import DocumentIndexRead
@@ -15,11 +16,6 @@ from ...types.environment_enum import EnvironmentEnum
 from ...types.paginated_document_index_read_list import PaginatedDocumentIndexReadList
 from .types.document_indexes_list_request_status import DocumentIndexesListRequestStatus
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 
@@ -61,8 +57,8 @@ class DocumentIndexesClient:
         client.document_indexes.list()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/document-indexes"),
+            method="GET",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/document-indexes"),
             params=jsonable_encoder(
                 remove_none_from_dict(
                     {
@@ -93,7 +89,7 @@ class DocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(PaginatedDocumentIndexReadList, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(PaginatedDocumentIndexReadList, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -169,8 +165,8 @@ class DocumentIndexesClient:
         if copy_documents_from_index_id is not OMIT:
             _request["copy_documents_from_index_id"] = copy_documents_from_index_id
         _response = self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/document-indexes"),
+            method="POST",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/document-indexes"),
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
@@ -195,7 +191,7 @@ class DocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -221,8 +217,8 @@ class DocumentIndexesClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(
+            method="GET",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -243,7 +239,7 @@ class DocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -294,8 +290,8 @@ class DocumentIndexesClient:
         if environment is not OMIT:
             _request["environment"] = environment
         _response = self._client_wrapper.httpx_client.request(
-            "PUT",
-            urllib.parse.urljoin(
+            method="PUT",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -322,7 +318,7 @@ class DocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -348,8 +344,8 @@ class DocumentIndexesClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "DELETE",
-            urllib.parse.urljoin(
+            method="DELETE",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -422,8 +418,8 @@ class DocumentIndexesClient:
         if environment is not OMIT:
             _request["environment"] = environment
         _response = self._client_wrapper.httpx_client.request(
-            "PATCH",
-            urllib.parse.urljoin(
+            method="PATCH",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -450,7 +446,7 @@ class DocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -495,8 +491,8 @@ class AsyncDocumentIndexesClient:
         await client.document_indexes.list()
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/document-indexes"),
+            method="GET",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/document-indexes"),
             params=jsonable_encoder(
                 remove_none_from_dict(
                     {
@@ -527,7 +523,7 @@ class AsyncDocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(PaginatedDocumentIndexReadList, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(PaginatedDocumentIndexReadList, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -603,8 +599,8 @@ class AsyncDocumentIndexesClient:
         if copy_documents_from_index_id is not OMIT:
             _request["copy_documents_from_index_id"] = copy_documents_from_index_id
         _response = await self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/document-indexes"),
+            method="POST",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/document-indexes"),
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
@@ -629,7 +625,7 @@ class AsyncDocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -655,8 +651,8 @@ class AsyncDocumentIndexesClient:
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(
+            method="GET",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -677,7 +673,7 @@ class AsyncDocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -728,8 +724,8 @@ class AsyncDocumentIndexesClient:
         if environment is not OMIT:
             _request["environment"] = environment
         _response = await self._client_wrapper.httpx_client.request(
-            "PUT",
-            urllib.parse.urljoin(
+            method="PUT",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -756,7 +752,7 @@ class AsyncDocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -782,8 +778,8 @@ class AsyncDocumentIndexesClient:
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "DELETE",
-            urllib.parse.urljoin(
+            method="DELETE",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -856,8 +852,8 @@ class AsyncDocumentIndexesClient:
         if environment is not OMIT:
             _request["environment"] = environment
         _response = await self._client_wrapper.httpx_client.request(
-            "PATCH",
-            urllib.parse.urljoin(
+            method="PATCH",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/document-indexes/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -884,7 +880,7 @@ class AsyncDocumentIndexesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentIndexRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:

vellum/resources/document_indexes/types/document_indexes_list_request_status.py

@@ -1,17 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class DocumentIndexesListRequestStatus(str, enum.Enum):
-    ACTIVE = "ACTIVE"
-    ARCHIVED = "ARCHIVED"
-
-    def visit(self, active: typing.Callable[[], T_Result], archived: typing.Callable[[], T_Result]) -> T_Result:
-        if self is DocumentIndexesListRequestStatus.ACTIVE:
-            return active()
-        if self is DocumentIndexesListRequestStatus.ARCHIVED:
-            return archived()
+DocumentIndexesListRequestStatus = typing.Union[typing.Literal["ACTIVE", "ARCHIVED"], typing.Any]

vellum/resources/documents/client.py

@@ -8,6 +8,7 @@ from ... import core
 from ...core.api_error import ApiError
 from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import pydantic_v1
 from ...core.remove_none_from_dict import remove_none_from_dict
 from ...core.request_options import RequestOptions
 from ...errors.bad_request_error import BadRequestError
@@ -18,11 +19,6 @@ from ...types.document_status import DocumentStatus
 from ...types.paginated_slim_document_list import PaginatedSlimDocumentList
 from ...types.upload_document_response import UploadDocumentResponse
 
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
 
@@ -62,8 +58,8 @@ class DocumentsClient:
         client.documents.list()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/documents"),
+            method="GET",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/documents"),
             params=jsonable_encoder(
                 remove_none_from_dict(
                     {
@@ -94,7 +90,55 @@ class DocumentsClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(PaginatedSlimDocumentList, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(PaginatedSlimDocumentList, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> DocumentRead:
+        """
+        Retrieve a Document via its Vellum-generated ID.
+
+        Parameters:
+            - id: str. A UUID string identifying this document.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from vellum.client import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.documents.retrieve(
+            id="id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            method="GET",
+            url=urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/", f"v1/documents/{jsonable_encoder(id)}"
+            ),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DocumentRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -118,8 +162,8 @@ class DocumentsClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "DELETE",
-            urllib.parse.urljoin(
+            method="DELETE",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/documents/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -188,8 +232,8 @@ class DocumentsClient:
         if metadata is not OMIT:
             _request["metadata"] = metadata
         _response = self._client_wrapper.httpx_client.request(
-            "PATCH",
-            urllib.parse.urljoin(
+            method="PATCH",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/documents/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -216,7 +260,7 @@ class DocumentsClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -267,11 +311,13 @@ class DocumentsClient:
         client = Vellum(
             api_key="YOUR_API_KEY",
         )
-        client.documents.upload()
+        client.documents.upload(
+            label="label",
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().documents}/", "v1/upload-document"),
+            method="POST",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_environment().documents}/", "v1/upload-document"),
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
@@ -317,13 +363,13 @@ class DocumentsClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(UploadDocumentResponse, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(UploadDocumentResponse, _response.json())  # type: ignore
         if _response.status_code == 400:
-            raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+            raise BadRequestError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 404:
-            raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+            raise NotFoundError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:
-            raise InternalServerError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+            raise InternalServerError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -366,8 +412,8 @@ class AsyncDocumentsClient:
         await client.documents.list()
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/documents"),
+            method="GET",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_environment().default}/", "v1/documents"),
             params=jsonable_encoder(
                 remove_none_from_dict(
                     {
@@ -398,7 +444,55 @@ class AsyncDocumentsClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(PaginatedSlimDocumentList, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(PaginatedSlimDocumentList, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def retrieve(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> DocumentRead:
+        """
+        Retrieve a Document via its Vellum-generated ID.
+
+        Parameters:
+            - id: str. A UUID string identifying this document.
+
+            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
+        ---
+        from vellum.client import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+        await client.documents.retrieve(
+            id="id",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            method="GET",
+            url=urllib.parse.urljoin(
+                f"{self._client_wrapper.get_environment().default}/", f"v1/documents/{jsonable_encoder(id)}"
+            ),
+            params=jsonable_encoder(
+                request_options.get("additional_query_parameters") if request_options is not None else None
+            ),
+            headers=jsonable_encoder(
+                remove_none_from_dict(
+                    {
+                        **self._client_wrapper.get_headers(),
+                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
+                    }
+                )
+            ),
+            timeout=request_options.get("timeout_in_seconds")
+            if request_options is not None and request_options.get("timeout_in_seconds") is not None
+            else self._client_wrapper.get_timeout(),
+            retries=0,
+            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DocumentRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -422,8 +516,8 @@ class AsyncDocumentsClient:
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "DELETE",
-            urllib.parse.urljoin(
+            method="DELETE",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/documents/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -492,8 +586,8 @@ class AsyncDocumentsClient:
         if metadata is not OMIT:
             _request["metadata"] = metadata
         _response = await self._client_wrapper.httpx_client.request(
-            "PATCH",
-            urllib.parse.urljoin(
+            method="PATCH",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/", f"v1/documents/{jsonable_encoder(id)}"
             ),
             params=jsonable_encoder(
@@ -520,7 +614,7 @@ class AsyncDocumentsClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(DocumentRead, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(DocumentRead, _response.json())  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -571,11 +665,13 @@ class AsyncDocumentsClient:
         client = AsyncVellum(
             api_key="YOUR_API_KEY",
         )
-        await client.documents.upload()
+        await client.documents.upload(
+            label="label",
+        )
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_environment().documents}/", "v1/upload-document"),
+            method="POST",
+            url=urllib.parse.urljoin(f"{self._client_wrapper.get_environment().documents}/", "v1/upload-document"),
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
@@ -621,13 +717,13 @@ class AsyncDocumentsClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(UploadDocumentResponse, _response.json())  # type: ignore
+            return pydantic_v1.parse_obj_as(UploadDocumentResponse, _response.json())  # type: ignore
         if _response.status_code == 400:
-            raise BadRequestError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+            raise BadRequestError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 404:
-            raise NotFoundError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+            raise NotFoundError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         if _response.status_code == 500:
-            raise InternalServerError(pydantic.parse_obj_as(typing.Any, _response.json()))  # type: ignore
+            raise InternalServerError(pydantic_v1.parse_obj_as(typing.Any, _response.json()))  # type: ignore
         try:
             _response_json = _response.json()
         except JSONDecodeError:

vellum/resources/folder_entities/client.py

@@ -44,8 +44,8 @@ class FolderEntitiesClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(
+            method="POST",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/",
                 f"v1/folders/{jsonable_encoder(folder_id)}/add-entity",
             ),
@@ -111,8 +111,8 @@ class AsyncFolderEntitiesClient:
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(
+            method="POST",
+            url=urllib.parse.urljoin(
                 f"{self._client_wrapper.get_environment().default}/",
                 f"v1/folders/{jsonable_encoder(folder_id)}/add-entity",
             ),