arbi 0.1.1__py3-none-any.whl → 0.2.0__py3-none-any.whl

arbi/resources/api/document/document.py

@@ -2,13 +2,23 @@
 
 from __future__ import annotations
 
-from typing import List, Union, Mapping, Optional, cast
+from typing import Union, Mapping, Optional, cast
 from datetime import date
 from typing_extensions import Literal
 
 import httpx
 
-from ...._types import NOT_GIVEN, Body, Query, Headers, NotGiven, FileTypes
+from ...._types import (
+    Body,
+    Omit,
+    Query,
+    Headers,
+    NotGiven,
+    FileTypes,
+    SequenceNotStr,
+    omit,
+    not_given,
+)
 from ...._utils import extract_files, maybe_transform, deepcopy_minimal, async_maybe_transform
 from ...._compat import cached_property
 from .annotation import (
@@ -26,7 +36,12 @@ from ...._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ....types.api import document_view_params, document_update_params, document_upload_params
+from ....types.api import (
+    document_view_params,
+    document_update_params,
+    document_upload_params,
+    document_upload_from_url_params,
+)
 from ...._base_client import make_request_options
 from ....types.api.doc_response import DocResponse
 from ....types.api.document_delete_response import DocumentDeleteResponse
@@ -65,15 +80,15 @@ class DocumentResource(SyncAPIResource):
         self,
         document_ext_id: str,
         *,
-        doc_date: Union[str, date, None] | NotGiven = NOT_GIVEN,
-        shared: Optional[bool] | NotGiven = NOT_GIVEN,
-        title: Optional[str] | NotGiven = NOT_GIVEN,
+        doc_date: Union[str, date, None] | Omit = omit,
+        shared: Optional[bool] | Omit = omit,
+        title: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocumentUpdateResponse:
         """Update document metadata such as title, date, or sharing status.
 
@@ -116,7 +131,7 @@ class DocumentResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocumentDeleteResponse:
         """Delete a document by its external ID.
 
@@ -151,7 +166,7 @@ class DocumentResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> object:
         """Download a document by its external ID.
 
@@ -186,7 +201,7 @@ class DocumentResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocResponse:
         """Retrieve document metadata by its external ID.
 
@@ -222,7 +237,7 @@ class DocumentResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocumentGetParsedResponse:
         """Retrieve the full parsed document to be handled by the frontend.
 
@@ -259,7 +274,7 @@ class DocumentResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocumentGetTagsResponse:
         """
         Get all tags applied to a specific document along with doctag metadata.
@@ -287,15 +302,15 @@ class DocumentResource(SyncAPIResource):
         self,
         *,
         workspace_ext_id: str,
-        files: List[FileTypes],
-        config_ext_id: Optional[str] | NotGiven = NOT_GIVEN,
-        shared: bool | NotGiven = NOT_GIVEN,
+        files: SequenceNotStr[FileTypes],
+        config_ext_id: Optional[str] | Omit = omit,
+        shared: bool | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> object:
         """Upload multiple documents to a workspace with encryption.
 
@@ -344,17 +359,70 @@ class DocumentResource(SyncAPIResource):
             cast_to=object,
         )
 
+    def upload_from_url(
+        self,
+        *,
+        urls: SequenceNotStr[str],
+        workspace_ext_id: str,
+        config_ext_id: Optional[str] | Omit = omit,
+        shared: bool | Omit = omit,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> object:
+        """
+        Download and upload documents from URLs to a workspace with encryption.
+        Documents are queued for processing, parsed, and indexed for vector search.
+
+        Args:
+          urls: URLs to download documents from
+
+          config_ext_id: Configuration to use for processing
+
+          shared: Whether the document should be shared with workspace members
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/api/document/upload-url",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "urls": urls,
+                        "workspace_ext_id": workspace_ext_id,
+                        "config_ext_id": config_ext_id,
+                        "shared": shared,
+                    },
+                    document_upload_from_url_params.DocumentUploadFromURLParams,
+                ),
+            ),
+            cast_to=object,
+        )
+
     def view(
         self,
         document_ext_id: str,
         *,
-        page: Optional[int] | NotGiven = NOT_GIVEN,
+        page: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> object:
         """View a document inline in the browser.
 
@@ -415,15 +483,15 @@ class AsyncDocumentResource(AsyncAPIResource):
         self,
         document_ext_id: str,
         *,
-        doc_date: Union[str, date, None] | NotGiven = NOT_GIVEN,
-        shared: Optional[bool] | NotGiven = NOT_GIVEN,
-        title: Optional[str] | NotGiven = NOT_GIVEN,
+        doc_date: Union[str, date, None] | Omit = omit,
+        shared: Optional[bool] | Omit = omit,
+        title: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocumentUpdateResponse:
         """Update document metadata such as title, date, or sharing status.
 
@@ -466,7 +534,7 @@ class AsyncDocumentResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocumentDeleteResponse:
         """Delete a document by its external ID.
 
@@ -501,7 +569,7 @@ class AsyncDocumentResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> object:
         """Download a document by its external ID.
 
@@ -536,7 +604,7 @@ class AsyncDocumentResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocResponse:
         """Retrieve document metadata by its external ID.
 
@@ -572,7 +640,7 @@ class AsyncDocumentResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocumentGetParsedResponse:
         """Retrieve the full parsed document to be handled by the frontend.
 
@@ -609,7 +677,7 @@ class AsyncDocumentResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DocumentGetTagsResponse:
         """
         Get all tags applied to a specific document along with doctag metadata.
@@ -637,15 +705,15 @@ class AsyncDocumentResource(AsyncAPIResource):
         self,
         *,
         workspace_ext_id: str,
-        files: List[FileTypes],
-        config_ext_id: Optional[str] | NotGiven = NOT_GIVEN,
-        shared: bool | NotGiven = NOT_GIVEN,
+        files: SequenceNotStr[FileTypes],
+        config_ext_id: Optional[str] | Omit = omit,
+        shared: bool | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> object:
         """Upload multiple documents to a workspace with encryption.
 
@@ -694,17 +762,70 @@ class AsyncDocumentResource(AsyncAPIResource):
             cast_to=object,
         )
 
