@smythos/sre 1.6.8 → 1.6.10

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 (239) hide show
  1. package/CHANGELOG +111 -111
  2. package/LICENSE +18 -18
  3. package/README.md +135 -135
  4. package/dist/bundle-analysis-lazy.html +4949 -0
  5. package/dist/bundle-analysis.html +4949 -0
  6. package/dist/index.js +2 -2
  7. package/dist/index.js.map +1 -1
  8. package/dist/types/Components/Triggers/Gmail.trigger.d.ts +58 -0
  9. package/dist/types/Components/Triggers/GmailTrigger.class.d.ts +44 -0
  10. package/dist/types/Components/Triggers/Trigger.class.d.ts +21 -0
  11. package/dist/types/Components/Triggers/WhatsApp.trigger.d.ts +22 -0
  12. package/dist/types/helpers/AIPerformanceAnalyzer.helper.d.ts +45 -0
  13. package/dist/types/helpers/AIPerformanceCollector.helper.d.ts +111 -0
  14. package/dist/types/subsystems/IO/Storage.service/connectors/AzureBlobStorage.class.d.ts +211 -0
  15. package/dist/types/subsystems/IO/VectorDB.service/connectors/WeaviateVectorDB.class.d.ts +187 -0
  16. package/dist/types/subsystems/PerformanceManager/Performance.service/PerformanceConnector.d.ts +102 -0
  17. package/dist/types/subsystems/PerformanceManager/Performance.service/connectors/LocalPerformanceConnector.class.d.ts +100 -0
  18. package/dist/types/subsystems/PerformanceManager/Performance.service/index.d.ts +22 -0
  19. package/dist/types/subsystems/Security/Credentials/Credentials.class.d.ts +2 -0
  20. package/dist/types/subsystems/Security/Credentials/ManagedOAuth2Credentials.class.d.ts +18 -0
  21. package/dist/types/subsystems/Security/Credentials/OAuth2Credentials.class.d.ts +14 -0
  22. package/dist/types/types/Performance.types.d.ts +468 -0
  23. package/dist/types/utils/package-manager.utils.d.ts +26 -0
  24. package/package.json +1 -1
  25. package/src/Components/APICall/APICall.class.ts +161 -161
  26. package/src/Components/APICall/AccessTokenManager.ts +166 -166
  27. package/src/Components/APICall/ArrayBufferResponse.helper.ts +58 -58
  28. package/src/Components/APICall/OAuth.helper.ts +447 -447
  29. package/src/Components/APICall/mimeTypeCategories.ts +46 -46
  30. package/src/Components/APICall/parseData.ts +167 -167
  31. package/src/Components/APICall/parseHeaders.ts +41 -41
  32. package/src/Components/APICall/parseProxy.ts +68 -68
  33. package/src/Components/APICall/parseUrl.ts +91 -91
  34. package/src/Components/APIEndpoint.class.ts +234 -234
  35. package/src/Components/APIOutput.class.ts +58 -58
  36. package/src/Components/AgentPlugin.class.ts +102 -102
  37. package/src/Components/Async.class.ts +155 -155
  38. package/src/Components/Await.class.ts +90 -90
  39. package/src/Components/Classifier.class.ts +158 -158
  40. package/src/Components/Component.class.ts +147 -147
  41. package/src/Components/ComponentHost.class.ts +38 -38
  42. package/src/Components/DataSourceCleaner.class.ts +92 -92
  43. package/src/Components/DataSourceIndexer.class.ts +181 -181
  44. package/src/Components/DataSourceLookup.class.ts +161 -161
  45. package/src/Components/ECMASandbox.class.ts +72 -72
  46. package/src/Components/FEncDec.class.ts +29 -29
  47. package/src/Components/FHash.class.ts +33 -33
  48. package/src/Components/FSign.class.ts +80 -80
  49. package/src/Components/FSleep.class.ts +25 -25
  50. package/src/Components/FTimestamp.class.ts +66 -66
  51. package/src/Components/FileStore.class.ts +78 -78
  52. package/src/Components/ForEach.class.ts +97 -97
  53. package/src/Components/GPTPlugin.class.ts +70 -70
  54. package/src/Components/GenAILLM.class.ts +586 -586
  55. package/src/Components/HuggingFace.class.ts +313 -313
  56. package/src/Components/Image/imageSettings.config.ts +70 -70
  57. package/src/Components/ImageGenerator.class.ts +483 -483
  58. package/src/Components/JSONFilter.class.ts +54 -54
  59. package/src/Components/LLMAssistant.class.ts +213 -213
  60. package/src/Components/LogicAND.class.ts +28 -28
  61. package/src/Components/LogicAtLeast.class.ts +85 -85
  62. package/src/Components/LogicAtMost.class.ts +86 -86
  63. package/src/Components/LogicOR.class.ts +29 -29
  64. package/src/Components/LogicXOR.class.ts +34 -34
  65. package/src/Components/MCPClient.class.ts +137 -137
  66. package/src/Components/MemoryDeleteKeyVal.class.ts +70 -70
  67. package/src/Components/MemoryReadKeyVal.class.ts +67 -67
  68. package/src/Components/MemoryWriteKeyVal.class.ts +62 -62
  69. package/src/Components/MemoryWriteObject.class.ts +97 -97
  70. package/src/Components/MultimodalLLM.class.ts +128 -128
  71. package/src/Components/OpenAPI.class.ts +72 -72
  72. package/src/Components/PromptGenerator.class.ts +122 -122
  73. package/src/Components/ScrapflyWebScrape.class.ts +183 -183
  74. package/src/Components/ServerlessCode.class.ts +123 -123
  75. package/src/Components/TavilyWebSearch.class.ts +103 -103
  76. package/src/Components/VisionLLM.class.ts +104 -104
  77. package/src/Components/ZapierAction.class.ts +127 -127
  78. package/src/Components/index.ts +97 -97
  79. package/src/Core/AgentProcess.helper.ts +240 -240
  80. package/src/Core/Connector.class.ts +123 -123
  81. package/src/Core/ConnectorsService.ts +197 -197
  82. package/src/Core/DummyConnector.ts +49 -49
  83. package/src/Core/HookService.ts +105 -105
  84. package/src/Core/SmythRuntime.class.ts +241 -241
  85. package/src/Core/SystemEvents.ts +16 -16
  86. package/src/Core/boot.ts +56 -56
  87. package/src/config.ts +15 -15
  88. package/src/constants.ts +126 -126
  89. package/src/data/hugging-face.params.json +579 -579
  90. package/src/helpers/AWSLambdaCode.helper.ts +624 -624
  91. package/src/helpers/BinaryInput.helper.ts +331 -331
  92. package/src/helpers/Conversation.helper.ts +1157 -1157
  93. package/src/helpers/ECMASandbox.helper.ts +64 -64
  94. package/src/helpers/JsonContent.helper.ts +97 -97
  95. package/src/helpers/LocalCache.helper.ts +97 -97
  96. package/src/helpers/Log.helper.ts +274 -274
  97. package/src/helpers/OpenApiParser.helper.ts +150 -150
  98. package/src/helpers/S3Cache.helper.ts +147 -147
  99. package/src/helpers/SmythURI.helper.ts +5 -5
  100. package/src/helpers/Sysconfig.helper.ts +95 -95
  101. package/src/helpers/TemplateString.helper.ts +243 -243
  102. package/src/helpers/TypeChecker.helper.ts +329 -329
  103. package/src/index.ts +3 -3
  104. package/src/index.ts.bak +3 -3
  105. package/src/subsystems/AgentManager/Agent.class.ts +1114 -1114
  106. package/src/subsystems/AgentManager/Agent.helper.ts +3 -3
  107. package/src/subsystems/AgentManager/AgentData.service/AgentDataConnector.ts +230 -230
  108. package/src/subsystems/AgentManager/AgentData.service/connectors/CLIAgentDataConnector.class.ts +66 -66
  109. package/src/subsystems/AgentManager/AgentData.service/connectors/LocalAgentDataConnector.class.ts +145 -145
  110. package/src/subsystems/AgentManager/AgentData.service/connectors/NullAgentData.class.ts +39 -39
  111. package/src/subsystems/AgentManager/AgentData.service/index.ts +18 -18
  112. package/src/subsystems/AgentManager/AgentLogger.class.ts +301 -301
  113. package/src/subsystems/AgentManager/AgentRequest.class.ts +51 -51
  114. package/src/subsystems/AgentManager/AgentRuntime.class.ts +557 -557
  115. package/src/subsystems/AgentManager/AgentSSE.class.ts +101 -101
  116. package/src/subsystems/AgentManager/AgentSettings.class.ts +52 -52
  117. package/src/subsystems/AgentManager/Component.service/ComponentConnector.ts +32 -32
  118. package/src/subsystems/AgentManager/Component.service/connectors/LocalComponentConnector.class.ts +60 -60
  119. package/src/subsystems/AgentManager/Component.service/index.ts +11 -11
  120. package/src/subsystems/AgentManager/EmbodimentSettings.class.ts +47 -47
  121. package/src/subsystems/AgentManager/ForkedAgent.class.ts +154 -154
  122. package/src/subsystems/AgentManager/OSResourceMonitor.ts +77 -77
  123. package/src/subsystems/ComputeManager/Code.service/CodeConnector.ts +98 -98
  124. package/src/subsystems/ComputeManager/Code.service/connectors/AWSLambdaCode.class.ts +171 -171
  125. package/src/subsystems/ComputeManager/Code.service/connectors/ECMASandbox.class.ts +131 -131
  126. package/src/subsystems/ComputeManager/Code.service/index.ts +13 -13
  127. package/src/subsystems/IO/CLI.service/CLIConnector.ts +47 -47
  128. package/src/subsystems/IO/CLI.service/index.ts +9 -9
  129. package/src/subsystems/IO/Log.service/LogConnector.ts +32 -32
  130. package/src/subsystems/IO/Log.service/connectors/ConsoleLog.class.ts +28 -28
  131. package/src/subsystems/IO/Log.service/index.ts +13 -13
  132. package/src/subsystems/IO/NKV.service/NKVConnector.ts +43 -43
  133. package/src/subsystems/IO/NKV.service/connectors/NKVLocalStorage.class.ts +234 -234
  134. package/src/subsystems/IO/NKV.service/connectors/NKVRAM.class.ts +204 -204
  135. package/src/subsystems/IO/NKV.service/connectors/NKVRedis.class.ts +182 -182
  136. package/src/subsystems/IO/NKV.service/index.ts +14 -14
  137. package/src/subsystems/IO/Router.service/RouterConnector.ts +21 -21
  138. package/src/subsystems/IO/Router.service/connectors/ExpressRouter.class.ts +48 -48
  139. package/src/subsystems/IO/Router.service/connectors/NullRouter.class.ts +40 -40
  140. package/src/subsystems/IO/Router.service/index.ts +11 -11
  141. package/src/subsystems/IO/Storage.service/SmythFS.class.ts +488 -488
  142. package/src/subsystems/IO/Storage.service/StorageConnector.ts +66 -66
  143. package/src/subsystems/IO/Storage.service/connectors/LocalStorage.class.ts +327 -327
  144. package/src/subsystems/IO/Storage.service/connectors/S3Storage.class.ts +482 -482
  145. package/src/subsystems/IO/Storage.service/index.ts +13 -13
  146. package/src/subsystems/IO/VectorDB.service/VectorDBConnector.ts +108 -108
  147. package/src/subsystems/IO/VectorDB.service/connectors/MilvusVectorDB.class.ts +465 -465
  148. package/src/subsystems/IO/VectorDB.service/connectors/PineconeVectorDB.class.ts +387 -387
  149. package/src/subsystems/IO/VectorDB.service/connectors/RAMVecrtorDB.class.ts +408 -408
  150. package/src/subsystems/IO/VectorDB.service/embed/BaseEmbedding.ts +107 -107
  151. package/src/subsystems/IO/VectorDB.service/embed/GoogleEmbedding.ts +118 -118
  152. package/src/subsystems/IO/VectorDB.service/embed/OpenAIEmbedding.ts +109 -109
  153. package/src/subsystems/IO/VectorDB.service/embed/index.ts +26 -26
  154. package/src/subsystems/IO/VectorDB.service/index.ts +14 -14
  155. package/src/subsystems/LLMManager/LLM.helper.ts +251 -251
  156. package/src/subsystems/LLMManager/LLM.inference.ts +345 -345
  157. package/src/subsystems/LLMManager/LLM.service/LLMConnector.ts +492 -492
  158. package/src/subsystems/LLMManager/LLM.service/LLMCredentials.helper.ts +171 -171
  159. package/src/subsystems/LLMManager/LLM.service/connectors/Anthropic.class.ts +666 -666
  160. package/src/subsystems/LLMManager/LLM.service/connectors/Bedrock.class.ts +407 -407
  161. package/src/subsystems/LLMManager/LLM.service/connectors/Echo.class.ts +92 -92
  162. package/src/subsystems/LLMManager/LLM.service/connectors/GoogleAI.class.ts +983 -983
  163. package/src/subsystems/LLMManager/LLM.service/connectors/Groq.class.ts +319 -319
  164. package/src/subsystems/LLMManager/LLM.service/connectors/Ollama.class.ts +361 -361
  165. package/src/subsystems/LLMManager/LLM.service/connectors/Perplexity.class.ts +257 -257
  166. package/src/subsystems/LLMManager/LLM.service/connectors/VertexAI.class.ts +430 -430
  167. package/src/subsystems/LLMManager/LLM.service/connectors/openai/OpenAIConnector.class.ts +503 -503
  168. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ChatCompletionsApiInterface.ts +524 -524
  169. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterface.ts +100 -100
  170. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterfaceFactory.ts +81 -81
  171. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ResponsesApiInterface.ts +1145 -1145
  172. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/constants.ts +13 -13
  173. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/index.ts +4 -4
  174. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/utils.ts +11 -11
  175. package/src/subsystems/LLMManager/LLM.service/connectors/openai/types.ts +32 -32
  176. package/src/subsystems/LLMManager/LLM.service/connectors/xAI.class.ts +478 -478
  177. package/src/subsystems/LLMManager/LLM.service/index.ts +47 -47
  178. package/src/subsystems/LLMManager/ModelsProvider.service/ModelsProviderConnector.ts +303 -303
  179. package/src/subsystems/LLMManager/ModelsProvider.service/connectors/JSONModelsProvider.class.ts +280 -271
  180. package/src/subsystems/LLMManager/ModelsProvider.service/index.ts +11 -11
  181. package/src/subsystems/LLMManager/custom-models.ts +854 -854
  182. package/src/subsystems/LLMManager/models.ts +2540 -2540
  183. package/src/subsystems/LLMManager/paramMappings.ts +69 -69
  184. package/src/subsystems/MemoryManager/Cache.service/CacheConnector.ts +86 -86
  185. package/src/subsystems/MemoryManager/Cache.service/connectors/LocalStorageCache.class.ts +297 -297
  186. package/src/subsystems/MemoryManager/Cache.service/connectors/RAMCache.class.ts +214 -214
  187. package/src/subsystems/MemoryManager/Cache.service/connectors/RedisCache.class.ts +252 -252
  188. package/src/subsystems/MemoryManager/Cache.service/connectors/S3Cache.class.ts +373 -373
  189. package/src/subsystems/MemoryManager/Cache.service/index.ts +15 -15
  190. package/src/subsystems/MemoryManager/LLMCache.ts +72 -72
  191. package/src/subsystems/MemoryManager/LLMContext.ts +124 -124
  192. package/src/subsystems/MemoryManager/LLMMemory.service/LLMMemoryConnector.ts +26 -26
  193. package/src/subsystems/MemoryManager/RuntimeContext.ts +277 -277
  194. package/src/subsystems/Security/AccessControl/ACL.class.ts +208 -208
  195. package/src/subsystems/Security/AccessControl/AccessCandidate.class.ts +82 -82
  196. package/src/subsystems/Security/AccessControl/AccessRequest.class.ts +52 -52
  197. package/src/subsystems/Security/Account.service/AccountConnector.ts +44 -44
  198. package/src/subsystems/Security/Account.service/connectors/DummyAccount.class.ts +130 -130
  199. package/src/subsystems/Security/Account.service/connectors/JSONFileAccount.class.ts +170 -170
  200. package/src/subsystems/Security/Account.service/connectors/MySQLAccount.class.ts +76 -76
  201. package/src/subsystems/Security/Account.service/index.ts +14 -14
  202. package/src/subsystems/Security/Credentials.helper.ts +62 -62
  203. package/src/subsystems/Security/ManagedVault.service/ManagedVaultConnector.ts +38 -38
  204. package/src/subsystems/Security/ManagedVault.service/connectors/NullManagedVault.class.ts +53 -53
  205. package/src/subsystems/Security/ManagedVault.service/connectors/SecretManagerManagedVault.ts +154 -154
  206. package/src/subsystems/Security/ManagedVault.service/index.ts +12 -12
  207. package/src/subsystems/Security/SecureConnector.class.ts +110 -110
  208. package/src/subsystems/Security/Vault.service/Vault.helper.ts +30 -30
  209. package/src/subsystems/Security/Vault.service/VaultConnector.ts +29 -29
  210. package/src/subsystems/Security/Vault.service/connectors/HashicorpVault.class.ts +46 -46
  211. package/src/subsystems/Security/Vault.service/connectors/JSONFileVault.class.ts +221 -221
  212. package/src/subsystems/Security/Vault.service/connectors/NullVault.class.ts +54 -54
  213. package/src/subsystems/Security/Vault.service/connectors/SecretsManager.class.ts +140 -140
  214. package/src/subsystems/Security/Vault.service/index.ts +12 -12
  215. package/src/types/ACL.types.ts +104 -104
  216. package/src/types/AWS.types.ts +10 -10
  217. package/src/types/Agent.types.ts +61 -61
  218. package/src/types/AgentLogger.types.ts +17 -17
  219. package/src/types/Cache.types.ts +1 -1
  220. package/src/types/Common.types.ts +2 -2
  221. package/src/types/LLM.types.ts +520 -520
  222. package/src/types/Redis.types.ts +8 -8
  223. package/src/types/SRE.types.ts +64 -64
  224. package/src/types/Security.types.ts +14 -14
  225. package/src/types/Storage.types.ts +5 -5
  226. package/src/types/VectorDB.types.ts +86 -86
  227. package/src/utils/base64.utils.ts +275 -275
  228. package/src/utils/cli.utils.ts +68 -68
  229. package/src/utils/data.utils.ts +322 -322
  230. package/src/utils/date-time.utils.ts +22 -22
  231. package/src/utils/general.utils.ts +238 -238
  232. package/src/utils/index.ts +12 -12
  233. package/src/utils/lazy-client.ts +261 -261
  234. package/src/utils/numbers.utils.ts +13 -13
  235. package/src/utils/oauth.utils.ts +35 -35
  236. package/src/utils/string.utils.ts +414 -414
  237. package/src/utils/url.utils.ts +19 -19
  238. package/src/utils/validation.utils.ts +74 -74
  239. package/dist/types/subsystems/LLMManager/ModelsProvider.service/connectors/SmythModelsProvider.class.d.ts +0 -39
@@ -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
+ }