@multiplayer-app/ai-agent-node 0.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/.env.example +45 -0
- package/README.md +611 -0
- package/config.example.json +73 -0
- package/dist/config.d.ts +35 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +44 -0
- package/dist/config.js.map +1 -0
- package/dist/helpers/AIHelper.d.ts +23 -0
- package/dist/helpers/AIHelper.d.ts.map +1 -0
- package/dist/helpers/AIHelper.js +326 -0
- package/dist/helpers/AIHelper.js.map +1 -0
- package/dist/helpers/AIHelper.test.d.ts +2 -0
- package/dist/helpers/AIHelper.test.d.ts.map +1 -0
- package/dist/helpers/AIHelper.test.js +332 -0
- package/dist/helpers/AIHelper.test.js.map +1 -0
- package/dist/helpers/ConfigHelper.d.ts +20 -0
- package/dist/helpers/ConfigHelper.d.ts.map +1 -0
- package/dist/helpers/ConfigHelper.js +118 -0
- package/dist/helpers/ConfigHelper.js.map +1 -0
- package/dist/helpers/ContextLimiter.d.ts +82 -0
- package/dist/helpers/ContextLimiter.d.ts.map +1 -0
- package/dist/helpers/ContextLimiter.js +165 -0
- package/dist/helpers/ContextLimiter.js.map +1 -0
- package/dist/helpers/FileHelper.d.ts +31 -0
- package/dist/helpers/FileHelper.d.ts.map +1 -0
- package/dist/helpers/FileHelper.js +175 -0
- package/dist/helpers/FileHelper.js.map +1 -0
- package/dist/helpers/SetupHelper.d.ts +5 -0
- package/dist/helpers/SetupHelper.d.ts.map +1 -0
- package/dist/helpers/SetupHelper.js +32 -0
- package/dist/helpers/SetupHelper.js.map +1 -0
- package/dist/helpers/index.d.ts +6 -0
- package/dist/helpers/index.d.ts.map +1 -0
- package/dist/helpers/index.js +6 -0
- package/dist/helpers/index.js.map +1 -0
- package/dist/index.d.ts +18 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +17 -0
- package/dist/index.js.map +1 -0
- package/dist/libs/index.d.ts +4 -0
- package/dist/libs/index.d.ts.map +1 -0
- package/dist/libs/index.js +4 -0
- package/dist/libs/index.js.map +1 -0
- package/dist/libs/kafka/config.d.ts +5 -0
- package/dist/libs/kafka/config.d.ts.map +1 -0
- package/dist/libs/kafka/config.js +5 -0
- package/dist/libs/kafka/config.js.map +1 -0
- package/dist/libs/kafka/consumer.d.ts +16 -0
- package/dist/libs/kafka/consumer.d.ts.map +1 -0
- package/dist/libs/kafka/consumer.js +126 -0
- package/dist/libs/kafka/consumer.js.map +1 -0
- package/dist/libs/kafka/index.d.ts +3 -0
- package/dist/libs/kafka/index.d.ts.map +1 -0
- package/dist/libs/kafka/index.js +3 -0
- package/dist/libs/kafka/index.js.map +1 -0
- package/dist/libs/kafka/kafka.d.ts +3 -0
- package/dist/libs/kafka/kafka.d.ts.map +1 -0
- package/dist/libs/kafka/kafka.js +24 -0
- package/dist/libs/kafka/kafka.js.map +1 -0
- package/dist/libs/kafka/producer.d.ts +11 -0
- package/dist/libs/kafka/producer.d.ts.map +1 -0
- package/dist/libs/kafka/producer.js +44 -0
- package/dist/libs/kafka/producer.js.map +1 -0
- package/dist/libs/logger/config.d.ts +5 -0
- package/dist/libs/logger/config.d.ts.map +1 -0
- package/dist/libs/logger/config.js +6 -0
- package/dist/libs/logger/config.js.map +1 -0
- package/dist/libs/logger/index.d.ts +10 -0
- package/dist/libs/logger/index.d.ts.map +1 -0
- package/dist/libs/logger/index.js +20 -0
- package/dist/libs/logger/index.js.map +1 -0
- package/dist/libs/logger/kafkajs-logger-creator.d.ts +12 -0
- package/dist/libs/logger/kafkajs-logger-creator.d.ts.map +1 -0
- package/dist/libs/logger/kafkajs-logger-creator.js +29 -0
- package/dist/libs/logger/kafkajs-logger-creator.js.map +1 -0
- package/dist/libs/logger/logger.d.ts +42 -0
- package/dist/libs/logger/logger.d.ts.map +1 -0
- package/dist/libs/logger/logger.js +44 -0
- package/dist/libs/logger/logger.js.map +1 -0
- package/dist/libs/s3/config.d.ts +7 -0
- package/dist/libs/s3/config.d.ts.map +1 -0
- package/dist/libs/s3/config.js +7 -0
- package/dist/libs/s3/config.js.map +1 -0
- package/dist/libs/s3/index.d.ts +4 -0
- package/dist/libs/s3/index.d.ts.map +1 -0
- package/dist/libs/s3/index.js +4 -0
- package/dist/libs/s3/index.js.map +1 -0
- package/dist/libs/s3/s3.lib.d.ts +25 -0
- package/dist/libs/s3/s3.lib.d.ts.map +1 -0
- package/dist/libs/s3/s3.lib.js +202 -0
- package/dist/libs/s3/s3.lib.js.map +1 -0
- package/dist/processors/ChatProcessor.d.ts +66 -0
- package/dist/processors/ChatProcessor.d.ts.map +1 -0
- package/dist/processors/ChatProcessor.js +610 -0
- package/dist/processors/ChatProcessor.js.map +1 -0
- package/dist/processors/ModelsProcessor.d.ts +11 -0
- package/dist/processors/ModelsProcessor.d.ts.map +1 -0
- package/dist/processors/ModelsProcessor.js +30 -0
- package/dist/processors/ModelsProcessor.js.map +1 -0
- package/dist/processors/index.d.ts +3 -0
- package/dist/processors/index.d.ts.map +1 -0
- package/dist/processors/index.js +3 -0
- package/dist/processors/index.js.map +1 -0
- package/dist/services/AIService.d.ts +48 -0
- package/dist/services/AIService.d.ts.map +1 -0
- package/dist/services/AIService.js +196 -0
- package/dist/services/AIService.js.map +1 -0
- package/dist/services/InternalEventsHandler.d.ts +21 -0
- package/dist/services/InternalEventsHandler.d.ts.map +1 -0
- package/dist/services/InternalEventsHandler.js +56 -0
- package/dist/services/InternalEventsHandler.js.map +1 -0
- package/dist/services/KafkaService.d.ts +35 -0
- package/dist/services/KafkaService.d.ts.map +1 -0
- package/dist/services/KafkaService.js +120 -0
- package/dist/services/KafkaService.js.map +1 -0
- package/dist/services/ModelFetcher.d.ts +54 -0
- package/dist/services/ModelFetcher.d.ts.map +1 -0
- package/dist/services/ModelFetcher.js +247 -0
- package/dist/services/ModelFetcher.js.map +1 -0
- package/dist/services/RedisService.d.ts +90 -0
- package/dist/services/RedisService.d.ts.map +1 -0
- package/dist/services/RedisService.js +236 -0
- package/dist/services/RedisService.js.map +1 -0
- package/dist/services/SocketService.d.ts +39 -0
- package/dist/services/SocketService.d.ts.map +1 -0
- package/dist/services/SocketService.js +128 -0
- package/dist/services/SocketService.js.map +1 -0
- package/dist/services/index.d.ts +7 -0
- package/dist/services/index.d.ts.map +1 -0
- package/dist/services/index.js +7 -0
- package/dist/services/index.js.map +1 -0
- package/dist/store/AgentStore.d.ts +48 -0
- package/dist/store/AgentStore.d.ts.map +1 -0
- package/dist/store/AgentStore.js +98 -0
- package/dist/store/AgentStore.js.map +1 -0
- package/dist/store/ArtifactStore.d.ts +13 -0
- package/dist/store/ArtifactStore.d.ts.map +1 -0
- package/dist/store/ArtifactStore.js +27 -0
- package/dist/store/ArtifactStore.js.map +1 -0
- package/dist/store/ConfigStore.d.ts +89 -0
- package/dist/store/ConfigStore.d.ts.map +1 -0
- package/dist/store/ConfigStore.js +214 -0
- package/dist/store/ConfigStore.js.map +1 -0
- package/dist/store/ConfigStore.test.d.ts +2 -0
- package/dist/store/ConfigStore.test.d.ts.map +1 -0
- package/dist/store/ConfigStore.test.js +259 -0
- package/dist/store/ConfigStore.test.js.map +1 -0
- package/dist/store/ModelStore.d.ts +44 -0
- package/dist/store/ModelStore.d.ts.map +1 -0
- package/dist/store/ModelStore.js +81 -0
- package/dist/store/ModelStore.js.map +1 -0
- package/dist/store/ModelStore.test.d.ts +2 -0
- package/dist/store/ModelStore.test.d.ts.map +1 -0
- package/dist/store/ModelStore.test.js +390 -0
- package/dist/store/ModelStore.test.js.map +1 -0
- package/dist/store/index.d.ts +5 -0
- package/dist/store/index.d.ts.map +1 -0
- package/dist/store/index.js +5 -0
- package/dist/store/index.js.map +1 -0
- package/dist/tools/generateChartTool.d.ts +24 -0
- package/dist/tools/generateChartTool.d.ts.map +1 -0
- package/dist/tools/generateChartTool.js +124 -0
- package/dist/tools/generateChartTool.js.map +1 -0
- package/dist/tools/proposeFormValuesTool.d.ts +35 -0
- package/dist/tools/proposeFormValuesTool.d.ts.map +1 -0
- package/dist/tools/proposeFormValuesTool.js +56 -0
- package/dist/tools/proposeFormValuesTool.js.map +1 -0
- package/package.json +71 -0
- package/src/config.ts +46 -0
- package/src/helpers/AIHelper.test.ts +375 -0
- package/src/helpers/AIHelper.ts +353 -0
- package/src/helpers/ConfigHelper.ts +130 -0
- package/src/helpers/ContextLimiter.ts +228 -0
- package/src/helpers/FileHelper.ts +197 -0
- package/src/helpers/SetupHelper.ts +35 -0
- package/src/helpers/index.ts +5 -0
- package/src/index.ts +18 -0
- package/src/libs/index.ts +3 -0
- package/src/libs/kafka/config.ts +4 -0
- package/src/libs/kafka/consumer.ts +161 -0
- package/src/libs/kafka/index.ts +2 -0
- package/src/libs/kafka/kafka.ts +27 -0
- package/src/libs/kafka/producer.ts +48 -0
- package/src/libs/logger/config.ts +4 -0
- package/src/libs/logger/index.ts +21 -0
- package/src/libs/logger/kafkajs-logger-creator.ts +28 -0
- package/src/libs/logger/logger.ts +60 -0
- package/src/libs/s3/config.ts +7 -0
- package/src/libs/s3/index.ts +3 -0
- package/src/libs/s3/s3.lib.ts +284 -0
- package/src/processors/ChatProcessor.ts +713 -0
- package/src/processors/ModelsProcessor.ts +34 -0
- package/src/processors/index.ts +2 -0
- package/src/services/AIService.ts +241 -0
- package/src/services/InternalEventsHandler.ts +61 -0
- package/src/services/KafkaService.ts +142 -0
- package/src/services/ModelFetcher.ts +286 -0
- package/src/services/RedisService.ts +285 -0
- package/src/services/SocketService.ts +153 -0
- package/src/services/index.ts +6 -0
- package/src/store/AgentStore.ts +138 -0
- package/src/store/ArtifactStore.ts +29 -0
- package/src/store/ConfigStore.test.ts +314 -0
- package/src/store/ConfigStore.ts +239 -0
- package/src/store/ModelStore.test.ts +473 -0
- package/src/store/ModelStore.ts +93 -0
- package/src/store/index.ts +4 -0
- package/src/tools/generateChartTool.ts +131 -0
- package/src/tools/proposeFormValuesTool.ts +67 -0
- package/tsconfig.json +24 -0
|
@@ -0,0 +1,202 @@
|
|
|
1
|
+
import { S3, GetObjectCommand, PutObjectCommand, ListObjectsV2Command, HeadBucketCommand, CreateBucketCommand, } from '@aws-sdk/client-s3';
|
|
2
|
+
import { S3RequestPresigner } from '@aws-sdk/s3-request-presigner';
|
|
3
|
+
import { createRequest } from '@aws-sdk/util-create-request';
|
|
4
|
+
import { formatUrl } from '@aws-sdk/util-format-url';
|
|
5
|
+
import { Upload } from '@aws-sdk/lib-storage';
|
|
6
|
+
import stream from 'stream';
|
|
7
|
+
import { S3_HOST, S3_PRESIGNED_URL_EXPIRES, AWS_REGION, } from './config';
|
|
8
|
+
import logger from '../logger';
|
|
9
|
+
const S3Client = new S3({
|
|
10
|
+
...S3_HOST ? {
|
|
11
|
+
endpoint: S3_HOST,
|
|
12
|
+
forcePathStyle: true,
|
|
13
|
+
}
|
|
14
|
+
: {},
|
|
15
|
+
});
|
|
16
|
+
const signer = new S3RequestPresigner({
|
|
17
|
+
...S3Client.config,
|
|
18
|
+
});
|
|
19
|
+
export const uploadFile = (Key, Bucket, Body, Expires) => {
|
|
20
|
+
return S3Client.putObject({
|
|
21
|
+
Key,
|
|
22
|
+
Bucket,
|
|
23
|
+
Body,
|
|
24
|
+
Expires,
|
|
25
|
+
});
|
|
26
|
+
};
|
|
27
|
+
export const downloadFile = (Key, Bucket) => {
|
|
28
|
+
const command = new GetObjectCommand({
|
|
29
|
+
Bucket,
|
|
30
|
+
Key,
|
|
31
|
+
});
|
|
32
|
+
return S3Client.send(command);
|
|
33
|
+
};
|
|
34
|
+
export const downloadFileAsString = async (Key, Bucket) => {
|
|
35
|
+
var _a;
|
|
36
|
+
try {
|
|
37
|
+
const command = new GetObjectCommand({
|
|
38
|
+
Bucket,
|
|
39
|
+
Key,
|
|
40
|
+
});
|
|
41
|
+
const response = await S3Client.send(command);
|
|
42
|
+
return (_a = response.Body) === null || _a === void 0 ? void 0 : _a.transformToString();
|
|
43
|
+
}
|
|
44
|
+
catch (err) {
|
|
45
|
+
if ((err === null || err === void 0 ? void 0 : err.name) === 'NoSuchKey') {
|
|
46
|
+
return undefined;
|
|
47
|
+
}
|
|
48
|
+
throw err;
|
|
49
|
+
}
|
|
50
|
+
};
|
|
51
|
+
export const downloadFileAsByteArray = async (Key, Bucket) => {
|
|
52
|
+
var _a;
|
|
53
|
+
try {
|
|
54
|
+
const command = new GetObjectCommand({
|
|
55
|
+
Bucket,
|
|
56
|
+
Key,
|
|
57
|
+
});
|
|
58
|
+
const response = await S3Client.send(command);
|
|
59
|
+
return (_a = response.Body) === null || _a === void 0 ? void 0 : _a.transformToByteArray();
|
|
60
|
+
}
|
|
61
|
+
catch (err) {
|
|
62
|
+
if ((err === null || err === void 0 ? void 0 : err.name) === 'NoSuchKey') {
|
|
63
|
+
return undefined;
|
|
64
|
+
}
|
|
65
|
+
throw err;
|
|
66
|
+
}
|
|
67
|
+
};
|
|
68
|
+
export const streamUpload = (Key, Bucket, ACL) => {
|
|
69
|
+
const passThroughStream = new stream.PassThrough();
|
|
70
|
+
const options = {
|
|
71
|
+
client: S3Client,
|
|
72
|
+
params: {
|
|
73
|
+
Bucket,
|
|
74
|
+
Key,
|
|
75
|
+
Body: passThroughStream,
|
|
76
|
+
},
|
|
77
|
+
leavePartsOnError: false,
|
|
78
|
+
};
|
|
79
|
+
if (ACL) {
|
|
80
|
+
options.params.ACL = ACL;
|
|
81
|
+
}
|
|
82
|
+
const upload = new Upload(options);
|
|
83
|
+
return {
|
|
84
|
+
writeStream: passThroughStream,
|
|
85
|
+
promise: upload.done(),
|
|
86
|
+
};
|
|
87
|
+
};
|
|
88
|
+
export const getPresignedUploadUrl = async (Key, Bucket, expiresIn) => {
|
|
89
|
+
const request = await createRequest(S3Client, new PutObjectCommand({ Key, Bucket }));
|
|
90
|
+
const signedUrl = formatUrl(await signer.presign(request, {
|
|
91
|
+
expiresIn: expiresIn || S3_PRESIGNED_URL_EXPIRES,
|
|
92
|
+
}));
|
|
93
|
+
return signedUrl;
|
|
94
|
+
};
|
|
95
|
+
export const getPresignedDownloadUrl = async (Key, Bucket) => {
|
|
96
|
+
const request = await createRequest(S3Client, new GetObjectCommand({ Key, Bucket }));
|
|
97
|
+
const signedUrl = formatUrl(await signer.presign(request, {
|
|
98
|
+
expiresIn: S3_PRESIGNED_URL_EXPIRES,
|
|
99
|
+
}));
|
|
100
|
+
return signedUrl;
|
|
101
|
+
};
|
|
102
|
+
export const getDownloadUrl = (Key, Bucket) => {
|
|
103
|
+
if (S3_HOST) {
|
|
104
|
+
return `${S3_HOST}/${Bucket}/${Key}`;
|
|
105
|
+
}
|
|
106
|
+
const regionString = AWS_REGION.includes('us-east-1') ? '' : '-' + AWS_REGION;
|
|
107
|
+
return `https://${Bucket}.s3${regionString}.amazonaws.com/${Key}`;
|
|
108
|
+
};
|
|
109
|
+
export const copy = (Bucket, KeyFrom, KeyTo) => {
|
|
110
|
+
return S3Client.copyObject({
|
|
111
|
+
CopySource: `/${Bucket}/${KeyFrom}`,
|
|
112
|
+
Bucket,
|
|
113
|
+
Key: KeyTo,
|
|
114
|
+
});
|
|
115
|
+
};
|
|
116
|
+
export const copyBetweenBuckets = (BucketFrom, KeyFrom, BucketTo, KeyTo) => {
|
|
117
|
+
return S3Client.copyObject({
|
|
118
|
+
CopySource: `/${BucketFrom}/${KeyFrom}`,
|
|
119
|
+
Bucket: BucketTo,
|
|
120
|
+
Key: KeyTo,
|
|
121
|
+
});
|
|
122
|
+
};
|
|
123
|
+
export const getObject = async (Bucket, Key) => {
|
|
124
|
+
return S3Client.getObject({
|
|
125
|
+
Bucket,
|
|
126
|
+
Key,
|
|
127
|
+
});
|
|
128
|
+
};
|
|
129
|
+
export const listObjectsByPrefix = async (Bucket, Prefix) => {
|
|
130
|
+
return S3Client.send(new ListObjectsV2Command({
|
|
131
|
+
Bucket,
|
|
132
|
+
Prefix,
|
|
133
|
+
}));
|
|
134
|
+
};
|
|
135
|
+
export const headObject = async (Bucket, Key) => {
|
|
136
|
+
return S3Client.headObject({
|
|
137
|
+
Bucket,
|
|
138
|
+
Key,
|
|
139
|
+
});
|
|
140
|
+
};
|
|
141
|
+
export const deleteObject = async (Bucket, Key) => {
|
|
142
|
+
return S3Client.deleteObject({
|
|
143
|
+
Bucket,
|
|
144
|
+
Key,
|
|
145
|
+
});
|
|
146
|
+
};
|
|
147
|
+
export const deleteObjectsByPrefix = async (Bucket, Prefix, ContinuationToken) => {
|
|
148
|
+
var _a, _b;
|
|
149
|
+
const objects = await S3Client.listObjectsV2({ Bucket, Prefix, ContinuationToken });
|
|
150
|
+
if (!((_a = objects.Contents) === null || _a === void 0 ? void 0 : _a.length))
|
|
151
|
+
return;
|
|
152
|
+
const deleteParams = {
|
|
153
|
+
Bucket,
|
|
154
|
+
Delete: { Objects: [] },
|
|
155
|
+
};
|
|
156
|
+
objects.Contents.forEach(({ Key }) => {
|
|
157
|
+
if (!Key)
|
|
158
|
+
return;
|
|
159
|
+
deleteParams.Delete.Objects.push({ Key });
|
|
160
|
+
});
|
|
161
|
+
logger.info(deleteParams, '[S3] Deleting objects');
|
|
162
|
+
const deletedResp = await S3Client.deleteObjects(deleteParams);
|
|
163
|
+
if ((_b = deletedResp.Errors) === null || _b === void 0 ? void 0 : _b.length) {
|
|
164
|
+
logger.error(deletedResp.Errors, '[S3] Error during deleteObjectsByPrefix');
|
|
165
|
+
}
|
|
166
|
+
if (objects.IsTruncated) {
|
|
167
|
+
await deleteObjectsByPrefix(Bucket, Prefix, objects.NextContinuationToken);
|
|
168
|
+
}
|
|
169
|
+
};
|
|
170
|
+
/**
|
|
171
|
+
* Ensures a bucket exists, creating it if it doesn't
|
|
172
|
+
*/
|
|
173
|
+
export const ensureBucketExists = async (Bucket) => {
|
|
174
|
+
var _a, _b;
|
|
175
|
+
try {
|
|
176
|
+
await S3Client.send(new HeadBucketCommand({ Bucket }));
|
|
177
|
+
// Bucket exists
|
|
178
|
+
return;
|
|
179
|
+
}
|
|
180
|
+
catch (error) {
|
|
181
|
+
// If bucket doesn't exist, create it
|
|
182
|
+
if (((_a = error.$metadata) === null || _a === void 0 ? void 0 : _a.httpStatusCode) === 404) {
|
|
183
|
+
try {
|
|
184
|
+
await S3Client.send(new CreateBucketCommand({ Bucket }));
|
|
185
|
+
logger.info({ Bucket }, '[S3] Created bucket');
|
|
186
|
+
}
|
|
187
|
+
catch (createError) {
|
|
188
|
+
// Ignore if bucket was created by another process
|
|
189
|
+
if (((_b = createError.$metadata) === null || _b === void 0 ? void 0 : _b.httpStatusCode) !== 409) {
|
|
190
|
+
logger.error({ Bucket, error: createError }, '[S3] Failed to create bucket');
|
|
191
|
+
throw createError;
|
|
192
|
+
}
|
|
193
|
+
}
|
|
194
|
+
}
|
|
195
|
+
else {
|
|
196
|
+
// Some other error (permissions, etc.)
|
|
197
|
+
logger.error({ Bucket, error }, '[S3] Failed to check bucket existence');
|
|
198
|
+
throw error;
|
|
199
|
+
}
|
|
200
|
+
}
|
|
201
|
+
};
|
|
202
|
+
//# sourceMappingURL=s3.lib.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"s3.lib.js","sourceRoot":"","sources":["../../../src/libs/s3/s3.lib.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,EAAE,EACF,gBAAgB,EAChB,gBAAgB,EAIhB,oBAAoB,EACpB,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAA;AAClE,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAA;AAC5D,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAA;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,sBAAsB,CAAA;AAC7C,OAAO,MAAM,MAAM,QAAQ,CAAA;AAC3B,OAAO,EACL,OAAO,EACP,wBAAwB,EACxB,UAAU,GACX,MAAM,UAAU,CAAA;AACjB,OAAO,MAAM,MAAM,WAAW,CAAA;AAE9B,MAAM,QAAQ,GAAG,IAAI,EAAE,CAAC;IACtB,GAAG,OAAO,CAAC,CAAC,CAAC;QACX,QAAQ,EAAE,OAAO;QACjB,cAAc,EAAE,IAAI;KACrB;QACC,CAAC,CAAC,EAAE;CACP,CAAC,CAAA;AAEF,MAAM,MAAM,GAAG,IAAI,kBAAkB,CAAC;IACpC,GAAG,QAAQ,CAAC,MAAM;CACnB,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,GAAW,EAAE,MAAc,EAAE,IAAS,EAAE,OAAc,EAAE,EAAE;IACnF,OAAO,QAAQ,CAAC,SAAS,CAAC;QACxB,GAAG;QACH,MAAM;QACN,IAAI;QACJ,OAAO;KACR,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,GAAW,EAAE,MAAc,EAA4B,EAAE;IACpF,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC;QACnC,MAAM;QACN,GAAG;KACJ,CAAC,CAAA;IAEF,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;AAC/B,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,oBAAoB,GAAG,KAAK,EACvC,GAAW,EACX,MAAc,EACe,EAAE;;IAC/B,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC;YACnC,MAAM;YACN,GAAG;SACJ,CAAC,CAAA;QAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC7C,OAAO,MAAA,QAAQ,CAAC,IAAI,0CAAE,iBAAiB,EAAE,CAAA;IAC3C,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,GAAW,aAAX,GAAG,uBAAH,GAAG,CAAU,IAAI,MAAK,WAAW,EAAE,CAAC;YACvC,OAAO,SAAS,CAAA;QAClB,CAAC;QACD,MAAM,GAAG,CAAA;IACX,CAAC;AACH,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,uBAAuB,GAAG,KAAK,EAAE,GAAW,EAAE,MAAc,EAAmC,EAAE;;IAC5G,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC;YACnC,MAAM;YACN,GAAG;SACJ,CAAC,CAAA;QAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC7C,OAAO,MAAA,QAAQ,CAAC,IAAI,0CAAE,oBAAoB,EAAE,CAAA;IAC9C,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,GAAW,aAAX,GAAG,uBAAH,GAAG,CAAU,IAAI,MAAK,WAAW,EAAE,CAAC;YACvC,OAAO,SAAS,CAAA;QAClB,CAAC;QACD,MAAM,GAAG,CAAA;IACX,CAAC;AACH,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,GAAW,EAAE,MAAc,EAAE,GAAqB,EAAE,EAAE;IACjF,MAAM,iBAAiB,GAAG,IAAI,MAAM,CAAC,WAAW,EAAE,CAAA;IAElD,MAAM,OAAO,GAAQ;QACnB,MAAM,EAAE,QAAQ;QAChB,MAAM,EAAE;YACN,MAAM;YACN,GAAG;YACH,IAAI,EAAE,iBAAiB;SACxB;QACD,iBAAiB,EAAE,KAAK;KACzB,CAAA;IAED,IAAI,GAAG,EAAE,CAAC;QACR,OAAO,CAAC,MAAM,CAAC,GAAG,GAAG,GAAG,CAAA;IAC1B,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,CAAA;IAElC,OAAO;QACL,WAAW,EAAE,iBAAiB;QAC9B,OAAO,EAAE,MAAM,CAAC,IAAI,EAA4C;KACjE,CAAA;AACH,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,KAAK,EACxC,GAAW,EACX,MAAc,EACd,SAAkB,EAClB,EAAE;IACF,MAAM,OAAO,GAAG,MAAM,aAAa,CACjC,QAAQ,EACR,IAAI,gBAAgB,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CACtC,CAAA;IAED,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,MAAM,CAAC,OAAO,CAC9C,OAAO,EACP;QACE,SAAS,EAAE,SAAS,IAAI,wBAAwB;KACjD,CACF,CAAC,CAAA;IAEF,OAAO,SAAS,CAAA;AAClB,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,uBAAuB,GAAG,KAAK,EAAE,GAAW,EAAE,MAAc,EAAE,EAAE;IAC3E,MAAM,OAAO,GAAG,MAAM,aAAa,CACjC,QAAQ,EACR,IAAI,gBAAgB,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CACtC,CAAA;IAED,MAAM,SAAS,GAAG,SAAS,CACzB,MAAM,MAAM,CAAC,OAAO,CAClB,OAAO,EACP;QACE,SAAS,EAAE,wBAAwB;KACpC,CACF,CACF,CAAA;IAED,OAAO,SAAS,CAAA;AAClB,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,GAAW,EAAE,MAAc,EAAE,EAAE;IAC5D,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,GAAG,OAAO,IAAI,MAAM,IAAI,GAAG,EAAE,CAAA;IACtC,CAAC;IAED,MAAM,YAAY,GAAG,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,UAAU,CAAA;IAC7E,OAAO,WAAW,MAAM,MAAM,YAAY,kBAAkB,GAAG,EAAE,CAAA;AACnE,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,IAAI,GAAG,CAClB,MAAc,EACd,OAAe,EACf,KAAa,EACb,EAAE;IACF,OAAO,QAAQ,CAAC,UAAU,CAAC;QACzB,UAAU,EAAE,IAAI,MAAM,IAAI,OAAO,EAAE;QACnC,MAAM;QACN,GAAG,EAAE,KAAK;KACX,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,UAAkB,EAClB,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,EAAE;IACF,OAAO,QAAQ,CAAC,UAAU,CAAC;QACzB,UAAU,EAAE,IAAI,UAAU,IAAI,OAAO,EAAE;QACvC,MAAM,EAAE,QAAQ;QAChB,GAAG,EAAE,KAAK;KACX,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,SAAS,GAAG,KAAK,EAC5B,MAAc,EACd,GAAW,EACsB,EAAE;IACnC,OAAO,QAAQ,CAAC,SAAS,CAAC;QACxB,MAAM;QACN,GAAG;KACJ,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,mBAAmB,GAAG,KAAK,EACtC,MAAc,EACd,MAAc,EACuB,EAAE;IACvC,OAAO,QAAQ,CAAC,IAAI,CAClB,IAAI,oBAAoB,CAAC;QACvB,MAAM;QACN,MAAM;KACP,CAAC,CACH,CAAA;AACH,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,KAAK,EAC7B,MAAc,EACd,GAAW,EACsB,EAAE;IACnC,OAAO,QAAQ,CAAC,UAAU,CAAC;QACzB,MAAM;QACN,GAAG;KACJ,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,KAAK,EAC/B,MAAc,EACd,GAAW,EACsB,EAAE;IACnC,OAAO,QAAQ,CAAC,YAAY,CAAC;QAC3B,MAAM;QACN,GAAG;KACJ,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,KAAK,EACxC,MAAc,EACd,MAAc,EACd,iBAA0B,EACX,EAAE;;IACjB,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAA;IACnF,IAAI,CAAC,CAAA,MAAA,OAAO,CAAC,QAAQ,0CAAE,MAAM,CAAA;QAAE,OAAM;IAErC,MAAM,YAAY,GAAG;QACnB,MAAM;QACN,MAAM,EAAE,EAAE,OAAO,EAAE,EAAuB,EAAE;KAC7C,CAAA;IAED,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE;QACnC,IAAI,CAAC,GAAG;YAAE,OAAM;QAChB,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;IACF,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,uBAAuB,CAAC,CAAA;IAClD,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,aAAa,CAAC,YAAY,CAAC,CAAA;IAC9D,IAAI,MAAA,WAAW,CAAC,MAAM,0CAAE,MAAM,EAAE,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,yCAAyC,CAAC,CAAA;IAC7E,CAAC;IACD,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,MAAM,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,qBAAqB,CAAC,CAAA;IAC5E,CAAC;AACH,CAAC,CAAA;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,KAAK,EAAE,MAAc,EAAiB,EAAE;;IACxE,IAAI,CAAC;QACH,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;QACtD,gBAAgB;QAChB,OAAM;IACR,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,qCAAqC;QACrC,IAAI,CAAA,MAAA,KAAK,CAAC,SAAS,0CAAE,cAAc,MAAK,GAAG,EAAE,CAAC;YAC5C,IAAI,CAAC;gBACH,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,mBAAmB,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;gBACxD,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,qBAAqB,CAAC,CAAA;YAChD,CAAC;YAAC,OAAO,WAAgB,EAAE,CAAC;gBAC1B,kDAAkD;gBAClD,IAAI,CAAA,MAAA,WAAW,CAAC,SAAS,0CAAE,cAAc,MAAK,GAAG,EAAE,CAAC;oBAClD,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,8BAA8B,CAAC,CAAA;oBAC5E,MAAM,WAAW,CAAA;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;aAAM,CAAC;YACN,uCAAuC;YACvC,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,uCAAuC,CAAC,CAAA;YACxE,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC;AACH,CAAC,CAAA"}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import type { AgentChatRepository, AgentMessageRepository } from '@multiplayer-app/ai-agent-db';
|
|
2
|
+
import { AgentArtifact, AgentChat, AgentChatResponse, AgentAttachment, AgentMessage, MessageRole, SendMessagePayload, SortOrder, SendMessagePayloadWithContent, AdditionalContext } from '@multiplayer-app/ai-agent-types';
|
|
3
|
+
import { ArtifactStore } from '../store';
|
|
4
|
+
import { GenerateTextOptions } from '../services';
|
|
5
|
+
import { PassThrough } from 'stream';
|
|
6
|
+
export declare class ChatProcessor {
|
|
7
|
+
private chatRepository;
|
|
8
|
+
private messageRepository;
|
|
9
|
+
private artifactStore;
|
|
10
|
+
constructor(chatRepository: AgentChatRepository, messageRepository: AgentMessageRepository, artifactStore: ArtifactStore);
|
|
11
|
+
private getTemporaryTitle;
|
|
12
|
+
listChats(params?: {
|
|
13
|
+
contextKey?: string;
|
|
14
|
+
limit?: number;
|
|
15
|
+
userId?: string;
|
|
16
|
+
sortField?: string;
|
|
17
|
+
sortOrder?: SortOrder;
|
|
18
|
+
}): Promise<AgentChatResponse[]>;
|
|
19
|
+
getChat(chatId: string): Promise<AgentChatResponse>;
|
|
20
|
+
deleteChat(chatId: string): Promise<void>;
|
|
21
|
+
upsertAndGetChat(payload: SendMessagePayload, excludeSocketId?: string): Promise<AgentChat>;
|
|
22
|
+
getChatTitle(payload: {
|
|
23
|
+
chatId?: string;
|
|
24
|
+
content?: string;
|
|
25
|
+
contextKey: string;
|
|
26
|
+
userId?: string;
|
|
27
|
+
}): Promise<string>;
|
|
28
|
+
createMessage(chat: AgentChat, role: MessageRole, messageData: Pick<AgentMessage, 'content' | 'agentName'>, attachments?: AgentAttachment[], excludeSocketId?: string): Promise<AgentMessage>;
|
|
29
|
+
createChat(payload: SendMessagePayload, excludeSocketId?: string): Promise<AgentChat>;
|
|
30
|
+
/**
|
|
31
|
+
* Persist a client-side action against a tool call (generic mechanism).
|
|
32
|
+
* This updates the message containing the tool call and optionally appends a System message.
|
|
33
|
+
*/
|
|
34
|
+
recordToolCallAction(params: {
|
|
35
|
+
chatId: string;
|
|
36
|
+
toolCallId: string;
|
|
37
|
+
action: string;
|
|
38
|
+
data: Record<string, unknown>;
|
|
39
|
+
systemMessage?: string;
|
|
40
|
+
excludeSocketId?: string;
|
|
41
|
+
}): Promise<{
|
|
42
|
+
ok: true;
|
|
43
|
+
}>;
|
|
44
|
+
private streamMessageStep;
|
|
45
|
+
private isApprovalPayload;
|
|
46
|
+
private processApprovalPayload;
|
|
47
|
+
processNewUserMessage(chat: AgentChat, payload: SendMessagePayloadWithContent & {
|
|
48
|
+
context?: AdditionalContext;
|
|
49
|
+
}, prevMessages: AgentMessage[], excludeSocketId?: string): Promise<{
|
|
50
|
+
userMessage: AgentMessage;
|
|
51
|
+
assistantMessage: AgentMessage;
|
|
52
|
+
agentOptions: Omit<GenerateTextOptions, "messages"> & {
|
|
53
|
+
name: string;
|
|
54
|
+
};
|
|
55
|
+
}>;
|
|
56
|
+
streamMessage(chat: AgentChat, payload: SendMessagePayload & {
|
|
57
|
+
context?: AdditionalContext;
|
|
58
|
+
}, excludeSocketId?: string, onAgentStart?: () => void): Promise<void>;
|
|
59
|
+
listArtifacts(chatId: string): AgentArtifact[];
|
|
60
|
+
private eventToChunk;
|
|
61
|
+
getMessageStream(chatId: string): PassThrough;
|
|
62
|
+
createMessageStream(payload: SendMessagePayload & {
|
|
63
|
+
context?: AdditionalContext;
|
|
64
|
+
}, excludeSocketId?: string): Promise<PassThrough>;
|
|
65
|
+
}
|
|
66
|
+
//# sourceMappingURL=ChatProcessor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ChatProcessor.d.ts","sourceRoot":"","sources":["../../src/processors/ChatProcessor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,mBAAmB,EAAE,sBAAsB,EAAC,MAAM,8BAA8B,CAAC;AAC9F,OAAO,EACL,aAAa,EACb,SAAS,EACT,iBAAiB,EACjB,eAAe,EACf,YAAY,EAIZ,WAAW,EACX,kBAAkB,EAClB,SAAS,EAIT,6BAA6B,EAC7B,iBAAiB,EAClB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAKzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAGlD,OAAO,EAAE,WAAW,EAAY,MAAM,QAAQ,CAAC;AAO/C,qBAAa,aAAa;IAEtB,OAAO,CAAC,cAAc;IACtB,OAAO,CAAC,iBAAiB;IACzB,OAAO,CAAC,aAAa;gBAFb,cAAc,EAAE,mBAAmB,EACnC,iBAAiB,EAAE,sBAAsB,EACzC,aAAa,EAAE,aAAa;IAGtC,OAAO,CAAC,iBAAiB;IAInB,SAAS,CAAC,MAAM,CAAC,EAAE;QACvB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,SAAS,CAAC;KACvB,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC;IA0B1B,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAYnD,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQzC,gBAAgB,CAAC,OAAO,EAAE,kBAAkB,EAAE,eAAe,CAAC,EAAE,MAAM;IAyBtE,YAAY,CAAC,OAAO,EAAE;QAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,MAAM,CAAC;IAkBb,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,SAAS,GAAG,WAAW,CAAC,EAAE,WAAW,CAAC,EAAE,eAAe,EAAE,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAgB7L,UAAU,CAAC,OAAO,EAAE,kBAAkB,EAAE,eAAe,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAuB3F;;;OAGG;IACG,oBAAoB,CAAC,MAAM,EAAE;QACjC,MAAM,EAAE,MAAM,CAAC;QACf,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC1B,GAAG,OAAO,CAAC;QAAE,EAAE,EAAE,IAAI,CAAA;KAAE,CAAC;YAiEX,iBAAiB;IAmJ/B,OAAO,CAAC,iBAAiB;YAIX,sBAAsB;IAyD9B,qBAAqB,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,6BAA6B,GAAG;QAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;KAAE,EAAE,YAAY,EAAE,YAAY,EAAE,EAAE,eAAe,CAAC,EAAE,MAAM;;;;;;;IAsBvK,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,kBAAkB,GAAG;QAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;KAAE,EAAE,eAAe,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAiDvK,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa,EAAE;IAI9C,OAAO,CAAC,YAAY;IA6BpB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,WAAW;IAgFvC,mBAAmB,CACrB,OAAO,EAAE,kBAAkB,GAAG;QAAE,OAAO,CAAC,EAAE,iBAAiB,CAAA;KAAE,EAC7D,eAAe,CAAC,EAAE,MAAM,GACzB,OAAO,CAAC,WAAW,CAAC;CAyDxB"}
|