select-ai 1.0.0.dev7__py3-none-any.whl → 1.0.0.dev9__py3-none-any.whl

select_ai/__init__.py

@@ -6,11 +6,6 @@
 # -----------------------------------------------------------------------------
 
 from .action import Action
-from .admin import (
-    create_credential,
-    disable_provider,
-    enable_provider,
-)
 from .async_profile import AsyncProfile
 from .base_profile import BaseProfile, ProfileAttributes
 from .conversation import (
@@ -18,6 +13,12 @@ from .conversation import (
     Conversation,
     ConversationAttributes,
 )
+from .credential import (
+    async_create_credential,
+    async_delete_credential,
+    create_credential,
+    delete_credential,
+)
 from .db import (
     async_connect,
     async_cursor,
@@ -39,6 +40,10 @@ from .provider import (
     OCIGenAIProvider,
     OpenAIProvider,
     Provider,
+    async_disable_provider,
+    async_enable_provider,
+    disable_provider,
+    enable_provider,
 )
 from .synthetic_data import (
     SyntheticDataAttributes,

select_ai/async_profile.py

@@ -55,7 +55,7 @@ class AsyncProfile(BaseProfile):
         :return: None
         :raises: oracledb.DatabaseError
         """
-        if self.profile_name is not None:
+        if self.profile_name:
             profile_exists = False
             try:
                 saved_attributes = await self._get_attributes(
@@ -75,7 +75,7 @@ class AsyncProfile(BaseProfile):
                         profile_name=self.profile_name
                     )
             except ProfileNotFoundError:
-                if self.attributes is None:
+                if self.attributes is None and self.description is None:
                     raise
             else:
                 if self.attributes is None:
@@ -91,10 +91,13 @@ class AsyncProfile(BaseProfile):
                 await self.create(
                     replace=self.replace, description=self.description
                 )
+        else:  # profile name is None:
+            if self.attributes is not None or self.description is not None:
+                raise ValueError("'profile_name' cannot be empty or None")
         return self
 
     @staticmethod
-    async def _get_profile_description(profile_name) -> str:
+    async def _get_profile_description(profile_name) -> Union[str, None]:
         """Get description of profile from USER_CLOUD_AI_PROFILES
 
         :param str profile_name: Name of profile
@@ -110,7 +113,10 @@ class AsyncProfile(BaseProfile):
             )
             profile = await cr.fetchone()
             if profile:
-                return await profile[1].read()
+                if profile[1] is not None:
+                    return await profile[1].read()
+                else:
+                    return None
             else:
                 raise ProfileNotFoundError(profile_name)
 
@@ -186,6 +192,12 @@ class AsyncProfile(BaseProfile):
          attributes
         :return: None
         """
+        if not isinstance(attributes, ProfileAttributes):
+            raise TypeError(
+                "'attributes' must be an object of type "
+                "select_ai.ProfileAttributes"
+            )
+
         self.attributes = attributes
         parameters = {
             "profile_name": self.profile_name,
@@ -206,13 +218,14 @@ class AsyncProfile(BaseProfile):
         :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(
@@ -222,7 +235,7 @@ class AsyncProfile(BaseProfile):
             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",
@@ -302,7 +315,7 @@ class AsyncProfile(BaseProfile):
                 )
 
     async def generate(
-        self, prompt, action=Action.SHOWSQL, params: Mapping = None
+        self, prompt: str, action=Action.SHOWSQL, params: Mapping = None
     ) -> Union[pandas.DataFrame, str, None]:
         """Asynchronously perform AI translation using this profile
 
@@ -312,6 +325,9 @@ class AsyncProfile(BaseProfile):
          conversation_id for context-aware chats
         :return: Union[pandas.DataFrame, str]
         """
+        if not prompt:
+            raise ValueError("prompt cannot be empty or None")
+
         parameters = {
             "prompt": prompt,
             "action": action,
@@ -431,6 +447,13 @@ class AsyncProfile(BaseProfile):
         :raises: oracledb.DatabaseError
 
         """
+        if synthetic_data_attributes is None:
+            raise ValueError("'synthetic_data_attributes' cannot be None")
+
+        if not isinstance(synthetic_data_attributes, SyntheticDataAttributes):
+            raise TypeError("'synthetic_data_attributes' must be an object "
+                            "of type select_ai.SyntheticDataAttributes")
+
         keyword_parameters = synthetic_data_attributes.prepare()
         keyword_parameters["profile_name"] = self.profile_name
         async with async_cursor() as cr:

select_ai/base_profile.py

@@ -73,10 +73,9 @@ class ProfileAttributes(SelectAIDataClass):
     vector_index_name: Optional[str] = None
 
     def __post_init__(self):
-        if not isinstance(self.provider, Provider):
+        if self.provider and not isinstance(self.provider, Provider):
             raise ValueError(
-                f"The arg `provider` must be an object of "
-                f"type select_ai.Provider"
+                f"'provider' must be an object of " f"type select_ai.Provider"
             )
 
     def json(self, exclude_null=True):
@@ -166,6 +165,11 @@ class BaseProfile(ABC):
     ):
         """Initialize a base profile"""
         self.profile_name = profile_name
+        if attributes and not isinstance(attributes, ProfileAttributes):
+            raise TypeError(
+                "'attributes' must be an object of type "
+                "select_ai.ProfileAttributes"
+            )
         self.attributes = attributes
         self.description = description
         self.merge = merge

select_ai/conversation.py

@@ -129,7 +129,10 @@ class Conversation(_BaseConversation):
             attributes = cr.fetchone()
             if attributes:
                 conversation_title = attributes[0]
-                description = attributes[1].read()  # Oracle.LOB
+                if attributes[1]:
+                    description = attributes[1].read()  # Oracle.LOB
+                else:
+                    description = None
                 retention_days = attributes[2]
                 return ConversationAttributes(
                     title=conversation_title,
@@ -154,7 +157,10 @@ class Conversation(_BaseConversation):
             for row in cr.fetchall():
                 conversation_id = row[0]
                 conversation_title = row[1]
-                description = row[2].read()  # Oracle.LOB
+                if row[2]:
+                    description = row[2].read()  # Oracle.LOB
+                else:
+                    description = None
                 retention_days = row[3]
                 attributes = ConversationAttributes(
                     title=conversation_title,
@@ -224,7 +230,10 @@ class AsyncConversation(_BaseConversation):
             attributes = await cr.fetchone()
             if attributes:
                 conversation_title = attributes[0]
-                description = await attributes[1].read()  # Oracle.AsyncLOB
+                if attributes[1]:
+                    description = await attributes[1].read()  # Oracle.AsyncLOB
+                else:
+                    description = None
                 retention_days = attributes[2]
                 return ConversationAttributes(
                     title=conversation_title,
@@ -250,7 +259,10 @@ class AsyncConversation(_BaseConversation):
             for row in rows:
                 conversation_id = row[0]
                 conversation_title = row[1]
-                description = await row[2].read()  # Oracle.AsyncLOB
+                if row[2]:
+                    description = await row[2].read()  # Oracle.AsyncLOB
+                else:
+                    description = None
                 retention_days = row[3]
                 attributes = ConversationAttributes(
                     title=conversation_title,

select_ai/credential.py

@@ -0,0 +1,135 @@
+# -----------------------------------------------------------------------------
+# Copyright (c) 2025, Oracle and/or its affiliates.
+#
+# Licensed under the Universal Permissive License v 1.0 as shown at
+# http://oss.oracle.com/licenses/upl.
+# -----------------------------------------------------------------------------
+
+from typing import Mapping
+
+import oracledb
+
+from .db import async_cursor, cursor
+
+__all__ = [
+    "async_create_credential",
+    "async_delete_credential",
+    "create_credential",
+    "delete_credential",
+]
+
+
+def _validate_credential(credential: Mapping[str, str]):
+    valid_keys = {
+        "credential_name",
+        "username",
+        "password",
+        "user_ocid",
+        "tenancy_ocid",
+        "private_key",
+        "fingerprint",
+        "comments",
+    }
+    for k in credential.keys():
+        if k.lower() not in valid_keys:
+            raise ValueError(
+                f"Invalid value {k}: {credential[k]} for credential object"
+            )
+
+
+async def async_create_credential(credential: Mapping, replace: bool = False):
+    """
+    Async API to create credential.
+
+    Creates a credential object using DBMS_CLOUD.CREATE_CREDENTIAL. if replace
+    is True, credential will be replaced if it already exists
+
+    """
+    _validate_credential(credential)
+    async with async_cursor() as cr:
+        try:
+            await cr.callproc(
+                "DBMS_CLOUD.CREATE_CREDENTIAL", keyword_parameters=credential
+            )
+        except oracledb.DatabaseError as e:
+            (error,) = e.args
+            # If already exists and replace is True then drop and recreate
+            if error.code == 20022 and replace:
+                await cr.callproc(
+                    "DBMS_CLOUD.DROP_CREDENTIAL",
+                    keyword_parameters={
+                        "credential_name": credential["credential_name"]
+                    },
+                )
+                await cr.callproc(
+                    "DBMS_CLOUD.CREATE_CREDENTIAL",
+                    keyword_parameters=credential,
+                )
+            else:
+                raise
+
+
+async def async_delete_credential(credential_name: str, force: bool = False):
+    """
+    Async API to create credential.
+
+    Deletes a credential object using DBMS_CLOUD.DROP_CREDENTIAL
+    """
+    async with async_cursor() as cr:
+        try:
+            await cr.callproc(
+                "DBMS_CLOUD.DROP_CREDENTIAL",
+                keyword_parameters={"credential_name": credential_name},
+            )
+        except oracledb.DatabaseError as e:
+            (error,) = e.args
+            if error.code == 20004 and force:  # does not exist
+                pass
+            else:
+                raise
+
+
+def create_credential(credential: Mapping, replace: bool = False):
+    """
+
+    Creates a credential object using DBMS_CLOUD.CREATE_CREDENTIAL. if replace
+    is True, credential will be replaced if it "already exists"
+
+    """
+    _validate_credential(credential)
+    with cursor() as cr:
+        try:
+            cr.callproc(
+                "DBMS_CLOUD.CREATE_CREDENTIAL", keyword_parameters=credential
+            )
+        except oracledb.DatabaseError as e:
+            (error,) = e.args
+            # If already exists and replace is True then drop and recreate
+            if error.code == 20022 and replace:
+                cr.callproc(
+                    "DBMS_CLOUD.DROP_CREDENTIAL",
+                    keyword_parameters={
+                        "credential_name": credential["credential_name"]
+                    },
+                )
+                cr.callproc(
+                    "DBMS_CLOUD.CREATE_CREDENTIAL",
+                    keyword_parameters=credential,
+                )
+            else:
+                raise
+
+
+def delete_credential(credential_name: str, force: bool = False):
+    with cursor() as cr:
+        try:
+            cr.callproc(
+                "DBMS_CLOUD.DROP_CREDENTIAL",
+                keyword_parameters={"credential_name": credential_name},
+            )
+        except oracledb.DatabaseError as e:
+            (error,) = e.args
+            if error.code == 20004 and force:  # does not exist
+                pass
+            else:
+                raise

select_ai/db.py

@@ -73,7 +73,7 @@ def is_connected() -> bool:
         return False
     try:
         return conn.ping() is None
-    except oracledb.DatabaseError:
+    except (oracledb.DatabaseError, oracledb.InterfaceError):
         return False
 
 
@@ -87,7 +87,7 @@ async def async_is_connected() -> bool:
         return False
     try:
         return await conn.ping() is None
-    except oracledb.DatabaseError:
+    except (oracledb.DatabaseError, oracledb.InterfaceError):
         return False
 
 

select_ai/profile.py

@@ -44,7 +44,7 @@ class Profile(BaseProfile):
         :return: None
         :raises: oracledb.DatabaseError
         """
-        if self.profile_name is not None:
+        if self.profile_name:
             profile_exists = False
             try:
                 saved_attributes = self._get_attributes(
@@ -64,7 +64,7 @@ class Profile(BaseProfile):
                         profile_name=self.profile_name
                     )
             except ProfileNotFoundError:
-                if self.attributes is None:
+                if self.attributes is None and self.description is None:
                     raise
             else:
                 if self.attributes is None:
@@ -78,20 +78,28 @@ class Profile(BaseProfile):
                         )
             if self.replace or not profile_exists:
                 self.create(replace=self.replace)
+        else:  # profile name is None
+            if self.attributes is not None or self.description is not None:
+                raise ValueError(
+                    "Attribute 'profile_name' cannot be empty or None"
+                )
 
     @staticmethod
-    def _get_profile_description(profile_name) -> str:
+    def _get_profile_description(profile_name) -> Union[str, None]:
         """Get description of profile from USER_CLOUD_AI_PROFILES
 
         :param str profile_name:
-        :return: str
+        :return: Union[str, None] profile description
         :raises: ProfileNotFoundError
         """
         with cursor() as cr:
             cr.execute(GET_USER_AI_PROFILE, profile_name=profile_name.upper())
             profile = cr.fetchone()
             if profile:
-                return profile[1].read()
+                if profile[1] is not None:
+                    return profile[1].read()
+                else:
+                    return None
             else:
                 raise ProfileNotFoundError(profile_name)
 
@@ -165,6 +173,11 @@ class Profile(BaseProfile):
          attributes
         :return: None
         """
+        if not isinstance(attributes, ProfileAttributes):
+            raise TypeError(
+                "'attributes' must be an object of type"
+                " select_ai.ProfileAttributes"
+            )
         self.attributes = attributes
         parameters = {
             "profile_name": self.profile_name,
@@ -182,7 +195,8 @@ class Profile(BaseProfile):
         :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(),
@@ -199,7 +213,7 @@ class Profile(BaseProfile):
             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",
@@ -283,6 +297,8 @@ class Profile(BaseProfile):
          conversation_id for context-aware chats
         :return: Union[pandas.DataFrame, str]
         """
+        if not prompt:
+            raise ValueError("prompt cannot be empty or None")
         parameters = {
             "prompt": prompt,
             "action": action,
@@ -393,6 +409,13 @@ class Profile(BaseProfile):
         :raises: oracledb.DatabaseError
 
         """
+        if synthetic_data_attributes is None:
+            raise ValueError("Param 'synthetic_data_attributes' cannot be None")
+
+        if not isinstance(synthetic_data_attributes, SyntheticDataAttributes):
+            raise TypeError("'synthetic_data_attributes' must be an object "
+                            "of type select_ai.SyntheticDataAttributes")
+
         keyword_parameters = synthetic_data_attributes.prepare()
         keyword_parameters["profile_name"] = self.profile_name
         with cursor() as cr:

select_ai/provider.py

@@ -5,11 +5,19 @@
 # http://oss.oracle.com/licenses/upl.
 # -----------------------------------------------------------------------------
 
-from dataclasses import dataclass, fields
-from typing import Optional
+from dataclasses import dataclass
+from typing import List, Optional, Union
 
 from select_ai._abc import SelectAIDataClass
 
+from .db import async_cursor, cursor
+from .sql import (
+    DISABLE_AI_PROFILE_DOMAIN_FOR_USER,
+    ENABLE_AI_PROFILE_DOMAIN_FOR_USER,
+    GRANT_PRIVILEGES_TO_USER,
+    REVOKE_PRIVILEGES_FROM_USER,
+)
+
 OPENAI = "openai"
 COHERE = "cohere"
 AZURE = "azure"
@@ -184,3 +192,97 @@ class AnthropicProvider(Provider):
 
     provider_name: str = ANTHROPIC
     provider_endpoint = "api.anthropic.com"
+
+
+async def async_enable_provider(
+    users: Union[str, List[str]], provider_endpoint: str = None
+):
+    """
+    Async API to enable AI profile for database users.
+
+    This method grants execute privilege on the packages DBMS_CLOUD,
+    DBMS_CLOUD_AI and DBMS_CLOUD_PIPELINE. It  also enables the database
+    user to invoke the AI Provider (LLM) endpoint
+
+    """
+    if isinstance(users, str):
+        users = [users]
+
+    async with async_cursor() as cr:
+        for user in users:
+            await cr.execute(GRANT_PRIVILEGES_TO_USER.format(user))
+            if provider_endpoint:
+                await cr.execute(
+                    ENABLE_AI_PROFILE_DOMAIN_FOR_USER,
+                    user=user,
+                    host=provider_endpoint,
+                )
+
+
+async def async_disable_provider(
+    users: Union[str, List[str]], provider_endpoint: str = None
+):
+    """
+    Async API to disable  AI profile for database users
+
+    Disables AI provider for the user. This method revokes execute privilege
+    on the packages DBMS_CLOUD, DBMS_CLOUD_AI and DBMS_CLOUD_PIPELINE. It
+    also disables the user to invoke the AI Provider (LLM) endpoint
+    """
+    if isinstance(users, str):
+        users = [users]
+
+    async with async_cursor() as cr:
+        for user in users:
+            await cr.execute(REVOKE_PRIVILEGES_FROM_USER.format(user))
+            if provider_endpoint:
+                await cr.execute(
+                    DISABLE_AI_PROFILE_DOMAIN_FOR_USER,
+                    user=user,
+                    host=provider_endpoint,
+                )
+
+
+def enable_provider(
+    users: Union[str, List[str]], provider_endpoint: str = None
+):
+    """
+    Enables AI profile for the user. This method grants execute privilege
+    on the packages DBMS_CLOUD, DBMS_CLOUD_AI and DBMS_CLOUD_PIPELINE. It
+    also enables the user to invoke the AI Provider (LLM) endpoint
+    """
+    if isinstance(users, str):
+        users = [users]
+
+    with cursor() as cr:
+        for user in users:
+            cr.execute(GRANT_PRIVILEGES_TO_USER.format(user))
+            if provider_endpoint:
+                cr.execute(
+                    ENABLE_AI_PROFILE_DOMAIN_FOR_USER,
+                    user=user,
+                    host=provider_endpoint,
+                )
+
+
+def disable_provider(
+    users: Union[str, List[str]], provider_endpoint: str = None
+):
+    """
+    Disables AI provider for the user. This method revokes execute privilege
+    on the packages DBMS_CLOUD, DBMS_CLOUD_AI and DBMS_CLOUD_PIPELINE. It
+    also disables the user to invoke the AI(LLM) endpoint
+
+    """
+    if isinstance(users, str):
+        users = [users]
+
+    with cursor() as cr:
+        for user in users:
+            cr.execute(REVOKE_PRIVILEGES_FROM_USER.format(user))
+            if provider_endpoint:
+                cr.execute(
+                    DISABLE_AI_PROFILE_DOMAIN_FOR_USER,
+                    user=user,
+                    host=provider_endpoint,
+                )

select_ai/synthetic_data.py

@@ -60,6 +60,11 @@ class SyntheticDataAttributes(SelectAIDataClass):
     record_count: Optional[int] = None
     user_prompt: Optional[str] = None
 
+    def __post_init__(self):
+        if self.params and not isinstance(self.params, SyntheticDataParams):
+            raise TypeError("'params' must be an object of"
+                            " type SyntheticDataParams'")
+
     def dict(self, exclude_null=True):
         attributes = {}
         for k, v in self.__dict__.items():

select_ai/vector_index.py

@@ -192,7 +192,7 @@ class VectorIndex(_BaseVectorIndex):
             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",
@@ -325,7 +325,10 @@ class VectorIndex(_BaseVectorIndex):
             )
             for row in cr.fetchall():
                 index_name = row[0]
-                description = row[1].read()  # Oracle.LOB
+                if row[1]:
+                    description = row[1].read()  # Oracle.LOB
+                else:
+                    description = None
                 attributes = cls._get_attributes(index_name=index_name)
                 yield cls(
                     index_name=index_name,
@@ -396,7 +399,7 @@ class AsyncVectorIndex(_BaseVectorIndex):
             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",
@@ -534,7 +537,10 @@ class AsyncVectorIndex(_BaseVectorIndex):
             rows = await cr.fetchall()
             for row in rows:
                 index_name = row[0]
-                description = await row[1].read()  # AsyncLOB
+                if row[1]:
+                    description = await row[1].read()  # AsyncLOB
+                else:
+                    description = None
                 attributes = await cls._get_attributes(index_name=index_name)
                 yield VectorIndex(
                     index_name=index_name,

select_ai/version.py

@@ -5,4 +5,4 @@
 # http://oss.oracle.com/licenses/upl.
 # -----------------------------------------------------------------------------
 
-__version__ = "1.0.0.dev7"
+__version__ = "1.0.0.dev9"

select_ai-1.0.0.dev9.dist-info/METADATA

@@ -0,0 +1,114 @@
+Metadata-Version: 2.4
+Name: select_ai
+Version: 1.0.0.dev9
+Summary: Select AI for Python
+Author-email: Abhishek Singh <abhishek.o.singh@oracle.com>
+Maintainer-email: Abhishek Singh <abhishek.o.singh@oracle.com>
+License-Expression: UPL-1.0
+Keywords: oracle,select-ai,adbs,autonomous database serverless
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Natural Language :: English
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Topic :: Database
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE.txt
+Requires-Dist: oracledb
+Requires-Dist: pandas==2.2.3
+Dynamic: license-file
+
+# Select AI for Python
+
+
+Select AI for Python enables you to ask questions of your database data using natural language (text-to-SQL), get generative AI responses using your trusted content (retrieval augmented generation), generate synthetic data using large language models, and other features – all from Python. With the general availability of Select AI Python, Python developers have access to the functionality of Select AI on Oracle Autonomous Database.
+
+Select AI for Python enables you to leverage the broader Python ecosystem in combination with generative AI and database functionality - bridging the gap between the DBMS_CLOUD_AI PL/SQL package and Python's rich ecosystem. It provides intuitive objects and methods for AI model interaction.
+
+
+## Installation
+
+Run
+```bash
+python3 -m pip install select_ai
+```
+
+## Samples
+
+Examples can be found in the [/samples][samples] directory
+
+### Basic Example
+
+```python
+import select_ai
+
+user = "<your_select_ai_user>"
+password = "<your_select_ai_password>"
+dsn = "<your_select_ai_db_connect_string>"
+
+select_ai.connect(user=user, password=password, dsn=dsn)
+profile = select_ai.Profile(profile_name="oci_ai_profile")
+# run_sql returns a pandas dataframe
+df = profile.run_sql(prompt="How many promotions?")
+print(df.columns)
+print(df)
+```
+
+### Async Example
+
+```python
+
+import asyncio
+
+import select_ai
+
+user = "<your_select_ai_user>"
+password = "<your_select_ai_password>"
+dsn = "<your_select_ai_db_connect_string>"
+
+# This example shows how to asynchronously run sql
+async def main():
+    await select_ai.async_connect(user=user, password=password, dsn=dsn)
+    async_profile = await select_ai.AsyncProfile(
+        profile_name="async_oci_ai_profile",
+    )
+    # run_sql returns a pandas df
+    df = await async_profile.run_sql("How many promotions?")
+    print(df)
+
+asyncio.run(main())
+
+```
+## Help
+
+Questions can be asked in [GitHub Discussions][ghdiscussions].
+
+Problem reports can be raised in [GitHub Issues][ghissues].
+
+## Contributing
+
+This project welcomes contributions from the community. Before submitting a pull request, please [review our contribution guide][contributing]
+
+## Security
+
+Please consult the [security guide][security] for our responsible security vulnerability disclosure process
+
+## License
+
+Copyright (c) 2025 Oracle and/or its affiliates.
+
+Released under the Universal Permissive License v1.0 as shown at
+<https://oss.oracle.com/licenses/upl/>.
+
+[contributing]: https://github.com/oracle/python-select-ai/blob/main/CONTRIBUTING.md
+[ghdiscussions]: https://github.com/oracle/python-select-ai/discussions
+[ghissues]: https://github.com/oracle/python-select-ai/issues
+[samples]: https://github.com/oracle/python-select-ai/tree/main/samples
+[security]: https://github.com/oracle/python-select-ai/blob/main/SECURITY.md

select_ai-1.0.0.dev9.dist-info/RECORD

@@ -0,0 +1,21 @@
+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/action.py,sha256=s78uhi5HHCLbTN2S5tLDSswTLUl40FZ3nESLka8CfZk,592
+select_ai/async_profile.py,sha256=3Ybsl-mKIxJDelIXcmGTBdzgBdkDacomTGMx0X0kSsw,18705
+select_ai/base_profile.py,sha256=hY9thC-7pHvMwh01W0t7OzOAyG8Fi8Lnd0KhZ7ouVNs,7005
+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=UH7aC9B5W2w4WmflX6N5FRWfSLwagXxgz62T9enpOgY,16059
+select_ai/provider.py,sha256=Sv0b-6SQJEoasPsnyHcWg1W8gS8nKldUW-83lL4MDm8,8285
+select_ai/sql.py,sha256=9yCdB1H_sYpjMLs-6SB8xf5-QpTwCAx4cIgIsElvly4,2786
+select_ai/synthetic_data.py,sha256=-l4eHLsTQRYTgJdsscszSWJlo9p75rLFMpH4XIodllE,3303
+select_ai/vector_index.py,sha256=C6a3jgZ6ffyel7fF5owHV_2G4T_E62pmGp1QXl28M4w,20475
+select_ai/version.py,sha256=LekbfzftsNILcAlsLq7T6_vw-Z7AVR2pVAHm9F3rPJU,348
+select_ai-1.0.0.dev9.dist-info/licenses/LICENSE.txt,sha256=_0VqOxSjO1hu6JexZDVzqUXSmzH1A53EOfyiJzXTBKc,1840
+select_ai-1.0.0.dev9.dist-info/METADATA,sha256=kmaLxaa2dvHdv5hSkETxFnBGPEm6v8ScH0huFPVnorQ,3857
+select_ai-1.0.0.dev9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+select_ai-1.0.0.dev9.dist-info/top_level.txt,sha256=u_QUAHDibro58Lvi_MU6a9Wc6VfQT8HEQm0cciMTP-c,10
+select_ai-1.0.0.dev9.dist-info/RECORD,,

select_ai/admin.py

@@ -1,116 +0,0 @@
-# -----------------------------------------------------------------------------
-# Copyright (c) 2025, Oracle and/or its affiliates.
-#
-# Licensed under the Universal Permissive License v 1.0 as shown at
-# http://oss.oracle.com/licenses/upl.
-# -----------------------------------------------------------------------------
-
-from typing import List, Mapping, Union
-
-import oracledb
-
-from .db import cursor
-from .sql import (
-    DISABLE_AI_PROFILE_DOMAIN_FOR_USER,
-    ENABLE_AI_PROFILE_DOMAIN_FOR_USER,
-    GRANT_PRIVILEGES_TO_USER,
-    REVOKE_PRIVILEGES_FROM_USER,
-)
-
-__all__ = [
-    "create_credential",
-    "disable_provider",
-    "enable_provider",
-]
-
-
-def create_credential(credential: Mapping, replace: bool = False):
-    """
-    Creates a credential object using DBMS_CLOUD.CREATE_CREDENTIAL
-
-    if replace is True, credential will be replaced if it "already exists"
-
-    """
-    valid_keys = {
-        "credential_name",
-        "username",
-        "password",
-        "user_ocid",
-        "tenancy_ocid",
-        "private_key",
-        "fingerprint",
-        "comments",
-    }
-    for k in credential.keys():
-        if k.lower() not in valid_keys:
-            raise ValueError(
-                f"Invalid value {k}: {credential[k]} for credential object"
-            )
-
-    with cursor() as cr:
-        try:
-            cr.callproc(
-                "DBMS_CLOUD.CREATE_CREDENTIAL", keyword_parameters=credential
-            )
-        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:
-                cr.callproc(
-                    "DBMS_CLOUD.DROP_CREDENTIAL",
-                    keyword_parameters={
-                        "credential_name": credential["credential_name"]
-                    },
-                )
-                cr.callproc(
-                    "DBMS_CLOUD.CREATE_CREDENTIAL",
-                    keyword_parameters=credential,
-                )
-            else:
-                raise
-
-
-def enable_provider(
-    users: Union[str, List[str]], provider_endpoint: str = None
-):
-    """
-    Enables AI profile for the user. This method grants execute privilege
-    on the packages DBMS_CLOUD, DBMS_CLOUD_AI and DBMS_CLOUD_PIPELINE. It
-    also enables the user to invoke the AI(LLM) endpoint hosted at a
-    certain domain
-    """
-    if isinstance(users, str):
-        users = [users]
-
-    with cursor() as cr:
-        for user in users:
-            cr.execute(GRANT_PRIVILEGES_TO_USER.format(user))
-            if provider_endpoint:
-                cr.execute(
-                    ENABLE_AI_PROFILE_DOMAIN_FOR_USER,
-                    user=user,
-                    host=provider_endpoint,
-                )
-
-
-def disable_provider(
-    users: Union[str, List[str]], provider_endpoint: str = None
-):
-    """
-    Disables AI provider for the user. This method revokes execute privilege
-    on the packages DBMS_CLOUD, DBMS_CLOUD_AI and DBMS_CLOUD_PIPELINE. It
-    also disables the user to invoke the AI(LLM) endpoint hosted at a
-    certain domain
-    """
-    if isinstance(users, str):
-        users = [users]
-
-    with cursor() as cr:
-        for user in users:
-            cr.execute(REVOKE_PRIVILEGES_FROM_USER.format(user))
-            if provider_endpoint:
-                cr.execute(
-                    DISABLE_AI_PROFILE_DOMAIN_FOR_USER,
-                    user=user,
-                    host=provider_endpoint,
-                )

select_ai-1.0.0.dev7.dist-info/METADATA

@@ -1,25 +0,0 @@
-Metadata-Version: 2.4
-Name: select_ai
-Version: 1.0.0.dev7
-Summary: Python API for Select AI
-Author-email: Abhishek Singh <abhishek.o.singh@oracle.com>
-Maintainer-email: Abhishek Singh <abhishek.o.singh@oracle.com>
-License-Expression: UPL-1.0
-Keywords: oracle,select-ai,adbs,autonomous database serverless
-Classifier: Development Status :: 3 - Alpha
-Classifier: Intended Audience :: Developers
-Classifier: Natural Language :: English
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: Implementation :: CPython
-Classifier: Topic :: Database
-Requires-Python: >=3.9
-License-File: LICENSE.txt
-Requires-Dist: oracledb
-Requires-Dist: pandas==2.2.3
-Dynamic: license-file

select_ai-1.0.0.dev7.dist-info/RECORD

@@ -1,21 +0,0 @@
-select_ai/__init__.py,sha256=LVdOh_JnWEXaXs0NSflA1f0YIWCUEpTBZCvWNTvSS74,1319
-select_ai/_abc.py,sha256=ccec6yu54w5QMix_EKc_BJ_0JatUDXKjfLW9KshOGak,2692
-select_ai/_enums.py,sha256=U0UavuE4pbRY-0-Qb8z-F7KfxU0YUfOgUahaSjDi0dU,416
-select_ai/action.py,sha256=s78uhi5HHCLbTN2S5tLDSswTLUl40FZ3nESLka8CfZk,592
-select_ai/admin.py,sha256=MMkXpvVJZUoloty8HiQnndPVPpuZch22Pnk8llUhUCc,3528
-select_ai/async_profile.py,sha256=7fbEW4rRLy3FcT8E7fabXwlLja-echoWEhlQHQaBCOM,17651
-select_ai/base_profile.py,sha256=Iyd5GdBNqEISiWCwYpGSKOGd394f-A5QgHPtuJ547CE,6791
-select_ai/conversation.py,sha256=jeZy7qtbnFTO9xTE3gNZLyxVI0Pnj_rpc0QZb1r-A00,9240
-select_ai/db.py,sha256=Q0MvoIBzE0fHP1d9oFjvnjfgKqdRk6ZCM6bERokcM3E,4669
-select_ai/errors.py,sha256=2T5eICWWjj7a907aRh7es5wTcvWm_wJiP8h0b3ipxVQ,2114
-select_ai/profile.py,sha256=t9TMAqdaVm9DMz-jdP6AZddce6s9XYhlB_HOrfaq5NM,14926
-select_ai/provider.py,sha256=7KNdU2BmUMzjPj2WAwtn0vEYPBYY83MvRTT1q2Ng5tM,5103
-select_ai/sql.py,sha256=9yCdB1H_sYpjMLs-6SB8xf5-QpTwCAx4cIgIsElvly4,2786
-select_ai/synthetic_data.py,sha256=lnjEIcUzr7lr2dO24OgxRhol-tiVOiKK-kEvFVJoiig,3078
-select_ai/vector_index.py,sha256=0-0xkXWALPm6RqIUoY3r-NEisHO3Ya3lKkIIRed0a_4,20335
-select_ai/version.py,sha256=baI0umotUCS0Xec8hkkCm0zLQafNjA_g3EUrE1upAAU,348
-select_ai-1.0.0.dev7.dist-info/licenses/LICENSE.txt,sha256=_0VqOxSjO1hu6JexZDVzqUXSmzH1A53EOfyiJzXTBKc,1840
-select_ai-1.0.0.dev7.dist-info/METADATA,sha256=B0UgNOCBaYZmtLyC1_5O_-ibVST5j_VEkmFCffliiCs,1016
-select_ai-1.0.0.dev7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-select_ai-1.0.0.dev7.dist-info/top_level.txt,sha256=u_QUAHDibro58Lvi_MU6a9Wc6VfQT8HEQm0cciMTP-c,10
-select_ai-1.0.0.dev7.dist-info/RECORD,,