select-ai 1.0.0.dev4__tar.gz

select_ai-1.0.0.dev4/LICENSE.txt

@@ -0,0 +1,35 @@
+Copyright (c) 2025, Oracle and/or its affiliates.
+
+The Universal Permissive License (UPL), Version 1.0
+
+Subject to the condition set forth below, permission is hereby granted to any
+person obtaining a copy of this software, associated documentation and/or data
+(collectively the "Software"), free of charge and under any and all copyright
+rights in the Software, and any and all patent rights owned or freely
+licensable by each licensor hereunder covering either (i) the unmodified
+Software as contributed to or provided by such licensor, or (ii) the Larger
+Works (as defined below), to deal in both
+
+(a) the Software, and
+(b) any piece of software and/or hardware listed in the lrgrwrks.txt file if
+one is included with the Software (each a "Larger Work" to which the Software
+is contributed by such licensors),
+
+without restriction, including without limitation the rights to copy, create
+derivative works of, display, perform, and distribute the Software and make,
+use, sell, offer for sale, import, export, have made, and have sold the
+Software and the Larger Work(s), and to sublicense the foregoing rights on
+either these or other terms.
+
+This license is subject to the following condition:
+The above copyright notice and either this complete permission notice or at
+a minimum a reference to the UPL must be included in all copies or
+substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

select_ai-1.0.0.dev4/MANIFEST.in

@@ -0,0 +1,2 @@
+recursive-include tests *.py
+recursive-include samples *.py

select_ai-1.0.0.dev4/PKG-INFO

@@ -0,0 +1,25 @@
+Metadata-Version: 2.4
+Name: select_ai
+Version: 1.0.0.dev4
+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.dev4/README.md

@@ -0,0 +1,14 @@
+# select-ai
+
+Python API for Select AI
+
+## Install
+
+Run
+```bash
+python3 -m pip install select_ai
+```
+
+## Samples
+
+Examples can be found in the samples directory

select_ai-1.0.0.dev4/pyproject.toml

@@ -0,0 +1,57 @@
+[build-system]
+requires = ["setuptools >= 77.0.3", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "select_ai"
+dynamic = ["version"]
+description = "Python API for Select AI"
+requires-python = ">=3.9"
+authors = [
+    {name="Abhishek Singh", email="abhishek.o.singh@oracle.com"}
+]
+maintainers =[
+    {name="Abhishek Singh", email="abhishek.o.singh@oracle.com"}
+]
+keywords = [
+	"oracle",
+	"select-ai",
+    "adbs",
+    "autonomous database serverless"
+]
+license = " UPL-1.0"
+license-files = ["LICENSE.txt"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "Natural Language :: English",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: Implementation :: CPython",
+    "Topic :: Database"
+]
+dependencies = [
+    "oracledb",
+    "pandas==2.2.3"
+]
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.dynamic]
+version = { attr = "select_ai.version.__version__" }
+
+[tool.black]
+line-length = 79
+target-version = ["py39", "py310", "py311", "py312", "py313"]
+required-version = 24
+
+[tool.ruff]
+line-length = 79
+target-version = "py39"
+per-file-ignores = { "__init__.py" = ["F401"] }

select_ai-1.0.0.dev4/samples/async_samples/conversation_chat_session.py

@@ -0,0 +1,45 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+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"
+    )
+    conversation_attributes = select_ai.ConversationAttributes(
+        title="History of Science",
+        description="LLM's understanding of history of science",
+    )
+    async_conversation = select_ai.AsyncConversation(
+        attributes=conversation_attributes
+    )
+
+    async with async_profile.chat_session(
+        conversation=async_conversation, delete=True
+    ) as async_session:
+        response = await async_session.chat(
+            prompt="What is importance of history of science ?"
+        )
+        print(response)
+        response = await async_session.chat(
+            prompt="Elaborate more on 'Learning from past mistakes'"
+        )
+        print(response)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/conversations_list.py

@@ -0,0 +1,26 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+async def main():
+    await select_ai.async_connect(user=user, password=password, dsn=dsn)
+    async for conversation in select_ai.AsyncConversation().list():
+        print(conversation.conversation_id)
+        print(conversation.attributes)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/profile_chat.py

@@ -0,0 +1,36 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+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"
+    )
+
+    # Asynchronously send multiple prompts
+    chat_tasks = [
+        async_profile.chat(prompt="What is OCI ?"),
+        async_profile.chat(prompt="What is OML4PY?"),
+        async_profile.chat(prompt="What is Autonomous Database ?"),
+    ]
+    for chat_task in asyncio.as_completed(chat_tasks):
+        result = await chat_task
+        print(result)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/profile_create.py

@@ -0,0 +1,47 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+from pprint import pformat
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+# This example shows how to asynchronously generate SQLs nad run SQLs
+async def main():
+    await select_ai.async_connect(user=user, password=password, dsn=dsn)
+    provider = select_ai.OCIGenAIProvider(
+        region="us-chicago-1", oci_apiformat="GENERIC"
+    )
+    profile_attributes = select_ai.ProfileAttributes(
+        credential_name="my_oci_ai_profile_key",
+        object_list=[{"owner": "SH"}],
+        provider=provider,
+    )
+    async_profile = await select_ai.AsyncProfile(
+        profile_name="async_oci_ai_profile",
+        attributes=profile_attributes,
+        description="MY OCI AI Profile",
+        replace=True,
+    )
+
+    profile_attributes = await async_profile.get_attributes()
+
+    print("Profile attributes are: ", profile_attributes)
+    print(
+        "Profile attributes as Python dict: ",
+        pformat(profile_attributes.dict(exclude_null=False)),
+    )
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/profile_explain_sql.py

