@smythos/sre 1.5.43 → 1.5.45

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