Repository navigation
Expand file tree
/
Copy pathai-history.ts
More file actions
112 lines (101 loc) · 3.09 KB
/
Copy pathai-history.ts
File metadata and controls
112 lines (101 loc) · 3.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import {
clearCache,
readCache,
writeCache,
type EvenStorage,
} from "./live-cache";
import type { AiCitationSource } from "./ai-tools";
const MAX_EXCERPTS = 3;
const MAX_TEXT_LENGTH = 160;
export type AiConversationExcerpt = {
readonly id: string;
readonly endedAt: string;
readonly user: string;
readonly assistant: string;
readonly sources?: readonly AiCitationSource[];
};
function normalizeSources(value: readonly AiCitationSource[] | undefined) {
if (!value) return undefined;
return value.flatMap((source) => {
try {
const url = new URL(source.url);
if (url.protocol !== "https:") return [];
return [{
title: source.title.replace(/\s+/g, " ").trim().slice(0, 160)
|| url.hostname,
url: url.toString(),
}];
} catch {
return [];
}
}).slice(0, 6);
}
function cleanExcerptText(value: string): string {
return value.replace(/\s+/g, " ").trim().slice(0, MAX_TEXT_LENGTH);
}
function normalizeExcerpt(
value: AiConversationExcerpt,
): AiConversationExcerpt {
return {
id: value.id.slice(0, 96),
endedAt: value.endedAt,
user: cleanExcerptText(value.user),
assistant: cleanExcerptText(value.assistant),
...(value.sources ? { sources: normalizeSources(value.sources) } : {}),
};
}
function isExcerpt(value: unknown): value is AiConversationExcerpt {
if (typeof value !== "object" || value === null) return false;
const item = value as Record<string, unknown>;
return typeof item.id === "string"
&& typeof item.endedAt === "string"
&& !Number.isNaN(Date.parse(item.endedAt))
&& typeof item.user === "string"
&& typeof item.assistant === "string"
&& (item.sources === undefined || (
Array.isArray(item.sources)
&& item.sources.length <= 6
&& item.sources.every((source) => (
typeof source === "object"
&& source !== null
&& typeof source.title === "string"
&& typeof source.url === "string"
))
));
}
function isHistory(value: unknown): value is readonly AiConversationExcerpt[] {
return Array.isArray(value)
&& value.length <= MAX_EXCERPTS
&& value.every(isExcerpt);
}
export function appendAiConversationExcerpt(
history: readonly AiConversationExcerpt[],
excerpt: AiConversationExcerpt,
): readonly AiConversationExcerpt[] {
const normalized = normalizeExcerpt(excerpt);
return [
normalized,
...history.filter(({ id }) => id !== normalized.id).map(normalizeExcerpt),
].slice(0, MAX_EXCERPTS);
}
export async function resolveAiConversationHistory(
storage: EvenStorage,
): Promise<readonly AiConversationExcerpt[]> {
const history = await readCache(storage, "ai-history", isHistory) ?? [];
return history.map(normalizeExcerpt);
}
export function writeAiConversationHistory(
storage: EvenStorage,
history: readonly AiConversationExcerpt[],
): Promise<boolean> {
return writeCache(
storage,
"ai-history",
history.slice(0, MAX_EXCERPTS).map(normalizeExcerpt),
);
}
export function clearAiConversationHistory(
storage: EvenStorage,
): Promise<boolean> {
return clearCache(storage, "ai-history");
}