aixplain
diff --git a/‎.github/workflows/main.yaml
Lines changed: 68 additions & 17 deletions b/‎.github/workflows/main.yaml
Lines changed: 68 additions & 17 deletions
diff --git a/‎.pre-commit-config.yaml
Lines changed: 1 addition & 1 deletion b/‎.pre-commit-config.yaml
Lines changed: 1 addition & 1 deletion
diff --git a/‎aixplain/enums/__init__.py
Lines changed: 1 addition & 0 deletions b/‎aixplain/enums/__init__.py
Lines changed: 1 addition & 0 deletions
diff --git a/‎aixplain/enums/asset_status.py
Lines changed: 1 addition & 0 deletions b/‎aixplain/enums/asset_status.py
Lines changed: 1 addition & 0 deletions
diff --git a/‎aixplain/enums/embedding_model.py
Lines changed: 1 addition & 3 deletions b/‎aixplain/enums/embedding_model.py
Lines changed: 1 addition & 3 deletions
diff --git a/‎aixplain/enums/index_stores.py
Lines changed: 14 additions & 0 deletions b/‎aixplain/enums/index_stores.py
Lines changed: 14 additions & 0 deletions
diff --git a/‎aixplain/enums/license.py
Lines changed: 1 addition & 1 deletion b/‎aixplain/enums/license.py
Lines changed: 1 addition & 1 deletion
diff --git a/‎aixplain/enums/supplier.py
Lines changed: 1 addition & 1 deletion b/‎aixplain/enums/supplier.py
Lines changed: 1 addition & 1 deletion
diff --git a/‎aixplain/exceptions/__init__.py
Lines changed: 116 additions & 0 deletions b/‎aixplain/exceptions/__init__.py
Lines changed: 116 additions & 0 deletions
@@ -11,27 +11,77 @@ on:
 jobs:
   setup-and-test:
     runs-on: ubuntu-latest
+    timeout-minutes: 45  # Global timeout fallback
     strategy:
       fail-fast: false
       matrix:
         test-suite: [
-          'tests/unit',
-          'tests/functional/file_asset',
-          'tests/functional/data_asset',
-          'tests/functional/benchmark',
-          'tests/functional/model',
-          'tests/functional/pipelines/run_test.py --pipeline_version 2.0 --sdk_version v1 --sdk_version_param PipelineFactory',
-          'tests/functional/pipelines/run_test.py --pipeline_version 2.0 --sdk_version v2 --sdk_version_param PipelineFactory',
-          'tests/functional/pipelines/run_test.py --pipeline_version 3.0 --sdk_version v1 --sdk_version_param PipelineFactory',
-          'tests/functional/pipelines/run_test.py --pipeline_version 3.0 --sdk_version v2 --sdk_version_param PipelineFactory',
-          'tests/functional/pipelines/designer_test.py',
-          'tests/functional/pipelines/create_test.py',
-          'tests/functional/finetune --sdk_version v1 --sdk_version_param FinetuneFactory',
-          'tests/functional/finetune --sdk_version v2 --sdk_version_param FinetuneFactory',
-          'tests/functional/general_assets',
-          'tests/functional/apikey',
-          'tests/functional/agent tests/functional/team_agent',
+          'unit',
+          'file_asset',
+          'data_asset',
+          'benchmark',
+          'model',
+          'pipeline_2.0_v1',
+          'pipeline_2.0_v2',
+          'pipeline_3.0_v1',
+          'pipeline_3.0_v2',
+          'pipeline_designer',
+          'pipeline_create',
+          'finetune_v1',
+          'finetune_v2',
+          'general_assets',
+          'apikey',
+          'agent_and_team_agent',
         ]
