dsh-titlecraft 0.1.0-rc.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +145 -0
- package/cordis.patch.yml +9 -0
- package/lib/client.d.ts +1 -0
- package/lib/client.d.ts.map +1 -0
- package/lib/client.js +1247 -0
- package/lib/client.js.map +1 -0
- package/lib/core.d.ts +76 -0
- package/lib/core.d.ts.map +1 -0
- package/lib/core.js +464 -0
- package/lib/core.js.map +1 -0
- package/lib/icons.json +1326 -0
- package/lib/index.d.ts +79 -0
- package/lib/index.d.ts.map +1 -0
- package/lib/index.js +364 -0
- package/lib/index.js.map +1 -0
- package/lib/refinement.d.ts +21 -0
- package/lib/refinement.d.ts.map +1 -0
- package/lib/refinement.js +329 -0
- package/lib/refinement.js.map +1 -0
- package/lib/scheduler.d.ts +17 -0
- package/lib/scheduler.d.ts.map +1 -0
- package/lib/scheduler.js +133 -0
- package/lib/scheduler.js.map +1 -0
- package/package.json +113 -0
- package/src/client.ts +1492 -0
- package/src/core.ts +516 -0
- package/src/icons.json +1326 -0
- package/src/index.ts +383 -0
- package/src/refinement.ts +356 -0
- package/src/scheduler.ts +125 -0
package/lib/index.d.ts
ADDED
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
import z from '@deepseek-ai/schemastery';
|
|
2
|
+
export declare const name = "titlecraft";
|
|
3
|
+
export declare const inject: string[];
|
|
4
|
+
export declare const SETTINGS_NAMESPACE = "titlecraft";
|
|
5
|
+
export declare const Config: z<Schemastery.ObjectS<{
|
|
6
|
+
schemaVersion: z<number, number>;
|
|
7
|
+
enabled: z<boolean, boolean>;
|
|
8
|
+
dateSource: z<string, string>;
|
|
9
|
+
types: z<string[], string[]>;
|
|
10
|
+
enabledTypes: z<string[], string[]>;
|
|
11
|
+
templatePreset: z<string, string>;
|
|
12
|
+
formatTemplate: z<string, string>;
|
|
13
|
+
iconMode: z<string, string>;
|
|
14
|
+
iconId: z<string, string>;
|
|
15
|
+
fontFamily: z<string, string>;
|
|
16
|
+
fontSize: z<number, number>;
|
|
17
|
+
sessionTitleLeftOffset: z<number, number>;
|
|
18
|
+
timezone: z<string, string>;
|
|
19
|
+
datePattern: z<string, string>;
|
|
20
|
+
instantEnabled: z<boolean, boolean>;
|
|
21
|
+
firstMessageLlmEnabled: z<boolean, boolean>;
|
|
22
|
+
refineEnabled: z<boolean, boolean>;
|
|
23
|
+
idleDelayMs: z<number, number>;
|
|
24
|
+
maxInputBytes: z<number, number>;
|
|
25
|
+
maxOutputTokens: z<number, number>;
|
|
26
|
+
timeoutMs: z<number, number>;
|
|
27
|
+
topicMaxChars: z<number, number>;
|
|
28
|
+
minConfidence: z<number, number>;
|
|
29
|
+
preferCheap: z<boolean, boolean>;
|
|
30
|
+
fallbackToSessionModel: z<boolean, boolean>;
|
|
31
|
+
modelProvider: z<string, string>;
|
|
32
|
+
modelName: z<string, string>;
|
|
33
|
+
respectManualTitles: z<boolean, boolean>;
|
|
34
|
+
topicGuidance: z<string, string>;
|
|
35
|
+
}>, Schemastery.ObjectT<{
|
|
36
|
+
schemaVersion: z<number, number>;
|
|
37
|
+
enabled: z<boolean, boolean>;
|
|
38
|
+
dateSource: z<string, string>;
|
|
39
|
+
types: z<string[], string[]>;
|
|
40
|
+
enabledTypes: z<string[], string[]>;
|
|
41
|
+
templatePreset: z<string, string>;
|
|
42
|
+
formatTemplate: z<string, string>;
|
|
43
|
+
iconMode: z<string, string>;
|
|
44
|
+
iconId: z<string, string>;
|
|
45
|
+
fontFamily: z<string, string>;
|
|
46
|
+
fontSize: z<number, number>;
|
|
47
|
+
sessionTitleLeftOffset: z<number, number>;
|
|
48
|
+
timezone: z<string, string>;
|
|
49
|
+
datePattern: z<string, string>;
|
|
50
|
+
instantEnabled: z<boolean, boolean>;
|
|
51
|
+
firstMessageLlmEnabled: z<boolean, boolean>;
|
|
52
|
+
refineEnabled: z<boolean, boolean>;
|
|
53
|
+
idleDelayMs: z<number, number>;
|
|
54
|
+
maxInputBytes: z<number, number>;
|
|
55
|
+
maxOutputTokens: z<number, number>;
|
|
56
|
+
timeoutMs: z<number, number>;
|
|
57
|
+
topicMaxChars: z<number, number>;
|
|
58
|
+
minConfidence: z<number, number>;
|
|
59
|
+
preferCheap: z<boolean, boolean>;
|
|
60
|
+
fallbackToSessionModel: z<boolean, boolean>;
|
|
61
|
+
modelProvider: z<string, string>;
|
|
62
|
+
modelName: z<string, string>;
|
|
63
|
+
respectManualTitles: z<boolean, boolean>;
|
|
64
|
+
topicGuidance: z<string, string>;
|
|
65
|
+
}>>;
|
|
66
|
+
export declare function apply(ctx: any, entryConfig?: {}): void;
|
|
67
|
+
export declare class BudgetRouter {
|
|
68
|
+
llm: any;
|
|
69
|
+
logger: any;
|
|
70
|
+
generation: number;
|
|
71
|
+
cache: any;
|
|
72
|
+
discovery: Promise<any> | undefined;
|
|
73
|
+
constructor(llm: any, logger: any);
|
|
74
|
+
invalidate(): void;
|
|
75
|
+
resolve(config: any, fallbackRoute: any): Promise<any>;
|
|
76
|
+
findCheap(): Promise<any>;
|
|
77
|
+
scanCheap(): Promise<any>;
|
|
78
|
+
}
|
|
79
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,0BAA0B,CAAA;AAexC,eAAO,MAAM,IAAI,eAAe,CAAA;AAChC,eAAO,MAAM,MAAM,UAAyE,CAAA;AAC5F,eAAO,MAAM,kBAAkB,eAAe,CAAA;AAC9C,eAAO,MAAM,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmCjB,CAAA;AAEF,wBAAgB,KAAK,CAAC,GAAG,KAAA,EAAE,WAAW,KAAK,QA6O1C;AAED,qBAAa,YAAY;IACf,GAAG,EAAE,GAAG,CAAA;IACR,MAAM,EAAE,GAAG,CAAA;IACX,UAAU,EAAE,MAAM,CAAA;IAClB,KAAK,EAAE,GAAG,CAAA;IACV,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,GAAG,SAAS,CAAA;gBAC/B,GAAG,KAAA,EAAE,MAAM,KAAA;IAKvB,UAAU;IAIJ,OAAO,CAAC,MAAM,KAAA,EAAE,aAAa,KAAA;IAS7B,SAAS;IAmBT,SAAS;CA6ChB"}
|
package/lib/index.js
ADDED
|
@@ -0,0 +1,364 @@
|
|
|
1
|
+
import z from '@deepseek-ai/schemastery';
|
|
2
|
+
import { Session } from '@deepseek-ai/dsh-session';
|
|
3
|
+
import { SessionTitleProviderId } from '@deepseek-ai/dsh-session-title';
|
|
4
|
+
import { DEFAULTS, formatSessionTitle, inferInstant, isSameTitle, normalizeConfig, projectNameOf, sessionDateTimestamp, } from './core.js';
|
|
5
|
+
import { refineTitle, summarizeSession, titleFromSummary, TitleSkippedError } from './refinement.js';
|
|
6
|
+
import { IdleRefinementScheduler } from './scheduler.js';
|
|
7
|
+
export const name = 'titlecraft';
|
|
8
|
+
export const inject = ['sessionTitle', 'sessions', 'sessionController', 'llm', 'connection'];
|
|
9
|
+
export const SETTINGS_NAMESPACE = 'titlecraft';
|
|
10
|
+
export const Config = z.object({
|
|
11
|
+
schemaVersion: z.number().step(1).min(1).max(2).default(2),
|
|
12
|
+
enabled: z.boolean().default(DEFAULTS.enabled),
|
|
13
|
+
dateSource: z.string().default(DEFAULTS.dateSource),
|
|
14
|
+
types: z.array(z.string()).default([...DEFAULTS.types]),
|
|
15
|
+
enabledTypes: z.array(z.string()).default(undefined),
|
|
16
|
+
templatePreset: z.string().default(DEFAULTS.templatePreset),
|
|
17
|
+
formatTemplate: z.string().default(DEFAULTS.formatTemplate),
|
|
18
|
+
iconMode: z.string().default(DEFAULTS.iconMode),
|
|
19
|
+
iconId: z.string().default(DEFAULTS.iconId),
|
|
20
|
+
fontFamily: z.string().default(DEFAULTS.fontFamily),
|
|
21
|
+
fontSize: z.number().step(1).min(12).max(20).default(DEFAULTS.fontSize),
|
|
22
|
+
sessionTitleLeftOffset: z.number().step(1).min(0).max(8).default(DEFAULTS.sessionTitleLeftOffset),
|
|
23
|
+
timezone: z.string().default(DEFAULTS.timezone),
|
|
24
|
+
datePattern: z.string().default(DEFAULTS.datePattern),
|
|
25
|
+
instantEnabled: z.boolean().default(DEFAULTS.instantEnabled),
|
|
26
|
+
firstMessageLlmEnabled: z.boolean().default(DEFAULTS.firstMessageLlmEnabled),
|
|
27
|
+
refineEnabled: z.boolean().default(DEFAULTS.refineEnabled),
|
|
28
|
+
idleDelayMs: z.number().step(1).min(250).max(600000).default(DEFAULTS.idleDelayMs),
|
|
29
|
+
maxInputBytes: z.number().step(1).min(512).max(65536).default(DEFAULTS.maxInputBytes),
|
|
30
|
+
maxOutputTokens: z.number().step(1).min(32).max(512).default(DEFAULTS.maxOutputTokens),
|
|
31
|
+
timeoutMs: z
|
|
32
|
+
.number()
|
|
33
|
+
.step(1)
|
|
34
|
+
.min(1000)
|
|
35
|
+
.max(2 ** 31 - 1)
|
|
36
|
+
.default(DEFAULTS.timeoutMs),
|
|
37
|
+
topicMaxChars: z.number().step(1).min(4).max(40).default(DEFAULTS.topicMaxChars),
|
|
38
|
+
minConfidence: z.number().min(0).max(1).default(DEFAULTS.minConfidence),
|
|
39
|
+
preferCheap: z.boolean().default(DEFAULTS.preferCheap),
|
|
40
|
+
fallbackToSessionModel: z.boolean().default(false),
|
|
41
|
+
modelProvider: z.string().default(DEFAULTS.modelProvider),
|
|
42
|
+
modelName: z.string().default(DEFAULTS.modelName),
|
|
43
|
+
respectManualTitles: z.boolean().default(DEFAULTS.respectManualTitles),
|
|
44
|
+
topicGuidance: z.string().default(DEFAULTS.topicGuidance),
|
|
45
|
+
});
|
|
46
|
+
export function apply(ctx, entryConfig = {}) {
|
|
47
|
+
const logger = ctx.logger(name);
|
|
48
|
+
let source = () => normalizeConfig(Config(entryConfig ?? {}));
|
|
49
|
+
let current = source();
|
|
50
|
+
const readConfig = () => current;
|
|
51
|
+
const router = new BudgetRouter(ctx.llm, logger);
|
|
52
|
+
const firstMessageAttempts = new WeakSet();
|
|
53
|
+
const firstMessageControllers = new Set();
|
|
54
|
+
const cancelFirstMessages = () => {
|
|
55
|
+
for (const controller of firstMessageControllers)
|
|
56
|
+
controller.abort();
|
|
57
|
+
firstMessageControllers.clear();
|
|
58
|
+
};
|
|
59
|
+
ctx.effect(() => cancelFirstMessages, 'titlecraft: cancel first-message requests');
|
|
60
|
+
const rpcFailure = (code, message) => ({ ok: false, error: { code, message, details: {} } });
|
|
61
|
+
const rpcHandler = async (endpoint, payload, signal) => {
|
|
62
|
+
if (endpoint === 'models') {
|
|
63
|
+
let timer;
|
|
64
|
+
let abort;
|
|
65
|
+
try {
|
|
66
|
+
signal.throwIfAborted();
|
|
67
|
+
const providers = (ctx.llm.listProviders?.() ?? [])
|
|
68
|
+
.map((entry) => entry?.id)
|
|
69
|
+
.filter((id) => typeof id === 'string' && id.length > 0);
|
|
70
|
+
const uniqueProviders = [...new Set(providers)];
|
|
71
|
+
const provider = typeof payload?.provider === 'string' ? payload.provider : '';
|
|
72
|
+
if (provider && !uniqueProviders.includes(provider))
|
|
73
|
+
return { ok: true, value: { providers: uniqueProviders, models: [] } };
|
|
74
|
+
const models = provider
|
|
75
|
+
? await Promise.race([
|
|
76
|
+
ctx.llm.listModels(provider),
|
|
77
|
+
new Promise((_, reject) => {
|
|
78
|
+
timer = setTimeout(() => reject(new Error('timeout')), 5000);
|
|
79
|
+
abort = () => reject(new Error('aborted'));
|
|
80
|
+
signal.addEventListener('abort', abort, { once: true });
|
|
81
|
+
}),
|
|
82
|
+
])
|
|
83
|
+
: [];
|
|
84
|
+
signal.throwIfAborted();
|
|
85
|
+
return {
|
|
86
|
+
ok: true,
|
|
87
|
+
value: {
|
|
88
|
+
providers: uniqueProviders,
|
|
89
|
+
models: [
|
|
90
|
+
...new Set((models ?? [])
|
|
91
|
+
.map((model) => model?.id)
|
|
92
|
+
.filter((id) => typeof id === 'string' && id.length > 0)),
|
|
93
|
+
],
|
|
94
|
+
},
|
|
95
|
+
};
|
|
96
|
+
}
|
|
97
|
+
catch {
|
|
98
|
+
return rpcFailure(signal.aborted ? 'aborted' : 'model-discovery-failed', 'Model discovery is unavailable.');
|
|
99
|
+
}
|
|
100
|
+
finally {
|
|
101
|
+
clearTimeout(timer);
|
|
102
|
+
if (abort)
|
|
103
|
+
signal.removeEventListener('abort', abort);
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
if (!payload || typeof payload !== 'object' || typeof payload.sessionId !== 'string')
|
|
107
|
+
return rpcFailure('invalid-request', 'A sessionId is required.');
|
|
108
|
+
let session = ctx.sessions.get(payload.sessionId);
|
|
109
|
+
try {
|
|
110
|
+
if (!session) {
|
|
111
|
+
const inspection = await ctx.sessionController.inspect(payload.sessionId, signal);
|
|
112
|
+
const detached = structuredClone(inspection);
|
|
113
|
+
session = Session.fromRestore(payload.sessionId, detached.events, detached.meta, detached.inheritedEventCount);
|
|
114
|
+
}
|
|
115
|
+
const config = readConfig();
|
|
116
|
+
if (endpoint === 'summarize') {
|
|
117
|
+
const decision = await summarizeSession(ctx, session, config, router, signal);
|
|
118
|
+
return {
|
|
119
|
+
ok: true,
|
|
120
|
+
value: {
|
|
121
|
+
summary: decision.summary,
|
|
122
|
+
type: decision.type,
|
|
123
|
+
iconId: decision.iconId || config.iconId,
|
|
124
|
+
confidence: decision.confidence,
|
|
125
|
+
},
|
|
126
|
+
};
|
|
127
|
+
}
|
|
128
|
+
if (endpoint === 'generate') {
|
|
129
|
+
const result = await titleFromSummary(ctx, session, config, router, payload, signal);
|
|
130
|
+
return { ok: true, value: result };
|
|
131
|
+
}
|
|
132
|
+
return rpcFailure('not-found', 'Unknown TitleCraft action.');
|
|
133
|
+
}
|
|
134
|
+
catch (error) {
|
|
135
|
+
if (signal.aborted)
|
|
136
|
+
return rpcFailure('aborted', 'The operation was cancelled.');
|
|
137
|
+
const reason = typeof error?.message === 'string' ? error.message : 'internal';
|
|
138
|
+
const known = [
|
|
139
|
+
'no-messages',
|
|
140
|
+
'no-route',
|
|
141
|
+
'route-unavailable',
|
|
142
|
+
'timeout',
|
|
143
|
+
'incomplete-output',
|
|
144
|
+
'uncertain-summary',
|
|
145
|
+
'generic-topic',
|
|
146
|
+
'invalid-json',
|
|
147
|
+
'invalid-schema',
|
|
148
|
+
'invalid-type',
|
|
149
|
+
'type-disabled',
|
|
150
|
+
'invalid-icon',
|
|
151
|
+
'invalid-summary-length',
|
|
152
|
+
'invalid-topic-length',
|
|
153
|
+
'invalid-topic-format',
|
|
154
|
+
'title-too-long',
|
|
155
|
+
];
|
|
156
|
+
const code = known.includes(reason) ? reason : 'generation-failed';
|
|
157
|
+
logger.warn(`title-workbench session=${String(payload.sessionId)
|
|
158
|
+
.replace(/[^\w.:-]/g, '_')
|
|
159
|
+
.slice(0, 120)} status=failed reason=${code}`);
|
|
160
|
+
return rpcFailure(code, 'Title generation could not be completed.');
|
|
161
|
+
}
|
|
162
|
+
};
|
|
163
|
+
ctx.effect(() => ctx.connection.rpc.handle('/titlecraft', rpcHandler), 'titlecraft: browser workbench rpc');
|
|
164
|
+
const scheduler = new IdleRefinementScheduler(ctx, readConfig, logger);
|
|
165
|
+
ctx.inject(['settings'], (settingsCtx) => {
|
|
166
|
+
const settings = settingsCtx.settings;
|
|
167
|
+
const validate = (value) => normalizeConfig(value);
|
|
168
|
+
const changed = () => {
|
|
169
|
+
current = normalizeConfig(source());
|
|
170
|
+
router.invalidate();
|
|
171
|
+
cancelFirstMessages();
|
|
172
|
+
scheduler.reconfigure();
|
|
173
|
+
};
|
|
174
|
+
try {
|
|
175
|
+
if (typeof settings?.register === 'function') {
|
|
176
|
+
const scope = settings.register(SETTINGS_NAMESPACE, Config, {
|
|
177
|
+
base: current,
|
|
178
|
+
applies: 'live',
|
|
179
|
+
validate,
|
|
180
|
+
});
|
|
181
|
+
source = () => scope.get();
|
|
182
|
+
changed();
|
|
183
|
+
const unwatch = scope.watch(changed);
|
|
184
|
+
return () => {
|
|
185
|
+
unwatch();
|
|
186
|
+
scheduler.reconfigure();
|
|
187
|
+
};
|
|
188
|
+
}
|
|
189
|
+
settings.installSection(ctx, SETTINGS_NAMESPACE, Config, current, {
|
|
190
|
+
setSource(next) {
|
|
191
|
+
source = next;
|
|
192
|
+
changed();
|
|
193
|
+
},
|
|
194
|
+
onChange: changed,
|
|
195
|
+
validate,
|
|
196
|
+
});
|
|
197
|
+
}
|
|
198
|
+
catch {
|
|
199
|
+
logger.warn('title-settings status=registration-failed');
|
|
200
|
+
}
|
|
201
|
+
return () => scheduler.reconfigure();
|
|
202
|
+
});
|
|
203
|
+
ctx.on('session/event', (session, event) => scheduler.onEvent(session, event));
|
|
204
|
+
ctx.on('session/disposed', (session) => scheduler.remove(session));
|
|
205
|
+
ctx.on('llm/adapters-updated', () => router.invalidate());
|
|
206
|
+
ctx.effect(() => () => scheduler.dispose(), 'titlecraft: dispose timers');
|
|
207
|
+
ctx.sessionTitle.register({
|
|
208
|
+
id: SessionTitleProviderId(name),
|
|
209
|
+
automatic: 'all-prompts',
|
|
210
|
+
generate: async (request) => {
|
|
211
|
+
const config = readConfig();
|
|
212
|
+
request.signal.throwIfAborted();
|
|
213
|
+
if (!config.enabled)
|
|
214
|
+
throw new TitleSkippedError('disabled');
|
|
215
|
+
const currentTitle = ctx.sessionTitle.get(request.session);
|
|
216
|
+
if (config.respectManualTitles && currentTitle?.source?.kind === 'user')
|
|
217
|
+
throw new TitleSkippedError('manual-title');
|
|
218
|
+
const refineReady = scheduler.consumeRefineReady(request.session.id);
|
|
219
|
+
if (refineReady && config.refineEnabled) {
|
|
220
|
+
if (request.messages.length === 1 && firstMessageAttempts.has(request.session))
|
|
221
|
+
throw new TitleSkippedError('first-message-already-attempted');
|
|
222
|
+
return refineTitle(ctx, request, config, router, logger, readConfig);
|
|
223
|
+
}
|
|
224
|
+
if (config.firstMessageLlmEnabled &&
|
|
225
|
+
request.messages.length === 1 &&
|
|
226
|
+
scheduler.hasLiveHuman(request.session.id) &&
|
|
227
|
+
(!currentTitle || currentTitle.source?.kind === 'fallback')) {
|
|
228
|
+
if (firstMessageAttempts.has(request.session))
|
|
229
|
+
throw new TitleSkippedError('first-message-already-attempted');
|
|
230
|
+
firstMessageAttempts.add(request.session);
|
|
231
|
+
const controller = new AbortController();
|
|
232
|
+
firstMessageControllers.add(controller);
|
|
233
|
+
try {
|
|
234
|
+
return await refineTitle(ctx, {
|
|
235
|
+
...request,
|
|
236
|
+
signal: AbortSignal.any([request.signal, controller.signal]),
|
|
237
|
+
}, config, router, logger, readConfig, 'first-message');
|
|
238
|
+
}
|
|
239
|
+
finally {
|
|
240
|
+
firstMessageControllers.delete(controller);
|
|
241
|
+
}
|
|
242
|
+
}
|
|
243
|
+
if (!config.instantEnabled)
|
|
244
|
+
throw new TitleSkippedError('instant-disabled');
|
|
245
|
+
// Once a provider has supplied a title, only semantic refinement may replace it.
|
|
246
|
+
if (currentTitle && currentTitle.source?.kind !== 'fallback')
|
|
247
|
+
throw new TitleSkippedError('await-refinement');
|
|
248
|
+
const latest = request.messages.at(-1);
|
|
249
|
+
if (!latest)
|
|
250
|
+
throw new TitleSkippedError('no-messages');
|
|
251
|
+
const decision = inferInstant(latest.text, {
|
|
252
|
+
config,
|
|
253
|
+
projectName: projectNameOf(request.session),
|
|
254
|
+
});
|
|
255
|
+
if (!decision)
|
|
256
|
+
throw new TitleSkippedError('await-refinement');
|
|
257
|
+
const title = formatSessionTitle({
|
|
258
|
+
createdAt: request.session.header.createdAt,
|
|
259
|
+
dateTimestamp: sessionDateTimestamp(request.session, config.dateSource),
|
|
260
|
+
...decision,
|
|
261
|
+
config,
|
|
262
|
+
});
|
|
263
|
+
if (!title)
|
|
264
|
+
throw new TitleSkippedError('invalid-complete-title');
|
|
265
|
+
if (currentTitle && isSameTitle(title, currentTitle.title))
|
|
266
|
+
throw new TitleSkippedError('unchanged');
|
|
267
|
+
return { title, messageSeqs: [latest.seq] };
|
|
268
|
+
},
|
|
269
|
+
});
|
|
270
|
+
logger.info('title-plugin status=ready version=complete-topic-v2');
|
|
271
|
+
}
|
|
272
|
+
export class BudgetRouter {
|
|
273
|
+
constructor(llm, logger) {
|
|
274
|
+
this.llm = llm;
|
|
275
|
+
this.logger = logger;
|
|
276
|
+
this.generation = 0;
|
|
277
|
+
}
|
|
278
|
+
invalidate() {
|
|
279
|
+
this.cache = undefined;
|
|
280
|
+
this.generation++;
|
|
281
|
+
}
|
|
282
|
+
async resolve(config, fallbackRoute) {
|
|
283
|
+
if (config.modelProvider && config.modelName)
|
|
284
|
+
return { provider: config.modelProvider, model: config.modelName };
|
|
285
|
+
if (config.preferCheap) {
|
|
286
|
+
const cheap = await this.findCheap();
|
|
287
|
+
if (cheap)
|
|
288
|
+
return cheap;
|
|
289
|
+
}
|
|
290
|
+
return fallbackRoute;
|
|
291
|
+
}
|
|
292
|
+
async findCheap() {
|
|
293
|
+
if (this.cache && Date.now() - this.cache.at < 120000)
|
|
294
|
+
return this.cache.route;
|
|
295
|
+
if (this.discovery)
|
|
296
|
+
return this.discovery;
|
|
297
|
+
let timer;
|
|
298
|
+
const pending = Promise.race([
|
|
299
|
+
this.scanCheap(),
|
|
300
|
+
new Promise((_, reject) => {
|
|
301
|
+
timer = setTimeout(() => {
|
|
302
|
+
this.generation++;
|
|
303
|
+
reject(new Error('model-discovery-timeout'));
|
|
304
|
+
}, 5000);
|
|
305
|
+
}),
|
|
306
|
+
]).finally(() => {
|
|
307
|
+
clearTimeout(timer);
|
|
308
|
+
if (this.discovery === pending)
|
|
309
|
+
this.discovery = undefined;
|
|
310
|
+
});
|
|
311
|
+
this.discovery = pending;
|
|
312
|
+
return pending;
|
|
313
|
+
}
|
|
314
|
+
async scanCheap() {
|
|
315
|
+
const started = this.generation;
|
|
316
|
+
const patterns = [
|
|
317
|
+
'flash',
|
|
318
|
+
'haiku',
|
|
319
|
+
'lite',
|
|
320
|
+
'mini',
|
|
321
|
+
'nano',
|
|
322
|
+
'fast',
|
|
323
|
+
'compact',
|
|
324
|
+
'small',
|
|
325
|
+
'turbo',
|
|
326
|
+
'budget',
|
|
327
|
+
];
|
|
328
|
+
let best;
|
|
329
|
+
try {
|
|
330
|
+
const providers = this.llm.listConfigurableProviders?.() ?? [];
|
|
331
|
+
for (const entry of providers) {
|
|
332
|
+
const provider = entry?.provider;
|
|
333
|
+
if (typeof provider !== 'string')
|
|
334
|
+
continue;
|
|
335
|
+
let models = [];
|
|
336
|
+
try {
|
|
337
|
+
models = await this.llm.listModels(provider);
|
|
338
|
+
}
|
|
339
|
+
catch {
|
|
340
|
+
continue;
|
|
341
|
+
}
|
|
342
|
+
for (const model of models ?? []) {
|
|
343
|
+
if (typeof model?.id !== 'string')
|
|
344
|
+
continue;
|
|
345
|
+
const rank = patterns.findIndex((pattern) => model.id.toLowerCase().includes(pattern));
|
|
346
|
+
if (rank < 0)
|
|
347
|
+
continue;
|
|
348
|
+
if (!best ||
|
|
349
|
+
rank < best.rank ||
|
|
350
|
+
(rank === best.rank && model.id.length < best.route.model.length))
|
|
351
|
+
best = { rank, route: { provider, model: model.id } };
|
|
352
|
+
}
|
|
353
|
+
}
|
|
354
|
+
}
|
|
355
|
+
catch {
|
|
356
|
+
this.logger.debug('budget model scan unavailable');
|
|
357
|
+
}
|
|
358
|
+
const route = best?.route;
|
|
359
|
+
if (this.generation === started)
|
|
360
|
+
this.cache = { at: Date.now(), route };
|
|
361
|
+
return route;
|
|
362
|
+
}
|
|
363
|
+
}
|
|
364
|
+
//# sourceMappingURL=index.js.map
|
package/lib/index.js.map
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,0BAA0B,CAAA;AACxC,OAAO,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAA;AAClD,OAAO,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAA;AACvE,OAAO,EACL,QAAQ,EACR,kBAAkB,EAClB,YAAY,EACZ,WAAW,EACX,eAAe,EACf,aAAa,EACb,oBAAoB,GACrB,MAAM,WAAW,CAAA;AAClB,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAA;AACpG,OAAO,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAA;AAExD,MAAM,CAAC,MAAM,IAAI,GAAG,YAAY,CAAA;AAChC,MAAM,CAAC,MAAM,MAAM,GAAG,CAAC,cAAc,EAAE,UAAU,EAAE,mBAAmB,EAAE,KAAK,EAAE,YAAY,CAAC,CAAA;AAC5F,MAAM,CAAC,MAAM,kBAAkB,GAAG,YAAY,CAAA;AAC9C,MAAM,CAAC,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;IAC1D,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;IAC9C,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;IACnD,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACvD,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC;IACpD,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC;IAC3D,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC;IAC3D,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;IAC/C,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;IAC3C,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;IACnD,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;IACvE,sBAAsB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAC;IACjG,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;IAC/C,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC;IACrD,cAAc,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC;IAC5D,sBAAsB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAC;IAC5E,aAAa,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;IAC1D,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC;IAClF,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;IACrF,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC;IACtF,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,IAAI,CAAC,CAAC,CAAC;SACP,GAAG,CAAC,IAAI,CAAC;SACT,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;SAChB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;IAC9B,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;IAChF,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;IACvE,WAAW,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC;IACtD,sBAAsB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC;IAClD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;IACzD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;IACjD,mBAAmB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAC;IACtE,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;CAC1D,CAAC,CAAA;AAEF,MAAM,UAAU,KAAK,CAAC,GAAG,EAAE,WAAW,GAAG,EAAE;IACzC,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;IAC/B,IAAI,MAAM,GAAG,GAAG,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC,CAAA;IAC7D,IAAI,OAAO,GAAG,MAAM,EAAE,CAAA;IACtB,MAAM,UAAU,GAAG,GAAG,EAAE,CAAC,OAAO,CAAA;IAChC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAA;IAChD,MAAM,oBAAoB,GAAG,IAAI,OAAO,EAAU,CAAA;IAClD,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAmB,CAAA;IAC1D,MAAM,mBAAmB,GAAG,GAAG,EAAE;QAC/B,KAAK,MAAM,UAAU,IAAI,uBAAuB;YAAE,UAAU,CAAC,KAAK,EAAE,CAAA;QACpE,uBAAuB,CAAC,KAAK,EAAE,CAAA;IACjC,CAAC,CAAA;IACD,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,2CAA2C,CAAC,CAAA;IAClF,MAAM,UAAU,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,CAAA;IAC5F,MAAM,UAAU,GAAG,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE;QACrD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,IAAI,KAAK,CAAA;YACT,IAAI,KAAK,CAAA;YACT,IAAI,CAAC;gBACH,MAAM,CAAC,cAAc,EAAE,CAAA;gBACvB,MAAM,SAAS,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,CAAC;qBAChD,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;qBACzB,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,EAAE,KAAK,QAAQ,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;gBAC1D,MAAM,eAAe,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC,CAAA;gBAC/C,MAAM,QAAQ,GAAG,OAAO,OAAO,EAAE,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAA;gBAC9E,IAAI,QAAQ,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC;oBACjD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAA;gBACxE,MAAM,MAAM,GAAQ,QAAQ;oBAC1B,CAAC,CAAC,MAAM,OAAO,CAAC,IAAI,CAAC;wBACjB,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC;wBAC5B,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;4BACxB,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;4BAC5D,KAAK,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAA;4BAC1C,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;wBACzD,CAAC,CAAC;qBACH,CAAC;oBACJ,CAAC,CAAC,EAAE,CAAA;gBACN,MAAM,CAAC,cAAc,EAAE,CAAA;gBACvB,OAAO;oBACL,EAAE,EAAE,IAAI;oBACR,KAAK,EAAE;wBACL,SAAS,EAAE,eAAe;wBAC1B,MAAM,EAAE;4BACN,GAAG,IAAI,GAAG,CACR,CAAC,MAAM,IAAI,EAAE,CAAC;iCACX,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;iCACzB,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,OAAO,EAAE,KAAK,QAAQ,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAC3D;yBACF;qBACF;iBACF,CAAA;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,UAAU,CACf,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,wBAAwB,EACrD,iCAAiC,CAClC,CAAA;YACH,CAAC;oBAAS,CAAC;gBACT,YAAY,CAAC,KAAK,CAAC,CAAA;gBACnB,IAAI,KAAK;oBAAE,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;YACvD,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ;YAClF,OAAO,UAAU,CAAC,iBAAiB,EAAE,0BAA0B,CAAC,CAAA;QAClE,IAAI,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QACjD,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,iBAAiB,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAA;gBACjF,MAAM,QAAQ,GAAG,eAAe,CAAC,UAAU,CAAC,CAAA;gBAC5C,OAAO,GAAG,OAAO,CAAC,WAAW,CAC3B,OAAO,CAAC,SAAS,EACjB,QAAQ,CAAC,MAAM,EACf,QAAQ,CAAC,IAAI,EACb,QAAQ,CAAC,mBAAmB,CAC7B,CAAA;YACH,CAAC;YACD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAA;YAC3B,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;gBAC7B,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAA;gBAC7E,OAAO;oBACL,EAAE,EAAE,IAAI;oBACR,KAAK,EAAE;wBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;wBACzB,IAAI,EAAE,QAAQ,CAAC,IAAI;wBACnB,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM;wBACxC,UAAU,EAAE,QAAQ,CAAC,UAAU;qBAChC;iBACF,CAAA;YACH,CAAC;YACD,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;gBAC5B,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAA;gBACpF,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,CAAA;YACpC,CAAC;YACD,OAAO,UAAU,CAAC,WAAW,EAAE,4BAA4B,CAAC,CAAA;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,MAAM,CAAC,OAAO;gBAAE,OAAO,UAAU,CAAC,SAAS,EAAE,8BAA8B,CAAC,CAAA;YAChF,MAAM,MAAM,GAAG,OAAO,KAAK,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAA;YAC9E,MAAM,KAAK,GAAG;gBACZ,aAAa;gBACb,UAAU;gBACV,mBAAmB;gBACnB,SAAS;gBACT,mBAAmB;gBACnB,mBAAmB;gBACnB,eAAe;gBACf,cAAc;gBACd,gBAAgB;gBAChB,cAAc;gBACd,eAAe;gBACf,cAAc;gBACd,wBAAwB;gBACxB,sBAAsB;gBACtB,sBAAsB;gBACtB,gBAAgB;aACjB,CAAA;YACD,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,mBAAmB,CAAA;YAClE,MAAM,CAAC,IAAI,CACT,2BAA2B,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC;iBACjD,OAAO,CAAC,WAAW,EAAE,GAAG,CAAC;iBACzB,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,yBAAyB,IAAI,EAAE,CAChD,CAAA;YACD,OAAO,UAAU,CAAC,IAAI,EAAE,0CAA0C,CAAC,CAAA;QACrE,CAAC;IACH,CAAC,CAAA;IACD,GAAG,CAAC,MAAM,CACR,GAAG,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,aAAa,EAAE,UAAU,CAAC,EAC1D,mCAAmC,CACpC,CAAA;IACD,MAAM,SAAS,GAAG,IAAI,uBAAuB,CAAC,GAAG,EAAE,UAAU,EAAE,MAAM,CAAC,CAAA;IACtE,GAAG,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE;QACvC,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,CAAA;QACrC,MAAM,QAAQ,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC,CAAA;QAClD,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,OAAO,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,CAAA;YACnC,MAAM,CAAC,UAAU,EAAE,CAAA;YACnB,mBAAmB,EAAE,CAAA;YACrB,SAAS,CAAC,WAAW,EAAE,CAAA;QACzB,CAAC,CAAA;QACD,IAAI,CAAC;YACH,IAAI,OAAO,QAAQ,EAAE,QAAQ,KAAK,UAAU,EAAE,CAAC;gBAC7C,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,kBAAkB,EAAE,MAAM,EAAE;oBAC1D,IAAI,EAAE,OAAO;oBACb,OAAO,EAAE,MAAM;oBACf,QAAQ;iBACT,CAAC,CAAA;gBACF,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAA;gBAC1B,OAAO,EAAE,CAAA;gBACT,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;gBACpC,OAAO,GAAG,EAAE;oBACV,OAAO,EAAE,CAAA;oBACT,SAAS,CAAC,WAAW,EAAE,CAAA;gBACzB,CAAC,CAAA;YACH,CAAC;YACD,QAAQ,CAAC,cAAc,CAAC,GAAG,EAAE,kBAAkB,EAAE,MAAM,EAAE,OAAO,EAAE;gBAChE,SAAS,CAAC,IAAI;oBACZ,MAAM,GAAG,IAAI,CAAA;oBACb,OAAO,EAAE,CAAA;gBACX,CAAC;gBACD,QAAQ,EAAE,OAAO;gBACjB,QAAQ;aACT,CAAC,CAAA;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAA;QAC1D,CAAC;QACD,OAAO,GAAG,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,CAAA;IACtC,CAAC,CAAC,CAAA;IACF,GAAG,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAA;IAC9E,GAAG,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;IAClE,GAAG,CAAC,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAA;IACzD,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,4BAA4B,CAAC,CAAA;IACzE,GAAG,CAAC,YAAY,CAAC,QAAQ,CAAC;QACxB,EAAE,EAAE,sBAAsB,CAAC,IAAI,CAAC;QAChC,SAAS,EAAE,aAAa;QACxB,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC1B,MAAM,MAAM,GAAG,UAAU,EAAE,CAAA;YAC3B,OAAO,CAAC,MAAM,CAAC,cAAc,EAAE,CAAA;YAC/B,IAAI,CAAC,MAAM,CAAC,OAAO;gBAAE,MAAM,IAAI,iBAAiB,CAAC,UAAU,CAAC,CAAA;YAC5D,MAAM,YAAY,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;YAC1D,IAAI,MAAM,CAAC,mBAAmB,IAAI,YAAY,EAAE,MAAM,EAAE,IAAI,KAAK,MAAM;gBACrE,MAAM,IAAI,iBAAiB,CAAC,cAAc,CAAC,CAAA;YAC7C,MAAM,WAAW,GAAG,SAAS,CAAC,kBAAkB,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;YACpE,IAAI,WAAW,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACxC,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC;oBAC5E,MAAM,IAAI,iBAAiB,CAAC,iCAAiC,CAAC,CAAA;gBAChE,OAAO,WAAW,CAAC,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC,CAAA;YACtE,CAAC;YACD,IACE,MAAM,CAAC,sBAAsB;gBAC7B,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC;gBAC7B,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC1C,CAAC,CAAC,YAAY,IAAI,YAAY,CAAC,MAAM,EAAE,IAAI,KAAK,UAAU,CAAC,EAC3D,CAAC;gBACD,IAAI,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC;oBAC3C,MAAM,IAAI,iBAAiB,CAAC,iCAAiC,CAAC,CAAA;gBAChE,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;gBACzC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAA;gBACxC,uBAAuB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAA;gBACvC,IAAI,CAAC;oBACH,OAAO,MAAM,WAAW,CACtB,GAAG,EACH;wBACE,GAAG,OAAO;wBACV,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;qBAC7D,EACD,MAAM,EACN,MAAM,EACN,MAAM,EACN,UAAU,EACV,eAAe,CAChB,CAAA;gBACH,CAAC;wBAAS,CAAC;oBACT,uBAAuB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAA;gBAC5C,CAAC;YACH,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,cAAc;gBAAE,MAAM,IAAI,iBAAiB,CAAC,kBAAkB,CAAC,CAAA;YAC3E,iFAAiF;YACjF,IAAI,YAAY,IAAI,YAAY,CAAC,MAAM,EAAE,IAAI,KAAK,UAAU;gBAC1D,MAAM,IAAI,iBAAiB,CAAC,kBAAkB,CAAC,CAAA;YACjD,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YACtC,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,iBAAiB,CAAC,aAAa,CAAC,CAAA;YACvD,MAAM,QAAQ,GAAG,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE;gBACzC,MAAM;gBACN,WAAW,EAAE,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC;aAC5C,CAAC,CAAA;YACF,IAAI,CAAC,QAAQ;gBAAE,MAAM,IAAI,iBAAiB,CAAC,kBAAkB,CAAC,CAAA;YAC9D,MAAM,KAAK,GAAG,kBAAkB,CAAC;gBAC/B,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS;gBAC3C,aAAa,EAAE,oBAAoB,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,UAAU,CAAC;gBACvE,GAAG,QAAQ;gBACX,MAAM;aACP,CAAC,CAAA;YACF,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,iBAAiB,CAAC,wBAAwB,CAAC,CAAA;YACjE,IAAI,YAAY,IAAI,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC,KAAK,CAAC;gBACxD,MAAM,IAAI,iBAAiB,CAAC,WAAW,CAAC,CAAA;YAC1C,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAA;QAC7C,CAAC;KACF,CAAC,CAAA;IACF,MAAM,CAAC,IAAI,CAAC,qDAAqD,CAAC,CAAA;AACpE,CAAC;AAED,MAAM,OAAO,YAAY;IAMvB,YAAY,GAAG,EAAE,MAAM;QACrB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAA;QACd,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;QACpB,IAAI,CAAC,UAAU,GAAG,CAAC,CAAA;IACrB,CAAC;IACD,UAAU;QACR,IAAI,CAAC,KAAK,GAAG,SAAS,CAAA;QACtB,IAAI,CAAC,UAAU,EAAE,CAAA;IACnB,CAAC;IACD,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,aAAa;QACjC,IAAI,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,SAAS;YAC1C,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,aAAa,EAAE,KAAK,EAAE,MAAM,CAAC,SAAS,EAAE,CAAA;QACpE,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAA;YACpC,IAAI,KAAK;gBAAE,OAAO,KAAK,CAAA;QACzB,CAAC;QACD,OAAO,aAAa,CAAA;IACtB,CAAC;IACD,KAAK,CAAC,SAAS;QACb,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,GAAG,MAAM;YAAE,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAA;QAC9E,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO,IAAI,CAAC,SAAS,CAAA;QACzC,IAAI,KAAK,CAAA;QACT,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;YAC3B,IAAI,CAAC,SAAS,EAAE;YAChB,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBACxB,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;oBACtB,IAAI,CAAC,UAAU,EAAE,CAAA;oBACjB,MAAM,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAA;gBAC9C,CAAC,EAAE,IAAI,CAAC,CAAA;YACV,CAAC,CAAC;SACH,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YACd,YAAY,CAAC,KAAK,CAAC,CAAA;YACnB,IAAI,IAAI,CAAC,SAAS,KAAK,OAAO;gBAAE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;QAC5D,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,SAAS,GAAG,OAAO,CAAA;QACxB,OAAO,OAAO,CAAA;IAChB,CAAC;IACD,KAAK,CAAC,SAAS;QACb,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAA;QAC/B,MAAM,QAAQ,GAAG;YACf,OAAO;YACP,OAAO;YACP,MAAM;YACN,MAAM;YACN,MAAM;YACN,MAAM;YACN,SAAS;YACT,OAAO;YACP,OAAO;YACP,QAAQ;SACT,CAAA;QACD,IAAI,IAAI,CAAA;QACR,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,yBAAyB,EAAE,EAAE,IAAI,EAAE,CAAA;YAC9D,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;gBAC9B,MAAM,QAAQ,GAAG,KAAK,EAAE,QAAQ,CAAA;gBAChC,IAAI,OAAO,QAAQ,KAAK,QAAQ;oBAAE,SAAQ;gBAC1C,IAAI,MAAM,GAAG,EAAE,CAAA;gBACf,IAAI,CAAC;oBACH,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAA;gBAC9C,CAAC;gBAAC,MAAM,CAAC;oBACP,SAAQ;gBACV,CAAC;gBACD,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;oBACjC,IAAI,OAAO,KAAK,EAAE,EAAE,KAAK,QAAQ;wBAAE,SAAQ;oBAC3C,MAAM,IAAI,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAA;oBACtF,IAAI,IAAI,GAAG,CAAC;wBAAE,SAAQ;oBACtB,IACE,CAAC,IAAI;wBACL,IAAI,GAAG,IAAI,CAAC,IAAI;wBAChB,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;wBAEjE,IAAI,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,EAAE,CAAA;gBACzD,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAA;QACpD,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK,CAAA;QACzB,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO;YAAE,IAAI,CAAC,KAAK,GAAG,EAAE,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAA;QACvE,OAAO,KAAK,CAAA;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
export declare class TitleSkippedError extends Error {
|
|
2
|
+
constructor(reason: any);
|
|
3
|
+
}
|
|
4
|
+
export declare function refineTitle(ctx: any, request: any, config: any, router: any, logger: any, readConfig?: () => any, phase?: string): Promise<{
|
|
5
|
+
title: any;
|
|
6
|
+
messageSeqs: any[];
|
|
7
|
+
model: any;
|
|
8
|
+
}>;
|
|
9
|
+
export declare function summarizeSession(ctx: any, session: any, config: any, router: any, signal: any): Promise<{
|
|
10
|
+
iconId?: any;
|
|
11
|
+
summary: any;
|
|
12
|
+
type: any;
|
|
13
|
+
confidence: any;
|
|
14
|
+
}>;
|
|
15
|
+
export declare function titleFromSummary(ctx: any, session: any, config: any, router: any, input: any, signal: any): Promise<{
|
|
16
|
+
title: any;
|
|
17
|
+
topic: string;
|
|
18
|
+
type: string;
|
|
19
|
+
iconId: any;
|
|
20
|
+
}>;
|
|
21
|
+
//# sourceMappingURL=refinement.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"refinement.d.ts","sourceRoot":"","sources":["../src/refinement.ts"],"names":[],"mappings":"AAaA,qBAAa,iBAAkB,SAAQ,KAAK;gBAC9B,MAAM,KAAA;CAInB;AAaD,wBAAsB,WAAW,CAC/B,GAAG,KAAA,EACH,OAAO,KAAA,EACP,MAAM,KAAA,EACN,MAAM,KAAA,EACN,MAAM,KAAA,EACN,UAAU,YAAe,EACzB,KAAK,SAAW;;;;GAoFjB;AAWD,wBAAsB,gBAAgB,CAAC,GAAG,KAAA,EAAE,OAAO,KAAA,EAAE,MAAM,KAAA,EAAE,MAAM,KAAA,EAAE,MAAM,KAAA;;;;;GAuE1E;AAED,wBAAsB,gBAAgB,CAAC,GAAG,KAAA,EAAE,OAAO,KAAA,EAAE,MAAM,KAAA,EAAE,MAAM,KAAA,EAAE,KAAK,KAAA,EAAE,MAAM,KAAA;;;;;GAwCjF"}
|