Jesse_Chen
47 lines
1.3 KiB
TypeScript
47 lines
1.3 KiB
TypeScript
import type { AgentExecutionReceipt, PublicActivityPhase, WorkflowReceipt } from "./consultation-agent-events.ts";
|
|
|
|
export type AgentActivityView = Readonly<{
|
|
phase: PublicActivityPhase;
|
|
label: string;
|
|
}>;
|
|
|
|
export type ChatMessage = {
|
|
readonly role: "user" | "assistant";
|
|
readonly text: string;
|
|
readonly suggestions?: readonly string[];
|
|
readonly techniqueTruth?: string;
|
|
readonly agentExecutionReceipt?: AgentExecutionReceipt;
|
|
readonly workflowReceipt?: WorkflowReceipt;
|
|
};
|
|
|
|
export type ChatMessageView = ChatMessage & {
|
|
readonly renderKey: string;
|
|
readonly state: "settled" | "streaming" | "thinking";
|
|
readonly activity?: AgentActivityView;
|
|
};
|
|
|
|
export function chatMessageViews(
|
|
messages: readonly ChatMessage[],
|
|
loading: boolean,
|
|
streamingText: string,
|
|
activity?: AgentActivityView,
|
|
): readonly ChatMessageView[] {
|
|
const settled = messages.map((message, index) => ({
|
|
...message,
|
|
renderKey: `message-${index}`,
|
|
state: "settled" as const,
|
|
}));
|
|
if (!loading || messages.at(-1)?.role === "assistant") return settled;
|
|
|
|
return [
|
|
...settled,
|
|
{
|
|
role: "assistant",
|
|
text: streamingText,
|
|
renderKey: `message-${messages.length}`,
|
|
state: streamingText ? "streaming" : "thinking",
|
|
activity,
|
|
},
|
|
];
|
|
}
|