@corgelabs/corge 0.0.1 → 1.0.1
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/dist/ai/index.d.ts +3 -0
- package/dist/ai/index.d.ts.map +1 -0
- package/dist/ai/index.js +3 -0
- package/dist/ai/index.js.map +1 -0
- package/dist/ai/schema.d.ts +83 -0
- package/dist/ai/schema.d.ts.map +1 -0
- package/dist/ai/schema.js +19 -0
- package/dist/ai/schema.js.map +1 -0
- package/dist/ai/service.d.ts +14 -0
- package/dist/ai/service.d.ts.map +1 -0
- package/dist/ai/service.js +266 -0
- package/dist/ai/service.js.map +1 -0
- package/dist/ai/utils.d.ts +10 -0
- package/dist/ai/utils.d.ts.map +1 -0
- package/dist/ai/utils.js +27 -0
- package/dist/ai/utils.js.map +1 -0
- package/dist/attachments/index.d.ts +4 -0
- package/dist/attachments/index.d.ts.map +1 -0
- package/dist/attachments/index.js +4 -0
- package/dist/attachments/index.js.map +1 -0
- package/dist/attachments/routes.d.ts +75 -0
- package/dist/attachments/routes.d.ts.map +1 -0
- package/dist/attachments/routes.js +18 -0
- package/dist/attachments/routes.js.map +1 -0
- package/dist/attachments/schema.d.ts +61 -0
- package/dist/attachments/schema.d.ts.map +1 -0
- package/dist/attachments/schema.js +30 -0
- package/dist/attachments/schema.js.map +1 -0
- package/dist/attachments/service.d.ts +27 -0
- package/dist/attachments/service.d.ts.map +1 -0
- package/dist/attachments/service.js +224 -0
- package/dist/attachments/service.js.map +1 -0
- package/dist/attachments/utils.d.ts +9 -0
- package/dist/attachments/utils.d.ts.map +1 -0
- package/dist/attachments/utils.js +80 -0
- package/dist/attachments/utils.js.map +1 -0
- package/dist/auth/index.d.ts +4 -0
- package/dist/auth/index.d.ts.map +1 -0
- package/dist/auth/index.js +4 -0
- package/dist/auth/index.js.map +1 -0
- package/dist/auth/oidc.d.ts +25 -0
- package/dist/auth/oidc.d.ts.map +1 -0
- package/dist/auth/oidc.js +298 -0
- package/dist/auth/oidc.js.map +1 -0
- package/dist/auth/recovery.d.ts +41 -0
- package/dist/auth/recovery.d.ts.map +1 -0
- package/dist/auth/recovery.js +250 -0
- package/dist/auth/recovery.js.map +1 -0
- package/dist/auth/routes.d.ts +226 -0
- package/dist/auth/routes.d.ts.map +1 -0
- package/dist/auth/routes.js +96 -0
- package/dist/auth/routes.js.map +1 -0
- package/dist/auth/schema.d.ts +158 -0
- package/dist/auth/schema.d.ts.map +1 -0
- package/dist/auth/schema.js +82 -0
- package/dist/auth/schema.js.map +1 -0
- package/dist/auth/service.d.ts +96 -0
- package/dist/auth/service.d.ts.map +1 -0
- package/dist/auth/service.js +129 -0
- package/dist/auth/service.js.map +1 -0
- package/dist/auth/session.d.ts +51 -0
- package/dist/auth/session.d.ts.map +1 -0
- package/dist/auth/session.js +344 -0
- package/dist/auth/session.js.map +1 -0
- package/dist/auth/utils.d.ts +17 -0
- package/dist/auth/utils.d.ts.map +1 -0
- package/dist/auth/utils.js +96 -0
- package/dist/auth/utils.js.map +1 -0
- package/dist/bin/corge.js +542 -0
- package/dist/chat/context.d.ts +10 -0
- package/dist/chat/context.d.ts.map +1 -0
- package/dist/chat/context.js +221 -0
- package/dist/chat/context.js.map +1 -0
- package/dist/chat/index.d.ts +4 -0
- package/dist/chat/index.d.ts.map +1 -0
- package/dist/chat/index.js +4 -0
- package/dist/chat/index.js.map +1 -0
- package/dist/chat/routes.d.ts +160 -0
- package/dist/chat/routes.d.ts.map +1 -0
- package/dist/chat/routes.js +210 -0
- package/dist/chat/routes.js.map +1 -0
- package/dist/chat/schema.d.ts +285 -0
- package/dist/chat/schema.d.ts.map +1 -0
- package/dist/chat/schema.js +74 -0
- package/dist/chat/schema.js.map +1 -0
- package/dist/chat/service.d.ts +64 -0
- package/dist/chat/service.d.ts.map +1 -0
- package/dist/chat/service.js +279 -0
- package/dist/chat/service.js.map +1 -0
- package/dist/chat/stream.d.ts +53 -0
- package/dist/chat/stream.d.ts.map +1 -0
- package/dist/chat/stream.js +339 -0
- package/dist/chat/stream.js.map +1 -0
- package/dist/chat/utils.d.ts +61 -0
- package/dist/chat/utils.d.ts.map +1 -0
- package/dist/chat/utils.js +285 -0
- package/dist/chat/utils.js.map +1 -0
- package/dist/cli/builder.d.ts +6 -0
- package/dist/cli/builder.d.ts.map +1 -0
- package/dist/cli/builder.js +91 -0
- package/dist/cli/builder.js.map +1 -0
- package/dist/cli/deployer.d.ts +2 -0
- package/dist/cli/deployer.d.ts.map +1 -0
- package/dist/cli/deployer.js +152 -0
- package/dist/cli/deployer.js.map +1 -0
- package/dist/cli/dev-server.d.ts +2 -0
- package/dist/cli/dev-server.d.ts.map +1 -0
- package/dist/cli/dev-server.js +70 -0
- package/dist/cli/dev-server.js.map +1 -0
- package/dist/cli/env.d.ts +4 -0
- package/dist/cli/env.d.ts.map +1 -0
- package/dist/cli/env.js +68 -0
- package/dist/cli/env.js.map +1 -0
- package/dist/cli/init.d.ts +6 -0
- package/dist/cli/init.d.ts.map +1 -0
- package/dist/cli/init.js +97 -0
- package/dist/cli/init.js.map +1 -0
- package/dist/config.d.ts +262 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +212 -0
- package/dist/config.js.map +1 -0
- package/dist/db/index.d.ts +3 -0
- package/dist/db/index.d.ts.map +1 -0
- package/dist/db/index.js +3 -0
- package/dist/db/index.js.map +1 -0
- package/dist/db/schema.d.ts +410 -0
- package/dist/db/schema.d.ts.map +1 -0
- package/dist/db/schema.js +294 -0
- package/dist/db/schema.js.map +1 -0
- package/dist/db/service.d.ts +42 -0
- package/dist/db/service.d.ts.map +1 -0
- package/dist/db/service.js +372 -0
- package/dist/db/service.js.map +1 -0
- package/dist/db/utils.d.ts +25 -0
- package/dist/db/utils.d.ts.map +1 -0
- package/dist/db/utils.js +188 -0
- package/dist/db/utils.js.map +1 -0
- package/dist/emails/index.d.ts +4 -0
- package/dist/emails/index.d.ts.map +1 -0
- package/dist/emails/index.js +4 -0
- package/dist/emails/index.js.map +1 -0
- package/dist/emails/routes.d.ts +191 -0
- package/dist/emails/routes.d.ts.map +1 -0
- package/dist/emails/routes.js +39 -0
- package/dist/emails/routes.js.map +1 -0
- package/dist/emails/schema.d.ts +237 -0
- package/dist/emails/schema.d.ts.map +1 -0
- package/dist/emails/schema.js +98 -0
- package/dist/emails/schema.js.map +1 -0
- package/dist/emails/service.d.ts +22 -0
- package/dist/emails/service.d.ts.map +1 -0
- package/dist/emails/service.js +337 -0
- package/dist/emails/service.js.map +1 -0
- package/dist/emails/utils.d.ts +49 -0
- package/dist/emails/utils.d.ts.map +1 -0
- package/dist/emails/utils.js +341 -0
- package/dist/emails/utils.js.map +1 -0
- package/dist/files/index.d.ts +4 -0
- package/dist/files/index.d.ts.map +1 -0
- package/dist/files/index.js +4 -0
- package/dist/files/index.js.map +1 -0
- package/dist/files/routes.d.ts +72 -0
- package/dist/files/routes.d.ts.map +1 -0
- package/dist/files/routes.js +38 -0
- package/dist/files/routes.js.map +1 -0
- package/dist/files/schema.d.ts +57 -0
- package/dist/files/schema.d.ts.map +1 -0
- package/dist/files/schema.js +39 -0
- package/dist/files/schema.js.map +1 -0
- package/dist/files/service.d.ts +27 -0
- package/dist/files/service.d.ts.map +1 -0
- package/dist/files/service.js +175 -0
- package/dist/files/service.js.map +1 -0
- package/dist/files/utils.d.ts +3 -0
- package/dist/files/utils.d.ts.map +1 -0
- package/dist/files/utils.js +63 -0
- package/dist/files/utils.js.map +1 -0
- package/dist/index.d.ts +875 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +9919 -0
- package/dist/index.js.map +7 -0
- package/dist/payments/billing.d.ts +29 -0
- package/dist/payments/billing.d.ts.map +1 -0
- package/dist/payments/billing.js +378 -0
- package/dist/payments/billing.js.map +1 -0
- package/dist/payments/index.d.ts +4 -0
- package/dist/payments/index.d.ts.map +1 -0
- package/dist/payments/index.js +4 -0
- package/dist/payments/index.js.map +1 -0
- package/dist/payments/routes.d.ts +340 -0
- package/dist/payments/routes.d.ts.map +1 -0
- package/dist/payments/routes.js +56 -0
- package/dist/payments/routes.js.map +1 -0
- package/dist/payments/schema.d.ts +412 -0
- package/dist/payments/schema.d.ts.map +1 -0
- package/dist/payments/schema.js +260 -0
- package/dist/payments/schema.js.map +1 -0
- package/dist/payments/service.d.ts +31 -0
- package/dist/payments/service.d.ts.map +1 -0
- package/dist/payments/service.js +322 -0
- package/dist/payments/service.js.map +1 -0
- package/dist/payments/utils.d.ts +61 -0
- package/dist/payments/utils.d.ts.map +1 -0
- package/dist/payments/utils.js +369 -0
- package/dist/payments/utils.js.map +1 -0
- package/dist/router/guard.d.ts +19 -0
- package/dist/router/guard.d.ts.map +1 -0
- package/dist/router/guard.js +243 -0
- package/dist/router/guard.js.map +1 -0
- package/dist/router/index.d.ts +3 -0
- package/dist/router/index.d.ts.map +1 -0
- package/dist/router/index.js +3 -0
- package/dist/router/index.js.map +1 -0
- package/dist/router/schema.d.ts +103 -0
- package/dist/router/schema.d.ts.map +1 -0
- package/dist/router/schema.js +282 -0
- package/dist/router/schema.js.map +1 -0
- package/dist/router/service.d.ts +25 -0
- package/dist/router/service.d.ts.map +1 -0
- package/dist/router/service.js +339 -0
- package/dist/router/service.js.map +1 -0
- package/dist/tasks/index.d.ts +4 -0
- package/dist/tasks/index.d.ts.map +1 -0
- package/dist/tasks/index.js +4 -0
- package/dist/tasks/index.js.map +1 -0
- package/dist/tasks/routes.d.ts +54 -0
- package/dist/tasks/routes.d.ts.map +1 -0
- package/dist/tasks/routes.js +195 -0
- package/dist/tasks/routes.js.map +1 -0
- package/dist/tasks/schema.d.ts +261 -0
- package/dist/tasks/schema.d.ts.map +1 -0
- package/dist/tasks/schema.js +35 -0
- package/dist/tasks/schema.js.map +1 -0
- package/dist/tasks/service.d.ts +32 -0
- package/dist/tasks/service.d.ts.map +1 -0
- package/dist/tasks/service.js +201 -0
- package/dist/tasks/service.js.map +1 -0
- package/dist/tasks/steps.d.ts +18 -0
- package/dist/tasks/steps.d.ts.map +1 -0
- package/dist/tasks/steps.js +365 -0
- package/dist/tasks/steps.js.map +1 -0
- package/dist/tasks/utils.d.ts +75 -0
- package/dist/tasks/utils.d.ts.map +1 -0
- package/dist/tasks/utils.js +311 -0
- package/dist/tasks/utils.js.map +1 -0
- package/dist/tasks/watcher.d.ts +13 -0
- package/dist/tasks/watcher.d.ts.map +1 -0
- package/dist/tasks/watcher.js +255 -0
- package/dist/tasks/watcher.js.map +1 -0
- package/dist/tasks/worker.d.ts +28 -0
- package/dist/tasks/worker.d.ts.map +1 -0
- package/dist/tasks/worker.js +326 -0
- package/dist/tasks/worker.js.map +1 -0
- package/dist/utils/index.d.ts +3 -0
- package/dist/utils/index.d.ts.map +1 -0
- package/dist/utils/index.js +3 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/schema.d.ts +79 -0
- package/dist/utils/schema.d.ts.map +1 -0
- package/dist/utils/schema.js +41 -0
- package/dist/utils/schema.js.map +1 -0
- package/dist/utils/service.d.ts +32 -0
- package/dist/utils/service.d.ts.map +1 -0
- package/dist/utils/service.js +384 -0
- package/dist/utils/service.js.map +1 -0
- package/docs/SDK_SPEC.md +1645 -0
- package/infra/firestore.base.indexes.json +138 -0
- package/infra/firestore.base.rules +152 -0
- package/infra/storage.base.rules +8 -0
- package/package.json +58 -13
- package/src/ai/index.ts +2 -0
- package/src/ai/schema.ts +64 -0
- package/src/ai/service.ts +317 -0
- package/src/ai/utils.ts +38 -0
- package/src/attachments/index.ts +3 -0
- package/src/attachments/routes.ts +28 -0
- package/src/attachments/schema.ts +45 -0
- package/src/attachments/service.ts +315 -0
- package/src/attachments/utils.ts +115 -0
- package/src/auth/index.ts +3 -0
- package/src/auth/oidc.ts +419 -0
- package/src/auth/recovery.ts +365 -0
- package/src/auth/routes.ts +142 -0
- package/src/auth/schema.ts +181 -0
- package/src/auth/service.ts +183 -0
- package/src/auth/session.ts +414 -0
- package/src/auth/utils.ts +106 -0
- package/src/chat/context.ts +310 -0
- package/src/chat/index.ts +3 -0
- package/src/chat/routes.ts +264 -0
- package/src/chat/schema.ts +295 -0
- package/src/chat/service.ts +360 -0
- package/src/chat/stream.ts +481 -0
- package/src/chat/utils.ts +370 -0
- package/src/cli/builder.ts +93 -0
- package/src/cli/deployer.ts +184 -0
- package/src/cli/dev-server.ts +77 -0
- package/src/cli/env.ts +89 -0
- package/src/cli/init.ts +121 -0
- package/src/config.ts +447 -0
- package/src/db/index.ts +2 -0
- package/src/db/schema.ts +409 -0
- package/src/db/service.ts +477 -0
- package/src/db/utils.ts +250 -0
- package/src/emails/index.ts +3 -0
- package/src/emails/routes.ts +64 -0
- package/src/emails/schema.ts +239 -0
- package/src/emails/service.ts +488 -0
- package/src/emails/utils.ts +449 -0
- package/src/files/index.ts +3 -0
- package/src/files/routes.ts +44 -0
- package/src/files/schema.ts +76 -0
- package/src/files/service.ts +238 -0
- package/src/files/utils.ts +77 -0
- package/src/index.ts +425 -0
- package/src/payments/billing.ts +488 -0
- package/src/payments/index.ts +3 -0
- package/src/payments/routes.ts +92 -0
- package/src/payments/schema.ts +401 -0
- package/src/payments/service.ts +445 -0
- package/src/payments/utils.ts +472 -0
- package/src/router/guard.ts +319 -0
- package/src/router/index.ts +2 -0
- package/src/router/schema.ts +458 -0
- package/src/router/service.ts +453 -0
- package/src/tasks/index.ts +11 -0
- package/src/tasks/routes.ts +230 -0
- package/src/tasks/schema.ts +310 -0
- package/src/tasks/service.ts +254 -0
- package/src/tasks/steps.ts +472 -0
- package/src/tasks/utils.ts +445 -0
- package/src/tasks/watcher.ts +307 -0
- package/src/tasks/worker.ts +461 -0
- package/src/utils/index.ts +2 -0
- package/src/utils/schema.ts +69 -0
- package/src/utils/service.ts +461 -0
- package/tests/e2e.test.ts +2531 -0
- package/tsconfig.base.json +12 -0
- package/README.md +0 -2
- package/index.d.ts +0 -1
- package/index.js +0 -1
|
@@ -0,0 +1,221 @@
|
|
|
1
|
+
import { readFile } from 'node:fs/promises';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
import { FieldValue } from 'firebase-admin/firestore';
|
|
4
|
+
import { CORE_CONFIG } from '../config';
|
|
5
|
+
import { Core } from '../index';
|
|
6
|
+
import { isContextItemVisible, isUserContextShape, mapFirestoreMessageDoc, PUBLIC_MESSAGE_FIELDS, RECALL_MESSAGE_FIELDS, toModelMessageFromDoc } from './utils';
|
|
7
|
+
let contextCatalog = [];
|
|
8
|
+
const markdownCache = new Map();
|
|
9
|
+
export class ChatContextEngine {
|
|
10
|
+
#core;
|
|
11
|
+
constructor(coreClient) {
|
|
12
|
+
this.#core = coreClient;
|
|
13
|
+
}
|
|
14
|
+
registerContextCatalog = (items = [], reset = false) => {
|
|
15
|
+
if (reset) {
|
|
16
|
+
contextCatalog = [];
|
|
17
|
+
markdownCache.clear();
|
|
18
|
+
}
|
|
19
|
+
contextCatalog = [...items];
|
|
20
|
+
};
|
|
21
|
+
#readContextMarkdown = async (relPath) => {
|
|
22
|
+
const cached = markdownCache.get(relPath);
|
|
23
|
+
if (cached !== undefined)
|
|
24
|
+
return cached;
|
|
25
|
+
const contextRoot = path.resolve(process.cwd(), CORE_CONFIG.context.dir);
|
|
26
|
+
const resolved = path.resolve(contextRoot, relPath);
|
|
27
|
+
if (!resolved.startsWith(contextRoot + path.sep) || !relPath.endsWith('.md')) {
|
|
28
|
+
return undefined;
|
|
29
|
+
}
|
|
30
|
+
try {
|
|
31
|
+
const content = await readFile(resolved, 'utf8');
|
|
32
|
+
markdownCache.set(relPath, content);
|
|
33
|
+
return content;
|
|
34
|
+
}
|
|
35
|
+
catch {
|
|
36
|
+
return undefined;
|
|
37
|
+
}
|
|
38
|
+
};
|
|
39
|
+
getContext = async (prompt, catalog = contextCatalog, options) => {
|
|
40
|
+
const trimmed = prompt.trim();
|
|
41
|
+
if (!trimmed || catalog.length === 0)
|
|
42
|
+
return [];
|
|
43
|
+
const validItems = catalog.filter((item) => item.path?.endsWith('.md') &&
|
|
44
|
+
!item.path.includes('..') &&
|
|
45
|
+
!path.isAbsolute(item.path) &&
|
|
46
|
+
isContextItemVisible(item, options?.audiences));
|
|
47
|
+
if (validItems.length === 0)
|
|
48
|
+
return [];
|
|
49
|
+
const maxSources = options?.maxSources ?? CORE_CONFIG.context.maxSources;
|
|
50
|
+
const userState = options?.userState;
|
|
51
|
+
const stateHint = userState ? ' Use the provided userState when judging relevance.' : '';
|
|
52
|
+
const questions = {};
|
|
53
|
+
validItems.forEach((item, idx) => {
|
|
54
|
+
questions[`doc_${idx}`] = {
|
|
55
|
+
type: 'noul',
|
|
56
|
+
instructions: `Is documentation file "${item.path}" (${item.description}) directly relevant to answering the user prompt?${stateHint}`,
|
|
57
|
+
criteria: {
|
|
58
|
+
true: `The user prompt requires or benefits from documentation in ${item.path} (${item.description})`,
|
|
59
|
+
false: 'The user prompt is unrelated to this documentation file'
|
|
60
|
+
}
|
|
61
|
+
};
|
|
62
|
+
});
|
|
63
|
+
try {
|
|
64
|
+
const decision = await this.#core.ai.decide({ prompt: trimmed, ...(userState ? { userState } : {}) }, questions);
|
|
65
|
+
const scored = validItems
|
|
66
|
+
.map((item, idx) => {
|
|
67
|
+
const ans = decision.answers[`doc_${idx}`];
|
|
68
|
+
return { item, score: ans?.type === 'noul' ? ans.noul : 0 };
|
|
69
|
+
})
|
|
70
|
+
.filter((x) => x.score >= 0.5)
|
|
71
|
+
.sort((a, b) => b.score - a.score)
|
|
72
|
+
.slice(0, maxSources);
|
|
73
|
+
if (scored.length === 0)
|
|
74
|
+
return [];
|
|
75
|
+
const loaded = await Promise.all(scored.map(async ({ item }) => {
|
|
76
|
+
const markdown = await this.#readContextMarkdown(item.path);
|
|
77
|
+
return markdown ? { path: item.path, description: item.description, markdown } : undefined;
|
|
78
|
+
}));
|
|
79
|
+
return loaded.filter((entry) => Boolean(entry));
|
|
80
|
+
}
|
|
81
|
+
catch {
|
|
82
|
+
return [];
|
|
83
|
+
}
|
|
84
|
+
};
|
|
85
|
+
#resolveTurnUserContext = async (opts) => {
|
|
86
|
+
const resolver = opts.agent?.resolveUserContext;
|
|
87
|
+
const role = opts.role ?? this.#core.internals.auth.resolveUserRole(opts.userId);
|
|
88
|
+
if (!resolver || !opts.userId || role === 'anon')
|
|
89
|
+
return undefined;
|
|
90
|
+
const timeout = AbortSignal.timeout(CORE_CONFIG.context.userContextTimeoutMs);
|
|
91
|
+
const signal = opts.abortSignal ? AbortSignal.any([opts.abortSignal, timeout]) : timeout;
|
|
92
|
+
let onAbort;
|
|
93
|
+
try {
|
|
94
|
+
const state = { env: this.#core.env, userId: opts.userId, role };
|
|
95
|
+
const turnCore = new Core(state);
|
|
96
|
+
const aborted = new Promise((resolve) => {
|
|
97
|
+
onAbort = () => resolve(undefined);
|
|
98
|
+
if (signal.aborted)
|
|
99
|
+
return resolve(undefined);
|
|
100
|
+
signal.addEventListener('abort', onAbort, { once: true });
|
|
101
|
+
});
|
|
102
|
+
const result = await Promise.race([
|
|
103
|
+
this.#core.internals.context.run(state, () => Promise.resolve(resolver({ userId: opts.userId, core: turnCore, signal }))),
|
|
104
|
+
aborted
|
|
105
|
+
]);
|
|
106
|
+
return isUserContextShape(result) ? result : undefined;
|
|
107
|
+
}
|
|
108
|
+
catch {
|
|
109
|
+
return undefined;
|
|
110
|
+
}
|
|
111
|
+
finally {
|
|
112
|
+
if (onAbort)
|
|
113
|
+
signal.removeEventListener('abort', onAbort);
|
|
114
|
+
}
|
|
115
|
+
};
|
|
116
|
+
#queryNearestMessages = async (baseQuery, vectorValue, limit, scope, fallbackChatId = '') => {
|
|
117
|
+
try {
|
|
118
|
+
await this.#core.internals.resilience.acquireLimit('firestore');
|
|
119
|
+
const snap = await baseQuery
|
|
120
|
+
.select(...RECALL_MESSAGE_FIELDS)
|
|
121
|
+
.findNearest({
|
|
122
|
+
vectorField: 'embedding',
|
|
123
|
+
queryVector: vectorValue,
|
|
124
|
+
limit,
|
|
125
|
+
distanceMeasure: 'COSINE',
|
|
126
|
+
distanceThreshold: 0.35
|
|
127
|
+
})
|
|
128
|
+
.get();
|
|
129
|
+
return snap.docs
|
|
130
|
+
.map((d) => {
|
|
131
|
+
const data = d.data();
|
|
132
|
+
const content = typeof data.content === 'string' ? data.content.trim() : '';
|
|
133
|
+
if (!content)
|
|
134
|
+
return undefined;
|
|
135
|
+
return {
|
|
136
|
+
id: d.id,
|
|
137
|
+
chatId: typeof data.chatId === 'string' ? data.chatId : fallbackChatId,
|
|
138
|
+
role: (data.role === 'assistant' ? 'assistant' : 'user'),
|
|
139
|
+
content,
|
|
140
|
+
scope
|
|
141
|
+
};
|
|
142
|
+
})
|
|
143
|
+
.filter((x) => Boolean(x));
|
|
144
|
+
}
|
|
145
|
+
catch {
|
|
146
|
+
return [];
|
|
147
|
+
}
|
|
148
|
+
};
|
|
149
|
+
#vectorRecall = async (prompt, userId, chatId, messageCount, slidingWindowSize) => {
|
|
150
|
+
const queryVector = await this.#core.ai.embed(prompt);
|
|
151
|
+
if (!queryVector)
|
|
152
|
+
return [];
|
|
153
|
+
const vectorValue = FieldValue.vector(queryVector);
|
|
154
|
+
const db = this.#core.db.firestore();
|
|
155
|
+
const inChatPromise = chatId && messageCount > slidingWindowSize
|
|
156
|
+
? this.#queryNearestMessages(db.collection('chat').doc(chatId).collection('messages'), vectorValue, 5, 'in_chat', chatId)
|
|
157
|
+
: Promise.resolve([]);
|
|
158
|
+
const crossChatPromise = userId
|
|
159
|
+
? this.#queryNearestMessages(db.collectionGroup('messages').where('userId', '==', userId), vectorValue, 3, 'cross_chat')
|
|
160
|
+
: Promise.resolve([]);
|
|
161
|
+
const [inChat, crossChat] = await Promise.all([inChatPromise, crossChatPromise]);
|
|
162
|
+
return [...inChat, ...crossChat];
|
|
163
|
+
};
|
|
164
|
+
retrieveContext = async (opts) => {
|
|
165
|
+
const { userId = this.#core.userId, chatId, messageCount = 0, prompt, useContext = true, slidingWindowSize = CORE_CONFIG.limits.chatMessagesLimit, agent, role, abortSignal } = opts;
|
|
166
|
+
const db = this.#core.db.firestore();
|
|
167
|
+
const userContextPromise = this.#resolveTurnUserContext({ userId, role, agent, abortSignal });
|
|
168
|
+
const slidingWindowPromise = (async () => {
|
|
169
|
+
if (!chatId)
|
|
170
|
+
return [];
|
|
171
|
+
try {
|
|
172
|
+
await this.#core.internals.resilience.acquireLimit('firestore');
|
|
173
|
+
const snap = await db
|
|
174
|
+
.collection('chat')
|
|
175
|
+
.doc(chatId)
|
|
176
|
+
.collection('messages')
|
|
177
|
+
.select(...PUBLIC_MESSAGE_FIELDS)
|
|
178
|
+
.orderBy('createdAt', 'desc')
|
|
179
|
+
.limit(slidingWindowSize)
|
|
180
|
+
.get();
|
|
181
|
+
return snap.docs.map((d) => mapFirestoreMessageDoc(d.id, d.data() ?? {})).reverse();
|
|
182
|
+
}
|
|
183
|
+
catch {
|
|
184
|
+
return [];
|
|
185
|
+
}
|
|
186
|
+
})();
|
|
187
|
+
const vectorRecallPromise = this.#vectorRecall(prompt, userId, chatId, messageCount, slidingWindowSize);
|
|
188
|
+
const infobasePromise = useContext !== false && prompt.trim()
|
|
189
|
+
? userContextPromise.then((userContext) => this.getContext(prompt, undefined, {
|
|
190
|
+
...(agent?.contextAudiences ? { audiences: agent.contextAudiences } : {}),
|
|
191
|
+
...(userContext ? { userState: userContext.state } : {})
|
|
192
|
+
}))
|
|
193
|
+
: Promise.resolve([]);
|
|
194
|
+
const [slidingWindowDocs, rawRecalled, infobaseContextEntries, userContext] = await Promise.all([
|
|
195
|
+
slidingWindowPromise,
|
|
196
|
+
vectorRecallPromise,
|
|
197
|
+
infobasePromise,
|
|
198
|
+
userContextPromise
|
|
199
|
+
]);
|
|
200
|
+
const windowIds = new Set(slidingWindowDocs.map((m) => m.id));
|
|
201
|
+
const seenRecallIds = new Set();
|
|
202
|
+
const recalledMemories = [];
|
|
203
|
+
for (const mem of rawRecalled) {
|
|
204
|
+
if (windowIds.has(mem.id) || seenRecallIds.has(mem.id))
|
|
205
|
+
continue;
|
|
206
|
+
seenRecallIds.add(mem.id);
|
|
207
|
+
recalledMemories.push(mem);
|
|
208
|
+
}
|
|
209
|
+
const slidingWindowMessages = slidingWindowDocs
|
|
210
|
+
.map(toModelMessageFromDoc)
|
|
211
|
+
.filter((m) => Boolean(m));
|
|
212
|
+
return {
|
|
213
|
+
slidingWindowDocs,
|
|
214
|
+
slidingWindowMessages,
|
|
215
|
+
recalledMemories,
|
|
216
|
+
infobaseContextEntries,
|
|
217
|
+
...(userContext ? { userContext } : {})
|
|
218
|
+
};
|
|
219
|
+
};
|
|
220
|
+
}
|
|
221
|
+
//# sourceMappingURL=context.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context.js","sourceRoot":"","sources":["../../src/chat/context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAA;AAC3C,OAAO,IAAI,MAAM,WAAW,CAAA;AAE5B,OAAO,EAAE,UAAU,EAAc,MAAM,0BAA0B,CAAA;AAEjE,OAAO,EAAE,WAAW,EAAe,MAAM,WAAW,CAAA;AAEpD,OAAO,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAc/B,OAAO,EACL,oBAAoB,EACpB,kBAAkB,EAClB,sBAAsB,EACtB,qBAAqB,EACrB,qBAAqB,EACrB,qBAAqB,EACtB,MAAM,SAAS,CAAA;AAEhB,IAAI,cAAc,GAA+B,EAAE,CAAA;AACnD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAA;AAE/C,MAAM,OAAO,iBAAiB;IACnB,KAAK,CAAM;IAEpB,YAAY,UAAgB;QAC1B,IAAI,CAAC,KAAK,GAAG,UAAU,CAAA;IACzB,CAAC;IAED,sBAAsB,GAAG,CAAC,KAAK,GAA+B,EAAE,EAAE,KAAK,GAAG,KAAK,EAAQ,EAAE;QACvF,IAAI,KAAK,EAAE,CAAC;YACV,cAAc,GAAG,EAAE,CAAA;YACnB,aAAa,CAAC,KAAK,EAAE,CAAA;QACvB,CAAC;QACD,cAAc,GAAG,CAAC,GAAG,KAAK,CAAC,CAAA;IAC7B,CAAC,CAAA;IAED,oBAAoB,GAAG,KAAK,EAAE,OAAe,EAA+B,EAAE;QAC5E,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QACzC,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO,MAAM,CAAA;QAEvC,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;QACxE,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA;QACnD,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7E,OAAO,SAAS,CAAA;QAClB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;YAChD,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;YACnC,OAAO,OAAO,CAAA;QAChB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAA;QAClB,CAAC;IACH,CAAC,CAAA;IAED,UAAU,GAAG,KAAK,EAChB,MAAc,EACd,OAAO,GAA+B,cAAc,EACpD,OAA2B,EACF,EAAE;QAC3B,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAA;QAC7B,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAA;QAE/C,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAC/B,CAAC,IAAI,EAAE,EAAE,CACP,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC;YAC1B,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACzB,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;YAC3B,oBAAoB,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,CAAC,CACjD,CAAA;QACD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAA;QAEtC,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,WAAW,CAAC,OAAO,CAAC,UAAU,CAAA;QACxE,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,CAAA;QACpC,MAAM,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,qDAAqD,CAAC,CAAC,CAAC,EAAE,CAAA;QACxF,MAAM,SAAS,GAAgC,EAAE,CAAA;QACjD,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;YAC/B,SAAS,CAAC,OAAO,GAAG,EAAE,CAAC,GAAG;gBACxB,IAAI,EAAE,MAAM;gBACZ,YAAY,EAAE,0BAA0B,IAAI,CAAC,IAAI,MAAM,IAAI,CAAC,WAAW,oDAAoD,SAAS,EAAE;gBACtI,QAAQ,EAAE;oBACR,IAAI,EAAE,8DAA8D,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,WAAW,GAAG;oBACrG,KAAK,EAAE,yDAAyD;iBACjE;aACF,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CACzC,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EACxD,SAAS,CACV,CAAA;YACD,MAAM,MAAM,GAAG,UAAU;iBACtB,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;gBACjB,MAAM,GAAG,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,GAAG,EAAE,CAAC,CAAA;gBAC1C,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;YAC7D,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,GAAG,CAAC;iBAC7B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;iBACjC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAA;YAEvB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAA;YAElC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAC9B,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,EAAqC,EAAE;gBAC/D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBAC3D,OAAO,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS,CAAA;YAC5F,CAAC,CAAC,CACH,CAAA;YACD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAyB,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAA;QACxE,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAA;QACX,CAAC;IACH,CAAC,CAAA;IAED,uBAAuB,GAAG,KAAK,EAAE,IAKhC,EAAoC,EAAE;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,EAAE,kBAAkB,CAAA;QAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QAChF,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,KAAK,MAAM;YAAE,OAAO,SAAS,CAAA;QAClE,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAA;QAC7E,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAA;QACxF,IAAI,OAAiC,CAAA;QACrC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAA;YAChE,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,CAAA;YAChC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAY,CAAC,OAAO,EAAE,EAAE;gBACjD,OAAO,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;gBAClC,IAAI,MAAM,CAAC,OAAO;oBAAE,OAAO,OAAO,CAAC,SAAS,CAAC,CAAA;gBAC7C,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC3D,CAAC,CAAC,CAAA;YACF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;gBAChC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,CAC3C,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC,CAC3E;gBACD,OAAO;aACR,CAAC,CAAA;YACF,OAAO,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAA;QACxD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAA;QAClB,CAAC;gBAAS,CAAC;YACT,IAAI,OAAO;gBAAE,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;QAC3D,CAAC;IACH,CAAC,CAAA;IAED,qBAAqB,GAAG,KAAK,EAC3B,SAAgB,EAChB,WAAwB,EACxB,KAAa,EACb,KAA+B,EAC/B,cAAc,GAAG,EAAE,EACe,EAAE;QACpC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,CAAC,YAAY,CAAC,WAAW,CAAC,CAAA;YAC/D,MAAM,IAAI,GAAG,MAAM,SAAS;iBACzB,MAAM,CAAC,GAAG,qBAAqB,CAAC;iBAChC,WAAW,CAAC;gBACX,WAAW,EAAE,WAAW;gBACxB,WAAW,EAAE,WAAW;gBACxB,KAAK;gBACL,eAAe,EAAE,QAAQ;gBACzB,iBAAiB,EAAE,IAAI;aACxB,CAAC;iBACD,GAAG,EAAE,CAAA;YACR,OAAO,IAAI,CAAC,IAAI;iBACb,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACT,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAA;gBACrB,MAAM,OAAO,GAAG,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAA;gBAC3E,IAAI,CAAC,OAAO;oBAAE,OAAO,SAAS,CAAA;gBAC9B,OAAO;oBACL,EAAE,EAAE,CAAC,CAAC,EAAE;oBACR,MAAM,EAAE,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,cAAc;oBACtE,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAoB;oBAC3E,OAAO;oBACP,KAAK;iBACN,CAAA;YACH,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,CAAC,EAA8B,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAA;QAC1D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAA;QACX,CAAC;IACH,CAAC,CAAA;IAED,aAAa,GAAG,KAAK,EACnB,MAAc,EACd,MAAe,EACf,MAA0B,EAC1B,YAAoB,EACpB,iBAAyB,EACS,EAAE;QACpC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;QACrD,IAAI,CAAC,WAAW;YAAE,OAAO,EAAE,CAAA;QAC3B,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC,WAAW,CAAC,CAAA;QAClD,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,CAAA;QAEpC,MAAM,aAAa,GACjB,MAAM,IAAI,YAAY,GAAG,iBAAiB;YACxC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CACxB,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,EACxD,WAAW,EACX,CAAC,EACD,SAAS,EACT,MAAM,CACP;YACH,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAA6B,CAAC,CAAA;QAEpD,MAAM,gBAAgB,GAAG,MAAM;YAC7B,CAAC,CAAC,IAAI,CAAC,qBAAqB,CACxB,EAAE,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,EAC5D,WAAW,EACX,CAAC,EACD,YAAY,CACb;YACH,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAA6B,CAAC,CAAA;QAElD,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC,CAAA;QAChF,OAAO,CAAC,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,CAAA;IAClC,CAAC,CAAA;IAED,eAAe,GAAG,KAAK,EAAE,IAA4B,EAAkC,EAAE;QACvF,MAAM,EACJ,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAC1B,MAAM,EACN,YAAY,GAAG,CAAC,EAChB,MAAM,EACN,UAAU,GAAG,IAAI,EACjB,iBAAiB,GAAG,WAAW,CAAC,MAAM,CAAC,iBAAiB,EACxD,KAAK,EACL,IAAI,EACJ,WAAW,EACZ,GAAG,IAAI,CAAA;QACR,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,CAAA;QAEpC,MAAM,kBAAkB,GAAG,IAAI,CAAC,uBAAuB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAA;QAE7F,MAAM,oBAAoB,GAAG,CAAC,KAAK,IAAkC,EAAE;YACrE,IAAI,CAAC,MAAM;gBAAE,OAAO,EAAE,CAAA;YACtB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,CAAC,YAAY,CAAC,WAAW,CAAC,CAAA;gBAC/D,MAAM,IAAI,GAAG,MAAM,EAAE;qBAClB,UAAU,CAAC,MAAM,CAAC;qBAClB,GAAG,CAAC,MAAM,CAAC;qBACX,UAAU,CAAC,UAAU,CAAC;qBACtB,MAAM,CAAC,GAAG,qBAAqB,CAAC;qBAChC,OAAO,CAAC,WAAW,EAAE,MAAM,CAAC;qBAC5B,KAAK,CAAC,iBAAiB,CAAC;qBACxB,GAAG,EAAE,CAAA;gBACR,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAA;YACrF,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,CAAA;YACX,CAAC;QACH,CAAC,CAAC,EAAE,CAAA;QAEJ,MAAM,mBAAmB,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,iBAAiB,CAAC,CAAA;QAEvG,MAAM,eAAe,GACnB,UAAU,KAAK,KAAK,IAAI,MAAM,CAAC,IAAI,EAAE;YACnC,CAAC,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,EAAE,CACtC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,SAAS,EAAE;gBACjC,GAAG,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzE,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACzD,CAAC,CACH;YACH,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,EAAoB,CAAC,CAAA;QAE3C,MAAM,CAAC,iBAAiB,EAAE,WAAW,EAAE,sBAAsB,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAC9F,oBAAoB;YACpB,mBAAmB;YACnB,eAAe;YACf,kBAAkB;SACnB,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;QAC7D,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAA;QACvC,MAAM,gBAAgB,GAA4B,EAAE,CAAA;QACpD,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC9B,IAAI,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAAE,SAAQ;YAChE,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;YACzB,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QAC5B,CAAC;QAED,MAAM,qBAAqB,GAAG,iBAAiB;aAC5C,GAAG,CAAC,qBAAqB,CAAC;aAC1B,MAAM,CAAC,CAAC,CAAC,EAAqB,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAA;QAE/C,OAAO;YACL,iBAAiB;YACjB,qBAAqB;YACrB,gBAAgB;YAChB,sBAAsB;YACtB,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACxC,CAAA;IACH,CAAC,CAAA;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/chat/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAA;AACxB,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAA;AACvC,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/chat/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAA;AACxB,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAA;AACvC,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA"}
|
|
@@ -0,0 +1,160 @@
|
|
|
1
|
+
import type { Context } from 'hono';
|
|
2
|
+
import type { z } from 'zod';
|
|
3
|
+
import { type AppEnv } from '../config';
|
|
4
|
+
import type { EntryId } from '../db';
|
|
5
|
+
import { openaiCompletionsBody, saveChatBody, streamChatBody, type WorkflowToolSpec } from './schema';
|
|
6
|
+
declare const listOpenaiModelsService: () => Promise<{
|
|
7
|
+
object: string;
|
|
8
|
+
data: {
|
|
9
|
+
id: string;
|
|
10
|
+
object: string;
|
|
11
|
+
created: number;
|
|
12
|
+
owned_by: string;
|
|
13
|
+
}[];
|
|
14
|
+
}>;
|
|
15
|
+
declare const openaiCompletionsService: ({ c, userId, input }: {
|
|
16
|
+
c: Context<AppEnv>;
|
|
17
|
+
userId: EntryId;
|
|
18
|
+
input?: z.infer<typeof openaiCompletionsBody>;
|
|
19
|
+
}) => Promise<Response | {
|
|
20
|
+
id: string;
|
|
21
|
+
object: string;
|
|
22
|
+
created: number;
|
|
23
|
+
model: string;
|
|
24
|
+
choices: {
|
|
25
|
+
index: number;
|
|
26
|
+
message: {
|
|
27
|
+
role: string;
|
|
28
|
+
content: string;
|
|
29
|
+
};
|
|
30
|
+
finish_reason: string;
|
|
31
|
+
}[];
|
|
32
|
+
usage: {
|
|
33
|
+
prompt_tokens: number;
|
|
34
|
+
completion_tokens: number;
|
|
35
|
+
total_tokens: number;
|
|
36
|
+
};
|
|
37
|
+
}>;
|
|
38
|
+
export declare const mcpRoute: (request: Request, userId: EntryId, mcpTools: Map<string, WorkflowToolSpec>) => Promise<Response>;
|
|
39
|
+
export declare const coreChatRoutes: {
|
|
40
|
+
'chat.list': {
|
|
41
|
+
handler: ({ c, userId }: {
|
|
42
|
+
c: Context<AppEnv>;
|
|
43
|
+
userId: EntryId;
|
|
44
|
+
}) => Promise<import("../db").PaginatedList<import("../db").WithBaseFields<{
|
|
45
|
+
userId: string;
|
|
46
|
+
agentId: string;
|
|
47
|
+
title: string;
|
|
48
|
+
model: string;
|
|
49
|
+
messageCount: number;
|
|
50
|
+
summary?: string | null | undefined;
|
|
51
|
+
activeGenerationId?: string | null | undefined;
|
|
52
|
+
}>>>;
|
|
53
|
+
};
|
|
54
|
+
'chat.get': {
|
|
55
|
+
handler: ({ c }: {
|
|
56
|
+
c: Context<AppEnv>;
|
|
57
|
+
}) => Promise<import("../db").WithBaseFields<{
|
|
58
|
+
userId: string;
|
|
59
|
+
agentId: string;
|
|
60
|
+
title: string;
|
|
61
|
+
model: string;
|
|
62
|
+
messageCount: number;
|
|
63
|
+
summary?: string | null | undefined;
|
|
64
|
+
activeGenerationId?: string | null | undefined;
|
|
65
|
+
}>>;
|
|
66
|
+
};
|
|
67
|
+
'chat.messages': {
|
|
68
|
+
handler: ({ c }: {
|
|
69
|
+
c: Context<AppEnv>;
|
|
70
|
+
}) => Promise<import("./schema").ChatMessagePublic[]>;
|
|
71
|
+
};
|
|
72
|
+
'chat.save': {
|
|
73
|
+
input: z.ZodObject<{
|
|
74
|
+
id: z.ZodOptional<z.ZodString>;
|
|
75
|
+
chatId: z.ZodOptional<z.ZodString>;
|
|
76
|
+
title: z.ZodOptional<z.ZodString>;
|
|
77
|
+
agentId: z.ZodOptional<z.ZodString>;
|
|
78
|
+
model: z.ZodOptional<z.ZodString>;
|
|
79
|
+
summary: z.ZodOptional<z.ZodNullable<z.ZodString>>;
|
|
80
|
+
}, z.core.$strict>;
|
|
81
|
+
handler: ({ userId, input }: {
|
|
82
|
+
userId: EntryId;
|
|
83
|
+
input: z.infer<typeof saveChatBody>;
|
|
84
|
+
}) => Promise<import("../db").WithBaseFields<{
|
|
85
|
+
userId: string;
|
|
86
|
+
agentId: string;
|
|
87
|
+
title: string;
|
|
88
|
+
model: string;
|
|
89
|
+
messageCount: number;
|
|
90
|
+
summary?: string | null | undefined;
|
|
91
|
+
activeGenerationId?: string | null | undefined;
|
|
92
|
+
}>>;
|
|
93
|
+
};
|
|
94
|
+
'chat.delete': {
|
|
95
|
+
handler: ({ c }: {
|
|
96
|
+
c: Context<AppEnv>;
|
|
97
|
+
}) => Promise<{
|
|
98
|
+
id: string;
|
|
99
|
+
deleted: boolean;
|
|
100
|
+
}>;
|
|
101
|
+
};
|
|
102
|
+
'chat.stream': {
|
|
103
|
+
input: z.ZodObject<{
|
|
104
|
+
id: z.ZodOptional<z.ZodString>;
|
|
105
|
+
chatId: z.ZodOptional<z.ZodString>;
|
|
106
|
+
agentId: z.ZodOptional<z.ZodString>;
|
|
107
|
+
title: z.ZodOptional<z.ZodString>;
|
|
108
|
+
model: z.ZodOptional<z.ZodString>;
|
|
109
|
+
message: z.ZodObject<{
|
|
110
|
+
id: z.ZodOptional<z.ZodString>;
|
|
111
|
+
role: z.ZodDefault<z.ZodLiteral<"user">>;
|
|
112
|
+
content: z.ZodOptional<z.ZodString>;
|
|
113
|
+
parts: z.ZodOptional<z.ZodArray<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
|
|
114
|
+
attachments: z.ZodOptional<z.ZodArray<z.ZodString>>;
|
|
115
|
+
}, z.core.$strip>;
|
|
116
|
+
}, z.core.$strict>;
|
|
117
|
+
handler: ({ c, userId, input }: {
|
|
118
|
+
c: Context<AppEnv>;
|
|
119
|
+
userId: EntryId;
|
|
120
|
+
input?: z.infer<typeof streamChatBody>;
|
|
121
|
+
}) => Promise<Response>;
|
|
122
|
+
};
|
|
123
|
+
'chat.cancel': {
|
|
124
|
+
handler: ({ c }: {
|
|
125
|
+
c: Context<AppEnv>;
|
|
126
|
+
}) => Promise<{
|
|
127
|
+
chatId: string;
|
|
128
|
+
cancelled: true;
|
|
129
|
+
}>;
|
|
130
|
+
};
|
|
131
|
+
'openai.models': {
|
|
132
|
+
handler: typeof listOpenaiModelsService;
|
|
133
|
+
};
|
|
134
|
+
'openai.completions': {
|
|
135
|
+
input: z.ZodObject<{
|
|
136
|
+
messages: z.ZodArray<z.ZodObject<{
|
|
137
|
+
role: z.ZodString;
|
|
138
|
+
content: z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodObject<{
|
|
139
|
+
type: z.ZodString;
|
|
140
|
+
text: z.ZodOptional<z.ZodString>;
|
|
141
|
+
image_url: z.ZodOptional<z.ZodObject<{
|
|
142
|
+
url: z.ZodOptional<z.ZodString>;
|
|
143
|
+
}, z.core.$strip>>;
|
|
144
|
+
file: z.ZodOptional<z.ZodObject<{
|
|
145
|
+
file_data: z.ZodOptional<z.ZodString>;
|
|
146
|
+
filename: z.ZodOptional<z.ZodString>;
|
|
147
|
+
}, z.core.$strip>>;
|
|
148
|
+
}, z.core.$loose>>, z.ZodNull]>>;
|
|
149
|
+
}, z.core.$loose>>;
|
|
150
|
+
stream: z.ZodOptional<z.ZodBoolean>;
|
|
151
|
+
stream_options: z.ZodOptional<z.ZodObject<{
|
|
152
|
+
include_usage: z.ZodOptional<z.ZodBoolean>;
|
|
153
|
+
}, z.core.$strip>>;
|
|
154
|
+
model: z.ZodOptional<z.ZodString>;
|
|
155
|
+
}, z.core.$loose>;
|
|
156
|
+
handler: typeof openaiCompletionsService;
|
|
157
|
+
};
|
|
158
|
+
};
|
|
159
|
+
export {};
|
|
160
|
+
//# sourceMappingURL=routes.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"routes.d.ts","sourceRoot":"","sources":["../../src/chat/routes.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,MAAM,CAAA;AACnC,OAAO,KAAK,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAC5B,OAAO,EAAe,KAAK,MAAM,EAAE,MAAM,WAAW,CAAA;AACpD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,OAAO,CAAA;AAEpC,OAAO,EAEL,qBAAqB,EACrB,YAAY,EACZ,cAAc,EAEd,KAAK,gBAAgB,EACtB,MAAM,UAAU,CAAA;AAGjB,QAAA,MAAM,uBAAuB;;;;;;;;EAW5B,CAAA;AA+DD,QAAA,MAAM,wBAAwB,yBAI3B;IACD,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;IAClB,MAAM,EAAE,OAAO,CAAA;IACf,KAAK,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAA;CAC9C;;;;;;;;YAsCsC,IAAI;YAAe,OAAO;;;;;;;;;EAMhE,CAAA;AAED,eAAO,MAAM,QAAQ,YACV,OAAO,UACR,OAAO,YACL,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,KACtC,OAAO,CAAC,QAAQ,CAkClB,CAAA;AAED,eAAO,MAAM,cAAc;IACzB,WAAW;QACT,OAAO,kBAAwB;YAAE,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;YAAC,MAAM,EAAE,OAAO,CAAA;SAAE;;;;;;;;;;IA0BxE,UAAU;QACR,OAAO,UAAgB;YAAE,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;SAAE;;;;;;;;;;IAK/C,eAAe;QACb,OAAO,UAAgB;YAAE,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;SAAE;;IAM/C,WAAW;QACT,KAAK;;;;;;;;QACL,OAAO,sBAAsB;YAAE,MAAM,EAAE,OAAO,CAAC;YAAC,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAA;SAAE;;;;;;;;;;IAGvF,aAAa;QACX,OAAO,UAAgB;YAAE,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;SAAE;;;;;IAM/C,aAAa;QACX,KAAK;;;;;;;;;;;;;;QACL,OAAO,yBAIJ;YACD,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;YAClB,MAAM,EAAE,OAAO,CAAA;YACf,KAAK,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAA;SACvC;;IAEH,aAAa;QACX,OAAO,UAAU;YAAE,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;SAAE;;;;;IAKzC,eAAe;QACb,OAAO;;IAET,oBAAoB;QAClB,KAAK;;;;;;;;;;;;;;;;;;;;;QACL,OAAO;;CAEV,CAAA"}
|
|
@@ -0,0 +1,210 @@
|
|
|
1
|
+
import { createMcpHandler, McpServer } from '@modelcontextprotocol/server';
|
|
2
|
+
import { CORE_CONFIG } from '../config';
|
|
3
|
+
import { core } from '../index';
|
|
4
|
+
import { listChatMessagesQuery, openaiCompletionsBody, saveChatBody, streamChatBody } from './schema';
|
|
5
|
+
import { extractLatestUserPrompt, formatUsage, sanitizeToolInputSchema } from './utils';
|
|
6
|
+
const listOpenaiModelsService = async () => {
|
|
7
|
+
const agents = core.internals.registry.getRegisteredAgents();
|
|
8
|
+
return {
|
|
9
|
+
object: 'list',
|
|
10
|
+
data: Object.keys(agents).map((id) => ({
|
|
11
|
+
id,
|
|
12
|
+
object: 'model',
|
|
13
|
+
created: Math.floor(Date.now() / 1000),
|
|
14
|
+
owned_by: CORE_CONFIG.appName
|
|
15
|
+
}))
|
|
16
|
+
};
|
|
17
|
+
};
|
|
18
|
+
const parseOpenaiMessages = (messages) => {
|
|
19
|
+
const parts = [];
|
|
20
|
+
const modelMessages = [];
|
|
21
|
+
for (const message of messages) {
|
|
22
|
+
if (message.role !== 'user' && message.role !== 'assistant')
|
|
23
|
+
continue;
|
|
24
|
+
const text = typeof message.content === 'string'
|
|
25
|
+
? message.content
|
|
26
|
+
: (message.content ?? [])
|
|
27
|
+
.filter((p) => p.type === 'text')
|
|
28
|
+
.map((p) => p.text ?? '')
|
|
29
|
+
.join('\n\n');
|
|
30
|
+
if (message.role === 'assistant' && text.trim()) {
|
|
31
|
+
modelMessages.push({ role: 'assistant', content: [{ type: 'text', text }] });
|
|
32
|
+
}
|
|
33
|
+
if (message.role !== 'user')
|
|
34
|
+
continue;
|
|
35
|
+
for (const p of Array.isArray(message.content) ? message.content : []) {
|
|
36
|
+
const dataUrl = p.image_url?.url ?? p.file?.file_data;
|
|
37
|
+
if (dataUrl?.startsWith('data:'))
|
|
38
|
+
parts.push({ dataUrl, filename: p.file?.filename ?? 'file' });
|
|
39
|
+
}
|
|
40
|
+
if (text.trim())
|
|
41
|
+
modelMessages.push({ role: 'user', content: [{ type: 'text', text }] });
|
|
42
|
+
}
|
|
43
|
+
return { parts, modelMessages };
|
|
44
|
+
};
|
|
45
|
+
const createOpenaiSseResponse = (turn, id, created, model, includeUsage) => {
|
|
46
|
+
const encoder = new TextEncoder();
|
|
47
|
+
const stream = new ReadableStream({
|
|
48
|
+
async start(controller) {
|
|
49
|
+
const send = (payload) => controller.enqueue(encoder.encode(`data: ${JSON.stringify(payload)}\n\n`));
|
|
50
|
+
const sendChunk = (choices, extra) => send({ id, object: 'chat.completion.chunk', created, model, choices, ...extra });
|
|
51
|
+
sendChunk([{ index: 0, delta: { role: 'assistant' }, finish_reason: null }]);
|
|
52
|
+
try {
|
|
53
|
+
for await (const delta of turn.textStream) {
|
|
54
|
+
if (delta)
|
|
55
|
+
sendChunk([{ index: 0, delta: { content: delta }, finish_reason: null }]);
|
|
56
|
+
}
|
|
57
|
+
sendChunk([{ index: 0, delta: {}, finish_reason: 'stop' }]);
|
|
58
|
+
if (includeUsage)
|
|
59
|
+
sendChunk([], { usage: formatUsage(await turn.usage) });
|
|
60
|
+
}
|
|
61
|
+
catch (error) {
|
|
62
|
+
console.error('[openai] stream failed:', error);
|
|
63
|
+
send({ error: { message: 'The agent failed to produce a reply. Please retry.' } });
|
|
64
|
+
}
|
|
65
|
+
controller.enqueue(encoder.encode('data: [DONE]\n\n'));
|
|
66
|
+
controller.close();
|
|
67
|
+
}
|
|
68
|
+
});
|
|
69
|
+
return new Response(stream, {
|
|
70
|
+
headers: { 'content-type': 'text/event-stream; charset=utf-8', 'cache-control': 'no-cache' }
|
|
71
|
+
});
|
|
72
|
+
};
|
|
73
|
+
const openaiCompletionsService = async ({ c, userId, input }) => {
|
|
74
|
+
const body = input ?? openaiCompletionsBody.parse(await c.req.json().catch(() => ({})));
|
|
75
|
+
const agents = core.internals.registry.getRegisteredAgents();
|
|
76
|
+
const entries = Object.entries(agents);
|
|
77
|
+
if (entries.length === 0)
|
|
78
|
+
return core.errors.httpError(404, 'No registered agents', 'NOT_FOUND');
|
|
79
|
+
const [defaultModel, defaultAgent] = entries[0];
|
|
80
|
+
const modelKey = body.model && body.model in agents ? body.model : defaultModel;
|
|
81
|
+
const agent = agents[modelKey] ?? defaultAgent;
|
|
82
|
+
const model = agent.name ?? modelKey;
|
|
83
|
+
await core.internals.context.enforceAgentAccess(c, agent, 'openai.completions');
|
|
84
|
+
const { parts, modelMessages } = parseOpenaiMessages(body.messages);
|
|
85
|
+
if (modelMessages.length === 0 && parts.length === 0) {
|
|
86
|
+
return core.errors.httpError(400, 'No supported messages', 'VALIDATION_ERROR');
|
|
87
|
+
}
|
|
88
|
+
const latestPrompt = extractLatestUserPrompt(modelMessages);
|
|
89
|
+
const role = core.internals.auth.resolveUserRole(userId, c.get('userRole'));
|
|
90
|
+
const { turn, userCreatedAtMs } = await core.internals.context.startChatStreamTurn({
|
|
91
|
+
userId,
|
|
92
|
+
role,
|
|
93
|
+
agent,
|
|
94
|
+
attachmentParts: parts,
|
|
95
|
+
rawUserText: latestPrompt,
|
|
96
|
+
signal: c.req.raw.signal,
|
|
97
|
+
priorMessages: modelMessages
|
|
98
|
+
});
|
|
99
|
+
const id = `chatcmpl-${crypto.randomUUID().replace(/-/g, '').slice(0, 24)}`;
|
|
100
|
+
const created = Math.floor(userCreatedAtMs / 1000);
|
|
101
|
+
if (!body.stream) {
|
|
102
|
+
const [text, usage] = await Promise.all([turn.text, turn.usage]);
|
|
103
|
+
return {
|
|
104
|
+
id,
|
|
105
|
+
object: 'chat.completion',
|
|
106
|
+
created,
|
|
107
|
+
model,
|
|
108
|
+
choices: [{ index: 0, message: { role: 'assistant', content: text }, finish_reason: 'stop' }],
|
|
109
|
+
usage: formatUsage(usage)
|
|
110
|
+
};
|
|
111
|
+
}
|
|
112
|
+
return createOpenaiSseResponse(turn, id, created, model, Boolean(body.stream_options?.include_usage));
|
|
113
|
+
};
|
|
114
|
+
export const mcpRoute = async (request, userId, mcpTools) => {
|
|
115
|
+
const role = core.internals.auth.resolveUserRole(userId);
|
|
116
|
+
const server = new McpServer({ name: CORE_CONFIG.appName, version: '1.0.0' });
|
|
117
|
+
for (const [name, toolSpec] of mcpTools) {
|
|
118
|
+
if (toolSpec.mcp !== true)
|
|
119
|
+
continue;
|
|
120
|
+
server.registerTool(name, {
|
|
121
|
+
description: (toolSpec.description || name).slice(0, 1024),
|
|
122
|
+
inputSchema: sanitizeToolInputSchema(toolSpec.input)
|
|
123
|
+
}, async (input) => {
|
|
124
|
+
try {
|
|
125
|
+
const step = core.internals.workflows.createStepRunner('workflow');
|
|
126
|
+
const result = await core.internals.context.run({ env: core.env, userId, role, step }, () => toolSpec.execute(input, userId));
|
|
127
|
+
return { content: [{ type: 'text', text: JSON.stringify(result) }] };
|
|
128
|
+
}
|
|
129
|
+
catch (err) {
|
|
130
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
131
|
+
return {
|
|
132
|
+
content: [
|
|
133
|
+
{
|
|
134
|
+
type: 'text',
|
|
135
|
+
text: JSON.stringify(core.errors.buildErrorBody(msg, 502, 'WORKFLOW_ERROR'))
|
|
136
|
+
}
|
|
137
|
+
],
|
|
138
|
+
isError: true
|
|
139
|
+
};
|
|
140
|
+
}
|
|
141
|
+
});
|
|
142
|
+
}
|
|
143
|
+
return createMcpHandler(() => server).fetch(request);
|
|
144
|
+
};
|
|
145
|
+
export const coreChatRoutes = {
|
|
146
|
+
'chat.list': {
|
|
147
|
+
handler: async ({ c, userId }) => core.db.list('chat', {
|
|
148
|
+
userId,
|
|
149
|
+
...(c?.req.query('full') === 'true'
|
|
150
|
+
? {}
|
|
151
|
+
: {
|
|
152
|
+
select: [
|
|
153
|
+
'id',
|
|
154
|
+
'userId',
|
|
155
|
+
'agentId',
|
|
156
|
+
'title',
|
|
157
|
+
'model',
|
|
158
|
+
'messageCount',
|
|
159
|
+
'summary',
|
|
160
|
+
'activeGenerationId',
|
|
161
|
+
'deleted',
|
|
162
|
+
'createdAt',
|
|
163
|
+
'updatedAt'
|
|
164
|
+
]
|
|
165
|
+
})
|
|
166
|
+
}, { bypassRbac: true })
|
|
167
|
+
},
|
|
168
|
+
'chat.get': {
|
|
169
|
+
handler: async ({ c }) => {
|
|
170
|
+
const chat = await core.db.read('chat', c.req.param('chatId') ?? '', false);
|
|
171
|
+
return chat ?? core.errors.httpError(404, 'Chat not found', 'NOT_FOUND');
|
|
172
|
+
}
|
|
173
|
+
},
|
|
174
|
+
'chat.messages': {
|
|
175
|
+
handler: async ({ c }) => {
|
|
176
|
+
const chatId = c.req.param('chatId') ?? '';
|
|
177
|
+
const parsedQuery = listChatMessagesQuery.parse(c.req.query());
|
|
178
|
+
return core.chat.messages(chatId, { limit: parsedQuery.limit, before: parsedQuery.before });
|
|
179
|
+
}
|
|
180
|
+
},
|
|
181
|
+
'chat.save': {
|
|
182
|
+
input: saveChatBody,
|
|
183
|
+
handler: ({ userId, input }) => core.internals.context.saveChat({ userId, input })
|
|
184
|
+
},
|
|
185
|
+
'chat.delete': {
|
|
186
|
+
handler: async ({ c }) => {
|
|
187
|
+
const chatId = c.req.param('chatId') ?? '';
|
|
188
|
+
await core.db.delete('chat', chatId);
|
|
189
|
+
return { id: chatId, deleted: true };
|
|
190
|
+
}
|
|
191
|
+
},
|
|
192
|
+
'chat.stream': {
|
|
193
|
+
input: streamChatBody,
|
|
194
|
+
handler: ({ c, userId, input }) => core.internals.context.streamChatSse({ c, userId, input })
|
|
195
|
+
},
|
|
196
|
+
'chat.cancel': {
|
|
197
|
+
handler: ({ c }) => {
|
|
198
|
+
const chatId = c.req.param('chatId') ?? '';
|
|
199
|
+
return core.chat.cancel(chatId);
|
|
200
|
+
}
|
|
201
|
+
},
|
|
202
|
+
'openai.models': {
|
|
203
|
+
handler: listOpenaiModelsService
|
|
204
|
+
},
|
|
205
|
+
'openai.completions': {
|
|
206
|
+
input: openaiCompletionsBody,
|
|
207
|
+
handler: openaiCompletionsService
|
|
208
|
+
}
|
|
209
|
+
};
|
|
210
|
+
//# sourceMappingURL=routes.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"routes.js","sourceRoot":"","sources":["../../src/chat/routes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,8BAA8B,CAAA;AAI1E,OAAO,EAAE,WAAW,EAAe,MAAM,WAAW,CAAA;AAEpD,OAAO,EAAE,IAAI,EAAE,MAAM,UAAU,CAAA;AAC/B,OAAO,EACL,qBAAqB,EACrB,qBAAqB,EACrB,YAAY,EACZ,cAAc,EAGf,MAAM,UAAU,CAAA;AACjB,OAAO,EAAE,uBAAuB,EAAE,WAAW,EAAE,uBAAuB,EAAE,MAAM,SAAS,CAAA;AAEvF,MAAM,uBAAuB,GAAG,KAAK,IAAI,EAAE;IACzC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,mBAAmB,EAAE,CAAA;IAC5D,OAAO;QACL,MAAM,EAAE,MAAM;QACd,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YACrC,EAAE;YACF,MAAM,EAAE,OAAO;YACf,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACtC,QAAQ,EAAE,WAAW,CAAC,OAAO;SAC9B,CAAC,CAAC;KACJ,CAAA;AACH,CAAC,CAAA;AAED,MAAM,mBAAmB,GAAG,CAC1B,QAA2D,EACI,EAAE;IACjE,MAAM,KAAK,GAAwB,EAAE,CAAA;IACrC,MAAM,aAAa,GAAmB,EAAE,CAAA;IACxC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;YAAE,SAAQ;QACrE,MAAM,IAAI,GACR,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;YACjC,CAAC,CAAC,OAAO,CAAC,OAAO;YACjB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;iBACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;iBACxB,IAAI,CAAC,MAAM,CAAC,CAAA;QACrB,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YAChD,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAA;QAC9E,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM;YAAE,SAAQ;QACrC,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;YACtE,MAAM,OAAO,GAAG,CAAC,CAAC,SAAS,EAAE,GAAG,IAAI,CAAC,CAAC,IAAI,EAAE,SAAS,CAAA;YACrD,IAAI,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,IAAI,MAAM,EAAE,CAAC,CAAA;QACjG,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,EAAE;YAAE,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAA;IAC1F,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,CAAA;AACjC,CAAC,CAAA;AAED,MAAM,uBAAuB,GAAG,CAC9B,IAAmC,EACnC,EAAU,EACV,OAAe,EACf,KAAa,EACb,YAAqB,EACX,EAAE;IACZ,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;IACjC,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,KAAK,CAAC,KAAK,CAAC,UAAU;YACpB,MAAM,IAAI,GAAG,CAAC,OAAgB,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAA;YAC7G,MAAM,SAAS,GAAG,CAAC,OAAkB,EAAE,KAA+B,EAAE,EAAE,CACxE,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,uBAAuB,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,KAAK,EAAE,CAAC,CAAA;YAElF,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;YAC5E,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBAC1C,IAAI,KAAK;wBAAE,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CAAA;gBACtF,CAAC;gBACD,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;gBAC3D,IAAI,YAAY;oBAAE,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,WAAW,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YAC3E,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAA;gBAC/C,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,oDAAoD,EAAE,EAAE,CAAC,CAAA;YACpF,CAAC;YACD,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAA;YACtD,UAAU,CAAC,KAAK,EAAE,CAAA;QACpB,CAAC;KACF,CAAC,CAAA;IACF,OAAO,IAAI,QAAQ,CAAC,MAAM,EAAE;QAC1B,OAAO,EAAE,EAAE,cAAc,EAAE,kCAAkC,EAAE,eAAe,EAAE,UAAU,EAAE;KAC7F,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,MAAM,wBAAwB,GAAG,KAAK,EAAE,EACtC,CAAC,EACD,MAAM,EACN,KAAK,EAKN,EAAE,EAAE;IACH,MAAM,IAAI,GAAG,KAAK,IAAI,qBAAqB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;IACvF,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,mBAAmB,EAAE,CAAA;IAC5D,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IACtC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,sBAAsB,EAAE,WAAW,CAAC,CAAA;IAChG,MAAM,CAAC,YAAY,EAAE,YAAY,CAAC,GAAG,OAAO,CAAC,CAAC,CAAE,CAAA;IAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,YAAY,CAAA;IAC/E,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,YAAY,CAAA;IAC9C,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,IAAI,QAAQ,CAAA;IAEpC,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,EAAE,KAAK,EAAE,oBAAoB,CAAC,CAAA;IAE/E,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,GAAG,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IACnE,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrD,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,uBAAuB,EAAE,kBAAkB,CAAC,CAAA;IAChF,CAAC;IAED,MAAM,YAAY,GAAG,uBAAuB,CAAC,aAAa,CAAC,CAAA;IAC3D,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAA;IAC3E,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,mBAAmB,CAAC;QACjF,MAAM;QACN,IAAI;QACJ,KAAK;QACL,eAAe,EAAE,KAAK;QACtB,WAAW,EAAE,YAAY;QACzB,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM;QACxB,aAAa,EAAE,aAAa;KAC7B,CAAC,CAAA;IACF,MAAM,EAAE,GAAG,YAAY,MAAM,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAA;IAC3E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,CAAA;IAElD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACjB,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;QAChE,OAAO;YACL,EAAE;YACF,MAAM,EAAE,iBAAiB;YACzB,OAAO;YACP,KAAK;YACL,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;YAC7F,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC;SAC1B,CAAA;IACH,CAAC;IAED,OAAO,uBAAuB,CAAC,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC,CAAA;AACvG,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,QAAQ,GAAG,KAAK,EAC3B,OAAgB,EAChB,MAAe,EACf,QAAuC,EACpB,EAAE;IACrB,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAA;IACxD,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAA;IAC7E,KAAK,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,QAAQ,EAAE,CAAC;QACxC,IAAI,QAAQ,CAAC,GAAG,KAAK,IAAI;YAAE,SAAQ;QACnC,MAAM,CAAC,YAAY,CACjB,IAAI,EACJ;YACE,WAAW,EAAE,CAAC,QAAQ,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;YAC1D,WAAW,EAAE,uBAAuB,CAAC,QAAQ,CAAC,KAAK,CAAC;SACrD,EACD,KAAK,EAAE,KAAK,EAAE,EAAE;YACd,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAA;gBAClE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,CAC1F,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAChC,CAAA;gBACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAA;YAC/E,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;gBAC5D,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,EAAE,GAAG,EAAE,gBAAgB,CAAC,CAAC;yBAC7E;qBACF;oBACD,OAAO,EAAE,IAAI;iBACd,CAAA;YACH,CAAC;QACH,CAAC,CACF,CAAA;IACH,CAAC;IACD,OAAO,gBAAgB,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;AACtD,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,cAAc,GAAG;IAC5B,WAAW,EAAE;QACX,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,MAAM,EAA2C,EAAE,EAAE,CACxE,IAAI,CAAC,EAAE,CAAC,IAAI,CACV,MAAM,EACN;YACE,MAAM;YACN,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,MAAM;gBACjC,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC;oBACE,MAAM,EAAE;wBACN,IAAI;wBACJ,QAAQ;wBACR,SAAS;wBACT,OAAO;wBACP,OAAO;wBACP,cAAc;wBACd,SAAS;wBACT,oBAAoB;wBACpB,SAAS;wBACT,WAAW;wBACX,WAAW;qBACZ;iBACF,CAAC;SACP,EACD,EAAE,UAAU,EAAE,IAAI,EAAE,CACrB;KACJ;IACD,UAAU,EAAE;QACV,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,EAA0B,EAAE,EAAE;YAC/C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,KAAK,CAAC,CAAA;YAC3E,OAAO,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,gBAAgB,EAAE,WAAW,CAAC,CAAA;QAC1E,CAAC;KACF;IACD,eAAe,EAAE;QACf,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,EAA0B,EAAE,EAAE;YAC/C,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAA;YAC1C,MAAM,WAAW,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAA;YAC9D,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,CAAC,CAAA;QAC7F,CAAC;KACF;IACD,WAAW,EAAE;QACX,KAAK,EAAE,YAAY;QACnB,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,KAAK,EAA4D,EAAE,EAAE,CACvF,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;KACrD;IACD,aAAa,EAAE;QACb,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,EAA0B,EAAE,EAAE;YAC/C,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAA;YAC1C,MAAM,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;YACpC,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAA;QACtC,CAAC;KACF;IACD,aAAa,EAAE;QACb,KAAK,EAAE,cAAc;QACrB,OAAO,EAAE,CAAC,EACR,CAAC,EACD,MAAM,EACN,KAAK,EAKN,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;KACjE;IACD,aAAa,EAAE;QACb,OAAO,EAAE,CAAC,EAAE,CAAC,EAA0B,EAAE,EAAE;YACzC,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAA;YAC1C,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;QACjC,CAAC;KACF;IACD,eAAe,EAAE;QACf,OAAO,EAAE,uBAAuB;KACjC;IACD,oBAAoB,EAAE;QACpB,KAAK,EAAE,qBAAqB;QAC5B,OAAO,EAAE,wBAAwB;KAClC;CACF,CAAA"}
|