Disable SelectAI

Author: gotsysdba

src/client/content/config/databases.py

@@ -15,6 +15,7 @@
 import streamlit as st
 from streamlit import session_state as state
 
+from client.content.config.oci import get_oci
 import client.utils.api_call as api_call
 import common.logging_config as logging_config
 
@@ -187,7 +188,10 @@ def main() -> None:
         # Select AI
         #############################################
         st.subheader("SelectAI", divider="red")
-        if state.database_config[name]["selectai"]:
+        if "oci_config" not in state.user_settings:
+            get_oci()
+        oci_auth_profile = state.user_settings["oci"]["auth_profile"]
+        if state.database_config[name]["selectai"] and state.oci_config[oci_auth_profile]["namespace"] :
             st.write("Tables eligible and enabled/disabled for SelectAI.")
             if state.database_config[name]["selectai_objects"]:
                 df = pd.DataFrame(

src/client/utils/st_common.py

@@ -15,6 +15,7 @@
 
 from client.content.config.models import get_models
 from client.content.config.databases import get_databases
+from client.content.config.oci import get_oci
 import client.utils.api_call as api_call
 
 import common.help_text as help_text
@@ -253,6 +254,14 @@ def update_settings():
             ("VectorSearch", "Use AI with Unstructured Data", disable_vector_search),
         ]
 
+        # SelectAI Requirements
+        if "oci_config" not in state.user_settings:
+            get_oci()
+        oci_auth_profile = state.user_settings["oci"]["auth_profile"]
+        if not state.oci_config[oci_auth_profile]["namespace"]:
+            logger.debug("SelectAI Disabled (OCI not configured.)")
+            st.warning("OCI is not fully configured.  Disabling SelectAI.", icon="⚠️")
+            tools = [t for t in tools if t[0] != "SelectAI"]
         # Vector Search Requirements
         get_models(model_type="embed")
         available_embed_models = list(state.embed_model_enabled.keys())

src/server/agents/chatbot.py

@@ -422,8 +422,5 @@ async def generate_response(state: AgentState, config: RunnableConfig) -> AgentS
 memory = MemorySaver()
 chatbot_graph = workflow.compile(checkpointer=memory)
 
-## This will write a graph.png file of the LangGraph; don't deliver uncommented
-from IPython.display import Image, display
-
-# display(Image(chatbot_graph.get_graph(xray=True).draw_mermaid_png(output_file_path="chatbot_graph.png")))
-chatbot_graph.get_graph(xray=True).print_ascii()
+## This will output the Graph in ascii; don't deliver uncommented
+# chatbot_graph.get_graph(xray=True).print_ascii()