feat(session_store): add asynchronous and synchronous methods for initializing ADK tables

- Introduced `init_adk_tables_async` and `init_adk_tables` methods to create necessary tables and indexes for Google ADK.
- Updated example server initialization to reflect new ADK table setup.
- Cleaned up comments and improved code formatting for better readability.

This enhancement supports the integration of Google ADK within the conversation service.

Author: TJKkking

agentrun/conversation_service/__session_store_async_template.py

@@ -84,6 +84,17 @@ async def init_langgraph_tables_async(self) -> None:
         await self._backend.init_search_index_async()
         await self._backend.init_checkpoint_tables_async()
 
+    async def init_adk_tables_async(self) -> None:
+        """创建 Google ADK 所需的全部表和索引（异步）。
+
+        包含核心表（Conversation + Event + 二级索引）、三级 State 表
+        （state / app_state / user_state）以及多元索引。
+        表或索引已存在时跳过，可重复调用。
+        """
+        await self._backend.init_core_tables_async()
+        await self._backend.init_state_tables_async()
+        await self._backend.init_search_index_async()
+
     # -------------------------------------------------------------------
     # Checkpoint 管理（LangGraph）（异步）
     # -------------------------------------------------------------------

agentrun/conversation_service/session_store.py

@@ -133,10 +133,6 @@ async def init_langgraph_tables_async(self) -> None:
         await self._backend.init_search_index_async()
         await self._backend.init_checkpoint_tables_async()
 
-    # -------------------------------------------------------------------
-    # Checkpoint 管理（LangGraph）（同步）
-    # -------------------------------------------------------------------
-
     def init_langgraph_tables(self) -> None:
         """创建 LangGraph 所需的全部表和索引（同步）。
 
@@ -148,10 +144,36 @@ def init_langgraph_tables(self) -> None:
         self._backend.init_search_index()
         self._backend.init_checkpoint_tables()
 
+    async def init_adk_tables_async(self) -> None:
+        """创建 Google ADK 所需的全部表和索引（异步）。
+
+        包含核心表（Conversation + Event + 二级索引）、三级 State 表
+        （state / app_state / user_state）以及多元索引。
+        表或索引已存在时跳过，可重复调用。
+        """
+        await self._backend.init_core_tables_async()
+        await self._backend.init_state_tables_async()
+        await self._backend.init_search_index_async()
+
     # -------------------------------------------------------------------
     # Checkpoint 管理（LangGraph）（异步）
     # -------------------------------------------------------------------
 
+    def init_adk_tables(self) -> None:
+        """创建 Google ADK 所需的全部表和索引（同步）。
+
+        包含核心表（Conversation + Event + 二级索引）、三级 State 表
+        （state / app_state / user_state）以及多元索引。
+        表或索引已存在时跳过，可重复调用。
+        """
+        self._backend.init_core_tables()
+        self._backend.init_state_tables()
+        self._backend.init_search_index()
+
+    # -------------------------------------------------------------------
+    # Checkpoint 管理（LangGraph）（同步）
+    # -------------------------------------------------------------------
+
     async def put_checkpoint_async(
         self,
         thread_id: str,
@@ -366,7 +388,7 @@ async def delete_thread_checkpoints_async(
         await self._backend.delete_thread_checkpoints_async(thread_id)
 
     # -------------------------------------------------------------------
-    # Checkpoint 清理（同步）
+    # Session 管理（异步）/ Session management (async)
     # -------------------------------------------------------------------
 
     def delete_thread_checkpoints(
@@ -377,7 +399,7 @@ def delete_thread_checkpoints(
         self._backend.delete_thread_checkpoints(thread_id)
 
     # -------------------------------------------------------------------
-    # Session 管理（异步）/ Session management (async)
+    # Session 管理（同步）/ Session management (async)
     # -------------------------------------------------------------------
 
     async def create_session_async(
@@ -426,10 +448,6 @@ async def create_session_async(
         await self._backend.put_session_async(session)
         return session
 
-    # -------------------------------------------------------------------
-    # Session 管理（同步）/ Session management (sync)
-    # -------------------------------------------------------------------
-
     def create_session(
         self,
         agent_id: str,

agentrun/conversation_service/utils.py

@@ -10,10 +10,8 @@
 from typing import Any, TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from tablestore import (
-        AsyncOTSClient,  # type: ignore[import-untyped]
-        OTSClient,
-    )
+    from tablestore import AsyncOTSClient  # type: ignore[import-untyped]
+    from tablestore import OTSClient
 
 # OTS 单个属性列值上限为 2MB，留 0.5MB 余量（按字符数计）
 MAX_COLUMN_SIZE: int = 1_500_000  # 1.5M 字符

examples/conversation_service_adk_server.py

@@ -1,7 +1,7 @@
 """Google ADK Agent Server —— 使用 OTSSessionService 持久化会话。
 
 集成步骤：
-  Step 1: 初始化 SessionStore（OTS 后端）
+  Step 1: 初始化 SessionStore + 创建 ADK 所需表和索引
   Step 2: 创建 OTSSessionService
   Step 3: 创建 ADK Agent + Runner，传入 session_service
   Step 4: 实现 invoke_agent，将 AgentRequest 转为 ADK 调用并流式输出
@@ -71,10 +71,10 @@ def get_session_state(tool_context: ToolContext) -> dict[str, Any]:
     return tool_context.state.to_dict()
 
 
-# ── Step 1: 初始化 SessionStore ──────────────────────────────
+# ── Step 1: 初始化 SessionStore + 创建 ADK 所需表和索引 ────
 
 store = SessionStore.from_memory_collection(MEMORY_COLLECTION_NAME)
-store.init_tables()
+store.init_adk_tables()
 
 # ── Step 2: 创建 OTSSessionService ──────────────────────────