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