Use error code to check for objects which already exists

Author: aosingh

samples/async/delete_ai_credential.py

@@ -6,9 +6,9 @@
 # -----------------------------------------------------------------------------
 
 # -----------------------------------------------------------------------------
-# async/create_ai_credential.py
+# async/delete_ai_credential
 #
-# Async API to create credential
+# Async API to delete credential
 # -----------------------------------------------------------------------------
 
 import asyncio

samples/delete_ai_credential.py

@@ -8,7 +8,7 @@
 # -----------------------------------------------------------------------------
 # delete_ai_credential.py
 #
-# Create a Database credential storing OCI Gen AI's credentials
+# Delete AI credential
 # -----------------------------------------------------------------------------
 import os
 

src/select_ai/admin.py

@@ -218,13 +218,14 @@ async def create(
         :return: None
         :raises: oracledb.DatabaseError
         """
+        if self.attributes is None:
+            raise AttributeError("Profile attributes cannot be None")
         parameters = {
             "profile_name": self.profile_name,
             "attributes": self.attributes.json(),
         }
         if description:
             parameters["description"] = description
-
         async with async_cursor() as cr:
             try:
                 await cr.callproc(
@@ -234,7 +235,7 @@ async def create(
             except oracledb.DatabaseError as e:
                 (error,) = e.args
                 # If already exists and replace is True then drop and recreate
-                if "already exists" in error.message.lower() and replace:
+                if error.code == 20046 and replace:
                     await self.delete(force=True)
                     await cr.callproc(
                         "DBMS_CLOUD_AI.CREATE_PROFILE",

src/select_ai/async_profile.py

@@ -213,7 +213,7 @@ def create(self, replace: Optional[int] = False) -> None:
             except oracledb.DatabaseError as e:
                 (error,) = e.args
                 # If already exists and replace is True then drop and recreate
-                if "already exists" in error.message.lower() and replace:
+                if error.code == 20046 and replace:
                     self.delete(force=True)
                     cr.callproc(
                         "DBMS_CLOUD_AI.CREATE_PROFILE",

src/select_ai/profile.py

@@ -192,7 +192,7 @@ def create(self, replace: Optional[bool] = False):
             except oracledb.DatabaseError as e:
                 (error,) = e.args
                 # If already exists and replace is True then drop and recreate
-                if "already exists" in error.message.lower() and replace:
+                if error.code == 20048 and replace:
                     self.delete(force=True)
                     cr.callproc(
                         "DBMS_CLOUD_AI.CREATE_VECTOR_INDEX",
@@ -399,7 +399,7 @@ async def create(self, replace: Optional[bool] = False) -> None:
             except oracledb.DatabaseError as e:
                 (error,) = e.args
                 # If already exists and replace is True then drop and recreate
-                if "already exists" in error.message.lower() and replace:
+                if error.code == 20048 and replace:
                     await self.delete(force=True)
                     await cr.callproc(
                         "DBMS_CLOUD_AI.CREATE_VECTOR_INDEX",