h2ogpte 1.6.38rc3__py3-none-any.whl → 1.6.40rc1__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.38rc3"
+__version__ = "1.6.40rc1"
 
 __all__ = [
     "H2OGPTE",

h2ogpte/h2ogpte.py

@@ -3335,6 +3335,54 @@ class H2OGPTE(H2OGPTESyncBase):
             )
         return [User(**d.to_dict()) for d in response]
 
+    def request_current_user_deletion(
+        self,
+    ) -> str:
+        """Request deletion of the current user account.
+
+        This creates a deletion request and returns a delete ID that must be used
+        to confirm the deletion within 5 minutes.
+
+        Returns:
+            str: Delete ID that must be used to confirm deletion.
+        """
+        header = self._get_auth_header()
+        with self._RESTClient(self) as rest_client:
+            response = _rest_to_client_exceptions(
+                lambda: rest_client.permission_api.request_current_user_deletion(
+                    _headers=header,
+                )
+            )
+        return response.delete_id
+
+    def confirm_current_user_deletion(
+        self,
+        delete_id: str,
+        timeout: Union[float, None] = None,
+    ) -> None:
+        """Confirm deletion of the current user data.
+
+        Args:
+            delete_id:
+                The delete ID returned from request_current_user_deletion().
+            timeout:
+                Timeout in seconds. Default is 300 seconds.
+        """
+        header = self._get_auth_header()
+        if timeout is None:
+            timeout = 300.0
+
+        with self._RESTClient(self) as rest_client:
+            _rest_to_client_exceptions(
+                lambda: rest_client.permission_api.confirm_current_user_deletion(
+                    confirm_user_deletion_request=rest.ConfirmUserDeletionRequest(
+                        delete_id=delete_id,
+                    ),
+                    timeout=timeout,
+                    _headers=header,
+                )
+            )
+
     def share_collection(
         self, collection_id: str, permission: SharePermission
     ) -> ShareResponseStatus:
@@ -6352,6 +6400,80 @@ class H2OGPTE(H2OGPTESyncBase):
             )
         return response.count
 
