h2ogpte 1.6.44rc6__py3-none-any.whl → 1.6.44rc9__py3-none-any.whl

h2ogpte/__init__.py

@@ -3,7 +3,7 @@ from h2ogpte.h2ogpte import H2OGPTE
 from h2ogpte.h2ogpte_async import H2OGPTEAsync
 from h2ogpte.session_async import SessionAsync
 
-__version__ = "1.6.44rc6"
+__version__ = "1.6.44rc9"
 
 __all__ = [
     "H2OGPTE",

h2ogpte/connectors.py

@@ -49,22 +49,28 @@ def create_ingest_job_from_cloud_storage(
 class S3Credential:
     def __init__(
         self,
-        access_key_id: str,
-        secret_access_key: str,
+        access_key_id: str = None,
+        secret_access_key: str = None,
         session_token: str = None,
         role_arn: str = None,
+        external_id: str = None,
+        use_irsa: bool = False,
     ):
         """
         Creates an object with S3 credentials.
-        :param access_key_id: Access Key ID
-        :param secret_access_key: Secret Access Key
+        :param access_key_id: Access Key ID (not required when using IRSA)
+        :param secret_access_key: Secret Access Key (not required when using IRSA)
         :param session_token: Session Token (optional)
         :param role_arn: Role providing access to the S3 storage
+        :param external_id: External ID for cross-account role assumption (optional)
+        :param use_irsa: Enable IAM Roles for Service Accounts (IRSA) - set to True to use pod service account
         """
         self.access_key_id = access_key_id
         self.secret_access_key = secret_access_key
         self.session_token = session_token
         self.role_arn = role_arn
+        self.external_id = external_id
+        self.use_irsa = use_irsa
 
 
 def create_ingest_job_from_s3(

h2ogpte/h2ogpte.py

@@ -1249,6 +1249,22 @@ class H2OGPTE(H2OGPTESyncBase):
             )
         return collection_id
 
+    def reset_all_collection_expirations(self):
+        """
+        Reset any expiry dates and inactivity intervals that have been set for all collections (admin only). This will also reset any archived collections back to active.
+        """
+
+        header = self._get_auth_header()
+        with self._RESTClient(self) as rest_client:
+            result = _get_result(
+                lambda: _rest_to_client_exceptions(
+                    lambda: rest_client.collection_api.reset_all_collection_expirations(
+                        _headers=header,
+                    )
+                )
+            )
+        return result
+
     def set_collection_size_limit(
         self, collection_id: str, limit: Union[int, str]
     ) -> str:

h2ogpte/h2ogpte_async.py

@@ -1449,6 +1449,22 @@ class H2OGPTEAsync:
             )
         return collection_id
 
+    async def reset_all_collection_expirations(self):
+        """
+        Reset any expiry dates and inactivity intervals that have been set for all collections (admin only). This will also reset any archived collections back to active.
+        """
+
+        header = await self._get_auth_header()
+        async with self._RESTClient(self) as rest_client:
+            result = _get_result(
+                await _rest_to_client_exceptions(
+                    rest_client.collection_api.reset_all_collection_expirations(
+                        _headers=header,
+                    )
+                )
+            )
+        return result
+
     async def set_collection_size_limit(
         self, collection_id: str, limit: Union[int, str]
     ) -> str:

h2ogpte/rest_async/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "1.6.44-dev6"
+__version__ = "1.6.44-dev9"
 
 # import apis into sdk package
 from h2ogpte.rest_async.api.api_keys_api import APIKeysApi

h2ogpte/rest_async/api/collections_api.py

@@ -9559,6 +9559,255 @@ class CollectionsApi:
 
 
 
+    @validate_call
+    async def reset_all_collection_expirations(
+        self,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> None:
+        """Reset all collection expiry dates and inactivity intervals that have been set.
+
+        Reset any expiry dates and inactivity intervals that have been set for all collections (admin only). This will also reset any archived collections back to active.
+
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._reset_all_collection_expirations_serialize(
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '204': None,
+            '401': "EndpointError",
+        }
+        response_data = await self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        await response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    async def reset_all_collection_expirations_with_http_info(
+        self,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[None]:
+        """Reset all collection expiry dates and inactivity intervals that have been set.
+
+        Reset any expiry dates and inactivity intervals that have been set for all collections (admin only). This will also reset any archived collections back to active.
+
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._reset_all_collection_expirations_serialize(
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '204': None,
+            '401': "EndpointError",
+        }
+        response_data = await self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        await response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    async def reset_all_collection_expirations_without_preload_content(
+        self,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Reset all collection expiry dates and inactivity intervals that have been set.
+
+        Reset any expiry dates and inactivity intervals that have been set for all collections (admin only). This will also reset any archived collections back to active.
+
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._reset_all_collection_expirations_serialize(
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '204': None,
+            '401': "EndpointError",
+        }
+        response_data = await self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _reset_all_collection_expirations_serialize(
+        self,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'bearerAuth'
+        ]
+
+        return self.api_client.param_serialize(
+            method='POST',
+            resource_path='/collections/reset_all_collection_expirations',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
     @validate_call
     async def reset_collection_prompt_settings(
         self,

h2ogpte/rest_async/api_client.py

@@ -90,7 +90,7 @@ class ApiClient:
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'OpenAPI-Generator/1.6.44-dev6/python'
+        self.user_agent = 'OpenAPI-Generator/1.6.44-dev9/python'
         self.client_side_validation = configuration.client_side_validation
 
     async def __aenter__(self):

h2ogpte/rest_async/configuration.py

@@ -499,7 +499,7 @@ class Configuration:
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: v1.0.0\n"\
-               "SDK Package Version: 1.6.44-dev6".\
+               "SDK Package Version: 1.6.44-dev9".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self) -> List[HostSetting]:

h2ogpte/rest_async/models/s3_credentials.py

@@ -17,20 +17,22 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
 class S3Credentials(BaseModel):
     """
-    The object with S3 credentials. If the object is not provided, only public buckets will be accessible.
+    The object with S3 credentials. If the object is not provided, only public buckets will be accessible. When use_irsa is true, access_key_id and secret_access_key are optional and the pod's service account will be used for authentication.
     """ # noqa: E501
-    access_key_id: StrictStr
-    secret_access_key: StrictStr
-    session_token: Optional[StrictStr] = None
-    role_arn: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["access_key_id", "secret_access_key", "session_token", "role_arn"]
+    use_irsa: Optional[StrictBool] = Field(default=False, description="Enable IAM Roles for Service Accounts (IRSA) to authenticate using the pod's service account. When enabled, explicit credentials are not required.")
+    access_key_id: Optional[StrictStr] = Field(default=None, description="AWS access key ID. Not required when using IRSA.")
+    secret_access_key: Optional[StrictStr] = Field(default=None, description="AWS secret access key. Not required when using IRSA.")
+    session_token: Optional[StrictStr] = Field(default=None, description="AWS session token for temporary credentials.")
+    role_arn: Optional[StrictStr] = Field(default=None, description="The Amazon Resource Name (ARN) of IAM Role that will be utilized for accessing the S3 storage. When IRSA is enabled, this role will be assumed from the service account identity (role chaining). When IRSA is disabled, it requires specification of other credentials to identify an identity the role will be associated with.")
+    external_id: Optional[StrictStr] = Field(default=None, description="Optional security token for assuming cross-account roles. This is used to prevent the 'confused deputy' problem when a third party assumes a role in your account. Only required if the role's trust policy requires an external ID.")
+    __properties: ClassVar[List[str]] = ["use_irsa", "access_key_id", "secret_access_key", "session_token", "role_arn", "external_id"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -83,10 +85,12 @@ class S3Credentials(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
+            "use_irsa": obj.get("use_irsa") if obj.get("use_irsa") is not None else False,
             "access_key_id": obj.get("access_key_id"),
             "secret_access_key": obj.get("secret_access_key"),
             "session_token": obj.get("session_token"),
-            "role_arn": obj.get("role_arn")
+            "role_arn": obj.get("role_arn"),
+            "external_id": obj.get("external_id")
         })
         return _obj
 

h2ogpte/rest_sync/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "1.6.44-dev6"
+__version__ = "1.6.44-dev9"
 
 # import apis into sdk package
 from h2ogpte.rest_sync.api.api_keys_api import APIKeysApi

h2ogpte/rest_sync/api/collections_api.py

@@ -9559,6 +9559,255 @@ class CollectionsApi:
 
 
 
+    @validate_call
+    def reset_all_collection_expirations(
+        self,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> None:
+        """Reset all collection expiry dates and inactivity intervals that have been set.
+
+        Reset any expiry dates and inactivity intervals that have been set for all collections (admin only). This will also reset any archived collections back to active.
+
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._reset_all_collection_expirations_serialize(
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '204': None,
+            '401': "EndpointError",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def reset_all_collection_expirations_with_http_info(
+        self,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[None]:
+        """Reset all collection expiry dates and inactivity intervals that have been set.
+
+        Reset any expiry dates and inactivity intervals that have been set for all collections (admin only). This will also reset any archived collections back to active.
+
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._reset_all_collection_expirations_serialize(
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '204': None,
+            '401': "EndpointError",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def reset_all_collection_expirations_without_preload_content(
+        self,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Reset all collection expiry dates and inactivity intervals that have been set.
+
+        Reset any expiry dates and inactivity intervals that have been set for all collections (admin only). This will also reset any archived collections back to active.
+
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._reset_all_collection_expirations_serialize(
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '204': None,
+            '401': "EndpointError",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _reset_all_collection_expirations_serialize(
+        self,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'bearerAuth'
+        ]
+
+        return self.api_client.param_serialize(
+            method='POST',
+            resource_path='/collections/reset_all_collection_expirations',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
     @validate_call
     def reset_collection_prompt_settings(
         self,

h2ogpte/rest_sync/api_client.py

@@ -90,7 +90,7 @@ class ApiClient:
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'OpenAPI-Generator/1.6.44-dev6/python'
+        self.user_agent = 'OpenAPI-Generator/1.6.44-dev9/python'
         self.client_side_validation = configuration.client_side_validation
 
     def __enter__(self):

h2ogpte/rest_sync/configuration.py

@@ -503,7 +503,7 @@ class Configuration:
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: v1.0.0\n"\
-               "SDK Package Version: 1.6.44-dev6".\
+               "SDK Package Version: 1.6.44-dev9".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self) -> List[HostSetting]:

h2ogpte/rest_sync/models/s3_credentials.py

@@ -17,20 +17,22 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
 class S3Credentials(BaseModel):
     """
-    The object with S3 credentials. If the object is not provided, only public buckets will be accessible.
+    The object with S3 credentials. If the object is not provided, only public buckets will be accessible. When use_irsa is true, access_key_id and secret_access_key are optional and the pod's service account will be used for authentication.
     """ # noqa: E501
-    access_key_id: StrictStr
-    secret_access_key: StrictStr
-    session_token: Optional[StrictStr] = None
-    role_arn: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["access_key_id", "secret_access_key", "session_token", "role_arn"]
+    use_irsa: Optional[StrictBool] = Field(default=False, description="Enable IAM Roles for Service Accounts (IRSA) to authenticate using the pod's service account. When enabled, explicit credentials are not required.")
+    access_key_id: Optional[StrictStr] = Field(default=None, description="AWS access key ID. Not required when using IRSA.")
+    secret_access_key: Optional[StrictStr] = Field(default=None, description="AWS secret access key. Not required when using IRSA.")
+    session_token: Optional[StrictStr] = Field(default=None, description="AWS session token for temporary credentials.")
+    role_arn: Optional[StrictStr] = Field(default=None, description="The Amazon Resource Name (ARN) of IAM Role that will be utilized for accessing the S3 storage. When IRSA is enabled, this role will be assumed from the service account identity (role chaining). When IRSA is disabled, it requires specification of other credentials to identify an identity the role will be associated with.")
+    external_id: Optional[StrictStr] = Field(default=None, description="Optional security token for assuming cross-account roles. This is used to prevent the 'confused deputy' problem when a third party assumes a role in your account. Only required if the role's trust policy requires an external ID.")
+    __properties: ClassVar[List[str]] = ["use_irsa", "access_key_id", "secret_access_key", "session_token", "role_arn", "external_id"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -83,10 +85,12 @@ class S3Credentials(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
+            "use_irsa": obj.get("use_irsa") if obj.get("use_irsa") is not None else False,
             "access_key_id": obj.get("access_key_id"),
             "secret_access_key": obj.get("secret_access_key"),
             "session_token": obj.get("session_token"),
-            "role_arn": obj.get("role_arn")
+            "role_arn": obj.get("role_arn"),
+            "external_id": obj.get("external_id")
         })
         return _obj
 

{h2ogpte-1.6.44rc6.dist-info → h2ogpte-1.6.44rc9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: h2ogpte
-Version: 1.6.44rc6
+Version: 1.6.44rc9
 Summary: Client library for Enterprise h2oGPTe
 Author-email: "H2O.ai, Inc." <support@h2o.ai>
 Project-URL: Source, https://github.com/h2oai/h2ogpte

{h2ogpte-1.6.44rc6.dist-info → h2ogpte-1.6.44rc9.dist-info}/RECORD

@@ -1,8 +1,8 @@
-h2ogpte/__init__.py,sha256=Ix9s7mV88gTJIRCD3Y9U_P5iDCrkHTPVWW1N5uB1WX0,1524
-h2ogpte/connectors.py,sha256=vkILsfW-tsLUn6KB6zgu_kjps4NMGgJpZ3OOaIjji04,8057
+h2ogpte/__init__.py,sha256=J9-R6VEV8qEaclMf-7AN60HdZgnQJE3eo0AJgCuENgA,1524
+h2ogpte/connectors.py,sha256=CRAEpkn9GotcCjWANfJjZ5Hq1cjGWJ4H_IO4eJgVWiI,8466
 h2ogpte/errors.py,sha256=XgLdfJO1fZ9Bf9rhUKpnvRzzvkNyan3Oc6WzGS6hCUA,1248
-h2ogpte/h2ogpte.py,sha256=K3W8gXrHr1K68244xIh9myDOR0mzdkQLTDnkIdxSTc4,308848
-h2ogpte/h2ogpte_async.py,sha256=QIiNhezH9d2FF7wowOcZdswvYSHJ-6OXUHe3BP1r4kg,328753
+h2ogpte/h2ogpte.py,sha256=AqwDkynS9kLDi231PgGiwKm8kzVUQZUYquMqMdiYEHQ,309478
+h2ogpte/h2ogpte_async.py,sha256=cfuKTU5T2rSYeYXDjTsJ_TbSAyglhoRr6kQoCc-8fgI,329391
 h2ogpte/h2ogpte_sync_base.py,sha256=ftsVzpMqEsyi0UACMI-7H_EIYEx9JEdEUImbyjWy_Hc,15285
 h2ogpte/session.py,sha256=BqJg3mWVeyz_ZLUJC_olzZzeLnRSaJwCH1NkXXfhg54,32608
 h2ogpte/session_async.py,sha256=UzNijTn3kZjAUYl_Jn5Oji4QrPTOpdX9KKByTmhLlK8,31354
@@ -41,17 +41,17 @@ h2ogpte/cli/ui/prompts.py,sha256=bJvRe_32KppQTK5bqnsrPh0RS4JaY9KkiV7y-3v8PMQ,538
 h2ogpte/cli/ui/status_bar.py,sha256=hs2MLvkg-y3Aiu3gWRtgMXf3jv3DGe7Y47ucgoBAP7Y,3852
 h2ogpte/cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 h2ogpte/cli/utils/file_manager.py,sha256=ghNDX6G3Dr0vFvBYjbqx5o7qxq-pN8Vo2Rp1vyITfLo,13988
-h2ogpte/rest_async/__init__.py,sha256=9reksIT3P7gEDM8krgRey-5hM5sug7v9h50N9XAtDi4,15203
-h2ogpte/rest_async/api_client.py,sha256=3WugXotxD2f57OMUqPq77lQOqyAmGnadbfazsZFnIOg,29510
+h2ogpte/rest_async/__init__.py,sha256=hVi4wgpvB75-g3mWK1adO4HqKuwVAGbddZwON05ggw4,15203
+h2ogpte/rest_async/api_client.py,sha256=vUz2Aoa0qI34oPlC0VZYdWA6kmVcDbIrq8SOjZaZeu4,29510
 h2ogpte/rest_async/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
-h2ogpte/rest_async/configuration.py,sha256=UOVg3Z_EFQzjEgd97BhcP1dbukSL-n51-EKxyc2dimc,19567
+h2ogpte/rest_async/configuration.py,sha256=ZFVITmWtExcHnI06BVaR8CCvcKjXuX1esUedxQ1ScV0,19567
 h2ogpte/rest_async/exceptions.py,sha256=aSDc-0lURtyQjf5HGa7_Ta0nATxKxfHW3huDA2Zdj6o,8370
 h2ogpte/rest_async/rest.py,sha256=mdjDwzJ1kiaYtONUfDRqKsRPw5-tG6eyZV2P1yBuwRo,9147
 h2ogpte/rest_async/api/__init__.py,sha256=R_x57GGyaSgxZyrJOyOt551TodbRSQf3T7VrraQc-84,973
 h2ogpte/rest_async/api/agents_api.py,sha256=_-7Y1h3ssEqqgjriUmuDnngbimbVifva75Z7ks3-wNw,177547
 h2ogpte/rest_async/api/api_keys_api.py,sha256=hgywNjCWaIrhFRJacKuPkJSKDEltIK8B9fi-4xMWLgs,58244
 h2ogpte/rest_async/api/chat_api.py,sha256=bJDJqTIXiTYW4anTkqE4XYuEqRDcRW8LcwlMgXyQQS4,295852
-h2ogpte/rest_async/api/collections_api.py,sha256=fvB_HNR41uaS986lzitfIZ350grq6cAqsPZ2ZsvNY0A,639458
+h2ogpte/rest_async/api/collections_api.py,sha256=kg6ZJyqNtnrqu2Q1cS8S-ViYPjXTEAynE2nlSQ-61HA,649556
 h2ogpte/rest_async/api/configurations_api.py,sha256=H9I2hukGC8ACin3cSUR3pAqtghJs0OUPJAzC9nOlkGY,127865
 h2ogpte/rest_async/api/document_ingestion_api.py,sha256=K8iIH9yPEwZlJdySLnEQkSen2uneR2kEUHZXzJyc0JA,489154
 h2ogpte/rest_async/api/documents_api.py,sha256=X-w5x4bX2wHXlxb_97XgWdgNi5qAkdksoqz7bHKG5fA,261595
@@ -164,7 +164,7 @@ h2ogpte/rest_async/models/reset_and_share_with_groups_request.py,sha256=6IwfFIjA
 h2ogpte/rest_async/models/role_create_request.py,sha256=UayaEOpH_Qx8AYff5cFALodAmbTaO4YphBsMM2lHHFI,4476
 h2ogpte/rest_async/models/role_info.py,sha256=gCcSiqjkux4zFt0cR9CQDaeWgSneVQJ11JgthdtQUms,4633
 h2ogpte/rest_async/models/roles_reset_request.py,sha256=W5cKp897NVOwlWX-GR9OZ1THtqWDhjmCibadKZSa9i4,4409
-h2ogpte/rest_async/models/s3_credentials.py,sha256=fAeFj3sXAcQq6hpPgs_A_PHRPaM6Dmu8D4Id7zMcFZs,4835
+h2ogpte/rest_async/models/s3_credentials.py,sha256=GS4bIxa8tc4E0pzVjemDYiWJjqXVrWfSN3dsqEUegtI,6304
 h2ogpte/rest_async/models/search_collection_chunks_request.py,sha256=KfMNH2VvZYLahzrOVHpjfCmUpWIWV_YUWczjBuZ02O4,5129
 h2ogpte/rest_async/models/self_test_result.py,sha256=LOiW5WOJSBKlG7KMZ2tOHfmSow0BbDXSBN8p9OlNNeA,5750
 h2ogpte/rest_async/models/set_collection_size_limit_request.py,sha256=f8OGSc2uSRcwj7yByFWHj5A1eG_jPXajMHxZDjGO3DM,4501
@@ -201,17 +201,17 @@ h2ogpte/rest_async/models/user_deletion_request.py,sha256=z7gD8XKOGwwg782TRzXJii
 h2ogpte/rest_async/models/user_info.py,sha256=ef59Eh9k42JUY3X2RnCrwYR7sc_8lXT1vRLGoNz3uTU,4489
 h2ogpte/rest_async/models/user_job_details.py,sha256=kzu8fLxVsRMgnyt6dLr0VWjlIoE3i1VRpGR9nDxFyk4,4985
 h2ogpte/rest_async/models/user_permission.py,sha256=1k74E7s2kD2waSZ79KPlgTupVYEacTKWMqcKxv2972A,4856
-h2ogpte/rest_sync/__init__.py,sha256=X9-WJtgMWJ2px2jeWaWkwKIu9dk8jNVcQe6g_EYPors,15042
-h2ogpte/rest_sync/api_client.py,sha256=2_uN89Z_ESjAmevS0hYHKzcq3GuxCSjr6NMA9O-XAlg,29397
+h2ogpte/rest_sync/__init__.py,sha256=TDrF5FTonsbsGxH76o6LhgFnIEJO0PN3l9hrpjPJpuA,15042
+h2ogpte/rest_sync/api_client.py,sha256=XxK-Au-xNVKnRCFoK4UxFu_P1PCfmklCV486CWwiwl0,29397
 h2ogpte/rest_sync/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
-h2ogpte/rest_sync/configuration.py,sha256=aeg0VuYA7yiV3xfBLBxbT7mzb9XO-SXg3DfEcTUjCPg,19850
+h2ogpte/rest_sync/configuration.py,sha256=3WQeOh-BPaYPudUv13dayS589uPKWlROAIHVwD-4HOo,19850
 h2ogpte/rest_sync/exceptions.py,sha256=aSDc-0lURtyQjf5HGa7_Ta0nATxKxfHW3huDA2Zdj6o,8370
 h2ogpte/rest_sync/rest.py,sha256=evRzviTYC_fsrpTtFlGvruXmquH9C0jDn-oQrGrE5A0,11314
 h2ogpte/rest_sync/api/__init__.py,sha256=ZuLQQtyiXnP5UOwTlIOYLGLQq1BG_0PEkzC9s698vjM,958
 h2ogpte/rest_sync/api/agents_api.py,sha256=zWzPXtqHbhBe6xVZZKYl93bFLDVzNw2KmAJG5xIW6i0,176763
 h2ogpte/rest_sync/api/api_keys_api.py,sha256=MueDC8Z4VUC61IVIBem0kD0Wpgh35MvhoilUtPVLrN0,57997
 h2ogpte/rest_sync/api/chat_api.py,sha256=9DoF08mTxi3D7sYRkvOM76qB4crS13lb5QmTXtJIV1U,294582
-h2ogpte/rest_sync/api/collections_api.py,sha256=3oTV2IuRV9l_8Wjb9ivQX7bQ6Ay6uFO6zIjbre1bSXs,636980
+h2ogpte/rest_sync/api/collections_api.py,sha256=DsM57mFCtEn7WjMen3-zbZz_fC0TH1-zttlXJGQwwCs,647030
 h2ogpte/rest_sync/api/configurations_api.py,sha256=B39KQB3fRknMZ8PtDNBxWVX2aVGTQSmaE2nq6Ya8VkE,127330
 h2ogpte/rest_sync/api/document_ingestion_api.py,sha256=i7bLMjtFOvtr9zwgR2F-uXY1AhmW918axnm0T1vxU90,488230
 h2ogpte/rest_sync/api/documents_api.py,sha256=qRDBXrQaZWp_hGxg-7mqAFUMeeWd1EB2M_xIlzHh6rw,260474
@@ -324,7 +324,7 @@ h2ogpte/rest_sync/models/reset_and_share_with_groups_request.py,sha256=6IwfFIjAA
 h2ogpte/rest_sync/models/role_create_request.py,sha256=UayaEOpH_Qx8AYff5cFALodAmbTaO4YphBsMM2lHHFI,4476
 h2ogpte/rest_sync/models/role_info.py,sha256=gCcSiqjkux4zFt0cR9CQDaeWgSneVQJ11JgthdtQUms,4633
 h2ogpte/rest_sync/models/roles_reset_request.py,sha256=W5cKp897NVOwlWX-GR9OZ1THtqWDhjmCibadKZSa9i4,4409
-h2ogpte/rest_sync/models/s3_credentials.py,sha256=fAeFj3sXAcQq6hpPgs_A_PHRPaM6Dmu8D4Id7zMcFZs,4835
+h2ogpte/rest_sync/models/s3_credentials.py,sha256=GS4bIxa8tc4E0pzVjemDYiWJjqXVrWfSN3dsqEUegtI,6304
 h2ogpte/rest_sync/models/search_collection_chunks_request.py,sha256=KfMNH2VvZYLahzrOVHpjfCmUpWIWV_YUWczjBuZ02O4,5129
 h2ogpte/rest_sync/models/self_test_result.py,sha256=LOiW5WOJSBKlG7KMZ2tOHfmSow0BbDXSBN8p9OlNNeA,5750
 h2ogpte/rest_sync/models/set_collection_size_limit_request.py,sha256=f8OGSc2uSRcwj7yByFWHj5A1eG_jPXajMHxZDjGO3DM,4501
@@ -361,8 +361,8 @@ h2ogpte/rest_sync/models/user_deletion_request.py,sha256=z7gD8XKOGwwg782TRzXJiiP
 h2ogpte/rest_sync/models/user_info.py,sha256=ef59Eh9k42JUY3X2RnCrwYR7sc_8lXT1vRLGoNz3uTU,4489
 h2ogpte/rest_sync/models/user_job_details.py,sha256=9cbhpgLMDpar-aTOaY5Ygud-8Kbi23cLNldTGab0Sd8,4984
 h2ogpte/rest_sync/models/user_permission.py,sha256=1k74E7s2kD2waSZ79KPlgTupVYEacTKWMqcKxv2972A,4856
-h2ogpte-1.6.44rc6.dist-info/METADATA,sha256=q-vXm3EGmmy8oyF83AhaQYJtMGKdogZgNLOuI5EAsGY,8615
-h2ogpte-1.6.44rc6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-h2ogpte-1.6.44rc6.dist-info/entry_points.txt,sha256=BlaqX2SXJanrOGqNYwnzvCxHGNadM7RBI4pW4rVo5z4,54
-h2ogpte-1.6.44rc6.dist-info/top_level.txt,sha256=vXV4JnNwFWFAqTWyHrH-cGIQqbCcEDG9-BbyNn58JpM,8
-h2ogpte-1.6.44rc6.dist-info/RECORD,,
+h2ogpte-1.6.44rc9.dist-info/METADATA,sha256=o8-5v3-pkp-Q_Xz1LaCxB7YAnfMvgEYsiFqbcyTO5Fs,8615
+h2ogpte-1.6.44rc9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+h2ogpte-1.6.44rc9.dist-info/entry_points.txt,sha256=BlaqX2SXJanrOGqNYwnzvCxHGNadM7RBI4pW4rVo5z4,54
+h2ogpte-1.6.44rc9.dist-info/top_level.txt,sha256=vXV4JnNwFWFAqTWyHrH-cGIQqbCcEDG9-BbyNn58JpM,8
+h2ogpte-1.6.44rc9.dist-info/RECORD,,