+        include:
+          - test-suite: 'unit'
+            path: 'tests/unit'
+            timeout: 45  # Tweaked timeout for each unit tests
+          - test-suite: 'file_asset'
+            path: 'tests/functional/file_asset'
+            timeout: 45
+          - test-suite: 'data_asset'
+            path: 'tests/functional/data_asset'
+            timeout: 45
+          - test-suite: 'benchmark'
+            path: 'tests/functional/benchmark'
+            timeout: 45
+          - test-suite: 'model'
+            path: 'tests/functional/model'
+            timeout: 45
+          - test-suite: 'pipeline_2.0_v1'
+            path: 'tests/functional/pipelines/run_test.py --pipeline_version 2.0 --sdk_version v1 --sdk_version_param PipelineFactory'
+            timeout: 45
+          - test-suite: 'pipeline_2.0_v2'
+            path: 'tests/functional/pipelines/run_test.py --pipeline_version 2.0 --sdk_version v2 --sdk_version_param PipelineFactory'
+            timeout: 45
+          - test-suite: 'pipeline_3.0_v1'
+            path: 'tests/functional/pipelines/run_test.py --pipeline_version 3.0 --sdk_version v1 --sdk_version_param PipelineFactory'
+            timeout: 45
+          - test-suite: 'pipeline_3.0_v2'
+            path: 'tests/functional/pipelines/run_test.py --pipeline_version 3.0 --sdk_version v2 --sdk_version_param PipelineFactory'
+            timeout: 45
+          - test-suite: 'pipeline_designer'
+            path: 'tests/functional/pipelines/designer_test.py'
+            timeout: 45
+          - test-suite: 'pipeline_create'
+            path: 'tests/functional/pipelines/create_test.py'
+            timeout: 45
+          - test-suite: 'finetune_v1'
+            path: 'tests/functional/finetune --sdk_version v1 --sdk_version_param FinetuneFactory'
+            timeout: 45
+          - test-suite: 'finetune_v2'
+            path: 'tests/functional/finetune --sdk_version v2 --sdk_version_param FinetuneFactory'
+            timeout: 45
+          - test-suite: 'general_assets'
+            path: 'tests/functional/general_assets'
+            timeout: 45
+          - test-suite: 'apikey'
+            path: 'tests/functional/apikey'
+            timeout: 45
+          - test-suite: 'agent_and_team_agent'
+            path: 'tests/functional/agent tests/functional/team_agent'
+            timeout: 45
     steps:
       - name: Checkout repository
         uses: actions/checkout@v4
@@ -62,4 +112,5 @@ jobs:
           fi
           
       - name: Run Tests
-        run: python -m pytest ${{ matrix.test-suite}}
+        timeout-minutes: ${{ matrix.timeout }}
+        run: python -m pytest ${{ matrix.path }}
@@ -22,4 +22,4 @@ repos:
     hooks:
       - id: flake8
         args: # arguments to configure flake8
-          - --ignore=E402,E501,E203
+          - --ignore=E402,E501,E203,W503
@@ -18,3 +18,4 @@
 from .database_source import DatabaseSourceType
 from .embedding_model import EmbeddingModel
 from .asset_status import AssetStatus
+from .index_stores import IndexStores
@@ -43,3 +43,4 @@ class AssetStatus(Text, Enum):
     COMPLETED = "completed"
     CANCELING = "canceling"
     CANCELED = "canceled"
+    DEPRECATED_DRAFT = "deprecated_draft"
@@ -20,16 +20,14 @@
 from enum import Enum
 
 
-class EmbeddingModel(Enum):
+class EmbeddingModel(str, Enum):
     SNOWFLAKE_ARCTIC_EMBED_M_LONG = "6658d40729985c2cf72f42ec"
     OPENAI_ADA002 = "6734c55df127847059324d9e"
     SNOWFLAKE_ARCTIC_EMBED_L_V2_0 = "678a4f8547f687504744960a"
     JINA_CLIP_V2_MULTIMODAL = "67c5f705d8f6a65d6f74d732"
     MULTILINGUAL_E5_LARGE = "67efd0772a0a850afa045af3"
     BGE_M3 = "67efd4f92a0a850afa045af7"
     AIXPLAIN_LEGAL_EMBEDDINGS = "681254b668e47e7844c1f15a"
-
-
 
 
     def __str__(self):
 
@@ -0,0 +1,14 @@
+from enum import Enum
+
+
+class IndexStores(Enum):
+    AIR = {"name": "air", "id": "66eae6656eb56311f2595011"}
+    VECTARA = {"name": "vectara", "id": "655e20f46eb563062a1aa301"}
+    GRAPHRAG = {"name": "graphrag", "id": "67dd6d487cbf0a57cf4b72f3"}
+    ZERO_ENTROPY = {"name": "zeroentropy", "id": "6807949168e47e7844c1f0c5"}
+
+    def __str__(self):
+        return self.value["name"]
+
+    def get_model_id(self):
+        return self.value["id"]
@@ -25,8 +25,8 @@
 from enum import Enum
 from urllib.parse import urljoin
 from aixplain.utils import config
-from aixplain.utils.request_utils import _request_with_retry
 from aixplain.utils.cache_utils import save_to_cache, load_from_cache, CACHE_FOLDER
+from aixplain.utils.request_utils import _request_with_retry
 
 CACHE_FILE = f"{CACHE_FOLDER}/licenses.json"
 
 
