getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-16 08:02:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Support multiple messages in agent.prompt() and agentLoop

Browse source 
- agentLoop now accepts AgentMessage[] instead of single message
- agent.prompt() accepts AgentMessage | AgentMessage[]
- Emits message_start/end for each message in the array
- AgentSession.prompt() builds array with hook message + user message
- TUI now receives events for before_agent_start injected messages
This commit is contained in:
Mario Zechner 2025-12-28 17:03:38 +01:00
parent 575c875475
commit 41af99cccf
4 changed files with 48 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

12
packages/agent/src/agent-loop.ts
View file

@ -26,7 +26,7 @@ import type {
* The prompt is added to the context and events are emitted for it.
*/
export function agentLoop(
prompt: AgentMessage,
prompts: AgentMessage[],
context: AgentContext,
config: AgentLoopConfig,
signal?: AbortSignal,
@ -35,16 +35,18 @@ export function agentLoop(
const stream = createAgentStream();
(async () => {
const newMessages: AgentMessage[] = [prompt];
const newMessages: AgentMessage[] = [...prompts];
const currentContext: AgentContext = {
...context,
messages: [...context.messages, prompt],
messages: [...context.messages, ...prompts],
};
stream.push({ type: "agent_start" });
stream.push({ type: "turn_start" });
stream.push({ type: "message_start", message: prompt });
stream.push({ type: "message_end", message: prompt });
for (const prompt of prompts) {
stream.push({ type: "message_start", message: prompt });
stream.push({ type: "message_end", message: prompt });
}
await runLoop(currentContext, newMessages, config, signal, stream, streamFn);
})();