+    def add_custom_agent_tool(
+        self, tool_type: str, tool_args: dict, custom_tool_path: Optional[str] = None
+    ) -> list:
+        header = self._get_auth_header()
+        custom_tool_path = str(custom_tool_path) if custom_tool_path else None
+        with self._RESTClient(self) as rest_client:
+            response = _rest_to_client_exceptions(
+                lambda: rest_client.agent_api.add_custom_agent_tool(
+                    file=custom_tool_path,
+                    tool_type=tool_type,
+                    tool_args=json.dumps(tool_args),
+                    custom_tool_path=custom_tool_path,
+                    filename=os.path.basename(custom_tool_path)
+                    if custom_tool_path
+                    else None,
+                    _headers=header,
+                )
+            )
+        return [obj.agent_custom_tool_id for obj in response]
+
+    def delete_custom_agent_tool(self, tool_ids: List[str]) -> int:
+        header = self._get_auth_header()
+        with self._RESTClient(self) as rest_client:
+            response = _rest_to_client_exceptions(
+                lambda: rest_client.agent_api.delete_custom_agent_tool(
+                    tool_ids=tool_ids,
+                    _headers=header,
+                )
+            )
+        return response.count
+
+    def update_custom_agent_tool(
+        self,
+        tool_id: str,
+        tool_args: dict,
+    ) -> str:
+        """Updates a custom agent tool's arguments.
+
+        Args:
+            tool_id: The ID of the tool to update
+            tool_args: New tool arguments
+
+        Returns:
+            The updated tool ID
+        """
+        header = self._get_auth_header()
+
+        # Build request body with only tool_args
+        request_obj = rest.UpdateCustomAgentToolRequest(tool_args=tool_args)
+
+        with self._RESTClient(self) as rest_client:
+            response = _rest_to_client_exceptions(
+                lambda: rest_client.agent_api.update_custom_agent_tool(
+                    tool_id=tool_id,
+                    update_custom_agent_tool_request=request_obj,
+                    _headers=header,
+                )
+            )
+        return response.agent_custom_tool_id
+
+    def get_custom_agent_tools(self) -> List[dict]:
+        """Gets all custom agent tools for the current user.
+
+        Returns:
+            List[dict]: A list of custom agent tools with their details.
+                Each tool contains: id, tool_name, tool_type, tool_args, owner_email
+        """
+        header = self._get_auth_header()
+        with self._RESTClient(self) as rest_client:
+            response = _rest_to_client_exceptions(
+                lambda: rest_client.agent_api.list_custom_agent_tools(_headers=header)
+            )
+        return response
+
     def set_role_priority(self, role_id: str, priority: int) -> UserRole:
         """Sets the priority for a role.
         Args:

h2ogpte/h2ogpte_async.py

@@ -14,6 +14,7 @@ from typing import Any, Dict, Iterable, List, Optional, Tuple, Union
 from collections import defaultdict
 from h2o_authn import AsyncTokenProvider
 from urllib.parse import quote
+import os
 
 from h2ogpte import rest_async as rest
 from h2ogpte.errors import (
@@ -3530,6 +3531,54 @@ class H2OGPTEAsync:
             )
         return [User(**d.to_dict()) for d in response]
 
+    async def request_current_user_deletion(
+        self,
+    ) -> str:
+        """Request deletion of the current user account.
+
+        This creates a deletion request and returns a delete ID that must be used
+        to confirm the deletion within 5 minutes.
+
+        Returns:
+            str: Delete ID that must be used to confirm deletion.
+        """
+        header = await self._get_auth_header()
+        async with self._RESTClient(self) as rest_client:
+            response = await _rest_to_client_exceptions(
+                rest_client.permission_api.request_current_user_deletion(
+                    _headers=header,
+                )
+            )
+        return response.delete_id
+
+    async def confirm_current_user_deletion(
+        self,
+        delete_id: str,
+        timeout: Union[float, None] = None,
+    ) -> None:
+        """Confirm deletion of the current user data.
+
+        Args:
+            delete_id:
+                The delete ID returned from request_current_user_deletion().
+            timeout:
+                Timeout in seconds. Default is 300 seconds.
+        """
+        header = await self._get_auth_header()
+        if timeout is None:
+            timeout = 300.0
+
+        async with self._RESTClient(self) as rest_client:
+            await _rest_to_client_exceptions(
+                rest_client.permission_api.confirm_current_user_deletion(
+                    confirm_user_deletion_request=rest.ConfirmUserDeletionRequest(
+                        delete_id=delete_id,
+                    ),
+                    timeout=timeout,
+                    _headers=header,
+                )
+            )
+
     async def share_collection(
         self, collection_id: str, permission: SharePermission
     ) -> ShareResponseStatus:
@@ -6596,6 +6645,80 @@ class H2OGPTEAsync:
             )
         return response.count
 
+    async def add_custom_agent_tool(
+        self, tool_type: str, tool_args: dict, custom_tool_path: Optional[str] = None
+    ) -> list:
+        header = await self._get_auth_header()
+        custom_tool_path = str(custom_tool_path) if custom_tool_path else None
+        async with self._RESTClient(self) as rest_client:
+            response = await _rest_to_client_exceptions(
+                rest_client.agent_api.add_custom_agent_tool(
+                    file=custom_tool_path,
+                    tool_type=tool_type,
+                    tool_args=json.dumps(tool_args),
+                    custom_tool_path=custom_tool_path,
+                    filename=os.path.basename(custom_tool_path)
+                    if custom_tool_path
+                    else None,
+                    _headers=header,
+                )
+            )
+        return [obj.agent_custom_tool_id for obj in response]
+
+    async def delete_custom_agent_tool(self, tool_ids: List[str]) -> int:
+        header = await self._get_auth_header()
+        async with self._RESTClient(self) as rest_client:
+            response = await _rest_to_client_exceptions(
+                rest_client.agent_api.delete_custom_agent_tool(
+                    tool_ids=tool_ids,
+                    _headers=header,
+                )
+            )
+        return response.count
+
+    async def update_custom_agent_tool(
+        self,
+        tool_id: str,
+        tool_args: dict,
+    ) -> str:
+        """Updates a custom agent tool's arguments.
+
+        Args:
+            tool_id: The ID of the tool to update
+            tool_args: New tool arguments
+
+        Returns:
+            The updated tool ID
+        """
+        header = await self._get_auth_header()
+
+        # Build request body with only tool_args
+        request_obj = rest.UpdateCustomAgentToolRequest(tool_args=tool_args)
+
+        async with self._RESTClient(self) as rest_client:
+            response = await _rest_to_client_exceptions(
+                rest_client.agent_api.update_custom_agent_tool(
+                    tool_id=tool_id,
+                    update_custom_agent_tool_request=request_obj,
+                    _headers=header,
+                )
+            )
+        return response.agent_custom_tool_id
+
+    async def get_custom_agent_tools(self) -> List[dict]:
+        """Gets all custom agent tools for the current user.
+
+        Returns:
+            List[dict]: A list of custom agent tools with their details.
+                Each tool contains: id, tool_name, tool_type, tool_args, owner_email
+        """
+        header = await self._get_auth_header()
+        async with self._RESTClient(self) as rest_client:
+            response = await _rest_to_client_exceptions(
+                rest_client.agent_api.list_custom_agent_tools(_headers=header)
+            )
+        return response
+
     async def set_role_priority(self, role_id: str, priority: int) -> UserRole:
         """Sets the priority for a role.
         Args:

