h2ogpte 1.6.38rc2__py3-none-any.whl → 1.6.38rc3__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.38rc2"
+__version__ = "1.6.38rc3"
 
 __all__ = [
     "H2OGPTE",

h2ogpte/h2ogpte.py

@@ -2286,6 +2286,111 @@ class H2OGPTE(H2OGPTESyncBase):
             )
         return self._wait_for_completion(response.id, timeout=timeout)
 
+    def list_secret_ids(self, connector_type: Optional[str] = None) -> List[str]:
+        """
+        List available secret IDs from the SecureStore for cloud storage connectors.
+
+        Args:
+            connector_type: Type of connector ('s3', 'gcs', 'azure_key', 'azure_sas')
+                           If None, returns secrets for all connector types.
+
+        Returns:
+            List of secret IDs available for the specified connector type
+        """
+        self._init_rest()
+        header = self._get_auth_header()
+        response = _rest_to_client_exceptions(
+            lambda: self._secrets_api.list_secret_ids(
+                connector_type=connector_type, _headers=header
+            )
+        )
+        return response.ids
+
+    def get_secret(self, secret_id: str) -> Union[Dict[str, Any], None]:
+        """Get a secret from the SecureStore by its ID.
+
+        Args:
+            secret_id: The ID of the secret to retrieve.
+
+        Returns:
+            The secret object corresponding to the provided ID.
+        """
+        self._init_rest()
+        header = self._get_auth_header()
+        response = _rest_to_client_exceptions(
+            lambda: self._secrets_api.get_secret(secret_id=secret_id, _headers=header)
+        )
+        return response.to_dict()
+
+    def get_secret_value(self, secret_id: str) -> Union[str, None]:
+        """Get the value of a secret from the SecureStore by its ID.
+
+        Args:
+            secret_id: The ID of the secret to retrieve.
+
+        Returns:
+            The value of the secret corresponding to the provided ID.
+        """
+        self._init_rest()
+        header = self._get_auth_header()
+        response = _rest_to_client_exceptions(
+            lambda: self._secrets_api.get_secret_value(
+                secret_id=secret_id, _headers=header
+            )
+        )
+        return response.value if response else None
+
+    def create_secret(self, secret: Dict[str, Any]) -> None:
+        """Create a new secret in the SecureStore.
+
+        Args:
+            secret: A dictionary containing the secret data to be stored.
+
+        Returns:
+            The ID of the newly created secret.
+        """
+        self._init_rest()
+        header = self._get_auth_header()
+        response = _rest_to_client_exceptions(
+            lambda: self._secrets_api.create_secret(secret=secret, _headers=header)
+        )
+        return response.id
+
+    def update_secret(self, secret_id: str, secret: Dict[str, Any]) -> None:
+        """Update an existing secret in the SecureStore.
+
+        Args:
+            secret_id: The ID of the secret to update.
+            secret: A dictionary containing the updated secret data.
+
+        Returns:
+            None
+        """
+        self._init_rest()
+        header = self._get_auth_header()
+        _rest_to_client_exceptions(
+            lambda: self._secrets_api.update_secret(
+                secret_id=secret_id, secret=secret, _headers=header
+            )
+        )
+
+    def delete_secret(self, secret_id: str) -> None:
+        """Delete a secret from the SecureStore by its ID.
+
+        Args:
+            secret_id: The ID of the secret to delete.
+
+        Returns:
+            None
+        """
+        self._init_rest()
+        header = self._get_auth_header()
+        _rest_to_client_exceptions(
+            lambda: self._secrets_api.delete_secret(
+                secret_id=secret_id, _headers=header
+            )
+        )
+
     def ingest_uploads(
         self,
         collection_id: str,

h2ogpte/h2ogpte_async.py

@@ -165,6 +165,7 @@ class H2OGPTEAsync:
             self.api_keys_api = None
             self.configuration_api = None
             self.agent_api = None
+            self.secrets_api = None
 
         async def __aenter__(self):
             if not self._h2ogpte._version_checked:
@@ -197,6 +198,7 @@ class H2OGPTEAsync:
             self.api_keys_api = rest.APIKeysApi(self._rest_client)
             self.configuration_api = rest.ConfigurationsApi(self._rest_client)
             self.agent_api = rest.AgentsApi(self._rest_client)
+            self.secrets_api = rest.SecretsApi(self._rest_client)
             return self
 
         async def __aexit__(self, exc_type, exc_value, traceback):
@@ -2479,6 +2481,107 @@ class H2OGPTEAsync:
             )
         return await self._wait_for_completion(response.id, timeout=timeout)
 
