select-ai 1.1.0rc1__py3-none-any.whl → 1.2.0rc1__py3-none-any.whl

select_ai/sql.py

@@ -7,11 +7,14 @@
 
 GRANT_PRIVILEGES_TO_USER = """
 DECLARE
-    TYPE array_t IS VARRAY(3) OF VARCHAR2(60);
+    TYPE array_t IS VARRAY(4) OF VARCHAR2(60);
     v_packages array_t;
 BEGIN
     v_packages := array_t(
-        'DBMS_CLOUD', 'DBMS_CLOUD_AI', 'DBMS_CLOUD_PIPELINE'
+        'DBMS_CLOUD',
+        'DBMS_CLOUD_AI',
+        'DBMS_CLOUD_AI_AGENT',
+        'DBMS_CLOUD_PIPELINE'
     );
     FOR i in 1..v_packages.count LOOP
         EXECUTE IMMEDIATE
@@ -22,11 +25,14 @@ END;
 
 REVOKE_PRIVILEGES_FROM_USER = """
 DECLARE
-    TYPE array_t IS VARRAY(3) OF VARCHAR2(60);
+    TYPE array_t IS VARRAY(4) OF VARCHAR2(60);
     v_packages array_t;
 BEGIN
     v_packages := array_t(
-        'DBMS_CLOUD', 'DBMS_CLOUD_AI', 'DBMS_CLOUD_PIPELINE'
+        'DBMS_CLOUD',
+        'DBMS_CLOUD_AI',
+        'DBMS_CLOUD_AI_AGENT',
+        'DBMS_CLOUD_PIPELINE'
     );
     FOR i in 1..v_packages.count LOOP
         EXECUTE IMMEDIATE

select_ai/vector_index.py

@@ -17,13 +17,23 @@ from select_ai._abc import SelectAIDataClass
 from select_ai._enums import StrEnum
 from select_ai.async_profile import AsyncProfile
 from select_ai.db import async_cursor, cursor
-from select_ai.errors import VectorIndexNotFoundError, ProfileNotFoundError
+from select_ai.errors import ProfileNotFoundError, VectorIndexNotFoundError
 from select_ai.profile import Profile
 from select_ai.sql import (
     GET_USER_VECTOR_INDEX_ATTRIBUTES,
     LIST_USER_VECTOR_INDEXES,
 )
 
+UNMODIFIABLE_VECTOR_INDEX_ATTRIBUTES = (
+    "location",
+    "chunk_size",
+    "chunk_overlap",
+    "pipeline_name",
+    "vector_dimension",
+    "vector_table_name",
+    "vector_distance_metric",
+)
+
 
 class VectorDBProvider(StrEnum):
     ORACLE = "oracle"
@@ -93,9 +103,14 @@ class VectorIndexAttributes(SelectAIDataClass):
         attributes = self.dict(exclude_null=exclude_null)
         attributes.pop("pipeline_name", None)
         # Currently, the following are unmodifiable
-        unmodifiable = ["location", "chunk_size", "chunk_overlap",
-                        "vector_dimension", "vector_table_name",
-                        "vector_distance_metric"]
+        unmodifiable = [
+            "location",
+            "chunk_size",
+            "chunk_overlap",
+            "vector_dimension",
+            "vector_table_name",
+            "vector_distance_metric",
+        ]
         if for_update:
             for key in unmodifiable:
                 attributes.pop(key, None)
@@ -165,8 +180,12 @@ class VectorIndex(_BaseVectorIndex):
         :return: select_ai.VectorIndexAttributes
         :raises: VectorIndexNotFoundError
         """
+        if index_name is None:
+            raise AttributeError("'index_name' is required")
         with cursor() as cr:
-            cr.execute(GET_USER_VECTOR_INDEX_ATTRIBUTES, index_name=index_name.upper())
+            cr.execute(
+                GET_USER_VECTOR_INDEX_ATTRIBUTES, index_name=index_name.upper()
+            )
             attributes = cr.fetchall()
             if attributes:
                 post_processed_attributes = {}
@@ -255,10 +274,18 @@ class VectorIndex(_BaseVectorIndex):
 
         """
         with cursor() as cr:
