# mcp/workspace.py

import streamlit as st

def get_workspace():
    # Use Streamlit session state for per-user saved searches
    if "saved_queries" not in st.session_state:
        st.session_state["saved_queries"] = []
    return st.session_state["saved_queries"]

def save_query(query, result):
    ws = get_workspace()
    # Only keep the last 10 searches for simplicity
    ws.insert(0, {"query": query, "result": result})
    ws[:] = ws[:10]  # Cap at 10