+    async def list_secret_ids(self, connector_type: Optional[str] = None) -> List[str]:
+        """
+        List available secret IDs from the SecureStore for cloud storage connectors.
+
+        Args:
+            connector_type: Type of connector ('s3', 'gcs', 'azure_key', 'azure_sas')
+                           If None, returns secrets for all connector types.
+
+        Returns:
+            List of secret IDs available for the specified connector type
+        """
+        await self._init_rest()
+        header = await self._get_auth_header()
+        response = await _rest_to_client_exceptions(
+            self._secrets_api.list_secret_ids(
+                connector_type=connector_type, _headers=header
+            )
+        )
+        return response.ids
+
+    async def get_secret(self, secret_id: str) -> Union[Dict[str, Any], None]:
+        """Get a secret from the SecureStore by its ID.
+
+        Args:
+            secret_id: The ID of the secret to retrieve.
+
+        Returns:
+            The secret object corresponding to the provided ID.
+        """
+        await self._init_rest()
+        header = await self._get_auth_header()
+        response = await _rest_to_client_exceptions(
+            self._secrets_api.get_secret(secret_id=secret_id, _headers=header)
+        )
+        return response.to_dict()
+
+    async def get_secret_value(self, secret_id: str) -> Union[str, None]:
+        """Get the value of a secret from the SecureStore by its ID.
+
+        Args:
+            secret_id: The ID of the secret to retrieve.
+
+        Returns:
+            The value of the secret corresponding to the provided ID.
+        """
+        await self._init_rest()
+        header = await self._get_auth_header()
+        response = await _rest_to_client_exceptions(
+            self._secrets_api.get_secret_value(secret_id=secret_id, _headers=header)
+        )
+        return response.value if response else None
+
+    async def create_secret(self, secret: Dict[str, Any]) -> None:
+        """Create a new secret in the SecureStore.
+
+        Args:
+            secret: A dictionary containing the secret data to be stored.
+
+        Returns:
+            The ID of the newly created secret.
+        """
+        await self._init_rest()
+        header = await self._get_auth_header()
+        response = await _rest_to_client_exceptions(
+            self._secrets_api.create_secret(secret=secret, _headers=header)
+        )
+        return response.id
+
+    async def update_secret(self, secret_id: str, secret: Dict[str, Any]) -> None:
+        """Update an existing secret in the SecureStore.
+
+        Args:
+            secret_id: The ID of the secret to update.
+            secret: A dictionary containing the updated secret data.
+
+        Returns:
+            None
+        """
+        await self._init_rest()
+        header = await self._get_auth_header()
+        await _rest_to_client_exceptions(
+            self._secrets_api.update_secret(
+                secret_id=secret_id, secret=secret, _headers=header
+            )
+        )
+
+    async def delete_secret(self, secret_id: str) -> None:
+        """Delete a secret from the SecureStore by its ID.
+
+        Args:
+            secret_id: The ID of the secret to delete.
+
+        Returns:
+            None
+        """
+        await self._init_rest()
+        header = await self._get_auth_header()
+        await _rest_to_client_exceptions(
+            self._secrets_api.delete_secret(secret_id=secret_id, _headers=header)
+        )
+
     async def ingest_uploads(
         self,
         collection_id: str,

h2ogpte/h2ogpte_sync_base.py

@@ -93,6 +93,7 @@ class H2OGPTESyncBase:
         self._api_keys_api = rest.APIKeysApi(self._rest_client)
         self._configuration_api = rest.ConfigurationsApi(self._rest_client)
         self._agent_api = rest.AgentsApi(self._rest_client)
+        self._secrets_api = rest.SecretsApi(self._rest_client)
 
     class _RESTClient:
         def __init__(self, h2ogpte):
@@ -110,6 +111,7 @@ class H2OGPTESyncBase:
             self.api_keys_api = h2ogpte._api_keys_api
             self.configuration_api = h2ogpte._configuration_api
             self.agent_api = h2ogpte._agent_api
+            self.secrets_api = h2ogpte._secrets_api
 
         def __enter__(self):
             return self

h2ogpte/rest_async/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "1.6.38-dev2"
+__version__ = "1.6.38-dev3"
 
 # import apis into sdk package
 from h2ogpte.rest_async.api.api_keys_api import APIKeysApi
@@ -28,6 +28,7 @@ from h2ogpte.rest_async.api.jobs_api import JobsApi
 from h2ogpte.rest_async.api.models_api import ModelsApi
 from h2ogpte.rest_async.api.permissions_api import PermissionsApi
 from h2ogpte.rest_async.api.prompt_templates_api import PromptTemplatesApi
+from h2ogpte.rest_async.api.secrets_api import SecretsApi
 from h2ogpte.rest_async.api.system_api import SystemApi
 from h2ogpte.rest_async.api.tags_api import TagsApi
 
@@ -79,6 +80,8 @@ from h2ogpte.rest_async.models.create_agent_key_request import CreateAgentKeyReq
 from h2ogpte.rest_async.models.create_agent_tool_key_associations_request import CreateAgentToolKeyAssociationsRequest
 from h2ogpte.rest_async.models.create_import_collection_to_collection_job_request import CreateImportCollectionToCollectionJobRequest
 from h2ogpte.rest_async.models.create_insert_document_to_collection_job_request import CreateInsertDocumentToCollectionJobRequest
+from h2ogpte.rest_async.models.create_secret201_response import CreateSecret201Response
+from h2ogpte.rest_async.models.create_secret_request import CreateSecretRequest
 from h2ogpte.rest_async.models.create_topic_model_job_request import CreateTopicModelJobRequest
 from h2ogpte.rest_async.models.delete_chat_sessions_job_request import DeleteChatSessionsJobRequest
 from h2ogpte.rest_async.models.delete_collections_job_request import DeleteCollectionsJobRequest
@@ -154,6 +157,7 @@ from h2ogpte.rest_async.models.update_collection_privacy_request import UpdateCo
 from h2ogpte.rest_async.models.update_default_prompt_template_visibility_request import UpdateDefaultPromptTemplateVisibilityRequest
 from h2ogpte.rest_async.models.update_prompt_template_privacy_request import UpdatePromptTemplatePrivacyRequest
 from h2ogpte.rest_async.models.update_qa_feedback_request import UpdateQAFeedbackRequest
+from h2ogpte.rest_async.models.update_secret_request import UpdateSecretRequest
 from h2ogpte.rest_async.models.uploaded_file import UploadedFile
 from h2ogpte.rest_async.models.usage_stats import UsageStats
 from h2ogpte.rest_async.models.usage_stats_per_model import UsageStatsPerModel

h2ogpte/rest_async/api/__init__.py

@@ -12,6 +12,7 @@ from h2ogpte.rest_async.api.jobs_api import JobsApi
 from h2ogpte.rest_async.api.models_api import ModelsApi
 from h2ogpte.rest_async.api.permissions_api import PermissionsApi
 from h2ogpte.rest_async.api.prompt_templates_api import PromptTemplatesApi
+from h2ogpte.rest_async.api.secrets_api import SecretsApi
 from h2ogpte.rest_async.api.system_api import SystemApi
 from h2ogpte.rest_async.api.tags_api import TagsApi