agent-respond

1import type { Experimental_Agent as Agent } from "ai";
2import { convertToModelMessages } from "ai";
3import { AGENTS, type agentId } from "@/registry/ai/agents/agents-registry";
4import type { AIUIMessage } from "@/registry/ai/messages";
5import { idToReadableText } from "@/registry/lib/id-to-readable-text";
6 
7export function agentRoute(mentions: { id: string; name: string }[]): agentId {
8	let mentionedAgentId = mentions?.[0]?.id;
9 
10	if (!mentionedAgentId) {
11		mentionedAgentId = AGENTS[0];
12	}
13 
14	return mentionedAgentId as agentId;
15}
16 
17export async function agentExecute({
18	agentId,
19	agent,
20	messages,
21}: {
22	agentId: string;
23	// biome-ignore lint/suspicious/noExplicitAny: Agent type is not known
24	agent: Agent<any, any, any>;
25	messages: AIUIMessage[];
26}) {
27	const agentStream = agent.stream({
28		messages: convertToModelMessages(messages),
29	});
30 
31	return agentStream.toUIMessageStreamResponse<AIUIMessage>({
32		originalMessages: messages,
33		sendReasoning: true,
34		messageMetadata: ({ part }) => {
35			if (part.type === "start") {
36				return {
37					timestamp: new Date().toISOString(),
38					agent: {
39						id: agentId,
40						name: idToReadableText(agentId),
41					},
42				};
43			}
44 
45			if (part.type === "finish") {
46				return {
47					outputTokens: part.totalUsage.outputTokens,
48				};
49			}
50		},
51	});
52}