-            cr.callproc(
-                "DBMS_CLOUD_AI.ENABLE_VECTOR_INDEX",
-                keyword_parameters={"index_name": self.index_name},
-            )
+            try:
+                cr.callproc(
+                    "DBMS_CLOUD_AI.ENABLE_VECTOR_INDEX",
+                    keyword_parameters={"index_name": self.index_name},
+                )
+            except oracledb.Error as e:
+                (error,) = e.args
+                # ORA-20000: Vector Index is already in the desired status
+                if error.code == 20000:
+                    pass
+                else:
+                    raise
 
     def disable(self):
         """This procedure disables a vector index object in the current
@@ -271,15 +298,23 @@ class VectorIndex(_BaseVectorIndex):
         :raises: oracledb.DatabaseError
         """
         with cursor() as cr:
-            cr.callproc(
-                "DBMS_CLOUD_AI.DISABLE_VECTOR_INDEX",
-                keyword_parameters={"index_name": self.index_name},
-            )
+            try:
+                cr.callproc(
+                    "DBMS_CLOUD_AI.DISABLE_VECTOR_INDEX",
+                    keyword_parameters={"index_name": self.index_name},
+                )
+            except oracledb.Error as e:
+                (error,) = e.args
+                # ORA-20000: Vector Index is already in the desired status
+                if error.code == 20000:
+                    pass
+                else:
+                    raise
 
     def set_attribute(
-            self,
-            attribute_name: str = None,
-            attribute_value: Union[str, int, float] = None
+        self,
+        attribute_name: str,
+        attribute_value: Union[str, int, float],
     ):
         """
         This procedure updates an existing vector store index with a specified
@@ -293,7 +328,7 @@ class VectorIndex(_BaseVectorIndex):
         parameters = {
             "index_name": self.index_name,
             "attribute_name": attribute_name,
-            "attribute_value": attribute_value
+            "attribute_value": attribute_value,
         }
         with cursor() as cr:
             cr.callproc(
@@ -301,7 +336,6 @@ class VectorIndex(_BaseVectorIndex):
                 keyword_parameters=parameters,
             )
 
-
     def set_attributes(
         self,
         attribute_name: Optional[str] = None,
@@ -504,10 +538,18 @@ class AsyncVectorIndex(_BaseVectorIndex):
 
         """
         async with async_cursor() as cr:
-            await cr.callproc(
-                "DBMS_CLOUD_AI.ENABLE_VECTOR_INDEX",
-                keyword_parameters={"index_name": self.index_name},
-            )
+            try:
+                await cr.callproc(
+                    "DBMS_CLOUD_AI.ENABLE_VECTOR_INDEX",
+                    keyword_parameters={"index_name": self.index_name},
+                )
+            except oracledb.DatabaseError as e:
+                (error,) = e.args
+                # ORA-20000: Vector Index is already in the desired status
+                if error.code == 20000:
+                    pass
+                else:
+                    raise
 
     async def disable(self) -> None:
         """This procedure disables a vector index object in the current
@@ -520,15 +562,20 @@ class AsyncVectorIndex(_BaseVectorIndex):
         :raises: oracledb.DatabaseError
         """
         async with async_cursor() as cr:
-            await cr.callproc(
-                "DBMS_CLOUD_AI.DISABLE_VECTOR_INDEX",
-                keyword_parameters={"index_name": self.index_name},
-            )
+            try:
+                await cr.callproc(
+                    "DBMS_CLOUD_AI.DISABLE_VECTOR_INDEX",
+                    keyword_parameters={"index_name": self.index_name},
+                )
+            except oracledb.Error as e:
+                (error,) = e.args
+                if error.code == 20000:
+                    pass
+                else:
+                    raise
 
     async def set_attribute(
-            self,
-            attribute_name: str,
-            attribute_value: Union[str, int, float]
+        self, attribute_name: str, attribute_value: Union[str, int, float]
     ) -> None:
         """
         This procedure updates an existing vector store index with a specified
@@ -538,9 +585,11 @@ class AsyncVectorIndex(_BaseVectorIndex):
         :param Union[str, int, float] attribute_value: Attribute Value
 
         """
-        parameters = {"index_name": self.index_name,
-                      "attribute_name": attribute_name,
-                      "attribute_value": attribute_value}
+        parameters = {
+            "index_name": self.index_name,
+            "attribute_name": attribute_name,
+            "attribute_value": attribute_value,
+        }
         setattr(self.attributes, attribute_name, attribute_value)
         async with async_cursor() as cr:
             await cr.callproc(
@@ -548,7 +597,6 @@ class AsyncVectorIndex(_BaseVectorIndex):
                 keyword_parameters=parameters,
             )
 