+    async def upload_from_url(
+        self,
+        *,
+        urls: SequenceNotStr[str],
+        workspace_ext_id: str,
+        config_ext_id: Optional[str] | Omit = omit,
+        shared: bool | Omit = omit,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> object:
+        """
+        Download and upload documents from URLs to a workspace with encryption.
+        Documents are queued for processing, parsed, and indexed for vector search.
+
+        Args:
+          urls: URLs to download documents from
+
+          config_ext_id: Configuration to use for processing
+
+          shared: Whether the document should be shared with workspace members
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/api/document/upload-url",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "urls": urls,
+                        "workspace_ext_id": workspace_ext_id,
+                        "config_ext_id": config_ext_id,
+                        "shared": shared,
+                    },
+                    document_upload_from_url_params.DocumentUploadFromURLParams,
+                ),
+            ),
+            cast_to=object,
+        )
+
     async def view(
         self,
         document_ext_id: str,
         *,
-        page: Optional[int] | NotGiven = NOT_GIVEN,
+        page: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> object:
         """View a document inline in the browser.
 
@@ -762,6 +883,9 @@ class DocumentResourceWithRawResponse:
         self.upload = to_raw_response_wrapper(
             document.upload,
         )
+        self.upload_from_url = to_raw_response_wrapper(
+            document.upload_from_url,
+        )
         self.view = to_raw_response_wrapper(
             document.view,
         )
@@ -796,6 +920,9 @@ class AsyncDocumentResourceWithRawResponse:
         self.upload = async_to_raw_response_wrapper(
             document.upload,
         )
+        self.upload_from_url = async_to_raw_response_wrapper(
+            document.upload_from_url,
+        )
         self.view = async_to_raw_response_wrapper(
             document.view,
         )
@@ -830,6 +957,9 @@ class DocumentResourceWithStreamingResponse:
         self.upload = to_streamed_response_wrapper(
             document.upload,
         )
+        self.upload_from_url = to_streamed_response_wrapper(
+            document.upload_from_url,
+        )
         self.view = to_streamed_response_wrapper(
             document.view,
         )
@@ -864,6 +994,9 @@ class AsyncDocumentResourceWithStreamingResponse:
         self.upload = async_to_streamed_response_wrapper(
             document.upload,
         )
+        self.upload_from_url = async_to_streamed_response_wrapper(
+            document.upload_from_url,
+        )
         self.view = async_to_streamed_response_wrapper(
             document.view,
         )

arbi/resources/api/health.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 import httpx
 
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import Body, Query, Headers, NotGiven, not_given
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
@@ -18,6 +18,8 @@ from ...types.api.health_check_app_response import HealthCheckAppResponse
 from ...types.api.health_get_models_response import HealthGetModelsResponse
 from ...types.api.health_check_models_response import HealthCheckModelsResponse
 from ...types.api.health_check_services_response import HealthCheckServicesResponse
+from ...types.api.health_retrieve_status_response import HealthRetrieveStatusResponse
+from ...types.api.health_retrieve_version_response import HealthRetrieveVersionResponse
 
 __all__ = ["HealthResource", "AsyncHealthResource"]
 
@@ -50,9 +52,13 @@ class HealthResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> HealthCheckAppResponse:
-        """Lightweight health check endpoint for the arbi-app itself."""
+        """Lightweight health check endpoint for the arbi-app itself.
+
+        Returns version
+        information along with health status.
+        """
         return self._get(
             "/api/health/app",
             options=make_request_options(
@@ -69,7 +75,7 @@ class HealthResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> HealthCheckModelsResponse:
         """
         Endpoint to check the health of various models hosted on the LiteLLM platform.
@@ -92,7 +98,7 @@ class HealthResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> HealthCheckServicesResponse:
         """
         Health check endpoint that verifies the status of the application and external
@@ -115,7 +121,7 @@ class HealthResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> HealthGetModelsResponse:
         """Returns available models with model_name and api_type fields"""
         return self._get(
@@ -126,6 +132,48 @@ class HealthResource(SyncAPIResource):
             cast_to=HealthGetModelsResponse,
         )
 
+    def retrieve_status(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> HealthRetrieveStatusResponse:
+        """
+        Consolidated health endpoint that returns status, version information, and
+        service health. This combines the functionality of /app, /version, and /services
+        endpoints.
+        """
+        return self._get(
+            "/api/health/",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=HealthRetrieveStatusResponse,
+        )
+
+    def retrieve_version(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> HealthRetrieveVersionResponse:
+        """Get version information for backend and frontend components."""
+        return self._get(
+            "/api/health/version",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=HealthRetrieveVersionResponse,
+        )
+
 
 class AsyncHealthResource(AsyncAPIResource):
     @cached_property
@@ -155,9 +203,13 @@ class AsyncHealthResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> HealthCheckAppResponse:
-        """Lightweight health check endpoint for the arbi-app itself."""
+        """Lightweight health check endpoint for the arbi-app itself.
+
+        Returns version
+        information along with health status.
+        """
         return await self._get(
             "/api/health/app",
             options=make_request_options(
@@ -174,7 +226,7 @@ class AsyncHealthResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> HealthCheckModelsResponse:
         """
         Endpoint to check the health of various models hosted on the LiteLLM platform.
@@ -197,7 +249,7 @@ class AsyncHealthResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> HealthCheckServicesResponse:
         """
         Health check endpoint that verifies the status of the application and external
@@ -220,7 +272,7 @@ class AsyncHealthResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> HealthGetModelsResponse:
         """Returns available models with model_name and api_type fields"""
         return await self._get(
@@ -231,6 +283,48 @@ class AsyncHealthResource(AsyncAPIResource):
             cast_to=HealthGetModelsResponse,
         )
 
+    async def retrieve_status(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> HealthRetrieveStatusResponse:
+        """
+        Consolidated health endpoint that returns status, version information, and
+        service health. This combines the functionality of /app, /version, and /services
+        endpoints.
+        """
+        return await self._get(
+            "/api/health/",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=HealthRetrieveStatusResponse,
+        )
+
+    async def retrieve_version(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> HealthRetrieveVersionResponse:
+        """Get version information for backend and frontend components."""
+        return await self._get(
+            "/api/health/version",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=HealthRetrieveVersionResponse,
+        )
+
 
 class HealthResourceWithRawResponse:
     def __init__(self, health: HealthResource) -> None:
@@ -248,6 +342,12 @@ class HealthResourceWithRawResponse:
         self.get_models = to_raw_response_wrapper(
             health.get_models,
         )
+        self.retrieve_status = to_raw_response_wrapper(
+            health.retrieve_status,
+        )
+        self.retrieve_version = to_raw_response_wrapper(
+            health.retrieve_version,
+        )
 
 
 class AsyncHealthResourceWithRawResponse:
@@ -266,6 +366,12 @@ class AsyncHealthResourceWithRawResponse:
         self.get_models = async_to_raw_response_wrapper(
             health.get_models,
         )
+        self.retrieve_status = async_to_raw_response_wrapper(
+            health.retrieve_status,
+        )
+        self.retrieve_version = async_to_raw_response_wrapper(
+            health.retrieve_version,
+        )
 
 
 class HealthResourceWithStreamingResponse:
@@ -284,6 +390,12 @@ class HealthResourceWithStreamingResponse:
         self.get_models = to_streamed_response_wrapper(
             health.get_models,
         )
+        self.retrieve_status = to_streamed_response_wrapper(
+            health.retrieve_status,
+        )
+        self.retrieve_version = to_streamed_response_wrapper(
+            health.retrieve_version,
+        )
 
 
 class AsyncHealthResourceWithStreamingResponse:
@@ -302,3 +414,9 @@ class AsyncHealthResourceWithStreamingResponse:
         self.get_models = async_to_streamed_response_wrapper(
             health.get_models,
         )
+        self.retrieve_status = async_to_streamed_response_wrapper(
+            health.retrieve_status,
+        )
+        self.retrieve_version = async_to_streamed_response_wrapper(
+            health.retrieve_version,
+        )