@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.
Files changed (210) hide show
  1. package/.env.example +45 -0
  2. package/README.md +611 -0
  3. package/config.example.json +73 -0
  4. package/dist/config.d.ts +35 -0
  5. package/dist/config.d.ts.map +1 -0
  6. package/dist/config.js +44 -0
  7. package/dist/config.js.map +1 -0
  8. package/dist/helpers/AIHelper.d.ts +23 -0
  9. package/dist/helpers/AIHelper.d.ts.map +1 -0
  10. package/dist/helpers/AIHelper.js +326 -0
  11. package/dist/helpers/AIHelper.js.map +1 -0
  12. package/dist/helpers/AIHelper.test.d.ts +2 -0
  13. package/dist/helpers/AIHelper.test.d.ts.map +1 -0
  14. package/dist/helpers/AIHelper.test.js +332 -0
  15. package/dist/helpers/AIHelper.test.js.map +1 -0
  16. package/dist/helpers/ConfigHelper.d.ts +20 -0
  17. package/dist/helpers/ConfigHelper.d.ts.map +1 -0
  18. package/dist/helpers/ConfigHelper.js +118 -0
  19. package/dist/helpers/ConfigHelper.js.map +1 -0
  20. package/dist/helpers/ContextLimiter.d.ts +82 -0
  21. package/dist/helpers/ContextLimiter.d.ts.map +1 -0
  22. package/dist/helpers/ContextLimiter.js +165 -0
  23. package/dist/helpers/ContextLimiter.js.map +1 -0
  24. package/dist/helpers/FileHelper.d.ts +31 -0
  25. package/dist/helpers/FileHelper.d.ts.map +1 -0
  26. package/dist/helpers/FileHelper.js +175 -0
  27. package/dist/helpers/FileHelper.js.map +1 -0
  28. package/dist/helpers/SetupHelper.d.ts +5 -0
  29. package/dist/helpers/SetupHelper.d.ts.map +1 -0
  30. package/dist/helpers/SetupHelper.js +32 -0
  31. package/dist/helpers/SetupHelper.js.map +1 -0
  32. package/dist/helpers/index.d.ts +6 -0
  33. package/dist/helpers/index.d.ts.map +1 -0
  34. package/dist/helpers/index.js +6 -0
  35. package/dist/helpers/index.js.map +1 -0
  36. package/dist/index.d.ts +18 -0
  37. package/dist/index.d.ts.map +1 -0
  38. package/dist/index.js +17 -0
  39. package/dist/index.js.map +1 -0
  40. package/dist/libs/index.d.ts +4 -0
  41. package/dist/libs/index.d.ts.map +1 -0
  42. package/dist/libs/index.js +4 -0
  43. package/dist/libs/index.js.map +1 -0
  44. package/dist/libs/kafka/config.d.ts +5 -0
  45. package/dist/libs/kafka/config.d.ts.map +1 -0
  46. package/dist/libs/kafka/config.js +5 -0
  47. package/dist/libs/kafka/config.js.map +1 -0
  48. package/dist/libs/kafka/consumer.d.ts +16 -0
  49. package/dist/libs/kafka/consumer.d.ts.map +1 -0
  50. package/dist/libs/kafka/consumer.js +126 -0
  51. package/dist/libs/kafka/consumer.js.map +1 -0
  52. package/dist/libs/kafka/index.d.ts +3 -0
  53. package/dist/libs/kafka/index.d.ts.map +1 -0
  54. package/dist/libs/kafka/index.js +3 -0
  55. package/dist/libs/kafka/index.js.map +1 -0
  56. package/dist/libs/kafka/kafka.d.ts +3 -0
  57. package/dist/libs/kafka/kafka.d.ts.map +1 -0
  58. package/dist/libs/kafka/kafka.js +24 -0
  59. package/dist/libs/kafka/kafka.js.map +1 -0
  60. package/dist/libs/kafka/producer.d.ts +11 -0
  61. package/dist/libs/kafka/producer.d.ts.map +1 -0
  62. package/dist/libs/kafka/producer.js +44 -0
  63. package/dist/libs/kafka/producer.js.map +1 -0
  64. package/dist/libs/logger/config.d.ts +5 -0
  65. package/dist/libs/logger/config.d.ts.map +1 -0
  66. package/dist/libs/logger/config.js +6 -0
  67. package/dist/libs/logger/config.js.map +1 -0
  68. package/dist/libs/logger/index.d.ts +10 -0
  69. package/dist/libs/logger/index.d.ts.map +1 -0
  70. package/dist/libs/logger/index.js +20 -0
  71. package/dist/libs/logger/index.js.map +1 -0
  72. package/dist/libs/logger/kafkajs-logger-creator.d.ts +12 -0
  73. package/dist/libs/logger/kafkajs-logger-creator.d.ts.map +1 -0
  74. package/dist/libs/logger/kafkajs-logger-creator.js +29 -0
  75. package/dist/libs/logger/kafkajs-logger-creator.js.map +1 -0
  76. package/dist/libs/logger/logger.d.ts +42 -0
  77. package/dist/libs/logger/logger.d.ts.map +1 -0
  78. package/dist/libs/logger/logger.js +44 -0
  79. package/dist/libs/logger/logger.js.map +1 -0
  80. package/dist/libs/s3/config.d.ts +7 -0
  81. package/dist/libs/s3/config.d.ts.map +1 -0
  82. package/dist/libs/s3/config.js +7 -0
  83. package/dist/libs/s3/config.js.map +1 -0
  84. package/dist/libs/s3/index.d.ts +4 -0
  85. package/dist/libs/s3/index.d.ts.map +1 -0
  86. package/dist/libs/s3/index.js +4 -0
  87. package/dist/libs/s3/index.js.map +1 -0
  88. package/dist/libs/s3/s3.lib.d.ts +25 -0
  89. package/dist/libs/s3/s3.lib.d.ts.map +1 -0
  90. package/dist/libs/s3/s3.lib.js +202 -0
  91. package/dist/libs/s3/s3.lib.js.map +1 -0
  92. package/dist/processors/ChatProcessor.d.ts +66 -0
  93. package/dist/processors/ChatProcessor.d.ts.map +1 -0
  94. package/dist/processors/ChatProcessor.js +610 -0
  95. package/dist/processors/ChatProcessor.js.map +1 -0
  96. package/dist/processors/ModelsProcessor.d.ts +11 -0
  97. package/dist/processors/ModelsProcessor.d.ts.map +1 -0
  98. package/dist/processors/ModelsProcessor.js +30 -0
  99. package/dist/processors/ModelsProcessor.js.map +1 -0
  100. package/dist/processors/index.d.ts +3 -0
  101. package/dist/processors/index.d.ts.map +1 -0
  102. package/dist/processors/index.js +3 -0
  103. package/dist/processors/index.js.map +1 -0
  104. package/dist/services/AIService.d.ts +48 -0
  105. package/dist/services/AIService.d.ts.map +1 -0
  106. package/dist/services/AIService.js +196 -0
  107. package/dist/services/AIService.js.map +1 -0
  108. package/dist/services/InternalEventsHandler.d.ts +21 -0
  109. package/dist/services/InternalEventsHandler.d.ts.map +1 -0
  110. package/dist/services/InternalEventsHandler.js +56 -0
  111. package/dist/services/InternalEventsHandler.js.map +1 -0
  112. package/dist/services/KafkaService.d.ts +35 -0
  113. package/dist/services/KafkaService.d.ts.map +1 -0
  114. package/dist/services/KafkaService.js +120 -0
  115. package/dist/services/KafkaService.js.map +1 -0
  116. package/dist/services/ModelFetcher.d.ts +54 -0
  117. package/dist/services/ModelFetcher.d.ts.map +1 -0
  118. package/dist/services/ModelFetcher.js +247 -0
  119. package/dist/services/ModelFetcher.js.map +1 -0
  120. package/dist/services/RedisService.d.ts +90 -0
  121. package/dist/services/RedisService.d.ts.map +1 -0
  122. package/dist/services/RedisService.js +236 -0
  123. package/dist/services/RedisService.js.map +1 -0
  124. package/dist/services/SocketService.d.ts +39 -0
  125. package/dist/services/SocketService.d.ts.map +1 -0
  126. package/dist/services/SocketService.js +128 -0
  127. package/dist/services/SocketService.js.map +1 -0
  128. package/dist/services/index.d.ts +7 -0
  129. package/dist/services/index.d.ts.map +1 -0
  130. package/dist/services/index.js +7 -0
  131. package/dist/services/index.js.map +1 -0
  132. package/dist/store/AgentStore.d.ts +48 -0
  133. package/dist/store/AgentStore.d.ts.map +1 -0
  134. package/dist/store/AgentStore.js +98 -0
  135. package/dist/store/AgentStore.js.map +1 -0
  136. package/dist/store/ArtifactStore.d.ts +13 -0
  137. package/dist/store/ArtifactStore.d.ts.map +1 -0
  138. package/dist/store/ArtifactStore.js +27 -0
  139. package/dist/store/ArtifactStore.js.map +1 -0
  140. package/dist/store/ConfigStore.d.ts +89 -0
  141. package/dist/store/ConfigStore.d.ts.map +1 -0
  142. package/dist/store/ConfigStore.js +214 -0
  143. package/dist/store/ConfigStore.js.map +1 -0
  144. package/dist/store/ConfigStore.test.d.ts +2 -0
  145. package/dist/store/ConfigStore.test.d.ts.map +1 -0
  146. package/dist/store/ConfigStore.test.js +259 -0
  147. package/dist/store/ConfigStore.test.js.map +1 -0
  148. package/dist/store/ModelStore.d.ts +44 -0
  149. package/dist/store/ModelStore.d.ts.map +1 -0
  150. package/dist/store/ModelStore.js +81 -0
  151. package/dist/store/ModelStore.js.map +1 -0
  152. package/dist/store/ModelStore.test.d.ts +2 -0
  153. package/dist/store/ModelStore.test.d.ts.map +1 -0
  154. package/dist/store/ModelStore.test.js +390 -0
  155. package/dist/store/ModelStore.test.js.map +1 -0
  156. package/dist/store/index.d.ts +5 -0
  157. package/dist/store/index.d.ts.map +1 -0
  158. package/dist/store/index.js +5 -0
  159. package/dist/store/index.js.map +1 -0
  160. package/dist/tools/generateChartTool.d.ts +24 -0
  161. package/dist/tools/generateChartTool.d.ts.map +1 -0
  162. package/dist/tools/generateChartTool.js +124 -0
  163. package/dist/tools/generateChartTool.js.map +1 -0
  164. package/dist/tools/proposeFormValuesTool.d.ts +35 -0
  165. package/dist/tools/proposeFormValuesTool.d.ts.map +1 -0
  166. package/dist/tools/proposeFormValuesTool.js +56 -0
  167. package/dist/tools/proposeFormValuesTool.js.map +1 -0
  168. package/package.json +71 -0
  169. package/src/config.ts +46 -0
  170. package/src/helpers/AIHelper.test.ts +375 -0
  171. package/src/helpers/AIHelper.ts +353 -0
  172. package/src/helpers/ConfigHelper.ts +130 -0
  173. package/src/helpers/ContextLimiter.ts +228 -0
  174. package/src/helpers/FileHelper.ts +197 -0
  175. package/src/helpers/SetupHelper.ts +35 -0
  176. package/src/helpers/index.ts +5 -0
  177. package/src/index.ts +18 -0
  178. package/src/libs/index.ts +3 -0
  179. package/src/libs/kafka/config.ts +4 -0
  180. package/src/libs/kafka/consumer.ts +161 -0
  181. package/src/libs/kafka/index.ts +2 -0
  182. package/src/libs/kafka/kafka.ts +27 -0
  183. package/src/libs/kafka/producer.ts +48 -0
  184. package/src/libs/logger/config.ts +4 -0
  185. package/src/libs/logger/index.ts +21 -0
  186. package/src/libs/logger/kafkajs-logger-creator.ts +28 -0
  187. package/src/libs/logger/logger.ts +60 -0
  188. package/src/libs/s3/config.ts +7 -0
  189. package/src/libs/s3/index.ts +3 -0
  190. package/src/libs/s3/s3.lib.ts +284 -0
  191. package/src/processors/ChatProcessor.ts +713 -0
  192. package/src/processors/ModelsProcessor.ts +34 -0
  193. package/src/processors/index.ts +2 -0
  194. package/src/services/AIService.ts +241 -0
  195. package/src/services/InternalEventsHandler.ts +61 -0
  196. package/src/services/KafkaService.ts +142 -0
  197. package/src/services/ModelFetcher.ts +286 -0
  198. package/src/services/RedisService.ts +285 -0
  199. package/src/services/SocketService.ts +153 -0
  200. package/src/services/index.ts +6 -0
  201. package/src/store/AgentStore.ts +138 -0
  202. package/src/store/ArtifactStore.ts +29 -0
  203. package/src/store/ConfigStore.test.ts +314 -0
  204. package/src/store/ConfigStore.ts +239 -0
  205. package/src/store/ModelStore.test.ts +473 -0
  206. package/src/store/ModelStore.ts +93 -0
  207. package/src/store/index.ts +4 -0
  208. package/src/tools/generateChartTool.ts +131 -0
  209. package/src/tools/proposeFormValuesTool.ts +67 -0
  210. 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"}