mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-08-12 22:00:57 +00:00
perf: full_command_text field in execute context
This commit is contained in:
@@ -75,6 +75,7 @@ class CommandManager:
|
|||||||
async def execute(
|
async def execute(
|
||||||
self,
|
self,
|
||||||
command_text: str,
|
command_text: str,
|
||||||
|
full_command_text: str,
|
||||||
query: pipeline_query.Query,
|
query: pipeline_query.Query,
|
||||||
session: provider_session.Session,
|
session: provider_session.Session,
|
||||||
) -> typing.AsyncGenerator[command_context.CommandReturn, None]:
|
) -> typing.AsyncGenerator[command_context.CommandReturn, None]:
|
||||||
@@ -89,6 +90,7 @@ class CommandManager:
|
|||||||
query_id=query.query_id,
|
query_id=query.query_id,
|
||||||
session=session,
|
session=session,
|
||||||
command_text=command_text,
|
command_text=command_text,
|
||||||
|
full_command_text=full_command_text,
|
||||||
command='',
|
command='',
|
||||||
crt_command='',
|
crt_command='',
|
||||||
params=command_text.split(' '),
|
params=command_text.split(' '),
|
||||||
|
|||||||
@@ -18,7 +18,9 @@ class CommandHandler(handler.MessageHandler):
|
|||||||
) -> typing.AsyncGenerator[entities.StageProcessResult, None]:
|
) -> typing.AsyncGenerator[entities.StageProcessResult, None]:
|
||||||
"""Process"""
|
"""Process"""
|
||||||
|
|
||||||
command_text = str(query.message_chain).strip()[1:]
|
full_command_text = str(query.message_chain).strip()
|
||||||
|
|
||||||
|
command_text = full_command_text[1:]
|
||||||
|
|
||||||
privilege = 1
|
privilege = 1
|
||||||
|
|
||||||
@@ -39,7 +41,7 @@ class CommandHandler(handler.MessageHandler):
|
|||||||
sender_id=query.sender_id,
|
sender_id=query.sender_id,
|
||||||
command=spt[0],
|
command=spt[0],
|
||||||
params=spt[1:] if len(spt) > 1 else [],
|
params=spt[1:] if len(spt) > 1 else [],
|
||||||
text_message=str(query.message_chain),
|
text_message=full_command_text,
|
||||||
is_admin=(privilege == 2),
|
is_admin=(privilege == 2),
|
||||||
query=query,
|
query=query,
|
||||||
)
|
)
|
||||||
@@ -62,7 +64,9 @@ class CommandHandler(handler.MessageHandler):
|
|||||||
|
|
||||||
session = await self.ap.sess_mgr.get_session(query)
|
session = await self.ap.sess_mgr.get_session(query)
|
||||||
|
|
||||||
async for ret in self.ap.cmd_mgr.execute(command_text=command_text, query=query, session=session):
|
async for ret in self.ap.cmd_mgr.execute(
|
||||||
|
command_text=command_text, full_command_text=full_command_text, query=query, session=session
|
||||||
|
):
|
||||||
if ret.error is not None:
|
if ret.error is not None:
|
||||||
query.resp_messages.append(
|
query.resp_messages.append(
|
||||||
provider_message.Message(
|
provider_message.Message(
|
||||||
|
|||||||
Reference in New Issue
Block a user