@@ -24,7 +24,7 @@
 import logging
 
 from aixplain.utils import config
-from aixplain.utils.file_utils import _request_with_retry
+from aixplain.utils.request_utils import _request_with_retry
 from enum import Enum
 from urllib.parse import urljoin
 import re
 
@@ -0,0 +1,116 @@
+"""
+Error message registry for aiXplain SDK.
+
+This module maintains a centralized registry of error messages used throughout the aiXplain ecosystem.
+It allows developers to look up existing error messages and reuse them instead of creating new ones.
+"""
+
+from aixplain.exceptions.types import (
+    AixplainBaseException,
+    AuthenticationError,
+    ValidationError,
+    ResourceError,
+    BillingError,
+    SupplierError,
+    NetworkError,
+    ServiceError,
+    InternalError,
+)
+
+
+def get_error_from_status_code(status_code: int, error_details: str = None) -> AixplainBaseException:
+    """
+    Map HTTP status codes to appropriate exception types.
+
+    Args:
+        status_code (int): The HTTP status code to map.
+        default_message (str, optional): The default message to use if no specific message is available.
+
+    Returns:
+        AixplainBaseException: An exception of the appropriate type.
+    """
+    try:
+        if isinstance(status_code, str):
+            status_code = int(status_code)
+    except Exception as e:
+        raise InternalError(f"Failed to get status code from {status_code}: {e}") from e
+
+    error_details = f"Details: {error_details}" if error_details else ""
+    if status_code == 400:
+        return ValidationError(
+            message=f"Bad request: Please verify the request payload and ensure it is correct. {error_details}".strip(),
+            status_code=status_code,
+        )
+    elif status_code == 401:
+        return AuthenticationError(
+            message=f"Unauthorized API key: Please verify the spelling of the API key and its current validity. {error_details}".strip(),
+            status_code=status_code,
+        )
+    elif status_code == 402:
+        return BillingError(
+            message=f"Payment required: Please ensure you have enough credits to run this asset. {error_details}".strip(),
+            status_code=status_code,
+        )
+    elif status_code == 403:
+        # 403 could be auth or resource, using ResourceError as a general 'forbidden'
+        return ResourceError(
+            message=f"Forbidden access: Please verify the API key and its current validity. {error_details}".strip(),
+            status_code=status_code,
+        )
+    elif status_code == 404:
+        # Added 404 mapping
+        return ResourceError(
+            message=f"Resource not found: Please verify the spelling of the resource and its current availability. {error_details}".strip(),
+            status_code=status_code,
+        )
+    elif status_code == 429:
+        # Using SupplierError for rate limiting as per your original function
+        return SupplierError(
+            message=f"Rate limit exceeded: Please try again later. {error_details}".strip(),
+            status_code=status_code,
+            retry_recommended=True,
+        )
+    elif status_code == 500:
+        return InternalError(
+            message=f"Internal server error: Please try again later. {error_details}".strip(),
+            status_code=status_code,
+            retry_recommended=True,
+        )
+    elif status_code == 503:
+        return ServiceError(
+            message=f"Service unavailable: Please try again later. {error_details}".strip(),
+            status_code=status_code,
+            retry_recommended=True,
+        )
+    elif status_code == 504:
+        return NetworkError(
+            message=f"Gateway timeout: Please try again later. {error_details}".strip(),
+            status_code=status_code,
+            retry_recommended=True,
+        )
+    elif 460 <= status_code < 470:
+        return ResourceError(
+            message=f"Subscription-related error: Please ensure that your subscription is active and has not expired. {error_details}".strip(),
+            status_code=status_code,
+        )
+    elif 470 <= status_code < 480:
+        return BillingError(
+            message=f"Billing-related error: Please ensure you have enough credits to run this asset. {error_details}".strip(),
+            status_code=status_code,
+        )
+    elif 480 <= status_code < 490:
+        return SupplierError(
+            message=f"Supplier-related error: Please ensure that the selected supplier provides the asset you are trying to access. {error_details}".strip(),
+            status_code=status_code,
+        )
+    elif 490 <= status_code < 500:
+        return ValidationError(
+            message=f"Validation-related error: Please verify the request payload and ensure it is correct. {error_details}".strip(),
+            status_code=status_code,
+        )
+    else:
+        # Catch-all for other client/server errors
+        category = "Client" if 400 <= status_code < 500 else "Server"
+        return InternalError(
+            message=f"Unspecified {category} Error (Status {status_code}) {error_details}".strip(), status_code=status_code
+        )