@smythos/sre 1.6.0 → 1.6.8

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 (237) hide show
  1. package/CHANGELOG +111 -111
  2. package/LICENSE +18 -18
  3. package/README.md +135 -135
  4. package/dist/index.js +43 -43
  5. package/dist/index.js.map +1 -1
  6. package/dist/types/index.d.ts +1 -0
  7. package/dist/types/subsystems/LLMManager/LLM.service/connectors/Ollama.class.d.ts +45 -0
  8. package/dist/types/subsystems/LLMManager/ModelsProvider.service/connectors/SmythModelsProvider.class.d.ts +39 -0
  9. package/dist/types/types/LLM.types.d.ts +2 -0
  10. package/package.json +2 -1
  11. package/src/Components/APICall/APICall.class.ts +161 -161
  12. package/src/Components/APICall/AccessTokenManager.ts +166 -166
  13. package/src/Components/APICall/ArrayBufferResponse.helper.ts +58 -58
  14. package/src/Components/APICall/OAuth.helper.ts +447 -447
  15. package/src/Components/APICall/mimeTypeCategories.ts +46 -46
  16. package/src/Components/APICall/parseData.ts +167 -167
  17. package/src/Components/APICall/parseHeaders.ts +41 -41
  18. package/src/Components/APICall/parseProxy.ts +68 -68
  19. package/src/Components/APICall/parseUrl.ts +91 -91
  20. package/src/Components/APIEndpoint.class.ts +234 -234
  21. package/src/Components/APIOutput.class.ts +58 -58
  22. package/src/Components/AgentPlugin.class.ts +102 -102
  23. package/src/Components/Async.class.ts +155 -155
  24. package/src/Components/Await.class.ts +90 -90
  25. package/src/Components/Classifier.class.ts +158 -158
  26. package/src/Components/Component.class.ts +147 -147
  27. package/src/Components/ComponentHost.class.ts +38 -38
  28. package/src/Components/DataSourceCleaner.class.ts +92 -92
  29. package/src/Components/DataSourceIndexer.class.ts +181 -181
  30. package/src/Components/DataSourceLookup.class.ts +161 -161
  31. package/src/Components/ECMASandbox.class.ts +72 -72
  32. package/src/Components/FEncDec.class.ts +29 -29
  33. package/src/Components/FHash.class.ts +33 -33
  34. package/src/Components/FSign.class.ts +80 -80
  35. package/src/Components/FSleep.class.ts +25 -25
  36. package/src/Components/FTimestamp.class.ts +66 -66
  37. package/src/Components/FileStore.class.ts +78 -78
  38. package/src/Components/ForEach.class.ts +97 -97
  39. package/src/Components/GPTPlugin.class.ts +70 -70
  40. package/src/Components/GenAILLM.class.ts +586 -586
  41. package/src/Components/HuggingFace.class.ts +314 -314
  42. package/src/Components/Image/imageSettings.config.ts +70 -70
  43. package/src/Components/ImageGenerator.class.ts +483 -483
  44. package/src/Components/JSONFilter.class.ts +54 -54
  45. package/src/Components/LLMAssistant.class.ts +213 -213
  46. package/src/Components/LogicAND.class.ts +28 -28
  47. package/src/Components/LogicAtLeast.class.ts +85 -85
  48. package/src/Components/LogicAtMost.class.ts +86 -86
  49. package/src/Components/LogicOR.class.ts +29 -29
  50. package/src/Components/LogicXOR.class.ts +34 -34
  51. package/src/Components/MCPClient.class.ts +137 -137
  52. package/src/Components/MemoryDeleteKeyVal.class.ts +70 -70
  53. package/src/Components/MemoryReadKeyVal.class.ts +67 -67
  54. package/src/Components/MemoryWriteKeyVal.class.ts +62 -62
  55. package/src/Components/MemoryWriteObject.class.ts +97 -97
  56. package/src/Components/MultimodalLLM.class.ts +128 -128
  57. package/src/Components/OpenAPI.class.ts +72 -72
  58. package/src/Components/PromptGenerator.class.ts +122 -122
  59. package/src/Components/ScrapflyWebScrape.class.ts +183 -183
  60. package/src/Components/ServerlessCode.class.ts +123 -123
  61. package/src/Components/TavilyWebSearch.class.ts +103 -103
  62. package/src/Components/VisionLLM.class.ts +104 -104
  63. package/src/Components/ZapierAction.class.ts +127 -127
  64. package/src/Components/index.ts +97 -97
  65. package/src/Core/AgentProcess.helper.ts +240 -240
  66. package/src/Core/Connector.class.ts +123 -123
  67. package/src/Core/ConnectorsService.ts +197 -197
  68. package/src/Core/DummyConnector.ts +49 -49
  69. package/src/Core/HookService.ts +105 -105
  70. package/src/Core/SmythRuntime.class.ts +241 -241
  71. package/src/Core/SystemEvents.ts +16 -16
  72. package/src/Core/boot.ts +56 -56
  73. package/src/config.ts +15 -15
  74. package/src/constants.ts +126 -126
  75. package/src/data/hugging-face.params.json +579 -579
  76. package/src/helpers/AWSLambdaCode.helper.ts +624 -599
  77. package/src/helpers/BinaryInput.helper.ts +331 -331
  78. package/src/helpers/Conversation.helper.ts +1157 -1157
  79. package/src/helpers/ECMASandbox.helper.ts +64 -64
  80. package/src/helpers/JsonContent.helper.ts +97 -97
  81. package/src/helpers/LocalCache.helper.ts +97 -97
  82. package/src/helpers/Log.helper.ts +274 -274
  83. package/src/helpers/OpenApiParser.helper.ts +150 -150
  84. package/src/helpers/S3Cache.helper.ts +147 -147
  85. package/src/helpers/SmythURI.helper.ts +5 -5
  86. package/src/helpers/Sysconfig.helper.ts +95 -95
  87. package/src/helpers/TemplateString.helper.ts +243 -243
  88. package/src/helpers/TypeChecker.helper.ts +329 -329
  89. package/src/index.ts +4 -3
  90. package/src/index.ts.bak +4 -3
  91. package/src/subsystems/AgentManager/Agent.class.ts +1114 -1114
  92. package/src/subsystems/AgentManager/Agent.helper.ts +3 -3
  93. package/src/subsystems/AgentManager/AgentData.service/AgentDataConnector.ts +230 -230
  94. package/src/subsystems/AgentManager/AgentData.service/connectors/CLIAgentDataConnector.class.ts +66 -66
  95. package/src/subsystems/AgentManager/AgentData.service/connectors/LocalAgentDataConnector.class.ts +145 -145
  96. package/src/subsystems/AgentManager/AgentData.service/connectors/NullAgentData.class.ts +39 -39
  97. package/src/subsystems/AgentManager/AgentData.service/index.ts +18 -18
  98. package/src/subsystems/AgentManager/AgentLogger.class.ts +301 -301
  99. package/src/subsystems/AgentManager/AgentRequest.class.ts +51 -51
  100. package/src/subsystems/AgentManager/AgentRuntime.class.ts +557 -557
  101. package/src/subsystems/AgentManager/AgentSSE.class.ts +101 -101
  102. package/src/subsystems/AgentManager/AgentSettings.class.ts +52 -52
  103. package/src/subsystems/AgentManager/Component.service/ComponentConnector.ts +32 -32
  104. package/src/subsystems/AgentManager/Component.service/connectors/LocalComponentConnector.class.ts +60 -60
  105. package/src/subsystems/AgentManager/Component.service/index.ts +11 -11
  106. package/src/subsystems/AgentManager/EmbodimentSettings.class.ts +47 -47
  107. package/src/subsystems/AgentManager/ForkedAgent.class.ts +154 -154
  108. package/src/subsystems/AgentManager/OSResourceMonitor.ts +77 -77
  109. package/src/subsystems/ComputeManager/Code.service/CodeConnector.ts +98 -98
  110. package/src/subsystems/ComputeManager/Code.service/connectors/AWSLambdaCode.class.ts +171 -172
  111. package/src/subsystems/ComputeManager/Code.service/connectors/ECMASandbox.class.ts +131 -131
  112. package/src/subsystems/ComputeManager/Code.service/index.ts +13 -13
  113. package/src/subsystems/IO/CLI.service/CLIConnector.ts +47 -47
  114. package/src/subsystems/IO/CLI.service/index.ts +9 -9
  115. package/src/subsystems/IO/Log.service/LogConnector.ts +32 -32
  116. package/src/subsystems/IO/Log.service/connectors/ConsoleLog.class.ts +28 -28
  117. package/src/subsystems/IO/Log.service/index.ts +13 -13
  118. package/src/subsystems/IO/NKV.service/NKVConnector.ts +43 -43
  119. package/src/subsystems/IO/NKV.service/connectors/NKVLocalStorage.class.ts +234 -234
  120. package/src/subsystems/IO/NKV.service/connectors/NKVRAM.class.ts +204 -204
  121. package/src/subsystems/IO/NKV.service/connectors/NKVRedis.class.ts +182 -182
  122. package/src/subsystems/IO/NKV.service/index.ts +14 -14
  123. package/src/subsystems/IO/Router.service/RouterConnector.ts +21 -21
  124. package/src/subsystems/IO/Router.service/connectors/ExpressRouter.class.ts +48 -48
  125. package/src/subsystems/IO/Router.service/connectors/NullRouter.class.ts +40 -40
  126. package/src/subsystems/IO/Router.service/index.ts +11 -11
  127. package/src/subsystems/IO/Storage.service/SmythFS.class.ts +488 -488
  128. package/src/subsystems/IO/Storage.service/StorageConnector.ts +66 -66
  129. package/src/subsystems/IO/Storage.service/connectors/LocalStorage.class.ts +327 -327
  130. package/src/subsystems/IO/Storage.service/connectors/S3Storage.class.ts +482 -482
  131. package/src/subsystems/IO/Storage.service/index.ts +13 -13
  132. package/src/subsystems/IO/VectorDB.service/VectorDBConnector.ts +108 -108
  133. package/src/subsystems/IO/VectorDB.service/connectors/MilvusVectorDB.class.ts +465 -465
  134. package/src/subsystems/IO/VectorDB.service/connectors/PineconeVectorDB.class.ts +387 -387
  135. package/src/subsystems/IO/VectorDB.service/connectors/RAMVecrtorDB.class.ts +408 -408
  136. package/src/subsystems/IO/VectorDB.service/embed/BaseEmbedding.ts +107 -107
  137. package/src/subsystems/IO/VectorDB.service/embed/GoogleEmbedding.ts +118 -118
  138. package/src/subsystems/IO/VectorDB.service/embed/OpenAIEmbedding.ts +109 -109
  139. package/src/subsystems/IO/VectorDB.service/embed/index.ts +26 -26
  140. package/src/subsystems/IO/VectorDB.service/index.ts +14 -14
  141. package/src/subsystems/LLMManager/LLM.helper.ts +251 -251
  142. package/src/subsystems/LLMManager/LLM.inference.ts +345 -345
  143. package/src/subsystems/LLMManager/LLM.service/LLMConnector.ts +492 -492
  144. package/src/subsystems/LLMManager/LLM.service/LLMCredentials.helper.ts +171 -171
  145. package/src/subsystems/LLMManager/LLM.service/connectors/Anthropic.class.ts +666 -666
  146. package/src/subsystems/LLMManager/LLM.service/connectors/Bedrock.class.ts +407 -407
  147. package/src/subsystems/LLMManager/LLM.service/connectors/Echo.class.ts +92 -92
  148. package/src/subsystems/LLMManager/LLM.service/connectors/GoogleAI.class.ts +983 -983
  149. package/src/subsystems/LLMManager/LLM.service/connectors/Groq.class.ts +319 -319
  150. package/src/subsystems/LLMManager/LLM.service/connectors/Ollama.class.ts +362 -0
  151. package/src/subsystems/LLMManager/LLM.service/connectors/Perplexity.class.ts +257 -257
  152. package/src/subsystems/LLMManager/LLM.service/connectors/VertexAI.class.ts +430 -430
  153. package/src/subsystems/LLMManager/LLM.service/connectors/openai/OpenAIConnector.class.ts +503 -503
  154. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ChatCompletionsApiInterface.ts +524 -524
  155. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterface.ts +100 -100
  156. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterfaceFactory.ts +81 -81
  157. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ResponsesApiInterface.ts +1145 -1145
  158. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/constants.ts +13 -13
  159. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/index.ts +4 -4
  160. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/utils.ts +11 -11
  161. package/src/subsystems/LLMManager/LLM.service/connectors/openai/types.ts +32 -32
  162. package/src/subsystems/LLMManager/LLM.service/connectors/xAI.class.ts +478 -478
  163. package/src/subsystems/LLMManager/LLM.service/index.ts +47 -44
  164. package/src/subsystems/LLMManager/ModelsProvider.service/ModelsProviderConnector.ts +303 -303
  165. package/src/subsystems/LLMManager/ModelsProvider.service/connectors/JSONModelsProvider.class.ts +271 -271
  166. package/src/subsystems/LLMManager/ModelsProvider.service/index.ts +11 -11
  167. package/src/subsystems/LLMManager/custom-models.ts +854 -854
  168. package/src/subsystems/LLMManager/models.ts +2540 -2540
  169. package/src/subsystems/LLMManager/paramMappings.ts +69 -69
  170. package/src/subsystems/MemoryManager/Cache.service/CacheConnector.ts +86 -86
  171. package/src/subsystems/MemoryManager/Cache.service/connectors/LocalStorageCache.class.ts +297 -297
  172. package/src/subsystems/MemoryManager/Cache.service/connectors/RAMCache.class.ts +214 -214
  173. package/src/subsystems/MemoryManager/Cache.service/connectors/RedisCache.class.ts +252 -252
  174. package/src/subsystems/MemoryManager/Cache.service/connectors/S3Cache.class.ts +373 -373
  175. package/src/subsystems/MemoryManager/Cache.service/index.ts +15 -15
  176. package/src/subsystems/MemoryManager/LLMCache.ts +72 -72
  177. package/src/subsystems/MemoryManager/LLMContext.ts +124 -124
  178. package/src/subsystems/MemoryManager/LLMMemory.service/LLMMemoryConnector.ts +26 -26
  179. package/src/subsystems/MemoryManager/RuntimeContext.ts +277 -277
  180. package/src/subsystems/Security/AccessControl/ACL.class.ts +208 -208
  181. package/src/subsystems/Security/AccessControl/AccessCandidate.class.ts +82 -82
  182. package/src/subsystems/Security/AccessControl/AccessRequest.class.ts +52 -52
  183. package/src/subsystems/Security/Account.service/AccountConnector.ts +44 -44
  184. package/src/subsystems/Security/Account.service/connectors/DummyAccount.class.ts +130 -130
  185. package/src/subsystems/Security/Account.service/connectors/JSONFileAccount.class.ts +170 -170
  186. package/src/subsystems/Security/Account.service/connectors/MySQLAccount.class.ts +76 -76
  187. package/src/subsystems/Security/Account.service/index.ts +14 -14
  188. package/src/subsystems/Security/Credentials.helper.ts +62 -62
  189. package/src/subsystems/Security/ManagedVault.service/ManagedVaultConnector.ts +38 -38
  190. package/src/subsystems/Security/ManagedVault.service/connectors/NullManagedVault.class.ts +53 -53
  191. package/src/subsystems/Security/ManagedVault.service/connectors/SecretManagerManagedVault.ts +154 -154
  192. package/src/subsystems/Security/ManagedVault.service/index.ts +12 -12
  193. package/src/subsystems/Security/SecureConnector.class.ts +110 -110
  194. package/src/subsystems/Security/Vault.service/Vault.helper.ts +30 -30
  195. package/src/subsystems/Security/Vault.service/VaultConnector.ts +29 -29
  196. package/src/subsystems/Security/Vault.service/connectors/HashicorpVault.class.ts +46 -46
  197. package/src/subsystems/Security/Vault.service/connectors/JSONFileVault.class.ts +221 -221
  198. package/src/subsystems/Security/Vault.service/connectors/NullVault.class.ts +54 -54
  199. package/src/subsystems/Security/Vault.service/connectors/SecretsManager.class.ts +140 -140
  200. package/src/subsystems/Security/Vault.service/index.ts +12 -12
  201. package/src/types/ACL.types.ts +104 -104
  202. package/src/types/AWS.types.ts +10 -10
  203. package/src/types/Agent.types.ts +61 -61
  204. package/src/types/AgentLogger.types.ts +17 -17
  205. package/src/types/Cache.types.ts +1 -1
  206. package/src/types/Common.types.ts +2 -2
  207. package/src/types/LLM.types.ts +520 -519
  208. package/src/types/Redis.types.ts +8 -8
  209. package/src/types/SRE.types.ts +64 -64
  210. package/src/types/Security.types.ts +14 -14
  211. package/src/types/Storage.types.ts +5 -5
  212. package/src/types/VectorDB.types.ts +86 -86
  213. package/src/utils/base64.utils.ts +275 -275
  214. package/src/utils/cli.utils.ts +68 -68
  215. package/src/utils/data.utils.ts +322 -322
  216. package/src/utils/date-time.utils.ts +22 -22
  217. package/src/utils/general.utils.ts +238 -238
  218. package/src/utils/index.ts +12 -12
  219. package/src/utils/lazy-client.ts +261 -261
  220. package/src/utils/numbers.utils.ts +13 -13
  221. package/src/utils/oauth.utils.ts +35 -35
  222. package/src/utils/string.utils.ts +414 -414
  223. package/src/utils/url.utils.ts +19 -19
  224. package/src/utils/validation.utils.ts +74 -74
  225. package/dist/bundle-analysis-lazy.html +0 -4949
  226. package/dist/bundle-analysis.html +0 -4949
  227. package/dist/types/Components/Triggers/GmailTrigger.class.d.ts +0 -13
  228. package/dist/types/Components/Triggers/Trigger.class.d.ts +0 -3
  229. package/dist/types/helpers/AIPerformanceAnalyzer.helper.d.ts +0 -45
  230. package/dist/types/helpers/AIPerformanceCollector.helper.d.ts +0 -111
  231. package/dist/types/subsystems/IO/Storage.service/connectors/AzureBlobStorage.class.d.ts +0 -211
  232. package/dist/types/subsystems/IO/VectorDB.service/connectors/WeaviateVectorDB.class.d.ts +0 -187
  233. package/dist/types/subsystems/PerformanceManager/Performance.service/PerformanceConnector.d.ts +0 -102
  234. package/dist/types/subsystems/PerformanceManager/Performance.service/connectors/LocalPerformanceConnector.class.d.ts +0 -100
  235. package/dist/types/subsystems/PerformanceManager/Performance.service/index.d.ts +0 -22
  236. package/dist/types/types/Performance.types.d.ts +0 -468
  237. package/dist/types/utils/package-manager.utils.d.ts +0 -26
