pygeai 0.6.0b6__py3-none-any.whl → 0.6.0b7__py3-none-any.whl

pygeai/assistant/clients.py

@@ -1,5 +1,3 @@
-import json
-from json import JSONDecodeError
 
 from pygeai import logger
 from pygeai.assistant.endpoints import GET_ASSISTANT_DATA_V1, CREATE_ASSISTANT_V1, UPDATE_ASSISTANT_V1, BEGIN_CONVERSATION_V1, \
@@ -8,6 +6,8 @@ from pygeai.assistant.data_analyst.endpoints import GET_DATA_ANALYST_STATUS_V1,
 from pygeai.core.base.clients import BaseClient
 from pygeai.core.common.decorators import handler_server_error
 from pygeai.core.common.exceptions import InvalidAPIResponseException
+from pygeai.core.utils.validators import validate_status_code
+from pygeai.core.utils.parsers import parse_json_response
 
 
 class AssistantClient(BaseClient):
@@ -31,12 +31,7 @@ class AssistantClient(BaseClient):
                 "detail": detail
             }
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to get assistant data for ID '{assistant_id}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to get assistant data for ID '{assistant_id}': {response.text}")
+        return parse_json_response(response, "get assistant data for ID", assistant_id=assistant_id)
 
     def create_assistant(
             self,
@@ -88,12 +83,7 @@ class AssistantClient(BaseClient):
             endpoint=CREATE_ASSISTANT_V1,
             data=data
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to create assistant with name '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to create assistant with name '{name}': {response.text}")
+        return parse_json_response(response, "create assistant with name", name=name)
 
     def update_assistant(
             self,
@@ -165,12 +155,7 @@ class AssistantClient(BaseClient):
             endpoint=endpoint,
             data=data
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to update assistant with ID '{assistant_id}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to update assistant with ID '{assistant_id}': {response.text}")
+        return parse_json_response(response, "update assistant", assistant_id=assistant_id)
 
     def delete_assistant(
             self,
@@ -186,12 +171,7 @@ class AssistantClient(BaseClient):
         response = self.api_service.delete(
             endpoint=endpoint
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to delete assistant with ID '{assistant_id}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to delete assistant with ID '{assistant_id}': {response.text}")
+        return parse_json_response(response, "delete assistant", assistant_id=assistant_id)
 
     def send_chat_request(
             self,
@@ -223,12 +203,7 @@ class AssistantClient(BaseClient):
             endpoint=SEND_CHAT_REQUEST_V1,
             data=data
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to send chat request to assistant '{assistant_name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to send chat request to assistant '{assistant_name}': {response.text}")
+        return parse_json_response(response, "send chat request to assistant", assistant_name=assistant_name)
 
     def get_request_status(
             self,
@@ -244,12 +219,7 @@ class AssistantClient(BaseClient):
         response = self.api_service.get(
             endpoint=endpoint
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to get request status for ID '{request_id}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to get request status for ID '{request_id}': {response.text}")
+        return parse_json_response(response, "get request status for ID", request_id=request_id)
 
     def cancel_request(
             self,
@@ -268,12 +238,7 @@ class AssistantClient(BaseClient):
                 "requestId": request_id
             }
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to cancel request with ID '{request_id}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to cancel request with ID '{request_id}': {response.text}")
+        return parse_json_response(response, "cancel request", request_id=request_id)
 
 
 

pygeai/assistant/data/clients.py

@@ -6,6 +6,7 @@ from pygeai.assistant.endpoints import GET_ASSISTANT_DATA_V1, CREATE_ASSISTANT_V
 from pygeai.assistant.data_analyst.endpoints import GET_DATA_ANALYST_STATUS_V1, EXTEND_DATA_ANALYST_DATASET_V1
 from pygeai.core.base.clients import BaseClient
 from pygeai.core.common.decorators import handler_server_error
+from pygeai.core.utils.validators import validate_status_code
 
 
 class ChatWithDataAssistantClient(AssistantClient):

pygeai/assistant/data_analyst/clients.py

@@ -1,10 +1,11 @@
-from json import JSONDecodeError
 from pathlib import Path
 
 from pygeai import logger
 from pygeai.assistant.clients import AssistantClient
 from pygeai.assistant.data_analyst.endpoints import GET_DATA_ANALYST_STATUS_V1, EXTEND_DATA_ANALYST_DATASET_V1
 from pygeai.core.common.exceptions import InvalidAPIResponseException
+from pygeai.core.utils.validators import validate_status_code
+from pygeai.core.utils.parsers import parse_json_response
 
 
 class DataAnalystAssistantClient(AssistantClient):
@@ -34,12 +35,7 @@ class DataAnalystAssistantClient(AssistantClient):
             headers=headers
         )
 
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to get status for assistant ID '{assistant_id}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to get status for assistant ID '{assistant_id}': {response.text}")
+        return parse_json_response(response, "get status for assistant ID", assistant_id=assistant_id)
 
     def extend_dataset(
         self,
@@ -69,12 +65,7 @@ class DataAnalystAssistantClient(AssistantClient):
                 endpoint=endpoint,
                 files=files
             )
-            try:
-                result = response.json()
-                return result
-            except JSONDecodeError as e:
-                logger.error(f"Unable to extend dataset for assistant ID '{assistant_id}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-                raise InvalidAPIResponseException(f"Unable to extend dataset for assistant ID '{assistant_id}': {response.text}")
+            return parse_json_response(response, "extend dataset for assistant ID", assistant_id=assistant_id)
         finally:
             for _, file_handle in files:
                 file_handle.close()

pygeai/assistant/rag/clients.py

@@ -1,6 +1,5 @@
 import json
 import logging
-from json import JSONDecodeError
 from pathlib import Path
 from typing import Any
 
@@ -12,6 +11,8 @@ from pygeai.core.base.clients import BaseClient
 import urllib.parse
 
 from pygeai.core.common.exceptions import InvalidAPIResponseException
+from pygeai.core.utils.validators import validate_status_code
+from pygeai.core.utils.parsers import parse_json_response
 
 
 class RAGAssistantClient(BaseClient):
@@ -21,22 +22,12 @@ class RAGAssistantClient(BaseClient):
 
     def get_assistants_from_project(self) -> dict:
         response = self.api_service.get(endpoint=GET_ASSISTANTS_FROM_PROJECT_V1)
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to get assistants from project: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to get assistants from project: {response.text}")
+        return parse_json_response(response, "get assistants from project")
 
     def get_assistant_data(self, name: str) -> dict:
         endpoint = GET_ASSISTANT_V1.format(name=name)
         response = self.api_service.get(endpoint=endpoint)
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to get assistant data for name '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to get assistant data for name '{name}': {response.text}")
+        return parse_json_response(response, "get assistant data for name", name=name)
 
     def create_assistant(
             self,
@@ -116,12 +107,7 @@ class RAGAssistantClient(BaseClient):
             endpoint=CREATE_ASSISTANT_V1,
             data=data
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to create assistant with name '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to create assistant with name '{name}': {response.text}")
+        return parse_json_response(response, "create assistant with name", name=name)
 
     def update_assistant(
             self,
@@ -154,23 +140,13 @@ class RAGAssistantClient(BaseClient):
             endpoint=endpoint,
             data=data
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to update assistant with name '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to update assistant with name '{name}': {response.text}")
+        return parse_json_response(response, "update assistant with name", name=name)
 
     def delete_assistant(self, name: str) -> dict:
         safe_name = self.get_url_safe_name(name)
         endpoint = DELETE_ASSISTANT_V1.format(name=safe_name)
         response = self.api_service.delete(endpoint=endpoint)
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to delete assistant with name '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to delete assistant with name '{name}': {response.text}")
+        return parse_json_response(response, "delete assistant with name", name=name)
 
     def get_documents(
             self,
@@ -187,34 +163,19 @@ class RAGAssistantClient(BaseClient):
                 "count": count
             }
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to get documents for assistant '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to get documents for assistant '{name}': {response.text}")
+        return parse_json_response(response, "get documents for assistant", name=name)
 
     def delete_all_documents(self, name: str) -> dict:
         safe_name = self.get_url_safe_name(name)
         endpoint = DELETE_ALL_DOCUMENTS_V1.format(name=safe_name)
         response = self.api_service.delete(endpoint=endpoint)
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to delete all documents for assistant '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to delete all documents for assistant '{name}': {response.text}")
+        return parse_json_response(response, "delete all documents for assistant", name=name)
 
     def retrieve_document(self, name: str, document_id: str) -> dict:
         safe_name = self.get_url_safe_name(name)
         endpoint = RETRIEVE_DOCUMENT_V1.format(name=safe_name, id=document_id)
         response = self.api_service.get(endpoint=endpoint)
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to retrieve document with ID '{document_id}' for assistant '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to retrieve document with ID '{document_id}' for assistant '{name}': {response.text}")
+        return parse_json_response(response, "retrieve document", document_id=document_id)
 
     def upload_document(
             self,
@@ -234,12 +195,7 @@ class RAGAssistantClient(BaseClient):
         else:
             raise ValueError("Invalid upload_type. Use 'binary' or 'multipart'.")
 
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to upload document for assistant '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to upload document for assistant '{name}': {response.text}")
+        return parse_json_response(response, "upload document for assistant", name=name)
 
     def _upload_binary_document(self, endpoint: str, file_path: str, content_type: str):
         """
@@ -310,22 +266,12 @@ class RAGAssistantClient(BaseClient):
         safe_name = self.get_url_safe_name(name)
         endpoint = DELETE_DOCUMENT_V1.format(name=safe_name, id=document_id)
         response = self.api_service.delete(endpoint=endpoint)
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to delete document with ID '{document_id}' for assistant '{name}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to delete document with ID '{document_id}' for assistant '{name}': {response.text}")
+        return parse_json_response(response, "delete document", document_id=document_id)
 
     def execute_query(self, query: dict) -> dict:
         response = self.api_service.post(
             endpoint=EXECUTE_QUERY_V1,
             data=query
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to execute query: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to execute query: {response.text}")
+        return parse_json_response(response, "execute query")
 

pygeai/auth/clients.py

@@ -1,9 +1,10 @@
-from json import JSONDecodeError
-
 from pygeai import logger
-from pygeai.auth.endpoints import GET_USER_PROFILE_INFO, GET_OAUTH2_ACCESS_TOKEN
+from pygeai.auth.endpoints import GET_USER_PROFILE_INFO, GET_OAUTH2_ACCESS_TOKEN, \
+    CREATE_PROJECT_API_TOKEN_V2, DELETE_PROJECT_API_TOKEN_V2, UPDATE_PROJECT_API_TOKEN_V2, GET_PROJECT_API_TOKEN_V2
 from pygeai.core.base.clients import BaseClient
-from pygeai.core.common.exceptions import InvalidAPIResponseException
+from pygeai.core.common.exceptions import InvalidAPIResponseException, APIResponseError
+from pygeai.core.utils.validators import validate_status_code
+from pygeai.core.utils.parsers import parse_json_response
 
 
 class AuthClient(BaseClient):
@@ -33,11 +34,8 @@ class AuthClient(BaseClient):
                 "password": password
             }
         )
-        try:
-            return response.json()
-        except JSONDecodeError as e:
-            logger.error(f"Unable to obtain Oauth2 access token: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to  obtain Oauth2 access token: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, "obtain Oauth2 access token")
 
     def get_user_profile_information(self, access_token: str) -> dict:
         """
@@ -48,8 +46,84 @@ class AuthClient(BaseClient):
         """
         self.api_service.token = access_token
         response = self.api_service.get(endpoint=GET_USER_PROFILE_INFO)
-        try:
-            return response.json()
-        except JSONDecodeError as e:
-            logger.error(f"Unable to retrieve user profile information: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to retrieve user profile information: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, "retrieve user profile information")
+
+    def create_project_api_token(
+            self,
+            project_id: str,
+            name: str,
+            description: str = None
+    ) -> dict:
+        """
+        Creates a new API token for a project.
+
+        :param project_id: str - The project identifier (required). Will be sent as header.
+        :param name: str - The name of the API token (required).
+        :param description: str - A description of the API token (optional).
+        :return: dict - The API response containing the created API token details in JSON format.
+        """
+        headers = {"project-id": project_id}
+        data = {"name": name}
+        if description:
+            data["description"] = description
+
+        response = self.api_service.post(
+            endpoint=CREATE_PROJECT_API_TOKEN_V2,
+            data=data,
+            headers=headers
+        )
+        validate_status_code(response)
+        return parse_json_response(response, "create project API token")
+
+    def delete_project_api_token(self, api_token_id: str) -> dict:
+        """
+        Revokes an API token by setting its status to "revoked".
+
+        :param api_token_id: str - The unique identifier of the API token to delete (required).
+        :return: dict - The API response confirming the deletion, in JSON format.
+        """
+        endpoint = DELETE_PROJECT_API_TOKEN_V2.format(id=api_token_id)
+        response = self.api_service.delete(endpoint=endpoint)
+        validate_status_code(response)
+        return parse_json_response(response, "delete project API token", api_token_id=api_token_id)
+
+    def update_project_api_token(
+            self,
+            api_token_id: str,
+            description: str = None,
+            status: str = None
+    ) -> dict:
+        """
+        Updates an existing API token's description and/or status.
+
+        :param api_token_id: str - The unique identifier of the API token to update (required).
+        :param description: str - A new description for the API token (optional).
+        :param status: str - The new status for the API token: 'active' or 'blocked' (optional).
+        :return: dict - The API response containing the update result messages in JSON format.
+        """
+        endpoint = UPDATE_PROJECT_API_TOKEN_V2.format(id=api_token_id)
+        data = {}
+        if description is not None:
+            data["description"] = description
+        if status is not None:
+            data["status"] = status
+
+        response = self.api_service.put(
+            endpoint=endpoint,
+            data=data
+        )
+        validate_status_code(response)
+        return parse_json_response(response, "update project API token", api_token_id=api_token_id)
+
+    def get_project_api_token(self, api_token_id: str) -> dict:
+        """
+        Retrieves data for a specific project API token.
+
+        :param api_token_id: str - The unique identifier of the API token (required).
+        :return: dict - The API response containing the API token details in JSON format.
+        """
+        endpoint = GET_PROJECT_API_TOKEN_V2.format(id=api_token_id)
+        response = self.api_service.get(endpoint=endpoint)
+        validate_status_code(response)
+        return parse_json_response(response, "get project API token", api_token_id=api_token_id)

pygeai/auth/endpoints.py

@@ -1,2 +1,6 @@
 GET_OAUTH2_ACCESS_TOKEN = "/oauth/access_token"      # POST -> Obtain an OAuth 2.0 access token
 GET_USER_PROFILE_INFO = "/openid/userinfo"          # GET -> Retrieve user profile information
+CREATE_PROJECT_API_TOKEN_V2 = "v2/projects/tokens"          # POST -> Create a new API token for a project
+DELETE_PROJECT_API_TOKEN_V2 = "v2/projects/tokens/{id}"     # DELETE -> Revoke an API token
+UPDATE_PROJECT_API_TOKEN_V2 = "v2/projects/tokens/{id}"     # PUT -> Update an API token
+GET_PROJECT_API_TOKEN_V2 = "v2/projects/tokens/{id}"        # GET -> Get data of a specific project API token

pygeai/chat/clients.py

@@ -6,6 +6,7 @@ from pygeai import logger
 from pygeai.chat.endpoints import CHAT_V1, GENERATE_IMAGE_V1
 from pygeai.core.base.clients import BaseClient
 from pygeai.core.common.exceptions import InvalidAPIResponseException
+from pygeai.core.utils.validators import validate_status_code
 
 
 class ChatClient(BaseClient):

pygeai/cli/commands/auth.py

@@ -92,6 +92,146 @@ get_user_profile_information_options = [
 ]
 
 
+def create_project_api_token(option_list: list):
+    project_id = None
+    name = None
+    description = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
+        if option_flag.name == "name":
+            name = option_arg
+        if option_flag.name == "description":
+            description = option_arg
+
+    if not (project_id and name):
+        raise MissingRequirementException("Cannot create project API token without project-id and name")
+
+    client = AuthClient()
+    result = client.create_project_api_token(
+        project_id=project_id,
+        name=name,
+        description=description
+    )
+    Console.write_stdout(f"Project API token created: \n{result}")
+
+
+create_project_api_token_options = [
+    Option(
+        "project_id",
+        ["--project-id", "--pid"],
+        "The project identifier (required).",
+        True
+    ),
+    Option(
+        "name",
+        ["--name", "-n"],
+        "The name of the API token (required).",
+        True
+    ),
+    Option(
+        "description",
+        ["--description", "-d"],
+        "A description of the API token (optional).",
+        True
+    ),
+]
+
+
+def delete_project_api_token(option_list: list):
+    api_token_id = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "api_token_id":
+            api_token_id = option_arg
+
+    if not api_token_id:
+        raise MissingRequirementException("Cannot delete project API token without api-token-id")
+
+    client = AuthClient()
+    result = client.delete_project_api_token(api_token_id=api_token_id)
+    Console.write_stdout(f"Project API token deleted: \n{result}")
+
+
+delete_project_api_token_options = [
+    Option(
+        "api_token_id",
+        ["--api-token-id", "--tid"],
+        "The unique identifier of the API token to delete (required).",
+        True
+    ),
+]
+
+
+def update_project_api_token(option_list: list):
+    api_token_id = None
+    description = None
+    status = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "api_token_id":
+            api_token_id = option_arg
+        if option_flag.name == "description":
+            description = option_arg
+        if option_flag.name == "status":
+            status = option_arg
+
+    if not api_token_id:
+        raise MissingRequirementException("Cannot update project API token without api-token-id")
+
+    client = AuthClient()
+    result = client.update_project_api_token(
+        api_token_id=api_token_id,
+        description=description,
+        status=status
+    )
+    Console.write_stdout(f"Project API token updated: \n{result}")
+
+
+update_project_api_token_options = [
+    Option(
+        "api_token_id",
+        ["--api-token-id", "--tid"],
+        "The unique identifier of the API token to update (required).",
+        True
+    ),
+    Option(
+        "description",
+        ["--description", "-d"],
+        "A new description for the API token (optional).",
+        True
+    ),
+    Option(
+        "status",
+        ["--status"],
+        "The new status for the API token: 'active' or 'blocked' (optional).",
+        True
+    ),
+]
+
+
+def get_project_api_token(option_list: list):
+    api_token_id = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "api_token_id":
+            api_token_id = option_arg
+
+    if not api_token_id:
+        raise MissingRequirementException("Cannot get project API token without api-token-id")
+
+    client = AuthClient()
+    result = client.get_project_api_token(api_token_id=api_token_id)
+    Console.write_stdout(f"Project API token details: \n{result}")
+
+
+get_project_api_token_options = [
+    Option(
+        "api_token_id",
+        ["--api-token-id", "--tid"],
+        "The unique identifier of the API token (required).",
+        True
+    ),
+]
+
+
 auth_commands = [
     Command(
         "help",
@@ -112,12 +252,48 @@ auth_commands = [
         get_oauth2_access_token_options
     ),
     Command(
-        "get_user_profile_information_options",
+        "get_user_profile_information",
         ["get-user-information", "get-user-info", "gui"],
         "Retrieve user profile information",
-        get_user_profile_information_options,
+        get_user_profile_information,
         ArgumentsEnum.REQUIRED,
         [],
         get_user_profile_information_options
     ),
+    Command(
+        "create_project_api_token",
+        ["create-project-api-token", "create-api-token", "cat"],
+        "Create a new API token for a project",
+        create_project_api_token,
+        ArgumentsEnum.REQUIRED,
+        [],
+        create_project_api_token_options
+    ),
+    Command(
+        "delete_project_api_token",
+        ["delete-project-api-token", "delete-api-token", "dat"],
+        "Revoke an API token",
+        delete_project_api_token,
+        ArgumentsEnum.REQUIRED,
+        [],
+        delete_project_api_token_options
+    ),
+    Command(
+        "update_project_api_token",
+        ["update-project-api-token", "update-api-token", "uat"],
+        "Update an existing API token",
+        update_project_api_token,
+        ArgumentsEnum.REQUIRED,
+        [],
+        update_project_api_token_options
+    ),
+    Command(
+        "get_project_api_token",
+        ["get-project-api-token", "get-api-token", "gat"],
+        "Get data of a specific project API token",
+        get_project_api_token,
+        ArgumentsEnum.REQUIRED,
+        [],
+        get_project_api_token_options
+    ),
 ]

pygeai/cli/commands/lab/ai_lab.py

@@ -472,7 +472,6 @@ def export_agent(option_list: list):
         Console.write_stdout(f"Agent spec: \n{result}")
 
 
-
 export_agent_options = [
     PROJECT_ID_OPTION,
     Option(
@@ -1242,7 +1241,6 @@ def export_tool(option_list: list):
         if option_flag.name == "file":
             file = option_arg
 
-
     if not tool_id:
         raise MissingRequirementException("Tool ID must be specified.")