Clément Fiere
105 lines
3.8 KiB
TypeScript
105 lines
3.8 KiB
TypeScript
import type { RawSessionEvent } from "@/lib/ws/client";
|
|
import type { PluginConversationMessage, PluginConversationTurn } from "./types";
|
|
|
|
export function eventField(event: RawSessionEvent, key: string): unknown {
|
|
if (!event.payload || typeof event.payload !== "object" || !(key in event.payload)) {
|
|
return undefined;
|
|
}
|
|
return Reflect.get(event.payload, key);
|
|
}
|
|
|
|
export function eventString(event: RawSessionEvent, key: string): string | null {
|
|
const value = eventField(event, key);
|
|
return typeof value === "string" && value !== "" ? value : null;
|
|
}
|
|
|
|
function compareConversationTimestamps(left: string, right: string): number {
|
|
const leftMilliseconds = Date.parse(left);
|
|
const rightMilliseconds = Date.parse(right);
|
|
if (Number.isFinite(leftMilliseconds) && Number.isFinite(rightMilliseconds)) {
|
|
if (leftMilliseconds !== rightMilliseconds)
|
|
return leftMilliseconds < rightMilliseconds ? -1 : 1;
|
|
const leftFraction = left.match(/\.(\d+)(?:Z|[+-]\d{2}:\d{2})$/)?.[1] ?? "";
|
|
const rightFraction = right.match(/\.(\d+)(?:Z|[+-]\d{2}:\d{2})$/)?.[1] ?? "";
|
|
const leftNanoseconds = leftFraction.padEnd(9, "0").slice(0, 9);
|
|
const rightNanoseconds = rightFraction.padEnd(9, "0").slice(0, 9);
|
|
if (leftNanoseconds !== rightNanoseconds) {
|
|
return leftNanoseconds < rightNanoseconds ? -1 : 1;
|
|
}
|
|
return 0;
|
|
}
|
|
if (left < right) return -1;
|
|
if (left > right) return 1;
|
|
return 0;
|
|
}
|
|
|
|
function compareConversationIds(left: string, right: string): number {
|
|
if (left < right) return -1;
|
|
if (left > right) return 1;
|
|
return 0;
|
|
}
|
|
|
|
export function compareConversationMessages(
|
|
left: Pick<PluginConversationMessage, "createdAt" | "id">,
|
|
right: Pick<PluginConversationMessage, "createdAt" | "id">,
|
|
sort: "asc" | "desc",
|
|
): number {
|
|
const created = compareConversationTimestamps(left.createdAt, right.createdAt);
|
|
const ordered = created === 0 ? compareConversationIds(left.id, right.id) : created;
|
|
return sort === "asc" ? ordered : -ordered;
|
|
}
|
|
|
|
export function eventMatchesTask(event: RawSessionEvent, taskId: string | null): boolean {
|
|
return taskId === null || event.task_id === taskId;
|
|
}
|
|
|
|
export function messageFromEvent(event: RawSessionEvent): PluginConversationMessage | null {
|
|
const id = eventString(event, "message_id");
|
|
const authorType = eventString(event, "author_type");
|
|
const contentValue = eventField(event, "content");
|
|
const content = typeof contentValue === "string" ? contentValue : null;
|
|
const createdAt = eventString(event, "created_at");
|
|
const explicitUpdatedAt = eventString(event, "updated_at");
|
|
const updatedAt = explicitUpdatedAt ?? (event.event_type === "message.added" ? createdAt : null);
|
|
if (
|
|
!id ||
|
|
(authorType !== "user" && authorType !== "agent") ||
|
|
content === null ||
|
|
!createdAt ||
|
|
!updatedAt
|
|
) {
|
|
return null;
|
|
}
|
|
const promptIndex = eventField(event, "prompt_index");
|
|
return {
|
|
id,
|
|
taskId: event.task_id,
|
|
sessionId: event.session_id,
|
|
turnId: eventString(event, "turn_id") ?? undefined,
|
|
authorType,
|
|
type: eventString(event, "message_type") ?? "message",
|
|
content,
|
|
createdAt,
|
|
updatedAt,
|
|
promptIndex: typeof promptIndex === "number" ? promptIndex : undefined,
|
|
senderTaskId: eventString(event, "sender_task_id") ?? undefined,
|
|
};
|
|
}
|
|
|
|
export function turnFromEvent(event: RawSessionEvent): PluginConversationTurn | null {
|
|
const id = eventString(event, "turn_id") ?? eventString(event, "id");
|
|
const startedAt = eventString(event, "started_at");
|
|
const explicitUpdatedAt = eventString(event, "updated_at");
|
|
const updatedAt =
|
|
explicitUpdatedAt ?? (event.event_type === "session.turn.started" ? startedAt : null);
|
|
if (!id || !startedAt || !updatedAt) return null;
|
|
return {
|
|
id,
|
|
taskId: event.task_id,
|
|
sessionId: event.session_id,
|
|
startedAt,
|
|
completedAt: eventString(event, "completed_at") ?? undefined,
|
|
updatedAt,
|
|
};
|
|
}
|