Skip to content

Commit b5e5d76

Browse files
committed
feat(conversation_service): add asynchronous and synchronous methods for initializing conversation search index
- Introduced `init_conversation_search_index_async` and `init_conversation_search_index` methods to create the conversation search index, allowing for repeated calls if the index already exists. - Updated references in `session_store.py` and `ots_backend.py` to utilize the new methods for initializing the conversation search index. This enhancement improves the flexibility and usability of the conversation service's indexing capabilities.
1 parent 012c958 commit b5e5d76

4 files changed

Lines changed: 27 additions & 6 deletions

File tree

agentrun/conversation_service/__ots_backend_async_template.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,13 @@ async def init_search_index_async(self) -> None:
181181
await self._create_conversation_search_index_async()
182182
await self._create_state_search_index_async()
183183

184+
async def init_conversation_search_index_async(self) -> None:
185+
"""仅创建 Conversation 多元索引(异步)。
186+
187+
索引已存在时跳过,可重复调用。
188+
"""
189+
await self._create_conversation_search_index_async()
190+
184191
async def init_checkpoint_tables_async(self) -> None:
185192
"""创建 LangGraph checkpoint 相关的 3 张表(异步)。
186193

agentrun/conversation_service/__session_store_async_template.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ async def init_langchain_tables_async(self) -> None:
7171
表或索引已存在时跳过,可重复调用。
7272
"""
7373
await self._backend.init_core_tables_async()
74-
await self._backend.init_search_index_async()
74+
await self._backend.init_conversation_search_index_async()
7575

7676
async def init_langgraph_tables_async(self) -> None:
7777
"""创建 LangGraph 所需的全部表和索引(异步)。
@@ -81,7 +81,7 @@ async def init_langgraph_tables_async(self) -> None:
8181
表或索引已存在时跳过,可重复调用。
8282
"""
8383
await self._backend.init_core_tables_async()
84-
await self._backend.init_search_index_async()
84+
await self._backend.init_conversation_search_index_async()
8585
await self._backend.init_checkpoint_tables_async()
8686

8787
async def init_adk_tables_async(self) -> None:

agentrun/conversation_service/ots_backend.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -249,6 +249,20 @@ def init_search_index(self) -> None:
249249
self._create_conversation_search_index()
250250
self._create_state_search_index()
251251

252+
async def init_conversation_search_index_async(self) -> None:
253+
"""仅创建 Conversation 多元索引(异步)。
254+
255+
索引已存在时跳过,可重复调用。
256+
"""
257+
await self._create_conversation_search_index_async()
258+
259+
def init_conversation_search_index(self) -> None:
260+
"""仅创建 Conversation 多元索引(同步)。
261+
262+
索引已存在时跳过,可重复调用。
263+
"""
264+
self._create_conversation_search_index()
265+
252266
async def init_checkpoint_tables_async(self) -> None:
253267
"""创建 LangGraph checkpoint 相关的 3 张表(异步)。
254268

agentrun/conversation_service/session_store.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ async def init_langchain_tables_async(self) -> None:
111111
表或索引已存在时跳过,可重复调用。
112112
"""
113113
await self._backend.init_core_tables_async()
114-
await self._backend.init_search_index_async()
114+
await self._backend.init_conversation_search_index_async()
115115

116116
def init_langchain_tables(self) -> None:
117117
"""创建 LangChain 所需的全部表和索引(同步)。
@@ -120,7 +120,7 @@ def init_langchain_tables(self) -> None:
120120
表或索引已存在时跳过,可重复调用。
121121
"""
122122
self._backend.init_core_tables()
123-
self._backend.init_search_index()
123+
self._backend.init_conversation_search_index()
124124

125125
async def init_langgraph_tables_async(self) -> None:
126126
"""创建 LangGraph 所需的全部表和索引(异步)。
@@ -130,7 +130,7 @@ async def init_langgraph_tables_async(self) -> None:
130130
表或索引已存在时跳过,可重复调用。
131131
"""
132132
await self._backend.init_core_tables_async()
133-
await self._backend.init_search_index_async()
133+
await self._backend.init_conversation_search_index_async()
134134
await self._backend.init_checkpoint_tables_async()
135135

136136
def init_langgraph_tables(self) -> None:
@@ -141,7 +141,7 @@ def init_langgraph_tables(self) -> None:
141141
表或索引已存在时跳过,可重复调用。
142142
"""
143143
self._backend.init_core_tables()
144-
self._backend.init_search_index()
144+
self._backend.init_conversation_search_index()
145145
self._backend.init_checkpoint_tables()
146146

147147
async def init_adk_tables_async(self) -> None:

0 commit comments

Comments
 (0)