mirror/nanobot
1
0
Fork 0
mirror of https://github.com/HKUDS/nanobot.git synced 2026-05-19 16:12:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(loop): extract _build_initial_messages

Browse Source
This commit is contained in:
chengyongru 2026-05-09 15:29:44 +08:00 committed by Xubin Ren
parent ce52070fcf
commit 0124c94d19
1 changed files with 29 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
nanobot/agent/loop.py
View File

@ -631,6 +631,32 @@ class AgentLoop:
return True return True
return False return False
def _build_initial_messages(
self,
msg: InboundMessage,
session: Session,
history: list[dict[str, Any]],
pending_ask_id: str | None,
pending_summary: Any,
) -> list[dict[str, Any]]:
"""Build the initial message list for the LLM turn."""
if pending_ask_id:
return ask_user_tool_result_messages(
self.context.build_system_prompt(channel=msg.channel),
history,
pending_ask_id,
image_generation_prompt(msg.content, msg.metadata),
)
return self.context.build_messages(
history=history,
current_message=image_generation_prompt(msg.content, msg.metadata),
session_summary=pending_summary,
media=msg.media if msg.media else None,
channel=msg.channel,
chat_id=self._runtime_chat_id(msg),
sender_id=msg.sender_id,
)
async def _dispatch_command_inline( async def _dispatch_command_inline(
self, self,
msg: InboundMessage, msg: InboundMessage,
@ -1213,23 +1239,9 @@ class AgentLoop:
history = session.get_history(**_hist_kwargs) history = session.get_history(**_hist_kwargs)
pending_ask_id = pending_ask_user_id(history) pending_ask_id = pending_ask_user_id(history)
if pending_ask_id: initial_messages = self._build_initial_messages(
initial_messages = ask_user_tool_result_messages( msg, session, history, pending_ask_id, pending
self.context.build_system_prompt(channel=msg.channel), )
history,
pending_ask_id,
image_generation_prompt(msg.content, msg.metadata),
)
else:
initial_messages = self.context.build_messages(
history=history,
current_message=image_generation_prompt(msg.content, msg.metadata),
session_summary=pending,
media=msg.media if msg.media else None,
channel=msg.channel,
chat_id=self._runtime_chat_id(msg),
sender_id=msg.sender_id,
)
_bus_progress = await self._build_bus_progress_callback(msg) _bus_progress = await self._build_bus_progress_callback(msg)
_on_retry_wait = await self._build_retry_wait_callback(msg) _on_retry_wait = await self._build_retry_wait_callback(msg)