h2ogpte 1.6.43rc9__py3-none-any.whl → 1.6.44__py3-none-any.whl

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_async/models/user_permission.py

@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -30,7 +30,8 @@ class UserPermission(BaseModel):
     name: StrictStr
     description: Optional[StrictStr] = None
     category: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["id", "name", "description", "category"]
+    dependencies: Optional[List[StrictStr]] = Field(default=None, description="List of permissions that this permission depends on.")
+    __properties: ClassVar[List[str]] = ["id", "name", "description", "category", "dependencies"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -86,7 +87,8 @@ class UserPermission(BaseModel):
             "id": obj.get("id"),
             "name": obj.get("name"),
             "description": obj.get("description"),
-            "category": obj.get("category")
+            "category": obj.get("category"),
+            "dependencies": obj.get("dependencies")
         })
         return _obj
 

h2ogpte/rest_sync/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "1.6.43-dev9"
+__version__ = "1.6.44"
 
 # 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/permissions_api.py

@@ -7082,7 +7082,7 @@ class PermissionsApi:
     @validate_call
     def remove_permission_from_role(
         self,
-        role_id: Annotated[StrictStr, Field(description="The unique identifier of an user.")],
+        role_id: Annotated[StrictStr, Field(description="The unique identifier of the role.")],
         permission_name: Annotated[StrictStr, Field(description="The permission name.")],
         _request_timeout: Union[
             None,
@@ -7101,7 +7101,7 @@ class PermissionsApi:
 
         Removes permission from a given role.
 
-        :param role_id: The unique identifier of an user. (required)
+        :param role_id: The unique identifier of the role. (required)
         :type role_id: str
         :param permission_name: The permission name. (required)
         :type permission_name: str
@@ -7155,7 +7155,7 @@ class PermissionsApi:
     @validate_call
     def remove_permission_from_role_with_http_info(
         self,
-        role_id: Annotated[StrictStr, Field(description="The unique identifier of an user.")],
+        role_id: Annotated[StrictStr, Field(description="The unique identifier of the role.")],
         permission_name: Annotated[StrictStr, Field(description="The permission name.")],
         _request_timeout: Union[
             None,
@@ -7174,7 +7174,7 @@ class PermissionsApi:
 
         Removes permission from a given role.
 
-        :param role_id: The unique identifier of an user. (required)
+        :param role_id: The unique identifier of the role. (required)
         :type role_id: str
         :param permission_name: The permission name. (required)
         :type permission_name: str
@@ -7228,7 +7228,7 @@ class PermissionsApi:
     @validate_call
     def remove_permission_from_role_without_preload_content(
         self,
-        role_id: Annotated[StrictStr, Field(description="The unique identifier of an user.")],
+        role_id: Annotated[StrictStr, Field(description="The unique identifier of the role.")],
         permission_name: Annotated[StrictStr, Field(description="The permission name.")],
         _request_timeout: Union[
             None,
@@ -7247,7 +7247,7 @@ class PermissionsApi:
 
         Removes permission from a given role.
 
-        :param role_id: The unique identifier of an user. (required)
+        :param role_id: The unique identifier of the role. (required)
         :type role_id: str
         :param permission_name: The permission name. (required)
         :type permission_name: str

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.43-dev9/python'
+        self.user_agent = 'OpenAPI-Generator/1.6.44/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.43-dev9".\
+               "SDK Package Version: 1.6.44".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self) -> List[HostSetting]:

h2ogpte/rest_sync/models/global_configuration_item.py

@@ -32,7 +32,11 @@ class GlobalConfigurationItem(BaseModel):
     can_overwrite: StrictBool
     upper_bound: Optional[Union[StrictFloat, StrictInt]] = None
     is_public: StrictBool
-    __properties: ClassVar[List[str]] = ["key_name", "string_value", "value_type", "can_overwrite", "upper_bound", "is_public"]
+    is_read_only: Optional[StrictBool] = None
+    is_encrypted: Optional[StrictBool] = None
+    description: Optional[StrictStr] = None
+    category: Optional[StrictStr] = None
+    __properties: ClassVar[List[str]] = ["key_name", "string_value", "value_type", "can_overwrite", "upper_bound", "is_public", "is_read_only", "is_encrypted", "description", "category"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -90,7 +94,11 @@ class GlobalConfigurationItem(BaseModel):
             "value_type": obj.get("value_type"),
             "can_overwrite": obj.get("can_overwrite"),
             "upper_bound": obj.get("upper_bound"),
-            "is_public": obj.get("is_public")
+            "is_public": obj.get("is_public"),
+            "is_read_only": obj.get("is_read_only"),
+            "is_encrypted": obj.get("is_encrypted"),
+            "description": obj.get("description"),
+            "category": obj.get("category")
         })
         return _obj
 

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/rest_sync/models/user_permission.py

@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -30,7 +30,8 @@ class UserPermission(BaseModel):
     name: StrictStr
     description: Optional[StrictStr] = None
     category: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["id", "name", "description", "category"]
+    dependencies: Optional[List[StrictStr]] = Field(default=None, description="List of permissions that this permission depends on.")
+    __properties: ClassVar[List[str]] = ["id", "name", "description", "category", "dependencies"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -86,7 +87,8 @@ class UserPermission(BaseModel):
             "id": obj.get("id"),
             "name": obj.get("name"),
             "description": obj.get("description"),
-            "category": obj.get("category")
+            "category": obj.get("category"),
+            "dependencies": obj.get("dependencies")
         })
         return _obj
 

h2ogpte/session.py

@@ -149,6 +149,7 @@ class Session:
                     ssl_context=ssl_context,
                     open_timeout=self._open_timeout,
                     close_timeout=self._close_timeout,
+                    max_size=5 * 1024 * 1024,  # 5 MB limit for large responses
                 )
                 return self._connection
             except (ConnectionClosedError, InvalidURI, InvalidHandshake) as e:

h2ogpte/session_async.py

@@ -478,6 +478,7 @@ class SessionAsync:
                     open_timeout=self._open_timeout,
                     close_timeout=self._close_timeout,
                     ssl=ssl_context,
+                    max_size=5 * 1024 * 1024,  # 5 MB limit for large responses
                 )
                 return self._websocket
             except (ConnectionClosedError, InvalidURI, InvalidHandshake) as e:

h2ogpte/types.py

@@ -123,6 +123,8 @@ class ChatMessageFull(BaseModel):
     type_list: Optional[List[ChatMessageMeta]] = []
     has_references: bool
     total_references: int
+    collection_id: Optional[str] = None
+    collection_name: Optional[str] = None
     error: Optional[str] = None
 
 
@@ -382,6 +384,7 @@ class UserPermission(BaseModel):
     name: str
     description: Optional[str] = None
     category: Optional[str] = None
+    dependencies: Optional[List[str]] = None
 
 
 class UserRole(BaseModel):

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

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