-
     async def set_attributes(
         self,
         attribute_name: Optional[str] = None,
@@ -563,8 +611,8 @@ class AsyncVectorIndex(_BaseVectorIndex):
 
         :param str attribute_name: Custom attribute name
         :param Union[str, int, float] attribute_value: Attribute Value
-        :param VectorIndexAttributes attributes: Specify multiple attributes
-         to update in a single API invocation
+        :param select_ai.VectorIndexAttributes attributes: Use this to
+         update multiple attribute values
         :return: None
         :raises: oracledb.DatabaseError
         """
@@ -643,5 +691,5 @@ class AsyncVectorIndex(_BaseVectorIndex):
                     index_name=index_name,
                     description=description,
                     attributes=attributes,
-                    profile=profile
+                    profile=profile,
                 )

select_ai/version.py

@@ -5,4 +5,4 @@
 # http://oss.oracle.com/licenses/upl.
 # -----------------------------------------------------------------------------
 
-__version__ = "1.1.0rc1"
+__version__ = "1.2.0rc1"

{select_ai-1.1.0rc1.dist-info → select_ai-1.2.0rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: select_ai
-Version: 1.1.0rc1
+Version: 1.2.0rc1
 Summary: Select AI for Python
 Author-email: Abhishek Singh <abhishek.o.singh@oracle.com>
 Maintainer-email: Abhishek Singh <abhishek.o.singh@oracle.com>

select_ai-1.2.0rc1.dist-info/RECORD

@@ -0,0 +1,28 @@
+select_ai/__init__.py,sha256=5oVcMxHqKPnIqzsPntFrqrMrTQ86bDrJsG8ARXuxN7U,1460
+select_ai/_abc.py,sha256=ccec6yu54w5QMix_EKc_BJ_0JatUDXKjfLW9KshOGak,2692
+select_ai/_enums.py,sha256=U0UavuE4pbRY-0-Qb8z-F7KfxU0YUfOgUahaSjDi0dU,416
+select_ai/_validations.py,sha256=flMtAhYE-H2hgLdKNf26Lfs0mzISJZOXPIvrAXjC1Og,3959
+select_ai/action.py,sha256=s78uhi5HHCLbTN2S5tLDSswTLUl40FZ3nESLka8CfZk,592
+select_ai/async_profile.py,sha256=Q5rEsr-XY8Snn31ePHlhRXkBT_gROpGMIi2Yhz9mFKg,18968
+select_ai/base_profile.py,sha256=Ip1IaKpvAWs1vma4HqciuEIo21WI6l_8swCiEDLW_jg,7261
+select_ai/conversation.py,sha256=fbYH7ROLwpyf1VzNtcrESG1S0_ZfLJ-uObbvbUATJUY,9622
+select_ai/credential.py,sha256=cDIsiPgjbvckbOfb2LeoztPUsDO-6cJ_S8DzYT3pm4k,4220
+select_ai/db.py,sha256=ipyXt5_wEXAQGl7PLVq82FBuePdA_e6b8ee_vhEAMks,4723
+select_ai/errors.py,sha256=06mKo0Z2lipzIsPIxVUXhbF3Vw4tRqzHOzZfv5xbyv4,3104
+select_ai/profile.py,sha256=CMotnQRwW-XkXTNrhxiaYnlsurjHksQOsb3PMqgbjPM,16368
+select_ai/provider.py,sha256=mMOWJIh3p3wxHsijW1vr6OvkXIk81HNyRyuIEsNijFE,8426
+select_ai/sql.py,sha256=UBINldCDBQ3iswtA9F4CB6JntvmT8AdYd6_gvB8twBE,2880
+select_ai/synthetic_data.py,sha256=fClMI6ue2rNqzjY9TfeLA0_w1iPOahb08Y-1-gYzwzs,3305
+select_ai/vector_index.py,sha256=Q2w4qnnlAXrqKagPUIOOf6MUdb_yAnsnU2HtiFYvPTI,25498
+select_ai/version.py,sha256=0HJ4C5R4gudaimbDn4c-WHa03LWqWKfEVUgiFEGzlvE,346
+select_ai/agent/__init__.py,sha256=cv5WLaiEKgBlVaQ-qRl7HS9WqSufmXmXLp26pKfkyoo,693
+select_ai/agent/core.py,sha256=ClaqSfmCum6WVN4iOvo6_w7mUjeUNm4Lt_RBkhDSEUw,6980
+select_ai/agent/sql.py,sha256=oc4PiKhtNAJedTuzux-Kv_KEcsiRasANwbG9EyibAg4,2000
+select_ai/agent/task.py,sha256=aQTiucS2uVsksleGJpCQbOi5XkFVqA2b8QqyB7t5cas,7270
+select_ai/agent/team.py,sha256=2AqWrK-l8UJ_rpGav5oeJPYVwSGa6ExSLKXOJh5GSiM,8689
+select_ai/agent/tool.py,sha256=9-ogSo13aQRJ2KPgo7KMImpJuh6YrHh2q35cJEm4SA8,18649
+select_ai-1.2.0rc1.dist-info/licenses/LICENSE.txt,sha256=_0VqOxSjO1hu6JexZDVzqUXSmzH1A53EOfyiJzXTBKc,1840
+select_ai-1.2.0rc1.dist-info/METADATA,sha256=fwD35WO6Qu9mzLZ10muLJi0wJapS3BXKls5rqn-CcxA,4489
+select_ai-1.2.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+select_ai-1.2.0rc1.dist-info/top_level.txt,sha256=u_QUAHDibro58Lvi_MU6a9Wc6VfQT8HEQm0cciMTP-c,10
+select_ai-1.2.0rc1.dist-info/RECORD,,

select_ai-1.1.0rc1.dist-info/RECORD

@@ -1,22 +0,0 @@
-select_ai/__init__.py,sha256=5oVcMxHqKPnIqzsPntFrqrMrTQ86bDrJsG8ARXuxN7U,1460
-select_ai/_abc.py,sha256=ccec6yu54w5QMix_EKc_BJ_0JatUDXKjfLW9KshOGak,2692
-select_ai/_enums.py,sha256=U0UavuE4pbRY-0-Qb8z-F7KfxU0YUfOgUahaSjDi0dU,416
-select_ai/_validations.py,sha256=flMtAhYE-H2hgLdKNf26Lfs0mzISJZOXPIvrAXjC1Og,3959
-select_ai/action.py,sha256=s78uhi5HHCLbTN2S5tLDSswTLUl40FZ3nESLka8CfZk,592
-select_ai/async_profile.py,sha256=KRJCcfdSIZjb3iRS7n3MjBEAFIQe-OfMLRgI4SWWPek,18967
-select_ai/base_profile.py,sha256=touS-DofLBjTIo0NUIPXQbDq96excGYS5y-E2o4y8y8,7261
-select_ai/conversation.py,sha256=fbYH7ROLwpyf1VzNtcrESG1S0_ZfLJ-uObbvbUATJUY,9622
-select_ai/credential.py,sha256=cDIsiPgjbvckbOfb2LeoztPUsDO-6cJ_S8DzYT3pm4k,4220
-select_ai/db.py,sha256=ipyXt5_wEXAQGl7PLVq82FBuePdA_e6b8ee_vhEAMks,4723
-select_ai/errors.py,sha256=2T5eICWWjj7a907aRh7es5wTcvWm_wJiP8h0b3ipxVQ,2114
-select_ai/profile.py,sha256=6s4v8AQmKXoK37lvKw47y-q57htQYHcIt3VP7V1mfLY,16366
-select_ai/provider.py,sha256=mMOWJIh3p3wxHsijW1vr6OvkXIk81HNyRyuIEsNijFE,8426
-select_ai/sql.py,sha256=9yCdB1H_sYpjMLs-6SB8xf5-QpTwCAx4cIgIsElvly4,2786
-select_ai/synthetic_data.py,sha256=fClMI6ue2rNqzjY9TfeLA0_w1iPOahb08Y-1-gYzwzs,3305
-select_ai/vector_index.py,sha256=Zt9JBmJZSlx6WkjVJptD6WlxxpsqDw17f7oYsN5PcM4,24089
-select_ai/version.py,sha256=RQndK0tg4yw5xR9GR03N2OsctaItraepy-h3KNw67RE,346
-select_ai-1.1.0rc1.dist-info/licenses/LICENSE.txt,sha256=_0VqOxSjO1hu6JexZDVzqUXSmzH1A53EOfyiJzXTBKc,1840
-select_ai-1.1.0rc1.dist-info/METADATA,sha256=UiSQXF3OpyHk6mn9kV0wkNK2X7WBgGNaOK8J6HKzWjc,4489
-select_ai-1.1.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-select_ai-1.1.0rc1.dist-info/top_level.txt,sha256=u_QUAHDibro58Lvi_MU6a9Wc6VfQT8HEQm0cciMTP-c,10
-select_ai-1.1.0rc1.dist-info/RECORD,,