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

pygeai/_docs/source/pygeai.tests.snippets.evaluation.plan.rst

@@ -0,0 +1,133 @@
+pygeai.tests.snippets.evaluation.plan package
+=============================================
+
+Submodules
+----------
+
+pygeai.tests.snippets.evaluation.plan.add\_plan\_system\_metric module
+----------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.add_plan_system_metric
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.complete\_workflow\_example module
+------------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.complete_workflow_example
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.create\_evaluation\_plan module
+---------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.create_evaluation_plan
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.create\_rag\_evaluation\_plan module
+--------------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.create_rag_evaluation_plan
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.delete\_evaluation\_plan module
+---------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.delete_evaluation_plan
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.delete\_plan\_system\_metric module
+-------------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.delete_plan_system_metric
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.execute\_evaluation\_plan module
+----------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.execute_evaluation_plan
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.get\_evaluation\_plan module
+------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.get_evaluation_plan
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.get\_plan\_system\_metric module
+----------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.get_plan_system_metric
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.get\_system\_metric module
+----------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.get_system_metric
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.list\_evaluation\_plans module
+--------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.list_evaluation_plans
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.list\_plan\_system\_metrics module
+------------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.list_plan_system_metrics
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.list\_system\_metrics module
+------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.list_system_metrics
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.update\_evaluation\_plan module
+---------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.update_evaluation_plan
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.plan.update\_plan\_system\_metric module
+-------------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan.update_plan_system_metric
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+Module contents
+---------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.plan
+   :members:
+   :show-inheritance:
+   :undoc-members:

pygeai/_docs/source/pygeai.tests.snippets.evaluation.result.rst

@@ -0,0 +1,37 @@
+pygeai.tests.snippets.evaluation.result package
+===============================================
+
+Submodules
+----------
+
+pygeai.tests.snippets.evaluation.result.complete\_workflow\_example module
+--------------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.result.complete_workflow_example
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.result.get\_evaluation\_result module
+----------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.result.get_evaluation_result
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.evaluation.result.list\_evaluation\_results module
+------------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.result.list_evaluation_results
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+Module contents
+---------------
+
+.. automodule:: pygeai.tests.snippets.evaluation.result
+   :members:
+   :show-inheritance:
+   :undoc-members:

pygeai/_docs/source/pygeai.tests.snippets.evaluation.rst

@@ -1,6 +1,16 @@
 pygeai.tests.snippets.evaluation package
 ========================================
 
+Subpackages
+-----------
+
+.. toctree::
+   :maxdepth: 4
+
+   pygeai.tests.snippets.evaluation.dataset
+   pygeai.tests.snippets.evaluation.plan
+   pygeai.tests.snippets.evaluation.result
+
 Module contents
 ---------------
 

pygeai/_docs/source/pygeai.tests.snippets.organization.rst

@@ -28,6 +28,22 @@ pygeai.tests.snippets.organization.export\_request\_data module
    :show-inheritance:
    :undoc-members:
 
+pygeai.tests.snippets.organization.get\_memberships module
+----------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.organization.get_memberships
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.organization.get\_organization\_members module
+--------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.organization.get_organization_members
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
 pygeai.tests.snippets.organization.get\_project\_data module
 ------------------------------------------------------------
 
@@ -44,6 +60,30 @@ pygeai.tests.snippets.organization.get\_project\_list module
    :show-inheritance:
    :undoc-members:
 
+pygeai.tests.snippets.organization.get\_project\_members module
+---------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.organization.get_project_members
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.organization.get\_project\_memberships module
+-------------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.organization.get_project_memberships
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
+pygeai.tests.snippets.organization.get\_project\_roles module
+-------------------------------------------------------------
+
+.. automodule:: pygeai.tests.snippets.organization.get_project_roles
+   :members:
+   :show-inheritance:
+   :undoc-members:
+
 pygeai.tests.snippets.organization.get\_project\_tokens module
 --------------------------------------------------------------
 

pygeai/_docs/source/pygeai.tests.snippets.rst

@@ -8,7 +8,9 @@ Subpackages
    :maxdepth: 4
 
    pygeai.tests.snippets.assistants
+   pygeai.tests.snippets.auth
    pygeai.tests.snippets.chat
+   pygeai.tests.snippets.dbg
    pygeai.tests.snippets.embeddings
    pygeai.tests.snippets.evaluation
    pygeai.tests.snippets.extras

pygeai/admin/clients.py

@@ -1,11 +1,11 @@
-import json
-from json import JSONDecodeError
 
 from pygeai import logger
 from pygeai.admin.endpoints import GET_API_TOKEN_VALIDATION_V1, GET_AUTHORIZED_ORGANIZATIONS_V1, \
     GET_AUTHORIZED_PROJECTS_V1, GET_PROJECT_VISIBILITY_V1, GET_PROJECT_API_TOKEN_V1
 from pygeai.core.base.clients import BaseClient
 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 AdminClient(BaseClient):
