Merge branch 'refactor/api_design' into 'main'

refactoring of APIs

See merge request adbs/python-select-ai!1

Author: aosingh

examples/3_show_sql.py

@@ -9,15 +9,20 @@
 
 def main():
     select_ai.connect(user=user, password=password, dsn=dsn)
-    oci_provider_attributes = select_ai.OCIGenAIProviderAttributes(
-        region="us-chicago-1",
+
+    provider = select_ai.OCIGenAIProvider(
+        region="us-chicago-1", oci_apiformat="GENERIC"
+    )
+
+    profile_attributes = select_ai.ProfileAttributes(
         credential_name="my_oci_ai_profile_key",
-        oci_apiformat="GENERIC",
         object_list=[{"owner": "SH"}],
+        provider=provider,
     )
+
     profile = select_ai.Profile(
         profile_name="oci_ai_profile",
-        attributes=oci_provider_attributes,
+        attributes=profile_attributes,
         description="MY OCI AI Profile",
         replace=True,
     )

examples/4_run_sql.py

@@ -9,22 +9,11 @@
 
 def main():
     select_ai.connect(user=user, password=password, dsn=dsn)
-    oci_provider_attributes = select_ai.OCIGenAIProviderAttributes(
-        region="us-chicago-1",
-        credential_name="my_oci_ai_profile_key",
-        oci_apiformat="GENERIC",
-        object_list=[{"owner": "SH"}],
-    )
-    profile = select_ai.Profile(
-        profile_name="oci_ai_profile",
-        attributes=oci_provider_attributes,
-        description="MY OCI AI Profile",
-        replace=True,
-    )
-    # profile.set_attributes(attributes=oci_provider_attributes)
+    profile = select_ai.Profile(profile_name="oci_ai_profile")
     profile.set_attribute(
         attribute_name="model", attribute_value="meta.llama-3.1-70b-instruct"
     )
+    print(profile)
     prompts = [
         "How many promotions are there in the sh database?",
         "How many products are there in the sh database ?",

examples/5_chat.py

@@ -9,15 +9,7 @@
 
 def main():
     select_ai.connect(user=user, password=password, dsn=dsn)
-    oci_provider_attributes = select_ai.OCIGenAIProviderAttributes(
-        model="meta.llama-3.1-70b-instruct"
-    )
-    profile = select_ai.Profile(
-        profile_name="oci_ai_profile",
-        attributes=oci_provider_attributes,
-        description="MY OCI AI Profile",
-        fetch_and_merge_attributes=True,
-    )
+    profile = select_ai.Profile(profile_name="oci_ai_profile")
     print(profile)
     print(profile.chat(prompt="What is OCI ?"))
 

examples/7_vector_index.py

@@ -9,9 +9,7 @@
 
 def main():
     select_ai.connect(user=user, password=password, dsn=dsn)
-    profile = select_ai.Profile(
-        profile_name="oci_ai_profile", fetch_and_merge_attributes=True
-    )
+    profile = select_ai.Profile(profile_name="oci_ai_profile", merge=True)
     print("fetched profile: ", profile)
     vector_index_attributes = select_ai.VectorIndexAttributes(
         location="https://objectstorage.us-ashburn-1.oraclecloud.com/n/dwcsdev/b/conda-environment/o/tenant1-pdb3/graph",

examples/8_generate_synthetic_data.py

@@ -7,17 +7,10 @@
 dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
 
 
+# This example show how to generate multi-table synthetic data
 def main():
     select_ai.connect(user=user, password=password, dsn=dsn)
-    oci_provider_attributes = select_ai.OCIGenAIProviderAttributes(
-        model="meta.llama-3.1-70b-instruct"
-    )
-    profile = select_ai.Profile(
-        profile_name="oci_ai_profile",
-        attributes=oci_provider_attributes,
-        description="MY OCI AI Profile",
-        fetch_and_merge_attributes=True,
-    )
+    profile = select_ai.Profile(profile_name="oci_ai_profile", merge=True)
     synthetic_data_params = select_ai.SyntheticDataParams(
         sample_rows=100, table_statistics=True, priority="HIGH"
     )

examples/9_conversation.py

@@ -6,13 +6,13 @@
 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",
-        fetch_and_merge_attributes=True,
-    )
+    profile = select_ai.Profile(profile_name="oci_ai_profile", merge=True)
     conversation_attributes = select_ai.ConversationAttributes(
         title="History of Science",
         description="LLM's understanding of history of science",

examples/async_examples/1_sql.py

@@ -8,19 +8,23 @@
 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)
 
-    oci_provider_attributes = select_ai.OCIGenAIProviderAttributes(
-        region="us-chicago-1",
+    provider = select_ai.OCIGenAIProvider(
+        region="us-chicago-1", oci_apiformat="GENERIC"
+    )
+
+    profile_attributes = select_ai.ProfileAttributes(
         credential_name="my_oci_ai_profile_key",
-        oci_apiformat="GENERIC",
         object_list=[{"owner": "SH"}],
+        provider=provider,
     )
 
     async_profile = await select_ai.AsyncProfile(
         profile_name="async_oci_ai_profile",
-        attributes=oci_provider_attributes,
+        attributes=profile_attributes,
         description="MY OCI AI Profile",
         replace=True,
     )

examples/async_examples/2_chat.py

@@ -10,16 +10,9 @@
 
 async def main():
     await select_ai.async_connect(user=user, password=password, dsn=dsn)
-    oci_provider_attributes = select_ai.OCIGenAIProviderAttributes(
-        model="meta.llama-3.1-70b-instruct"
-    )
     async_profile = await select_ai.AsyncProfile(
-        profile_name="async_oci_ai_profile",
-        attributes=oci_provider_attributes,
-        description="MY OCI AI Profile",
-        fetch_and_merge_attributes=True,
+        profile_name="async_oci_ai_profile"
     )
-
     # Asynchronously send multiple prompts
     chat_tasks = [
         async_profile.chat(prompt="What is OCI ?"),

examples/async_examples/3_pipeline.py

@@ -11,8 +11,7 @@
 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",
-        fetch_and_merge_attributes=True,
+        profile_name="async_oci_ai_profile"
     )
     prompt_specifications = [
         ("What is Oracle Autonomous Database?", select_ai.Action.CHAT),

examples/async_examples/4_conversation.py

@@ -11,8 +11,7 @@
 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",
-        fetch_and_merge_attributes=True,
+        profile_name="async_oci_ai_profile"
     )
     conversation_attributes = select_ai.ConversationAttributes(
         title="History of Science",