h2ogpte/rest_async/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "1.6.38-dev3"
+__version__ = "1.6.40-dev1"
 
 # import apis into sdk package
 from h2ogpte.rest_async.api.api_keys_api import APIKeysApi
@@ -48,6 +48,7 @@ from h2ogpte.rest_async.models.api_key_create_request import APIKeyCreateRequest
 from h2ogpte.rest_async.models.api_key_info import APIKeyInfo
 from h2ogpte.rest_async.models.api_key_result import APIKeyResult
 from h2ogpte.rest_async.models.api_key_update_expiry_request import APIKeyUpdateExpiryRequest
+from h2ogpte.rest_async.models.add_custom_agent_tool201_response_inner import AddCustomAgentTool201ResponseInner
 from h2ogpte.rest_async.models.agent_key import AgentKey
 from h2ogpte.rest_async.models.agent_server_directory_file_stats import AgentServerDirectoryFileStats
 from h2ogpte.rest_async.models.agent_server_directory_stats import AgentServerDirectoryStats
@@ -74,10 +75,12 @@ from h2ogpte.rest_async.models.collection_change_request import CollectionChange
 from h2ogpte.rest_async.models.collection_create_request import CollectionCreateRequest
 from h2ogpte.rest_async.models.collection_settings import CollectionSettings
 from h2ogpte.rest_async.models.collection_update_request import CollectionUpdateRequest
+from h2ogpte.rest_async.models.confirm_user_deletion_request import ConfirmUserDeletionRequest
 from h2ogpte.rest_async.models.count import Count
 from h2ogpte.rest_async.models.count_with_queue_details import CountWithQueueDetails
 from h2ogpte.rest_async.models.create_agent_key_request import CreateAgentKeyRequest
 from h2ogpte.rest_async.models.create_agent_tool_key_associations_request import CreateAgentToolKeyAssociationsRequest
+from h2ogpte.rest_async.models.create_agent_tool_request import CreateAgentToolRequest
 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
@@ -112,6 +115,7 @@ from h2ogpte.rest_async.models.ingest_upload_body import IngestUploadBody
 from h2ogpte.rest_async.models.job_details import JobDetails
 from h2ogpte.rest_async.models.job_details_status import JobDetailsStatus
 from h2ogpte.rest_async.models.key_association import KeyAssociation
+from h2ogpte.rest_async.models.list_custom_agent_tools200_response_inner import ListCustomAgentTools200ResponseInner
 from h2ogpte.rest_async.models.match_collection_chunks_request import MatchCollectionChunksRequest
 from h2ogpte.rest_async.models.message_vote_update_request import MessageVoteUpdateRequest
 from h2ogpte.rest_async.models.model import Model
@@ -154,6 +158,8 @@ from h2ogpte.rest_async.models.update_agent_tool_preference_request import Updat
 from h2ogpte.rest_async.models.update_collection_expiry_date_request import UpdateCollectionExpiryDateRequest
 from h2ogpte.rest_async.models.update_collection_inactivity_interval_request import UpdateCollectionInactivityIntervalRequest
 from h2ogpte.rest_async.models.update_collection_privacy_request import UpdateCollectionPrivacyRequest
+from h2ogpte.rest_async.models.update_custom_agent_tool200_response import UpdateCustomAgentTool200Response
+from h2ogpte.rest_async.models.update_custom_agent_tool_request import UpdateCustomAgentToolRequest
 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
@@ -164,6 +170,7 @@ from h2ogpte.rest_async.models.usage_stats_per_model import UsageStatsPerModel
 from h2ogpte.rest_async.models.usage_stats_per_model_and_user import UsageStatsPerModelAndUser
 from h2ogpte.rest_async.models.usage_stats_per_user import UsageStatsPerUser
 from h2ogpte.rest_async.models.user_configuration_item import UserConfigurationItem
+from h2ogpte.rest_async.models.user_deletion_request import UserDeletionRequest
 from h2ogpte.rest_async.models.user_info import UserInfo
 from h2ogpte.rest_async.models.user_job_details import UserJobDetails
 from h2ogpte.rest_async.models.user_permission import UserPermission