xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-09 15:26:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: get bot uuid api

Browse Source
This commit is contained in:
Junyan Qin
2025-07-13 17:44:20 +08:00
parent 4b57771eb1
commit 9f269d1614
4 changed files with 32 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
pkg/plugin/handler.py
View File

@@ -13,7 +13,6 @@ from langbot_plugin.entities.io.actions.enums import (
LangBotToRuntimeAction,
PluginToRuntimeAction,
)
import langbot_plugin.api.entities.context as event_context_module
import langbot_plugin.api.entities.builtin.platform.message as platform_message
from ..entity.persistence import plugin as persistence_plugin
@@ -68,21 +67,21 @@ class RuntimeConnectionHandler(handler.Handler):
@self.action(PluginToRuntimeAction.REPLY_MESSAGE)
async def reply_message(data: dict[str, Any]) -> handler.ActionResponse:
"""Reply message"""
eid = data['eid']
query_id = data['query_id']
message_chain = data['message_chain']
quote_origin = data['quote_origin']
if eid not in event_context_module.cached_event_contexts:
if query_id not in self.ap.query_pool.cached_queries:
return handler.ActionResponse.error(
message=f'Event context with eid {eid} not found',
message=f'Query with query_id {query_id} not found',
)
event_context = event_context_module.cached_event_contexts[eid]
query = self.ap.query_pool.cached_queries[query_id]
message_chain_obj = platform_message.MessageChain.model_validate(message_chain)
await event_context.event.query.adapter.reply_message(
event_context.event.query.message_event,
await query.adapter.reply_message(
query.message_event,
message_chain_obj,
quote_origin,
)
@@ -91,6 +90,23 @@ class RuntimeConnectionHandler(handler.Handler):
data={},
)
@self.action(PluginToRuntimeAction.GET_BOT_UUID)
async def get_bot_uuid(data: dict[str, Any]) -> handler.ActionResponse:
"""Get bot uuid"""
query_id = data['query_id']
if query_id not in self.ap.query_pool.cached_queries:
return handler.ActionResponse.error(
message=f'Query with query_id {query_id} not found',
)
query = self.ap.query_pool.cached_queries[query_id]
return handler.ActionResponse.success(
data={
'bot_uuid': query.bot_uuid,
},
)
async def ping(self) -> dict[str, Any]:
"""Ping the runtime"""
return await self.call_action(