@@ -0,0 +1,29 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+# This example shows how to asynchronously ask the LLM to explain 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",
+    )
+    response = await async_profile.explain_sql("How many promotions")
+    print(response)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/profile_pipeline.py

@@ -0,0 +1,49 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+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"
+    )
+    prompt_specifications = [
+        ("What is Oracle Autonomous Database?", select_ai.Action.CHAT),
+        ("Generate SQL to list all customers?", select_ai.Action.SHOWSQL),
+        (
+            "Explain the query: SELECT * FROM sh.products",
+            select_ai.Action.EXPLAINSQL,
+        ),
+        ("Explain the query: SELECT * FROM sh.products", "INVALID ACTION"),
+    ]
+
+    # 1. Multiple prompts are sent in a single roundtrip to the Database
+    # 2. Results are returned as soon as Database has executed all prompts
+    # 3. Application doesn't have to wait on one response before sending
+    #    the next prompts
+    # 4. Fewer round trips and database is kept busy
+    # 5. Efficient network usage
+    results = await async_profile.run_pipeline(
+        prompt_specifications, continue_on_error=True
+    )
+    for i, result in enumerate(results):
+        print(
+            f"Result {i} for prompt '{prompt_specifications[i][0]}' is: {result}"
+        )
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/profile_run_sql.py

@@ -0,0 +1,30 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("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)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/profile_show_sql.py

@@ -0,0 +1,28 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+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",
+    )
+    response = await async_profile.show_sql("How many promotions")
+    print(response)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/profile_sql_concurrent_tasks.py

@@ -0,0 +1,36 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+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",
+    )
+    sql_tasks = [
+        async_profile.show_sql(prompt="How many customers?"),
+        async_profile.run_sql(prompt="How many promotions"),
+        async_profile.explain_sql(prompt="How many promotions"),
+    ]
+
+    # Collect results from multiple asynchronous tasks
+    for sql_task in asyncio.as_completed(sql_tasks):
+        result = await sql_task
+        print(result)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/profiles_list.py

@@ -0,0 +1,30 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+async def main():
+    await select_ai.async_connect(user=user, password=password, dsn=dsn)
+    async_profile = await select_ai.AsyncProfile()
+    # matches the start of string
+    async for fetched_profile in async_profile.list(
+        profile_name_pattern="^oci"
+    ):
+        p = await fetched_profile
+        print(p.profile_name)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/vector_index_create.py

@@ -0,0 +1,51 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+async def main():
+    await select_ai.async_connect(user=user, password=password, dsn=dsn)
+
+    provider = select_ai.OCIGenAIProvider(
+        region="us-chicago-1", oci_apiformat="GENERIC"
+    )
+    profile_attributes = select_ai.ProfileAttributes(
+        credential_name="my_oci_ai_profile_key",
+        provider=provider,
+    )
+    async_profile = await select_ai.AsyncProfile(
+        profile_name="async_oci_vector_ai_profile",
+        attributes=profile_attributes,
+        description="MY OCI AI Profile",
+        replace=True,
+    )
+
+    vector_index_attributes = select_ai.OracleVectorIndexAttributes(
+        location="https://objectstorage.us-ashburn-1.oraclecloud.com/n/dwcsdev/b/conda-environment/o/tenant1-pdb3/graph",
+        object_storage_credential_name="my_oci_ai_profile_key",
+    )
+
+    async_vector_index = select_ai.AsyncVectorIndex(
+        index_name="test_vector_index",
+        attributes=vector_index_attributes,
+        description="Vector index for conda environments",
+        profile=async_profile,
+    )
+    await async_vector_index.create(replace=True)
+    print("created vector index: test_vector_index")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/async_samples/vector_index_list.py

@@ -0,0 +1,27 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import asyncio
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+async def main():
+    await select_ai.async_connect(user=user, password=password, dsn=dsn)
+    vector_index = select_ai.AsyncVectorIndex()
+    async for index in vector_index.list(index_name_pattern="^test"):
+        print("Vector index", index.index_name)
+        print("Vector index profile", index.profile)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

select_ai-1.0.0.dev4/samples/conversation_chat_session.py

@@ -0,0 +1,46 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+# This example shows how to have a context-aware
+# conversation
+
+
+def main():
+    select_ai.connect(user=user, password=password, dsn=dsn)
+    profile = select_ai.Profile(profile_name="oci_ai_profile")
+    conversation_attributes = select_ai.ConversationAttributes(
+        title="History of Science",
+        description="LLM's understanding of history of science",
+    )
+    conversation = select_ai.Conversation(attributes=conversation_attributes)
+    with profile.chat_session(
+        conversation=conversation, delete=True
+    ) as session:
+        print(
+            "Conversation ID for this session is:",
+            conversation.conversation_id,
+        )
+        response = session.chat(
+            prompt="What is importance of history of science ?"
+        )
+        print(response)
+        response = session.chat(
+            prompt="Elaborate more on 'Learning from past mistakes'"
+        )
+        print(response)
+
+
+if __name__ == "__main__":
+    main()

select_ai-1.0.0.dev4/samples/conversation_create.py

@@ -0,0 +1,35 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import os
+
+import select_ai
+
+user = os.getenv("SELECT_AI_USER")
+password = os.getenv("SELECT_AI_PASSWORD")
+dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
+
+
+# This example shows how to create a conversation
+def main():
+    select_ai.connect(user=user, password=password, dsn=dsn)
+    conversation_attributes = select_ai.ConversationAttributes(
+        title="History of Science",
+        description="LLM's understanding of history of science",
+    )
+
+    conversation = select_ai.Conversation(attributes=conversation_attributes)
+    conversation.create()
+
+    print(
+        "Created conversation with conversation id: ",
+        conversation.conversation_id,
+    )
+
+
+if __name__ == "__main__":
+    main()