gen: (GPT-5-Codex) #file:chat_with_tools_agent.py に sidebar から checkpoint_type を選択できる UI を追加してください。現在サポートされている sqlite, cosmosdb, memory についても enum で型定義してください。

Author: ks6088ts

template_langgraph/services/streamlits/pages/chat_with_tools_agent.py

@@ -4,6 +4,7 @@
 import uuid
 from base64 import b64encode
 from dataclasses import dataclass
+from enum import Enum
 
 import streamlit as st
 from audio_recorder_streamlit import audio_recorder
@@ -24,7 +25,23 @@
 from template_langgraph.speeches.tts import TtsWrapper
 from template_langgraph.tools.common import get_default_tools
 
-checkpoint_type = "cosmosdb"  # "cosmosdb" or "sqlite"
+
+class CheckpointType(str, Enum):
+    SQLITE = "sqlite"
+    COSMOSDB = "cosmosdb"
+    MEMORY = "memory"
+    NONE = "none"
+
+
+DEFAULT_CHECKPOINT_TYPE = CheckpointType.NONE
+CHECKPOINT_LABELS = {
+    CheckpointType.COSMOSDB.value: "Cosmos DB",
+    CheckpointType.SQLITE.value: "SQLite",
+    CheckpointType.MEMORY.value: "メモリ",
+    CheckpointType.NONE.value: "なし",
+}
+
+
 store_conn = sqlite3.connect("store.sqlite", check_same_thread=False)
 thread_id = str(uuid.uuid4())
 
@@ -72,15 +89,25 @@ def ensure_session_state_defaults(tool_names: list[str]) -> None:
     st.session_state.setdefault("chat_history", [])
     st.session_state.setdefault("input_output_mode", "テキスト")
     st.session_state.setdefault("selected_tool_names", tool_names)
+    st.session_state.setdefault("checkpoint_type", DEFAULT_CHECKPOINT_TYPE.value)
+
+
+def get_selected_checkpoint_type() -> CheckpointType:
+    raw_value = st.session_state.get("checkpoint_type", DEFAULT_CHECKPOINT_TYPE.value)
+    try:
+        checkpoint = CheckpointType(raw_value)
+    except ValueError:
+        st.session_state["checkpoint_type"] = DEFAULT_CHECKPOINT_TYPE.value
+        return DEFAULT_CHECKPOINT_TYPE
+    return checkpoint
 
 
 def get_checkpointer():
-    if checkpoint_type == "sqlite":
+    checkpoint_type = get_selected_checkpoint_type()
+    if checkpoint_type is CheckpointType.SQLITE:
         conn = sqlite3.connect("checkpoints.sqlite", check_same_thread=False)
         return SqliteSaver(conn=conn)
-    if checkpoint_type == "cosmosdb":
-        import os
-
+    if checkpoint_type is CheckpointType.COSMOSDB:
         from template_langgraph.tools.cosmosdb_tool import get_cosmosdb_settings
 
         settings = get_cosmosdb_settings()
@@ -91,13 +118,13 @@ def get_checkpointer():
             database_name=settings.cosmosdb_database_name,
             container_name="checkpoints",
         )
-    if checkpoint_type == "memory":
+    if checkpoint_type is CheckpointType.MEMORY:
         return InMemorySaver()
     return None
 
 
 def ensure_agent_graph(selected_tools: list) -> None:
-    signature = tuple(tool.name for tool in selected_tools)
+    signature = (tuple(tool.name for tool in selected_tools), get_selected_checkpoint_type().value)
     graph_signature = st.session_state.get("graph_tools_signature")
     if "graph" not in st.session_state or graph_signature != signature:
         st.session_state["graph"] = ChatWithToolsAgent(
@@ -133,6 +160,23 @@ def build_sidebar() -> tuple[str, AudioSettings | None]:
         if input_mode == "音声":
             audio_settings = render_audio_controls()
 
+        st.divider()
+        st.subheader("チェックポイント")
+
+        checkpoint_options = [checkpoint.value for checkpoint in CheckpointType]
+        current_checkpoint_value = st.session_state["checkpoint_type"]
+        if current_checkpoint_value not in checkpoint_options:
+            current_checkpoint_value = DEFAULT_CHECKPOINT_TYPE.value
+            st.session_state["checkpoint_type"] = current_checkpoint_value
+        checkpoint_index = checkpoint_options.index(current_checkpoint_value)
+        selected_checkpoint_value = st.selectbox(
+            "保存方法",
+            options=checkpoint_options,
+            index=checkpoint_index,
+            format_func=lambda value: CHECKPOINT_LABELS.get(value, value),
+        )
+        st.session_state["checkpoint_type"] = selected_checkpoint_value
+
         st.divider()
         st.subheader("使用するツール")