@@ -1,408 +1,408 @@
1
- //==[ SRE: RAMVectorDB ]======================
2
-
3
- import { ACL } from '@sre/Security/AccessControl/ACL.class';
4
- import { IAccessCandidate, IACL, TAccessLevel } from '@sre/types/ACL.types';
5
- import { AccessRequest } from '@sre/Security/AccessControl/AccessRequest.class';
6
- import { AccessCandidate } from '@sre/Security/AccessControl/AccessCandidate.class';
7
- import { SecureConnector } from '@sre/Security/SecureConnector.class';
8
- import { VectorDBConnector, DeleteTarget } from '../VectorDBConnector';
9
- import {
10
- DatasourceDto,
11
- IStorageVectorDataSource,
12
- IStorageVectorNamespace,
13
- IVectorDataSourceDto,
14
- QueryOptions,
15
- VectorsResultData,
16
- } from '@sre/types/VectorDB.types';
17
- import { ConnectorService } from '@sre/Core/ConnectorsService';
18
- import { Logger } from '@sre/helpers/Log.helper';
19
- import { AccountConnector } from '@sre/Security/Account.service/AccountConnector';
20
- import { OpenAIEmbeds } from '@sre/IO/VectorDB.service/embed/OpenAIEmbedding';
21
- import crypto from 'crypto';
22
- import { BaseEmbedding, TEmbeddings } from '../embed/BaseEmbedding';
23
- import { EmbeddingsFactory } from '../embed';
24
- import { chunkText } from '@sre/utils/string.utils';
25
- import { jsonrepair } from 'jsonrepair';
26
-
27
- const console = Logger('RAM VectorDB');
28
-
29
- interface VectorData {
30
- id: string;
31
- values: number[];
32
- datasource: string;
33
- metadata?: { [key: string]: any };
34
- }
35
-
36
- export type RAMVectorDBConfig = {
37
- /**
38
- * The embeddings model to use
39
- */
40
- embeddings?: TEmbeddings;
41
- };
42
-
43
- /**
44
- * RAM Vector Database - stores everything in memory
45
- * Data structure:
46
- * - vectors: namespace -> VectorData[]
47
- * - namespaces: namespace -> IStorageVectorNamespace
48
- * - datasources: namespace -> datasourceId -> IStorageVectorDataSource
49
- * - acls: resourceId -> IACL
50
- */
51
-
52
- export class RAMVectorDB extends VectorDBConnector {
53
- public name = 'RAMVec';
54
- public id = 'ram';
55
- //private openaiApiKey: string;
56
- private accountConnector: AccountConnector;
57
- //private embeddingsProvider: OpenAIEmbeds;
58
-
59
- // In-memory storage (shared across connector instances)
60
- private static vectors: Record<string, VectorData[]> = {};
61
- private static namespaces: Record<string, IStorageVectorNamespace> = {};
62
- private static datasources: Record<string, Record<string, IStorageVectorDataSource>> = {};
63
- private static acls: Record<string, IACL> = {};
64
-
65
- //embedder
66
- public embedder: BaseEmbedding;
67
-
68
- constructor(protected _settings: RAMVectorDBConfig) {
69
- super(_settings);
70
-
71
- this.accountConnector = ConnectorService.getAccountConnector();
72
-
73
- if (!_settings.embeddings) {
74
- _settings.embeddings = { provider: 'OpenAI', model: 'text-embedding-3-large', params: { dimensions: 1024 } };
75
- }
76
- if (!_settings.embeddings.params) _settings.embeddings.params = { dimensions: 1024 };
77
- if (!_settings.embeddings.params?.dimensions) _settings.embeddings.params.dimensions = 1024;
78
-
79
- this.embedder = EmbeddingsFactory.create(_settings.embeddings.provider, _settings.embeddings);
80
- }
81
-
82
- public async getResourceACL(resourceId: string, candidate: IAccessCandidate): Promise<ACL> {
83
- //const teamId = await this.accountConnector.getCandidateTeam(AccessCandidate.clone(candidate));
84
- const preparedNs = this.constructNsName(candidate as AccessCandidate, resourceId);
85
- const acl = RAMVectorDB.acls[preparedNs];
86
- const exists = !!acl;
87
-
88
- if (!exists) {
89
- //the resource does not exist yet, we grant write access to the candidate in order to allow the resource creation
90
- return new ACL().addAccess(candidate.role, candidate.id, TAccessLevel.Owner);
91
- }
92
- return ACL.from(acl);
93
- }
94
-
95
- @SecureConnector.AccessControl
96
- protected async createNamespace(acRequest: AccessRequest, namespace: string, metadata?: { [key: string]: any }): Promise<void> {
97
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
98
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
99
-
100
- if (!RAMVectorDB.namespaces[preparedNs]) {
101
- const nsData = {
102
- namespace: preparedNs,
103
- displayName: namespace,
104
- candidateId: acRequest.candidate.id,
105
- candidateRole: acRequest.candidate.role,
106
- metadata: {
107
- ...metadata,
108
- storageType: 'RAM',
109
- },
110
- };
111
-
112
- // Store namespace metadata in memory
113
- RAMVectorDB.namespaces[preparedNs] = nsData;
114
-
115
- // Initialize namespace vectors storage
116
- RAMVectorDB.vectors[preparedNs] = [];
117
-
118
- // Initialize datasources storage for this namespace
119
- RAMVectorDB.datasources[preparedNs] = {};
120
- }
121
-
122
- // Store ACL in memory
123
- const acl = new ACL().addAccess(acRequest.candidate.role, acRequest.candidate.id, TAccessLevel.Owner).ACL;
124
- RAMVectorDB.acls[preparedNs] = acl;
125
-
126
- return new Promise<void>((resolve) => resolve());
127
- }
128
-
129
- @SecureConnector.AccessControl
130
- protected async namespaceExists(acRequest: AccessRequest, namespace: string): Promise<boolean> {
131
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
132
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
133
- return !!RAMVectorDB.namespaces[preparedNs];
134
- }
135
-
136
- @SecureConnector.AccessControl
137
- protected async getNamespace(acRequest: AccessRequest, namespace: string): Promise<IStorageVectorNamespace> {
138
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
139
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
140
- const nsData = RAMVectorDB.namespaces[preparedNs];
141
- if (!nsData) {
142
- throw new Error(`Namespace ${namespace} not found`);
143
- }
144
- return nsData;
145
- }
146
-
147
- @SecureConnector.AccessControl
148
- protected async listNamespaces(acRequest: AccessRequest): Promise<IStorageVectorNamespace[]> {
149
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
150
-
151
- // Filter namespaces by team
152
- return Object.values(RAMVectorDB.namespaces).filter((ns) => ns.candidateId === acRequest.candidate.id);
153
- }
154
-
155
- @SecureConnector.AccessControl
156
- protected async deleteNamespace(acRequest: AccessRequest, namespace: string): Promise<void> {
157
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
158
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
159
-
160
- // Delete from memory
161
- delete RAMVectorDB.vectors[preparedNs];
162
- delete RAMVectorDB.namespaces[preparedNs];
163
- delete RAMVectorDB.datasources[preparedNs];
164
- delete RAMVectorDB.acls[preparedNs];
165
- }
166
-
167
- @SecureConnector.AccessControl
168
- protected async search(
169
- acRequest: AccessRequest,
170
- namespace: string,
171
- query: string | number[],
172
- options: QueryOptions = {}
173
- ): Promise<VectorsResultData> {
174
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
175
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
176
-
177
- if (!RAMVectorDB.namespaces[preparedNs]) {
178
- throw new Error('Namespace does not exist');
179
- }
180
-
181
- // Get query vector
182
- let queryVector = query;
183
- if (typeof query === 'string') {
184
- queryVector = await this.embedder.embedText(query, acRequest.candidate as AccessCandidate);
185
- }
186
-
187
- // Search in namespace data
188
- const namespaceData = RAMVectorDB.vectors[preparedNs] || [];
189
- const results: Array<{ id: string; score: number; values: number[]; metadata?: any; text: string }> = [];
190
-
191
- for (const vector of namespaceData) {
192
- const similarity = this.cosineSimilarity(queryVector as number[], vector.values);
193
-
194
- let userMetadata = undefined;
195
- if (options.includeMetadata) {
196
- if (vector.metadata?.[this.USER_METADATA_KEY]) {
197
- try {
198
- userMetadata = JSON.parse(vector.metadata[this.USER_METADATA_KEY]);
199
- } catch {
200
- userMetadata = vector.metadata[this.USER_METADATA_KEY];
201
- }
202
- } else {
203
- userMetadata = {}; // Return empty object when no metadata exists, like Milvus
204
- }
205
- }
206
-
207
- results.push({
208
- id: vector.id,
209
- score: similarity,
210
- values: vector.values,
211
- text: vector.metadata?.text as string | undefined,
212
- metadata: options.includeMetadata ? userMetadata : undefined,
213
- });
214
- }
215
-
216
- // Sort by similarity (highest first) and limit results
217
- const topK = options.topK || 10;
218
- const sortedResults = results.sort((a, b) => b.score - a.score).slice(0, topK);
219
-
220
- return sortedResults;
221
- }
222
-
223
- @SecureConnector.AccessControl
224
- protected async insert(
225
- acRequest: AccessRequest,
226
- namespace: string,
227
- sourceWrapper: IVectorDataSourceDto | IVectorDataSourceDto[]
228
- ): Promise<string[]> {
229
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
230
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
231
-
232
- sourceWrapper = Array.isArray(sourceWrapper) ? sourceWrapper : [sourceWrapper];
233
-
234
- // make sure that all sources are of the same type (source.source)
235
- if (sourceWrapper.some((s) => this.embedder.detectSourceType(s.source) !== this.embedder.detectSourceType(sourceWrapper[0].source))) {
236
- throw new Error('All sources must be of the same type');
237
- }
238
-
239
- const sourceType = this.embedder.detectSourceType(sourceWrapper[0].source);
240
- if (sourceType === 'unknown' || sourceType === 'url') throw new Error('Invalid source type');
241
-
242
- const transformedSource = await this.embedder.transformSource(sourceWrapper, sourceType, acRequest.candidate as AccessCandidate);
243
-
244
- if (!RAMVectorDB.vectors[preparedNs]) {
245
- RAMVectorDB.vectors[preparedNs] = [];
246
- }
247
-
248
- const insertedIds: string[] = [];
249
-
250
- for (const source of transformedSource) {
251
- const vectorData: VectorData = {
252
- id: source.id,
253
- values: source.source as number[],
254
- datasource: source.metadata?.datasourceId || 'unknown',
255
- metadata: source.metadata,
256
- };
257
-
258
- // Check if vector with this ID already exists and update it
259
- const existingIndex = RAMVectorDB.vectors[preparedNs].findIndex((v) => v.id === source.id);
260
- if (existingIndex >= 0) {
261
- RAMVectorDB.vectors[preparedNs][existingIndex] = vectorData;
262
- } else {
263
- RAMVectorDB.vectors[preparedNs].push(vectorData);
264
- }
265
-
266
- insertedIds.push(source.id);
267
- }
268
-
269
- return insertedIds;
270
- }
271
-
272
- @SecureConnector.AccessControl
273
- protected async delete(acRequest: AccessRequest, namespace: string, deleteTarget: DeleteTarget): Promise<void> {
274
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
275
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
276
-
277
- const isDeleteByFilter = typeof deleteTarget === 'object' && !Array.isArray(deleteTarget);
278
-
279
- if (isDeleteByFilter) {
280
- // Handle delete by filter (e.g., by datasourceId)
281
- if ('datasourceId' in deleteTarget && deleteTarget.datasourceId) {
282
- if (RAMVectorDB.vectors[preparedNs]) {
283
- RAMVectorDB.vectors[preparedNs] = RAMVectorDB.vectors[preparedNs].filter(
284
- (vector) => vector.datasource !== deleteTarget.datasourceId
285
- );
286
- }
287
- } else {
288
- throw new Error('Unsupported delete filter');
289
- }
290
- } else {
291
- // Handle delete by ID(s)
292
- const ids = Array.isArray(deleteTarget) ? deleteTarget : [deleteTarget];
293
- if (RAMVectorDB.vectors[preparedNs]) {
294
- RAMVectorDB.vectors[preparedNs] = RAMVectorDB.vectors[preparedNs].filter((vector) => !ids.includes(vector.id));
295
- }
296
- }
297
- }
298
-
299
- @SecureConnector.AccessControl
300
- protected async createDatasource(acRequest: AccessRequest, namespace: string, datasource: DatasourceDto): Promise<IStorageVectorDataSource> {
301
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
302
- const acl = new ACL().addAccess(acRequest.candidate.role, acRequest.candidate.id, TAccessLevel.Owner);
303
- const dsId = datasource.id || crypto.randomUUID();
304
-
305
- const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
306
- const chunkedText = chunkText(datasource.text, {
307
- chunkSize: datasource.chunkSize,
308
- chunkOverlap: datasource.chunkOverlap,
309
- });
310
- const label = datasource.label || 'Untitled';
311
- const ids = Array.from({ length: chunkedText.length }, (_, i) => `${dsId}_${crypto.randomUUID()}`);
312
- const source: IVectorDataSourceDto[] = chunkedText.map((doc, i) => {
313
- return {
314
- id: ids[i],
315
- source: doc,
316
- metadata: {
317
- acl: acl.serializedACL,
318
- namespaceId: formattedNs,
319
- datasourceId: dsId,
320
- datasourceLabel: label,
321
- [this.USER_METADATA_KEY]: datasource.metadata ? jsonrepair(JSON.stringify(datasource.metadata)) : undefined,
322
- },
323
- };
324
- });
325
-
326
- const _vIds = await this.insert(acRequest, namespace, source);
327
-
328
- const dsData: IStorageVectorDataSource = {
329
- namespaceId: formattedNs,
330
- candidateId: acRequest.candidate.id,
331
- candidateRole: acRequest.candidate.role,
332
- name: datasource.label || 'Untitled',
333
- metadata: datasource.metadata ? jsonrepair(JSON.stringify(datasource.metadata)) : undefined,
334
- text: datasource.text,
335
- vectorIds: _vIds,
336
- id: dsId,
337
- };
338
-
339
- // Store datasource metadata in memory
340
- if (!RAMVectorDB.datasources[formattedNs]) {
341
- RAMVectorDB.datasources[formattedNs] = {};
342
- }
343
- RAMVectorDB.datasources[formattedNs][dsId] = dsData;
344
-
345
- return dsData;
346
- }
347
-
348
- @SecureConnector.AccessControl
349
- protected async deleteDatasource(acRequest: AccessRequest, namespace: string, datasourceId: string): Promise<void> {
350
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
351
- const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
352
-
353
- // Delete all vectors belonging to this datasource using a filter by datasourceId,
354
- // so we remove vectors from any prior insert/update operations sharing the same datasource id
355
- await this.delete(acRequest, namespace, { datasourceId });
356
-
357
- // Delete datasource metadata (if present)
358
- if (RAMVectorDB.datasources[formattedNs] && RAMVectorDB.datasources[formattedNs][datasourceId]) {
359
- delete RAMVectorDB.datasources[formattedNs][datasourceId];
360
- }
361
- }
362
-
363
- @SecureConnector.AccessControl
364
- protected async listDatasources(acRequest: AccessRequest, namespace: string): Promise<IStorageVectorDataSource[]> {
365
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
366
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
367
-
368
- const namespaceDatasources = RAMVectorDB.datasources[preparedNs] || {};
369
- return Object.values(namespaceDatasources);
370
- }
371
-
372
- @SecureConnector.AccessControl
373
- protected async getDatasource(acRequest: AccessRequest, namespace: string, datasourceId: string): Promise<IStorageVectorDataSource | undefined> {
374
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
375
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
376
-
377
- const datasource = RAMVectorDB.datasources[preparedNs]?.[datasourceId];
378
- return datasource; // Return undefined if not found, like MilvusVectorDB
379
- }
380
-
381
- /**
382
- * Calculate cosine similarity between two vectors
383
- */
384
- private cosineSimilarity(vecA: number[], vecB: number[]): number {
385
- if (vecA.length !== vecB.length) {
386
- throw new Error('Vectors must have the same length');
387
- }
388
-
389
- let dotProduct = 0;
390
- let normA = 0;
391
- let normB = 0;
392
-
393
- for (let i = 0; i < vecA.length; i++) {
394
- dotProduct += vecA[i] * vecB[i];
395
- normA += vecA[i] * vecA[i];
396
- normB += vecB[i] * vecB[i];
397
- }
398
-
399
- normA = Math.sqrt(normA);
400
- normB = Math.sqrt(normB);
401
-
402
- if (normA === 0 || normB === 0) {
403
- return 0;
404
- }
405
-
406
- return dotProduct / (normA * normB);
407
- }
408
- }
1
+ //==[ SRE: RAMVectorDB ]======================
2
+
3
+ import { ACL } from '@sre/Security/AccessControl/ACL.class';
4
+ import { IAccessCandidate, IACL, TAccessLevel } from '@sre/types/ACL.types';
5
+ import { AccessRequest } from '@sre/Security/AccessControl/AccessRequest.class';
6
+ import { AccessCandidate } from '@sre/Security/AccessControl/AccessCandidate.class';
7
+ import { SecureConnector } from '@sre/Security/SecureConnector.class';
8
+ import { VectorDBConnector, DeleteTarget } from '../VectorDBConnector';
9
+ import {
10
+ DatasourceDto,
11
+ IStorageVectorDataSource,
12
+ IStorageVectorNamespace,
13
+ IVectorDataSourceDto,
14
+ QueryOptions,
15
+ VectorsResultData,
16
+ } from '@sre/types/VectorDB.types';
17
+ import { ConnectorService } from '@sre/Core/ConnectorsService';
18
+ import { Logger } from '@sre/helpers/Log.helper';
19
+ import { AccountConnector } from '@sre/Security/Account.service/AccountConnector';
20
+ import { OpenAIEmbeds } from '@sre/IO/VectorDB.service/embed/OpenAIEmbedding';
21
+ import crypto from 'crypto';
22
+ import { BaseEmbedding, TEmbeddings } from '../embed/BaseEmbedding';
23
+ import { EmbeddingsFactory } from '../embed';
24
+ import { chunkText } from '@sre/utils/string.utils';
25
+ import { jsonrepair } from 'jsonrepair';
26
+
27
+ const console = Logger('RAM VectorDB');
28
+
29
+ interface VectorData {
30
+ id: string;
31
+ values: number[];
32
+ datasource: string;
33
+ metadata?: { [key: string]: any };
34
+ }
35
+
36
+ export type RAMVectorDBConfig = {
37
+ /**
38
+ * The embeddings model to use
39
+ */
40
+ embeddings?: TEmbeddings;
41
+ };
42
+
43
+ /**
44
+ * RAM Vector Database - stores everything in memory
45
+ * Data structure:
46
+ * - vectors: namespace -> VectorData[]
47
+ * - namespaces: namespace -> IStorageVectorNamespace
48
+ * - datasources: namespace -> datasourceId -> IStorageVectorDataSource
49
+ * - acls: resourceId -> IACL
50
+ */
51
+
52
+ export class RAMVectorDB extends VectorDBConnector {
53
+ public name = 'RAMVec';
54
+ public id = 'ram';
55
+ //private openaiApiKey: string;
56
+ private accountConnector: AccountConnector;
57
+ //private embeddingsProvider: OpenAIEmbeds;
58
+
59
+ // In-memory storage (shared across connector instances)
60
+ private static vectors: Record<string, VectorData[]> = {};
61
+ private static namespaces: Record<string, IStorageVectorNamespace> = {};
62
+ private static datasources: Record<string, Record<string, IStorageVectorDataSource>> = {};
63
+ private static acls: Record<string, IACL> = {};
64
+
65
+ //embedder
66
+ public embedder: BaseEmbedding;
67
+
68
+ constructor(protected _settings: RAMVectorDBConfig) {
69
+ super(_settings);
70
+
71
+ this.accountConnector = ConnectorService.getAccountConnector();
72
+
73
+ if (!_settings.embeddings) {
74
+ _settings.embeddings = { provider: 'OpenAI', model: 'text-embedding-3-large', params: { dimensions: 1024 } };
75
+ }
76
+ if (!_settings.embeddings.params) _settings.embeddings.params = { dimensions: 1024 };
77
+ if (!_settings.embeddings.params?.dimensions) _settings.embeddings.params.dimensions = 1024;
78
+
79
+ this.embedder = EmbeddingsFactory.create(_settings.embeddings.provider, _settings.embeddings);
80
+ }
81
+
82
+ public async getResourceACL(resourceId: string, candidate: IAccessCandidate): Promise<ACL> {
83
+ //const teamId = await this.accountConnector.getCandidateTeam(AccessCandidate.clone(candidate));
84
+ const preparedNs = this.constructNsName(candidate as AccessCandidate, resourceId);
85
+ const acl = RAMVectorDB.acls[preparedNs];
86
+ const exists = !!acl;
87
+
88
+ if (!exists) {
89
+ //the resource does not exist yet, we grant write access to the candidate in order to allow the resource creation
90
+ return new ACL().addAccess(candidate.role, candidate.id, TAccessLevel.Owner);
91
+ }
92
+ return ACL.from(acl);
93
+ }
94
+
95
+ @SecureConnector.AccessControl
96
+ protected async createNamespace(acRequest: AccessRequest, namespace: string, metadata?: { [key: string]: any }): Promise<void> {
97
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
98
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
99
+
100
+ if (!RAMVectorDB.namespaces[preparedNs]) {
101
+ const nsData = {
102
+ namespace: preparedNs,
103
+ displayName: namespace,
104
+ candidateId: acRequest.candidate.id,
105
+ candidateRole: acRequest.candidate.role,
106
+ metadata: {
107
+ ...metadata,
108
+ storageType: 'RAM',
109
+ },
110
+ };
111
+
112
+ // Store namespace metadata in memory
113
+ RAMVectorDB.namespaces[preparedNs] = nsData;
114
+
115
+ // Initialize namespace vectors storage
116
+ RAMVectorDB.vectors[preparedNs] = [];
117
+
118
+ // Initialize datasources storage for this namespace
119
+ RAMVectorDB.datasources[preparedNs] = {};
120
+ }
121
+
122
+ // Store ACL in memory
123
+ const acl = new ACL().addAccess(acRequest.candidate.role, acRequest.candidate.id, TAccessLevel.Owner).ACL;
124
+ RAMVectorDB.acls[preparedNs] = acl;
125
+
126
+ return new Promise<void>((resolve) => resolve());
127
+ }
128
+
129
+ @SecureConnector.AccessControl
130
+ protected async namespaceExists(acRequest: AccessRequest, namespace: string): Promise<boolean> {
131
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
132
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
133
+ return !!RAMVectorDB.namespaces[preparedNs];
134
+ }
135
+
136
+ @SecureConnector.AccessControl
137
+ protected async getNamespace(acRequest: AccessRequest, namespace: string): Promise<IStorageVectorNamespace> {
138
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
139
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
140
+ const nsData = RAMVectorDB.namespaces[preparedNs];
141
+ if (!nsData) {
142
+ throw new Error(`Namespace ${namespace} not found`);
143
+ }
144
+ return nsData;
145
+ }
146
+
147
+ @SecureConnector.AccessControl
148
+ protected async listNamespaces(acRequest: AccessRequest): Promise<IStorageVectorNamespace[]> {
149
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
150
+
151
+ // Filter namespaces by team
152
+ return Object.values(RAMVectorDB.namespaces).filter((ns) => ns.candidateId === acRequest.candidate.id);
153
+ }
154
+
155
+ @SecureConnector.AccessControl
156
+ protected async deleteNamespace(acRequest: AccessRequest, namespace: string): Promise<void> {
157
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
158
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
159
+
160
+ // Delete from memory
161
+ delete RAMVectorDB.vectors[preparedNs];
162
+ delete RAMVectorDB.namespaces[preparedNs];
163
+ delete RAMVectorDB.datasources[preparedNs];
164
+ delete RAMVectorDB.acls[preparedNs];
165
+ }
166
+
167
+ @SecureConnector.AccessControl
168
+ protected async search(
169
+ acRequest: AccessRequest,
170
+ namespace: string,
171
+ query: string | number[],
172
+ options: QueryOptions = {}
173
+ ): Promise<VectorsResultData> {
174
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
175
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
176
+
177
+ if (!RAMVectorDB.namespaces[preparedNs]) {
178
+ throw new Error('Namespace does not exist');
179
+ }
180
+
181
+ // Get query vector
182
+ let queryVector = query;
183
+ if (typeof query === 'string') {
184
+ queryVector = await this.embedder.embedText(query, acRequest.candidate as AccessCandidate);
185
+ }
186
+
187
+ // Search in namespace data
188
+ const namespaceData = RAMVectorDB.vectors[preparedNs] || [];
189
+ const results: Array<{ id: string; score: number; values: number[]; metadata?: any; text: string }> = [];
190
+
191
+ for (const vector of namespaceData) {
192
+ const similarity = this.cosineSimilarity(queryVector as number[], vector.values);
193
+
194
+ let userMetadata = undefined;
195
+ if (options.includeMetadata) {
196
+ if (vector.metadata?.[this.USER_METADATA_KEY]) {
197
+ try {
198
+ userMetadata = JSON.parse(vector.metadata[this.USER_METADATA_KEY]);
199
+ } catch {
200
+ userMetadata = vector.metadata[this.USER_METADATA_KEY];
201
+ }
202
+ } else {
203
+ userMetadata = {}; // Return empty object when no metadata exists, like Milvus
204
+ }
205
+ }
206
+
207
+ results.push({
208
+ id: vector.id,
209
+ score: similarity,
210
+ values: vector.values,
211
+ text: vector.metadata?.text as string | undefined,
212
+ metadata: options.includeMetadata ? userMetadata : undefined,
213
+ });
214
+ }
215
+
216
+ // Sort by similarity (highest first) and limit results
217
+ const topK = options.topK || 10;
218
+ const sortedResults = results.sort((a, b) => b.score - a.score).slice(0, topK);
219
+
220
+ return sortedResults;
221
+ }
222
+
223
+ @SecureConnector.AccessControl
224
+ protected async insert(
225
+ acRequest: AccessRequest,
226
+ namespace: string,
227
+ sourceWrapper: IVectorDataSourceDto | IVectorDataSourceDto[]
228
+ ): Promise<string[]> {
229
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
230
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
231
+
232
+ sourceWrapper = Array.isArray(sourceWrapper) ? sourceWrapper : [sourceWrapper];
233
+
234
+ // make sure that all sources are of the same type (source.source)
235
+ if (sourceWrapper.some((s) => this.embedder.detectSourceType(s.source) !== this.embedder.detectSourceType(sourceWrapper[0].source))) {
236
+ throw new Error('All sources must be of the same type');
237
+ }
238
+
239
+ const sourceType = this.embedder.detectSourceType(sourceWrapper[0].source);
240
+ if (sourceType === 'unknown' || sourceType === 'url') throw new Error('Invalid source type');
241
+
242
+ const transformedSource = await this.embedder.transformSource(sourceWrapper, sourceType, acRequest.candidate as AccessCandidate);
243
+
244
+ if (!RAMVectorDB.vectors[preparedNs]) {
245
+ RAMVectorDB.vectors[preparedNs] = [];
246
+ }
247
+
248
+ const insertedIds: string[] = [];
249
+
250
+ for (const source of transformedSource) {
251
+ const vectorData: VectorData = {
252
+ id: source.id,
253
+ values: source.source as number[],
254
+ datasource: source.metadata?.datasourceId || 'unknown',
255
+ metadata: source.metadata,
256
+ };
257
+
258
+ // Check if vector with this ID already exists and update it
259
+ const existingIndex = RAMVectorDB.vectors[preparedNs].findIndex((v) => v.id === source.id);
260
+ if (existingIndex >= 0) {
261
+ RAMVectorDB.vectors[preparedNs][existingIndex] = vectorData;
262
+ } else {
263
+ RAMVectorDB.vectors[preparedNs].push(vectorData);
264
+ }
265
+
266
+ insertedIds.push(source.id);
267
+ }
268
+
269
+ return insertedIds;
270
+ }
271
+
272
+ @SecureConnector.AccessControl
273
+ protected async delete(acRequest: AccessRequest, namespace: string, deleteTarget: DeleteTarget): Promise<void> {
274
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
275
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
276
+
277
+ const isDeleteByFilter = typeof deleteTarget === 'object' && !Array.isArray(deleteTarget);
278
+
279
+ if (isDeleteByFilter) {
280
+ // Handle delete by filter (e.g., by datasourceId)
281
+ if ('datasourceId' in deleteTarget && deleteTarget.datasourceId) {
282
+ if (RAMVectorDB.vectors[preparedNs]) {
283
+ RAMVectorDB.vectors[preparedNs] = RAMVectorDB.vectors[preparedNs].filter(
284
+ (vector) => vector.datasource !== deleteTarget.datasourceId
285
+ );
286
+ }
287
+ } else {
288
+ throw new Error('Unsupported delete filter');
289
+ }
290
+ } else {
291
+ // Handle delete by ID(s)
292
+ const ids = Array.isArray(deleteTarget) ? deleteTarget : [deleteTarget];
293
+ if (RAMVectorDB.vectors[preparedNs]) {
294
+ RAMVectorDB.vectors[preparedNs] = RAMVectorDB.vectors[preparedNs].filter((vector) => !ids.includes(vector.id));
295
+ }
296
+ }
297
+ }
298
+
299
+ @SecureConnector.AccessControl
300
+ protected async createDatasource(acRequest: AccessRequest, namespace: string, datasource: DatasourceDto): Promise<IStorageVectorDataSource> {
301
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
302
+ const acl = new ACL().addAccess(acRequest.candidate.role, acRequest.candidate.id, TAccessLevel.Owner);
303
+ const dsId = datasource.id || crypto.randomUUID();
304
+
305
+ const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
306
+ const chunkedText = chunkText(datasource.text, {
307
+ chunkSize: datasource.chunkSize,
308
+ chunkOverlap: datasource.chunkOverlap,
309
+ });
310
+ const label = datasource.label || 'Untitled';
311
+ const ids = Array.from({ length: chunkedText.length }, (_, i) => `${dsId}_${crypto.randomUUID()}`);
312
+ const source: IVectorDataSourceDto[] = chunkedText.map((doc, i) => {
313
+ return {
314
+ id: ids[i],
315
+ source: doc,
316
+ metadata: {
317
+ acl: acl.serializedACL,
318
+ namespaceId: formattedNs,
319
+ datasourceId: dsId,
320
+ datasourceLabel: label,
321
+ [this.USER_METADATA_KEY]: datasource.metadata ? jsonrepair(JSON.stringify(datasource.metadata)) : undefined,
322
+ },
323
+ };
324
+ });
325
+
326
+ const _vIds = await this.insert(acRequest, namespace, source);
327
+
328
+ const dsData: IStorageVectorDataSource = {
329
+ namespaceId: formattedNs,
330
+ candidateId: acRequest.candidate.id,
331
+ candidateRole: acRequest.candidate.role,
332
+ name: datasource.label || 'Untitled',
333
+ metadata: datasource.metadata ? jsonrepair(JSON.stringify(datasource.metadata)) : undefined,
334
+ text: datasource.text,
335
+ vectorIds: _vIds,
336
+ id: dsId,
337
+ };
338
+
339
+ // Store datasource metadata in memory
340
+ if (!RAMVectorDB.datasources[formattedNs]) {
341
+ RAMVectorDB.datasources[formattedNs] = {};
342
+ }
343
+ RAMVectorDB.datasources[formattedNs][dsId] = dsData;
344
+
345
+ return dsData;
346
+ }
347
+
348
+ @SecureConnector.AccessControl
349
+ protected async deleteDatasource(acRequest: AccessRequest, namespace: string, datasourceId: string): Promise<void> {
350
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
351
+ const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
352
+
353
+ // Delete all vectors belonging to this datasource using a filter by datasourceId,
354
+ // so we remove vectors from any prior insert/update operations sharing the same datasource id
355
+ await this.delete(acRequest, namespace, { datasourceId });
356
+
357
+ // Delete datasource metadata (if present)
358
+ if (RAMVectorDB.datasources[formattedNs] && RAMVectorDB.datasources[formattedNs][datasourceId]) {
359
+ delete RAMVectorDB.datasources[formattedNs][datasourceId];
360
+ }
361
+ }
362
+
363
+ @SecureConnector.AccessControl
364
+ protected async listDatasources(acRequest: AccessRequest, namespace: string): Promise<IStorageVectorDataSource[]> {
365
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
366
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
367
+
368
+ const namespaceDatasources = RAMVectorDB.datasources[preparedNs] || {};
369
+ return Object.values(namespaceDatasources);
370
+ }
371
+
372
+ @SecureConnector.AccessControl
373
+ protected async getDatasource(acRequest: AccessRequest, namespace: string, datasourceId: string): Promise<IStorageVectorDataSource | undefined> {
374
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
375
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
376
+
377
+ const datasource = RAMVectorDB.datasources[preparedNs]?.[datasourceId];
378
+ return datasource; // Return undefined if not found, like MilvusVectorDB
379
+ }
380
+
381
+ /**
382
+ * Calculate cosine similarity between two vectors
383
+ */
384
+ private cosineSimilarity(vecA: number[], vecB: number[]): number {
385
+ if (vecA.length !== vecB.length) {
386
+ throw new Error('Vectors must have the same length');
387
+ }
388
+
389
+ let dotProduct = 0;
390
+ let normA = 0;
391
+ let normB = 0;
392
+
393
+ for (let i = 0; i < vecA.length; i++) {
394
+ dotProduct += vecA[i] * vecB[i];
395
+ normA += vecA[i] * vecA[i];
396
+ normB += vecB[i] * vecB[i];
397
+ }
398
+
399
+ normA = Math.sqrt(normA);
400
+ normB = Math.sqrt(normB);
401
+
402
+ if (normA === 0 || normB === 0) {
403
+ return 0;
404
+ }
405
+
406
+ return dotProduct / (normA * normB);
407
+ }
408
+ }