半成品,为了保存记录,请勿使用。
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
from dotenv import load_dotenv
|
||||
|
||||
# 加载.env文件
|
||||
load_dotenv()
|
||||
|
||||
@@ -108,48 +109,6 @@ def rename_session_widget(agent: Agent) -> None:
|
||||
st.session_state.session_edit_mode = False
|
||||
st.rerun()
|
||||
|
||||
|
||||
def session_selector_widget(agent: Agent, model_id: str) -> None:
|
||||
"""Display a session selector in the sidebar"""
|
||||
|
||||
if agent.storage:
|
||||
agent_sessions = agent.storage.get_all_sessions()
|
||||
# Get session names if available, otherwise use IDs
|
||||
session_options = []
|
||||
for session in agent_sessions:
|
||||
session_id = session.session_id
|
||||
session_name = (
|
||||
session.session_data.get("session_name", None)
|
||||
if session.session_data
|
||||
else None
|
||||
)
|
||||
display_name = session_name if session_name else session_id
|
||||
session_options.append({"id": session_id, "display": display_name})
|
||||
|
||||
# Display session selector
|
||||
#selected_session = st.sidebar.selectbox(
|
||||
# "会话",
|
||||
# options=[s["display"] for s in session_options],
|
||||
# key="session_selector",
|
||||
#)
|
||||
# Find the selected session ID
|
||||
#selected_session_id = next(
|
||||
# s["id"] for s in session_options if s["display"] == selected_session
|
||||
#)
|
||||
if len(session_options) > 0:
|
||||
selected_session_id = session_options[0]["id"]
|
||||
|
||||
if st.session_state["agentic_rag_agent_session_id"] != selected_session_id:
|
||||
logger.info(
|
||||
f"---*--- Loading {model_id} run: {selected_session_id} ---*---"
|
||||
)
|
||||
st.session_state["agentic_rag_agent"] = get_agentic_rag_agent(
|
||||
model_id=model_id,
|
||||
session_id=selected_session_id,
|
||||
)
|
||||
st.rerun()
|
||||
|
||||
|
||||
def about_widget() -> None:
|
||||
"""Display an about section in the sidebar"""
|
||||
st.sidebar.markdown("---")
|
||||
|
||||
Reference in New Issue
Block a user