@@ -17,12 +17,8 @@ class AdminClient(BaseClient):
         :return: dict - The API response containing organization and project information in JSON format.
         """
         response = self.api_service.get(endpoint=GET_API_TOKEN_VALIDATION_V1)
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to validate API token: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to validate API token: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, "validate API token")
 
     def get_authorized_organizations(self) -> dict:
         """
@@ -31,12 +27,8 @@ class AdminClient(BaseClient):
         :return: dict - The API response containing the list of authorized organizations in JSON format.
         """
         response = self.api_service.get(endpoint=GET_AUTHORIZED_ORGANIZATIONS_V1)
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to retrieve authorized organizations: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to retrieve authorized organizations: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, "retrieve authorized organizations")
 
     def get_authorized_projects_by_organization(
             self,
@@ -54,12 +46,8 @@ class AdminClient(BaseClient):
                 "organization": organization
             }
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to retrieve authorized projects for organization '{organization}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to retrieve authorized projects for organization {organization}: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, "retrieve authorized projects for organization", organization=organization)
 
     def get_project_visibility(
             self,
@@ -90,12 +78,8 @@ class AdminClient(BaseClient):
                 "accessToken": access_token
             }
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to retrieve project visibility for organization '{organization}' and project '{project}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to retrieve project visibility for organization {organization} and project {project}: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, "retrieve project visibility", organization=organization, project=project)
 
     def get_project_api_token(
             self,
@@ -127,9 +111,5 @@ class AdminClient(BaseClient):
                 "accessToken": access_token
             }
         )
-        try:
-            result = response.json()
-            return result
-        except JSONDecodeError as e:
-            logger.error(f"Unable to retrieve project API token for organization '{organization}' and project '{project}': JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to retrieve project API token for organization {organization} and project {project}: {response.text}")
+        validate_status_code(response)
+        return parse_json_response(response, "retrieve project API token", organization=organization, project=project)

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,8 @@ 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}")
+        validate_status_code(response)
+        return parse_json_response(response, "get assistant data for ID", assistant_id=assistant_id)
 
     def create_assistant(
             self,
@@ -88,12 +84,8 @@ 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}")
+        validate_status_code(response)
+        return parse_json_response(response, "create assistant with name", name=name)
 
     def update_assistant(
             self,
@@ -165,12 +157,8 @@ 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}")
+        validate_status_code(response)
+        return parse_json_response(response, "update assistant", assistant_id=assistant_id)
 
     def delete_assistant(
             self,
@@ -186,12 +174,8 @@ 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}")
+        validate_status_code(response)
+        return parse_json_response(response, "delete assistant", assistant_id=assistant_id)
 
     def send_chat_request(
             self,
@@ -223,12 +207,8 @@ 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}")
+        validate_status_code(response)
+        return parse_json_response(response, "send chat request to assistant", assistant_name=assistant_name)
 
     def get_request_status(
             self,
@@ -244,12 +224,8 @@ 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}")
+        validate_status_code(response)
+        return parse_json_response(response, "get request status for ID", request_id=request_id)
 
     def cancel_request(
             self,
@@ -268,12 +244,8 @@ 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}")
+        validate_status_code(response)
+        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,8 @@ 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}")
+        validate_status_code(response)
+        return parse_json_response(response, "get status for assistant ID", assistant_id=assistant_id)
 
     def extend_dataset(
         self,
@@ -69,12 +66,8 @@ 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}")
+            validate_status_code(response)
+            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,14 @@ 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}")
+        validate_status_code(response)
+        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}")
+        validate_status_code(response)
+        return parse_json_response(response, "get assistant data for name", name=name)
 
     def create_assistant(
             self,
@@ -116,12 +109,8 @@ 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}")
+        validate_status_code(response)
+        return parse_json_response(response, "create assistant with name", name=name)
 
     def update_assistant(
             self,
@@ -154,23 +143,15 @@ 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}")
+        validate_status_code(response)
+        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}")
+        validate_status_code(response)
+        return parse_json_response(response, "delete assistant with name", name=name)
 
     def get_documents(
             self,
@@ -187,34 +168,22 @@ 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}")
+        validate_status_code(response)
+        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}")
+        validate_status_code(response)
+        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}")
+        validate_status_code(response)
+        return parse_json_response(response, "retrieve document", document_id=document_id)
 
     def upload_document(
             self,
@@ -234,12 +203,8 @@ 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}")
+        validate_status_code(response)
+        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 +275,14 @@ 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}")
+        validate_status_code(response)
+        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}")
+        validate_status_code(response)
+        return parse_json_response(response, "execute query")