半成品,为了保存记录,请勿使用。

This commit is contained in:
2025-04-09 20:44:26 +08:00
parent b6b697efdb
commit 211db332c0
3 changed files with 320 additions and 52 deletions
+1 -42
View File
@@ -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("---")