Skip to content

Fix when last VS is dropped and VS should be disabled #292

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Oct 8, 2025
Merged

Fix when last VS is dropped and VS should be disabled #292

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 14 additions & 9 deletions src/client/utils/st_common.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -253,7 +253,7 @@ def ll_sidebar() -> None:
def tools_sidebar() -> None:
"""SelectAI Sidebar Settings, conditional if all sorts of bs setup"""

def update_set_tool():
def _update_set_tool():
"""Update user settings as to which tool is being used"""
state.client_settings["vector_search"]["enabled"] = state.selected_tool == "Vector Search"
state.client_settings["selectai"]["enabled"] = state.selected_tool == "SelectAI"
Expand Down Expand Up @@ -298,14 +298,19 @@ def update_set_tool():

# Vector Search Requirements
embed_models_enabled = enabled_models_lookup("embed")

def _disable_vector_search(reason):
"""Disable Vector Store, and make sure prompt is reset"""
state.client_settings["vector_search"]["enabled"] = False
logger.debug("Vector Search Disabled (%s)", reason)
st.warning(f"{reason}. Disabling Vector Search.", icon="⚠️")
tools[:] = [t for t in tools if t[0] != "Vector Search"]
switch_prompt("sys", "Basic Example")

if not embed_models_enabled:
logger.debug("Vector Search Disabled (no Embedding Models)")
st.warning("No embedding models are configured and/or enabled. Disabling Vector Search.", icon="⚠️")
tools = [t for t in tools if t[0] != "Vector Search"]
_disable_vector_search("No embedding models are configured and/or enabled.")
elif not database_lookup[db_alias].get("vector_stores"):
logger.debug("Vector Search Disabled (Database has no vector stores.)")
st.warning("Database has no Vector Stores. Disabling Vector Search.", icon="⚠️")
tools = [t for t in tools if t[0] != "Vector Search"]
_disable_vector_search("Database has no vector stores")

tool_box = [name for name, _, disabled in tools if not disabled]
if len(tool_box) > 1:
Expand All @@ -324,10 +329,10 @@ def update_set_tool():
tool_box,
index=tool_index,
label_visibility="collapsed",
on_change=update_set_tool,
on_change=_update_set_tool,
key="selected_tool",
)
if state.selected_tool == "None":
if state.selected_tool is None:
switch_prompt("sys", "Basic Example")


Expand Down
Loading