@smythos/sre 1.5.45 → 1.5.50

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