@smythos/sre 1.6.0 → 1.6.8

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (237) hide show
  1. package/CHANGELOG +111 -111
  2. package/LICENSE +18 -18
  3. package/README.md +135 -135
  4. package/dist/index.js +43 -43
  5. package/dist/index.js.map +1 -1
  6. package/dist/types/index.d.ts +1 -0
  7. package/dist/types/subsystems/LLMManager/LLM.service/connectors/Ollama.class.d.ts +45 -0
  8. package/dist/types/subsystems/LLMManager/ModelsProvider.service/connectors/SmythModelsProvider.class.d.ts +39 -0
  9. package/dist/types/types/LLM.types.d.ts +2 -0
  10. package/package.json +2 -1
  11. package/src/Components/APICall/APICall.class.ts +161 -161
  12. package/src/Components/APICall/AccessTokenManager.ts +166 -166
  13. package/src/Components/APICall/ArrayBufferResponse.helper.ts +58 -58
  14. package/src/Components/APICall/OAuth.helper.ts +447 -447
  15. package/src/Components/APICall/mimeTypeCategories.ts +46 -46
  16. package/src/Components/APICall/parseData.ts +167 -167
  17. package/src/Components/APICall/parseHeaders.ts +41 -41
  18. package/src/Components/APICall/parseProxy.ts +68 -68
  19. package/src/Components/APICall/parseUrl.ts +91 -91
  20. package/src/Components/APIEndpoint.class.ts +234 -234
  21. package/src/Components/APIOutput.class.ts +58 -58
  22. package/src/Components/AgentPlugin.class.ts +102 -102
  23. package/src/Components/Async.class.ts +155 -155
  24. package/src/Components/Await.class.ts +90 -90
  25. package/src/Components/Classifier.class.ts +158 -158
  26. package/src/Components/Component.class.ts +147 -147
  27. package/src/Components/ComponentHost.class.ts +38 -38
  28. package/src/Components/DataSourceCleaner.class.ts +92 -92
  29. package/src/Components/DataSourceIndexer.class.ts +181 -181
  30. package/src/Components/DataSourceLookup.class.ts +161 -161
  31. package/src/Components/ECMASandbox.class.ts +72 -72
  32. package/src/Components/FEncDec.class.ts +29 -29
  33. package/src/Components/FHash.class.ts +33 -33
  34. package/src/Components/FSign.class.ts +80 -80
  35. package/src/Components/FSleep.class.ts +25 -25
  36. package/src/Components/FTimestamp.class.ts +66 -66
  37. package/src/Components/FileStore.class.ts +78 -78
  38. package/src/Components/ForEach.class.ts +97 -97
  39. package/src/Components/GPTPlugin.class.ts +70 -70
  40. package/src/Components/GenAILLM.class.ts +586 -586
  41. package/src/Components/HuggingFace.class.ts +314 -314
  42. package/src/Components/Image/imageSettings.config.ts +70 -70
  43. package/src/Components/ImageGenerator.class.ts +483 -483
  44. package/src/Components/JSONFilter.class.ts +54 -54
  45. package/src/Components/LLMAssistant.class.ts +213 -213
  46. package/src/Components/LogicAND.class.ts +28 -28
  47. package/src/Components/LogicAtLeast.class.ts +85 -85
  48. package/src/Components/LogicAtMost.class.ts +86 -86
  49. package/src/Components/LogicOR.class.ts +29 -29
  50. package/src/Components/LogicXOR.class.ts +34 -34
  51. package/src/Components/MCPClient.class.ts +137 -137
  52. package/src/Components/MemoryDeleteKeyVal.class.ts +70 -70
  53. package/src/Components/MemoryReadKeyVal.class.ts +67 -67
  54. package/src/Components/MemoryWriteKeyVal.class.ts +62 -62
  55. package/src/Components/MemoryWriteObject.class.ts +97 -97
  56. package/src/Components/MultimodalLLM.class.ts +128 -128
  57. package/src/Components/OpenAPI.class.ts +72 -72
  58. package/src/Components/PromptGenerator.class.ts +122 -122
  59. package/src/Components/ScrapflyWebScrape.class.ts +183 -183
  60. package/src/Components/ServerlessCode.class.ts +123 -123
  61. package/src/Components/TavilyWebSearch.class.ts +103 -103
  62. package/src/Components/VisionLLM.class.ts +104 -104
  63. package/src/Components/ZapierAction.class.ts +127 -127
  64. package/src/Components/index.ts +97 -97
  65. package/src/Core/AgentProcess.helper.ts +240 -240
  66. package/src/Core/Connector.class.ts +123 -123
  67. package/src/Core/ConnectorsService.ts +197 -197
  68. package/src/Core/DummyConnector.ts +49 -49
  69. package/src/Core/HookService.ts +105 -105
  70. package/src/Core/SmythRuntime.class.ts +241 -241
  71. package/src/Core/SystemEvents.ts +16 -16
  72. package/src/Core/boot.ts +56 -56
  73. package/src/config.ts +15 -15
  74. package/src/constants.ts +126 -126
  75. package/src/data/hugging-face.params.json +579 -579
  76. package/src/helpers/AWSLambdaCode.helper.ts +624 -599
  77. package/src/helpers/BinaryInput.helper.ts +331 -331
  78. package/src/helpers/Conversation.helper.ts +1157 -1157
  79. package/src/helpers/ECMASandbox.helper.ts +64 -64
  80. package/src/helpers/JsonContent.helper.ts +97 -97
  81. package/src/helpers/LocalCache.helper.ts +97 -97
  82. package/src/helpers/Log.helper.ts +274 -274
  83. package/src/helpers/OpenApiParser.helper.ts +150 -150
  84. package/src/helpers/S3Cache.helper.ts +147 -147
  85. package/src/helpers/SmythURI.helper.ts +5 -5
  86. package/src/helpers/Sysconfig.helper.ts +95 -95
  87. package/src/helpers/TemplateString.helper.ts +243 -243
  88. package/src/helpers/TypeChecker.helper.ts +329 -329
  89. package/src/index.ts +4 -3
  90. package/src/index.ts.bak +4 -3
  91. package/src/subsystems/AgentManager/Agent.class.ts +1114 -1114
  92. package/src/subsystems/AgentManager/Agent.helper.ts +3 -3
  93. package/src/subsystems/AgentManager/AgentData.service/AgentDataConnector.ts +230 -230
  94. package/src/subsystems/AgentManager/AgentData.service/connectors/CLIAgentDataConnector.class.ts +66 -66
  95. package/src/subsystems/AgentManager/AgentData.service/connectors/LocalAgentDataConnector.class.ts +145 -145
  96. package/src/subsystems/AgentManager/AgentData.service/connectors/NullAgentData.class.ts +39 -39
  97. package/src/subsystems/AgentManager/AgentData.service/index.ts +18 -18
  98. package/src/subsystems/AgentManager/AgentLogger.class.ts +301 -301
  99. package/src/subsystems/AgentManager/AgentRequest.class.ts +51 -51
  100. package/src/subsystems/AgentManager/AgentRuntime.class.ts +557 -557
  101. package/src/subsystems/AgentManager/AgentSSE.class.ts +101 -101
  102. package/src/subsystems/AgentManager/AgentSettings.class.ts +52 -52
  103. package/src/subsystems/AgentManager/Component.service/ComponentConnector.ts +32 -32
  104. package/src/subsystems/AgentManager/Component.service/connectors/LocalComponentConnector.class.ts +60 -60
  105. package/src/subsystems/AgentManager/Component.service/index.ts +11 -11
  106. package/src/subsystems/AgentManager/EmbodimentSettings.class.ts +47 -47
  107. package/src/subsystems/AgentManager/ForkedAgent.class.ts +154 -154
  108. package/src/subsystems/AgentManager/OSResourceMonitor.ts +77 -77
  109. package/src/subsystems/ComputeManager/Code.service/CodeConnector.ts +98 -98
  110. package/src/subsystems/ComputeManager/Code.service/connectors/AWSLambdaCode.class.ts +171 -172
  111. package/src/subsystems/ComputeManager/Code.service/connectors/ECMASandbox.class.ts +131 -131
  112. package/src/subsystems/ComputeManager/Code.service/index.ts +13 -13
  113. package/src/subsystems/IO/CLI.service/CLIConnector.ts +47 -47
  114. package/src/subsystems/IO/CLI.service/index.ts +9 -9
  115. package/src/subsystems/IO/Log.service/LogConnector.ts +32 -32
  116. package/src/subsystems/IO/Log.service/connectors/ConsoleLog.class.ts +28 -28
  117. package/src/subsystems/IO/Log.service/index.ts +13 -13
  118. package/src/subsystems/IO/NKV.service/NKVConnector.ts +43 -43
  119. package/src/subsystems/IO/NKV.service/connectors/NKVLocalStorage.class.ts +234 -234
  120. package/src/subsystems/IO/NKV.service/connectors/NKVRAM.class.ts +204 -204
  121. package/src/subsystems/IO/NKV.service/connectors/NKVRedis.class.ts +182 -182
  122. package/src/subsystems/IO/NKV.service/index.ts +14 -14
  123. package/src/subsystems/IO/Router.service/RouterConnector.ts +21 -21
  124. package/src/subsystems/IO/Router.service/connectors/ExpressRouter.class.ts +48 -48
  125. package/src/subsystems/IO/Router.service/connectors/NullRouter.class.ts +40 -40
  126. package/src/subsystems/IO/Router.service/index.ts +11 -11
  127. package/src/subsystems/IO/Storage.service/SmythFS.class.ts +488 -488
  128. package/src/subsystems/IO/Storage.service/StorageConnector.ts +66 -66
  129. package/src/subsystems/IO/Storage.service/connectors/LocalStorage.class.ts +327 -327
  130. package/src/subsystems/IO/Storage.service/connectors/S3Storage.class.ts +482 -482
  131. package/src/subsystems/IO/Storage.service/index.ts +13 -13
  132. package/src/subsystems/IO/VectorDB.service/VectorDBConnector.ts +108 -108
  133. package/src/subsystems/IO/VectorDB.service/connectors/MilvusVectorDB.class.ts +465 -465
  134. package/src/subsystems/IO/VectorDB.service/connectors/PineconeVectorDB.class.ts +387 -387
  135. package/src/subsystems/IO/VectorDB.service/connectors/RAMVecrtorDB.class.ts +408 -408
  136. package/src/subsystems/IO/VectorDB.service/embed/BaseEmbedding.ts +107 -107
  137. package/src/subsystems/IO/VectorDB.service/embed/GoogleEmbedding.ts +118 -118
  138. package/src/subsystems/IO/VectorDB.service/embed/OpenAIEmbedding.ts +109 -109
  139. package/src/subsystems/IO/VectorDB.service/embed/index.ts +26 -26
  140. package/src/subsystems/IO/VectorDB.service/index.ts +14 -14
  141. package/src/subsystems/LLMManager/LLM.helper.ts +251 -251
  142. package/src/subsystems/LLMManager/LLM.inference.ts +345 -345
  143. package/src/subsystems/LLMManager/LLM.service/LLMConnector.ts +492 -492
  144. package/src/subsystems/LLMManager/LLM.service/LLMCredentials.helper.ts +171 -171
  145. package/src/subsystems/LLMManager/LLM.service/connectors/Anthropic.class.ts +666 -666
  146. package/src/subsystems/LLMManager/LLM.service/connectors/Bedrock.class.ts +407 -407
  147. package/src/subsystems/LLMManager/LLM.service/connectors/Echo.class.ts +92 -92
  148. package/src/subsystems/LLMManager/LLM.service/connectors/GoogleAI.class.ts +983 -983
  149. package/src/subsystems/LLMManager/LLM.service/connectors/Groq.class.ts +319 -319
  150. package/src/subsystems/LLMManager/LLM.service/connectors/Ollama.class.ts +362 -0
  151. package/src/subsystems/LLMManager/LLM.service/connectors/Perplexity.class.ts +257 -257
  152. package/src/subsystems/LLMManager/LLM.service/connectors/VertexAI.class.ts +430 -430
  153. package/src/subsystems/LLMManager/LLM.service/connectors/openai/OpenAIConnector.class.ts +503 -503
  154. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ChatCompletionsApiInterface.ts +524 -524
  155. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterface.ts +100 -100
  156. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/OpenAIApiInterfaceFactory.ts +81 -81
  157. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/ResponsesApiInterface.ts +1145 -1145
  158. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/constants.ts +13 -13
  159. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/index.ts +4 -4
  160. package/src/subsystems/LLMManager/LLM.service/connectors/openai/apiInterfaces/utils.ts +11 -11
  161. package/src/subsystems/LLMManager/LLM.service/connectors/openai/types.ts +32 -32
  162. package/src/subsystems/LLMManager/LLM.service/connectors/xAI.class.ts +478 -478
  163. package/src/subsystems/LLMManager/LLM.service/index.ts +47 -44
  164. package/src/subsystems/LLMManager/ModelsProvider.service/ModelsProviderConnector.ts +303 -303
  165. package/src/subsystems/LLMManager/ModelsProvider.service/connectors/JSONModelsProvider.class.ts +271 -271
  166. package/src/subsystems/LLMManager/ModelsProvider.service/index.ts +11 -11
  167. package/src/subsystems/LLMManager/custom-models.ts +854 -854
  168. package/src/subsystems/LLMManager/models.ts +2540 -2540
  169. package/src/subsystems/LLMManager/paramMappings.ts +69 -69
  170. package/src/subsystems/MemoryManager/Cache.service/CacheConnector.ts +86 -86
  171. package/src/subsystems/MemoryManager/Cache.service/connectors/LocalStorageCache.class.ts +297 -297
  172. package/src/subsystems/MemoryManager/Cache.service/connectors/RAMCache.class.ts +214 -214
  173. package/src/subsystems/MemoryManager/Cache.service/connectors/RedisCache.class.ts +252 -252
  174. package/src/subsystems/MemoryManager/Cache.service/connectors/S3Cache.class.ts +373 -373
  175. package/src/subsystems/MemoryManager/Cache.service/index.ts +15 -15
  176. package/src/subsystems/MemoryManager/LLMCache.ts +72 -72
  177. package/src/subsystems/MemoryManager/LLMContext.ts +124 -124
  178. package/src/subsystems/MemoryManager/LLMMemory.service/LLMMemoryConnector.ts +26 -26
  179. package/src/subsystems/MemoryManager/RuntimeContext.ts +277 -277
  180. package/src/subsystems/Security/AccessControl/ACL.class.ts +208 -208
  181. package/src/subsystems/Security/AccessControl/AccessCandidate.class.ts +82 -82
  182. package/src/subsystems/Security/AccessControl/AccessRequest.class.ts +52 -52
  183. package/src/subsystems/Security/Account.service/AccountConnector.ts +44 -44
  184. package/src/subsystems/Security/Account.service/connectors/DummyAccount.class.ts +130 -130
  185. package/src/subsystems/Security/Account.service/connectors/JSONFileAccount.class.ts +170 -170
  186. package/src/subsystems/Security/Account.service/connectors/MySQLAccount.class.ts +76 -76
  187. package/src/subsystems/Security/Account.service/index.ts +14 -14
  188. package/src/subsystems/Security/Credentials.helper.ts +62 -62
  189. package/src/subsystems/Security/ManagedVault.service/ManagedVaultConnector.ts +38 -38
  190. package/src/subsystems/Security/ManagedVault.service/connectors/NullManagedVault.class.ts +53 -53
  191. package/src/subsystems/Security/ManagedVault.service/connectors/SecretManagerManagedVault.ts +154 -154
  192. package/src/subsystems/Security/ManagedVault.service/index.ts +12 -12
  193. package/src/subsystems/Security/SecureConnector.class.ts +110 -110
  194. package/src/subsystems/Security/Vault.service/Vault.helper.ts +30 -30
  195. package/src/subsystems/Security/Vault.service/VaultConnector.ts +29 -29
  196. package/src/subsystems/Security/Vault.service/connectors/HashicorpVault.class.ts +46 -46
  197. package/src/subsystems/Security/Vault.service/connectors/JSONFileVault.class.ts +221 -221
  198. package/src/subsystems/Security/Vault.service/connectors/NullVault.class.ts +54 -54
  199. package/src/subsystems/Security/Vault.service/connectors/SecretsManager.class.ts +140 -140
  200. package/src/subsystems/Security/Vault.service/index.ts +12 -12
  201. package/src/types/ACL.types.ts +104 -104
  202. package/src/types/AWS.types.ts +10 -10
  203. package/src/types/Agent.types.ts +61 -61
  204. package/src/types/AgentLogger.types.ts +17 -17
  205. package/src/types/Cache.types.ts +1 -1
  206. package/src/types/Common.types.ts +2 -2
  207. package/src/types/LLM.types.ts +520 -519
  208. package/src/types/Redis.types.ts +8 -8
  209. package/src/types/SRE.types.ts +64 -64
  210. package/src/types/Security.types.ts +14 -14
  211. package/src/types/Storage.types.ts +5 -5
  212. package/src/types/VectorDB.types.ts +86 -86
  213. package/src/utils/base64.utils.ts +275 -275
  214. package/src/utils/cli.utils.ts +68 -68
  215. package/src/utils/data.utils.ts +322 -322
  216. package/src/utils/date-time.utils.ts +22 -22
  217. package/src/utils/general.utils.ts +238 -238
  218. package/src/utils/index.ts +12 -12
  219. package/src/utils/lazy-client.ts +261 -261
  220. package/src/utils/numbers.utils.ts +13 -13
  221. package/src/utils/oauth.utils.ts +35 -35
  222. package/src/utils/string.utils.ts +414 -414
  223. package/src/utils/url.utils.ts +19 -19
  224. package/src/utils/validation.utils.ts +74 -74
  225. package/dist/bundle-analysis-lazy.html +0 -4949
  226. package/dist/bundle-analysis.html +0 -4949
  227. package/dist/types/Components/Triggers/GmailTrigger.class.d.ts +0 -13
  228. package/dist/types/Components/Triggers/Trigger.class.d.ts +0 -3
  229. package/dist/types/helpers/AIPerformanceAnalyzer.helper.d.ts +0 -45
  230. package/dist/types/helpers/AIPerformanceCollector.helper.d.ts +0 -111
  231. package/dist/types/subsystems/IO/Storage.service/connectors/AzureBlobStorage.class.d.ts +0 -211
  232. package/dist/types/subsystems/IO/VectorDB.service/connectors/WeaviateVectorDB.class.d.ts +0 -187
  233. package/dist/types/subsystems/PerformanceManager/Performance.service/PerformanceConnector.d.ts +0 -102
  234. package/dist/types/subsystems/PerformanceManager/Performance.service/connectors/LocalPerformanceConnector.class.d.ts +0 -100
  235. package/dist/types/subsystems/PerformanceManager/Performance.service/index.d.ts +0 -22
  236. package/dist/types/types/Performance.types.d.ts +0 -468
  237. package/dist/types/utils/package-manager.utils.d.ts +0 -26
@@ -1,387 +1,387 @@
1
- //==[ SRE: S3Storage ]======================
2
- import { ACL } from '@sre/Security/AccessControl/ACL.class';
3
- import { IAccessCandidate, IACL, TAccessLevel, TAccessRole } from '@sre/types/ACL.types';
4
- import { AccessRequest } from '@sre/Security/AccessControl/AccessRequest.class';
5
- import { AccessCandidate } from '@sre/Security/AccessControl/AccessCandidate.class';
6
- import { SecureConnector } from '@sre/Security/SecureConnector.class';
7
- import { VectorDBConnector, DeleteTarget } from '../VectorDBConnector';
8
- import {
9
- DatasourceDto,
10
- IStorageVectorDataSource,
11
- IStorageVectorNamespace,
12
- IVectorDataSourceDto,
13
- QueryOptions,
14
- VectorsResultData,
15
- } from '@sre/types/VectorDB.types';
16
- import { Pinecone } from '@pinecone-database/pinecone';
17
- import { ConnectorService } from '@sre/Core/ConnectorsService';
18
- import { Logger } from '@sre/helpers/Log.helper';
19
- import { NKVConnector } from '@sre/IO/NKV.service/NKVConnector';
20
- import { AccountConnector } from '@sre/Security/Account.service/AccountConnector';
21
- import { JSONContentHelper } from '@sre/helpers/JsonContent.helper';
22
- import { CacheConnector } from '@sre/MemoryManager/Cache.service/CacheConnector';
23
- import crypto from 'crypto';
24
- import { BaseEmbedding, TEmbeddings } from '../embed/BaseEmbedding';
25
- import { EmbeddingsFactory, SupportedProviders, SupportedModels } from '../embed';
26
- import { chunkText } from '@sre/utils/string.utils';
27
- import { jsonrepair } from 'jsonrepair';
28
-
29
- const console = Logger('Pinecone VectorDB');
30
-
31
- export type PineconeConfig = {
32
- /**
33
- * The Pinecone API key
34
- */
35
- apiKey: string;
36
- /**
37
- * The Pinecone index name
38
- */
39
- indexName: string;
40
- /**
41
- * The embeddings model to use
42
- */
43
- embeddings?: TEmbeddings;
44
- };
45
- export class PineconeVectorDB extends VectorDBConnector {
46
- public name = 'PineconeVectorDB';
47
- public id = 'pinecone';
48
- private client: Pinecone;
49
- private indexName: string;
50
- private cache: CacheConnector;
51
- private accountConnector: AccountConnector;
52
- private nkvConnector: NKVConnector;
53
- public embedder: BaseEmbedding;
54
-
55
- constructor(protected _settings: PineconeConfig) {
56
- super(_settings);
57
- if (!_settings.apiKey) {
58
- console.warn('Missing Pinecone API key : returning empty Pinecone connector');
59
- return;
60
- }
61
- if (!_settings.indexName) {
62
- console.warn('Missing Pinecone index name : returning empty Pinecone connector');
63
- return;
64
- }
65
-
66
- this.client = new Pinecone({
67
- apiKey: _settings.apiKey,
68
- });
69
- console.info('Pinecone client initialized');
70
- console.info('Pinecone index name:', _settings.indexName);
71
- this.indexName = _settings.indexName;
72
- this.accountConnector = ConnectorService.getAccountConnector();
73
- this.cache = ConnectorService.getCacheConnector();
74
- this.nkvConnector = ConnectorService.getNKVConnector();
75
- if (!_settings.embeddings) {
76
- _settings.embeddings = { provider: 'OpenAI', model: 'text-embedding-3-large', params: { dimensions: 1024 } };
77
- }
78
- if (!_settings.embeddings.params) _settings.embeddings.params = { dimensions: 1024 };
79
- if (!_settings.embeddings.params?.dimensions) _settings.embeddings.params.dimensions = 1024;
80
-
81
- this.embedder = EmbeddingsFactory.create(_settings.embeddings.provider, _settings.embeddings);
82
- }
83
-
84
- public async getResourceACL(resourceId: string, candidate: IAccessCandidate): Promise<ACL> {
85
- //const teamId = await this.accountConnector.getCandidateTeam(AccessCandidate.clone(candidate));
86
- const preparedNs = this.constructNsName(candidate as AccessCandidate, resourceId);
87
- const acl = await this.getACL(AccessCandidate.clone(candidate), preparedNs);
88
- const exists = !!acl;
89
-
90
- if (!exists) {
91
- //the resource does not exist yet, we grant write access to the candidate in order to allow the resource creation
92
- return new ACL().addAccess(candidate.role, candidate.id, TAccessLevel.Owner);
93
- }
94
- return ACL.from(acl);
95
- }
96
-
97
- @SecureConnector.AccessControl
98
- protected async createNamespace(acRequest: AccessRequest, namespace: string, metadata?: { [key: string]: any }): Promise<void> {
99
- //* Since Pinecone does not create explicit namespaces,
100
- //* we create a zero or dummy vector in the namespace to trigger the namespace creation and filter it out
101
-
102
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
103
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
104
-
105
- const acl = new ACL().addAccess(acRequest.candidate.role, acRequest.candidate.id, TAccessLevel.Owner);
106
-
107
- const nsData = {
108
- acl: acl.serializedACL,
109
- displayName: namespace,
110
- ...metadata,
111
- };
112
- await this.client
113
- .Index(this.indexName)
114
- .namespace(preparedNs)
115
- .upsert([
116
- {
117
- id: `_reserved_${preparedNs}`,
118
- values: this.embedder.dummyVector,
119
- metadata: {
120
- isSkeletonVector: true,
121
- ...nsData,
122
- },
123
- },
124
- ]);
125
-
126
- await this.setACL(acRequest, preparedNs, acl.ACL);
127
-
128
- return new Promise<void>((resolve) => resolve());
129
- }
130
-
131
- @SecureConnector.AccessControl
132
- protected async namespaceExists(acRequest: AccessRequest, namespace: string): Promise<boolean> {
133
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
134
- const stats = await this.client.Index(this.indexName).describeIndexStats();
135
- return Object.keys(stats.namespaces).includes(this.constructNsName(acRequest.candidate as AccessCandidate, namespace));
136
- }
137
-
138
- @SecureConnector.AccessControl
139
- protected async deleteNamespace(acRequest: AccessRequest, namespace: string): Promise<void> {
140
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
141
- //const candidate = AccessCandidate.team(teamId);
142
- const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
143
-
144
- await this.client
145
- .Index(this.indexName)
146
- .namespace(this.constructNsName(acRequest.candidate as AccessCandidate, namespace))
147
- .deleteAll()
148
- .catch((e) => {
149
- if (e?.name == 'PineconeNotFoundError') {
150
- console.warn(`Namespace ${namespace} does not exist and was requested to be deleted`);
151
- return;
152
- }
153
- throw e;
154
- });
155
-
156
- await this.deleteACL(AccessCandidate.clone(acRequest.candidate), namespace);
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
-
168
- const pineconeIndex = this.client.Index(this.indexName).namespace(this.constructNsName(acRequest.candidate as AccessCandidate, namespace));
169
- let _vector = query;
170
- if (typeof query === 'string') {
171
- _vector = await this.embedder.embedText(query, acRequest.candidate as AccessCandidate);
172
- }
173
-
174
- const topK = (options.topK || 10) + 1; //* we increment one in case it included the skeleton vector
175
-
176
- const results = await pineconeIndex.query({
177
- topK,
178
- vector: _vector as number[],
179
- includeMetadata: true,
180
- includeValues: true,
181
- });
182
-
183
- let matches = [];
184
-
185
- for (const match of results.matches) {
186
- if (match.metadata?.isSkeletonVector) continue;
187
-
188
- if (match.metadata?.[this.USER_METADATA_KEY]) {
189
- match.metadata[this.USER_METADATA_KEY] = JSONContentHelper.create(match.metadata[this.USER_METADATA_KEY].toString()).tryParse();
190
- }
191
-
192
- matches.push({
193
- id: match.id,
194
- values: match.values,
195
- text: match.metadata?.text as string | undefined,
196
- metadata: match.metadata?.[this.USER_METADATA_KEY] as Record<string, any> | undefined,
197
- score: match.score,
198
- });
199
- }
200
-
201
- // in the case where we did not filter out the skeleton vector, we need to remove the extra match from the results
202
- return matches.slice(0, options.topK);
203
- }
204
-
205
- @SecureConnector.AccessControl
206
- protected async insert(
207
- acRequest: AccessRequest,
208
- namespace: string,
209
- sourceWrapper: IVectorDataSourceDto | IVectorDataSourceDto[]
210
- ): Promise<string[]> {
211
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
212
- sourceWrapper = Array.isArray(sourceWrapper) ? sourceWrapper : [sourceWrapper];
213
-
214
- // make sure that all sources are of the same type (source.source)
215
- if (sourceWrapper.some((s) => this.embedder.detectSourceType(s.source) !== this.embedder.detectSourceType(sourceWrapper[0].source))) {
216
- throw new Error('All sources must be of the same type');
217
- }
218
-
219
- const sourceType = this.embedder.detectSourceType(sourceWrapper[0].source);
220
- if (sourceType === 'unknown' || sourceType === 'url') throw new Error('Invalid source type');
221
- const transformedSource = await this.embedder.transformSource(sourceWrapper, sourceType, acRequest.candidate as AccessCandidate);
222
- const preparedSource = transformedSource.map((s) => ({
223
- id: s.id,
224
- values: s.source as number[],
225
- metadata: s.metadata,
226
- }));
227
-
228
- // await pineconeStore.addDocuments(chunks, ids);
229
- await this.client
230
- .Index(this.indexName)
231
- .namespace(this.constructNsName(acRequest.candidate as AccessCandidate, namespace))
232
- .upsert(preparedSource);
233
-
234
- const accessCandidate = acRequest.candidate;
235
-
236
- const isNewNs = !(await this.requester(AccessCandidate.clone(accessCandidate)).namespaceExists(namespace));
237
- if (isNewNs) {
238
- let acl = new ACL().addAccess(accessCandidate.role, accessCandidate.id, TAccessLevel.Owner).ACL;
239
- await this.setACL(acRequest, namespace, acl);
240
- }
241
-
242
- return preparedSource.map((s) => s.id);
243
- }
244
-
245
- @SecureConnector.AccessControl
246
- protected async delete(acRequest: AccessRequest, namespace: string, deleteTarget: DeleteTarget): Promise<void> {
247
- const isDeleteByFilter = typeof deleteTarget === 'object';
248
-
249
- if (isDeleteByFilter) {
250
- // TODO: handle delete by filter logic
251
- } else {
252
- const _ids = Array.isArray(deleteTarget) ? deleteTarget : [deleteTarget];
253
-
254
- const res = await this.client
255
- .Index(this.indexName)
256
- .namespace(this.constructNsName(acRequest.candidate as AccessCandidate, namespace))
257
- .deleteMany(_ids);
258
- }
259
- }
260
-
261
- @SecureConnector.AccessControl
262
- protected async createDatasource(acRequest: AccessRequest, namespace: string, datasource: DatasourceDto): Promise<IStorageVectorDataSource> {
263
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
264
- const acl = new ACL().addAccess(acRequest.candidate.role, acRequest.candidate.id, TAccessLevel.Owner);
265
- const dsId = datasource.id || crypto.randomUUID();
266
-
267
- const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
268
- const chunkedText = chunkText(datasource.text, {
269
- chunkSize: datasource.chunkSize,
270
- chunkOverlap: datasource.chunkOverlap,
271
- });
272
- const label = datasource.label || 'Untitled';
273
- const ids = Array.from({ length: chunkedText.length }, (_, i) => `${dsId}_${crypto.randomUUID()}`);
274
- const source: IVectorDataSourceDto[] = chunkedText.map((doc, i) => {
275
- return {
276
- id: ids[i],
277
- source: doc,
278
- metadata: {
279
- acl: acl.serializedACL,
280
- namespaceId: formattedNs,
281
- datasourceId: dsId,
282
- datasourceLabel: label,
283
- user_metadata: datasource.metadata ? jsonrepair(JSON.stringify(datasource.metadata)) : undefined,
284
- },
285
- };
286
- });
287
-
288
- const _vIds = await this.insert(acRequest, namespace, source);
289
-
290
- const dsData: IStorageVectorDataSource = {
291
- namespaceId: formattedNs,
292
- candidateId: acRequest.candidate.id,
293
- candidateRole: acRequest.candidate.role,
294
- name: datasource.label || 'Untitled',
295
- metadata: datasource.metadata ? jsonrepair(JSON.stringify(datasource.metadata)) : undefined,
296
- text: datasource.text,
297
- vectorIds: _vIds,
298
- id: dsId,
299
- };
300
- // const url = `smythfs://${teamId}.team/_datasources/${dsId}.json`;
301
- // await SmythFS.Instance.write(url, JSON.stringify(dsData), AccessCandidate.team(teamId));
302
- await this.nkvConnector
303
- .requester(acRequest.candidate as AccessCandidate)
304
- .set(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`, dsId, JSON.stringify(dsData));
305
- // return { id: dsId, vectorIds: _vIds };
306
- return dsData;
307
- }
308
-
309
- @SecureConnector.AccessControl
310
- protected async deleteDatasource(acRequest: AccessRequest, namespace: string, datasourceId: string): Promise<void> {
311
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
312
- const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
313
- // const url = `smythfs://${teamId}.team/_datasources/${dsId}.json`;
314
- // await SmythFS.Instance.delete(url, AccessCandidate.team(teamId));
315
- let ds: IStorageVectorDataSource = JSONContentHelper.create(
316
- (
317
- await this.nkvConnector
318
- .requester(acRequest.candidate as AccessCandidate)
319
- .get(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`, datasourceId)
320
- )?.toString()
321
- ).tryParse();
322
-
323
- if (!ds || typeof ds !== 'object') {
324
- throw new Error(`Data source not found with id: ${datasourceId}`);
325
- }
326
-
327
- await this.delete(acRequest, namespace, ds.vectorIds || []);
328
-
329
- await this.nkvConnector
330
- .requester(acRequest.candidate as AccessCandidate)
331
- .delete(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`, datasourceId);
332
- }
333
-
334
- @SecureConnector.AccessControl
335
- protected async listDatasources(acRequest: AccessRequest, namespace: string): Promise<IStorageVectorDataSource[]> {
336
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
337
- const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
338
- return (
339
- await this.nkvConnector
340
- .requester(acRequest.candidate as AccessCandidate)
341
- .list(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`)
342
- ).map((ds) => {
343
- return JSONContentHelper.create(ds.data?.toString()).tryParse() as IStorageVectorDataSource;
344
- });
345
- }
346
-
347
- @SecureConnector.AccessControl
348
- protected async getDatasource(acRequest: AccessRequest, namespace: string, datasourceId: string): Promise<IStorageVectorDataSource> {
349
- //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
350
- const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
351
- return JSONContentHelper.create(
352
- (
353
- await this.nkvConnector
354
- .requester(acRequest.candidate as AccessCandidate)
355
- .get(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`, datasourceId)
356
- )?.toString()
357
- ).tryParse() as IStorageVectorDataSource;
358
- }
359
-
360
- private async setACL(acRequest: AccessRequest, preparedNs: string, acl: IACL): Promise<void> {
361
- await this.cache
362
- .requester(AccessCandidate.clone(acRequest.candidate))
363
- .set(`vectorDB:pinecone:namespace:${preparedNs}:acl`, JSON.stringify(acl));
364
- }
365
-
366
- private async getACL(ac: AccessCandidate, preparedNs: string): Promise<ACL | null | undefined> {
367
- let aclRes = await this.cache.requester(ac).get(`vectorDB:pinecone:namespace:${preparedNs}:acl`);
368
- const acl = JSONContentHelper.create(aclRes?.toString?.()).tryParse();
369
- return acl;
370
- }
371
-
372
- private async deleteACL(ac: AccessCandidate, preparedNs: string): Promise<void> {
373
- this.cache.requester(AccessCandidate.clone(ac)).delete(`vectorDB:pinecone:namespace:${preparedNs}:acl`);
374
- }
375
-
376
- public constructNsName(candidate: AccessCandidate, name: string) {
377
- const joinedName = name.trim().replace(/\s/g, '_').toLowerCase();
378
- let prefix = candidate.id;
379
-
380
- if (candidate.role !== TAccessRole.Team) {
381
- //DO NOT add role prefix for teams to preserve backward compatibility
382
- prefix = candidate.role[0] + '_' + candidate.id;
383
- }
384
-
385
- return `${prefix}_${joinedName}`;
386
- }
387
- }
1
+ //==[ SRE: S3Storage ]======================
2
+ import { ACL } from '@sre/Security/AccessControl/ACL.class';
3
+ import { IAccessCandidate, IACL, TAccessLevel, TAccessRole } from '@sre/types/ACL.types';
4
+ import { AccessRequest } from '@sre/Security/AccessControl/AccessRequest.class';
5
+ import { AccessCandidate } from '@sre/Security/AccessControl/AccessCandidate.class';
6
+ import { SecureConnector } from '@sre/Security/SecureConnector.class';
7
+ import { VectorDBConnector, DeleteTarget } from '../VectorDBConnector';
8
+ import {
9
+ DatasourceDto,
10
+ IStorageVectorDataSource,
11
+ IStorageVectorNamespace,
12
+ IVectorDataSourceDto,
13
+ QueryOptions,
14
+ VectorsResultData,
15
+ } from '@sre/types/VectorDB.types';
16
+ import { Pinecone } from '@pinecone-database/pinecone';
17
+ import { ConnectorService } from '@sre/Core/ConnectorsService';
18
+ import { Logger } from '@sre/helpers/Log.helper';
19
+ import { NKVConnector } from '@sre/IO/NKV.service/NKVConnector';
20
+ import { AccountConnector } from '@sre/Security/Account.service/AccountConnector';
21
+ import { JSONContentHelper } from '@sre/helpers/JsonContent.helper';
22
+ import { CacheConnector } from '@sre/MemoryManager/Cache.service/CacheConnector';
23
+ import crypto from 'crypto';
24
+ import { BaseEmbedding, TEmbeddings } from '../embed/BaseEmbedding';
25
+ import { EmbeddingsFactory, SupportedProviders, SupportedModels } from '../embed';
26
+ import { chunkText } from '@sre/utils/string.utils';
27
+ import { jsonrepair } from 'jsonrepair';
28
+
29
+ const console = Logger('Pinecone VectorDB');
30
+
31
+ export type PineconeConfig = {
32
+ /**
33
+ * The Pinecone API key
34
+ */
35
+ apiKey: string;
36
+ /**
37
+ * The Pinecone index name
38
+ */
39
+ indexName: string;
40
+ /**
41
+ * The embeddings model to use
42
+ */
43
+ embeddings?: TEmbeddings;
44
+ };
45
+ export class PineconeVectorDB extends VectorDBConnector {
46
+ public name = 'PineconeVectorDB';
47
+ public id = 'pinecone';
48
+ private client: Pinecone;
49
+ private indexName: string;
50
+ private cache: CacheConnector;
51
+ private accountConnector: AccountConnector;
52
+ private nkvConnector: NKVConnector;
53
+ public embedder: BaseEmbedding;
54
+
55
+ constructor(protected _settings: PineconeConfig) {
56
+ super(_settings);
57
+ if (!_settings.apiKey) {
58
+ console.warn('Missing Pinecone API key : returning empty Pinecone connector');
59
+ return;
60
+ }
61
+ if (!_settings.indexName) {
62
+ console.warn('Missing Pinecone index name : returning empty Pinecone connector');
63
+ return;
64
+ }
65
+
66
+ this.client = new Pinecone({
67
+ apiKey: _settings.apiKey,
68
+ });
69
+ console.info('Pinecone client initialized');
70
+ console.info('Pinecone index name:', _settings.indexName);
71
+ this.indexName = _settings.indexName;
72
+ this.accountConnector = ConnectorService.getAccountConnector();
73
+ this.cache = ConnectorService.getCacheConnector();
74
+ this.nkvConnector = ConnectorService.getNKVConnector();
75
+ if (!_settings.embeddings) {
76
+ _settings.embeddings = { provider: 'OpenAI', model: 'text-embedding-3-large', params: { dimensions: 1024 } };
77
+ }
78
+ if (!_settings.embeddings.params) _settings.embeddings.params = { dimensions: 1024 };
79
+ if (!_settings.embeddings.params?.dimensions) _settings.embeddings.params.dimensions = 1024;
80
+
81
+ this.embedder = EmbeddingsFactory.create(_settings.embeddings.provider, _settings.embeddings);
82
+ }
83
+
84
+ public async getResourceACL(resourceId: string, candidate: IAccessCandidate): Promise<ACL> {
85
+ //const teamId = await this.accountConnector.getCandidateTeam(AccessCandidate.clone(candidate));
86
+ const preparedNs = this.constructNsName(candidate as AccessCandidate, resourceId);
87
+ const acl = await this.getACL(AccessCandidate.clone(candidate), preparedNs);
88
+ const exists = !!acl;
89
+
90
+ if (!exists) {
91
+ //the resource does not exist yet, we grant write access to the candidate in order to allow the resource creation
92
+ return new ACL().addAccess(candidate.role, candidate.id, TAccessLevel.Owner);
93
+ }
94
+ return ACL.from(acl);
95
+ }
96
+
97
+ @SecureConnector.AccessControl
98
+ protected async createNamespace(acRequest: AccessRequest, namespace: string, metadata?: { [key: string]: any }): Promise<void> {
99
+ //* Since Pinecone does not create explicit namespaces,
100
+ //* we create a zero or dummy vector in the namespace to trigger the namespace creation and filter it out
101
+
102
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
103
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
104
+
105
+ const acl = new ACL().addAccess(acRequest.candidate.role, acRequest.candidate.id, TAccessLevel.Owner);
106
+
107
+ const nsData = {
108
+ acl: acl.serializedACL,
109
+ displayName: namespace,
110
+ ...metadata,
111
+ };
112
+ await this.client
113
+ .Index(this.indexName)
114
+ .namespace(preparedNs)
115
+ .upsert([
116
+ {
117
+ id: `_reserved_${preparedNs}`,
118
+ values: this.embedder.dummyVector,
119
+ metadata: {
120
+ isSkeletonVector: true,
121
+ ...nsData,
122
+ },
123
+ },
124
+ ]);
125
+
126
+ await this.setACL(acRequest, preparedNs, acl.ACL);
127
+
128
+ return new Promise<void>((resolve) => resolve());
129
+ }
130
+
131
+ @SecureConnector.AccessControl
132
+ protected async namespaceExists(acRequest: AccessRequest, namespace: string): Promise<boolean> {
133
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
134
+ const stats = await this.client.Index(this.indexName).describeIndexStats();
135
+ return Object.keys(stats.namespaces).includes(this.constructNsName(acRequest.candidate as AccessCandidate, namespace));
136
+ }
137
+
138
+ @SecureConnector.AccessControl
139
+ protected async deleteNamespace(acRequest: AccessRequest, namespace: string): Promise<void> {
140
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
141
+ //const candidate = AccessCandidate.team(teamId);
142
+ const preparedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
143
+
144
+ await this.client
145
+ .Index(this.indexName)
146
+ .namespace(this.constructNsName(acRequest.candidate as AccessCandidate, namespace))
147
+ .deleteAll()
148
+ .catch((e) => {
149
+ if (e?.name == 'PineconeNotFoundError') {
150
+ console.warn(`Namespace ${namespace} does not exist and was requested to be deleted`);
151
+ return;
152
+ }
153
+ throw e;
154
+ });
155
+
156
+ await this.deleteACL(AccessCandidate.clone(acRequest.candidate), namespace);
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
+
168
+ const pineconeIndex = this.client.Index(this.indexName).namespace(this.constructNsName(acRequest.candidate as AccessCandidate, namespace));
169
+ let _vector = query;
170
+ if (typeof query === 'string') {
171
+ _vector = await this.embedder.embedText(query, acRequest.candidate as AccessCandidate);
172
+ }
173
+
174
+ const topK = (options.topK || 10) + 1; //* we increment one in case it included the skeleton vector
175
+
176
+ const results = await pineconeIndex.query({
177
+ topK,
178
+ vector: _vector as number[],
179
+ includeMetadata: true,
180
+ includeValues: true,
181
+ });
182
+
183
+ let matches = [];
184
+
185
+ for (const match of results.matches) {
186
+ if (match.metadata?.isSkeletonVector) continue;
187
+
188
+ if (match.metadata?.[this.USER_METADATA_KEY]) {
189
+ match.metadata[this.USER_METADATA_KEY] = JSONContentHelper.create(match.metadata[this.USER_METADATA_KEY].toString()).tryParse();
190
+ }
191
+
192
+ matches.push({
193
+ id: match.id,
194
+ values: match.values,
195
+ text: match.metadata?.text as string | undefined,
196
+ metadata: match.metadata?.[this.USER_METADATA_KEY] as Record<string, any> | undefined,
197
+ score: match.score,
198
+ });
199
+ }
200
+
201
+ // in the case where we did not filter out the skeleton vector, we need to remove the extra match from the results
202
+ return matches.slice(0, options.topK);
203
+ }
204
+
205
+ @SecureConnector.AccessControl
206
+ protected async insert(
207
+ acRequest: AccessRequest,
208
+ namespace: string,
209
+ sourceWrapper: IVectorDataSourceDto | IVectorDataSourceDto[]
210
+ ): Promise<string[]> {
211
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
212
+ sourceWrapper = Array.isArray(sourceWrapper) ? sourceWrapper : [sourceWrapper];
213
+
214
+ // make sure that all sources are of the same type (source.source)
215
+ if (sourceWrapper.some((s) => this.embedder.detectSourceType(s.source) !== this.embedder.detectSourceType(sourceWrapper[0].source))) {
216
+ throw new Error('All sources must be of the same type');
217
+ }
218
+
219
+ const sourceType = this.embedder.detectSourceType(sourceWrapper[0].source);
220
+ if (sourceType === 'unknown' || sourceType === 'url') throw new Error('Invalid source type');
221
+ const transformedSource = await this.embedder.transformSource(sourceWrapper, sourceType, acRequest.candidate as AccessCandidate);
222
+ const preparedSource = transformedSource.map((s) => ({
223
+ id: s.id,
224
+ values: s.source as number[],
225
+ metadata: s.metadata,
226
+ }));
227
+
228
+ // await pineconeStore.addDocuments(chunks, ids);
229
+ await this.client
230
+ .Index(this.indexName)
231
+ .namespace(this.constructNsName(acRequest.candidate as AccessCandidate, namespace))
232
+ .upsert(preparedSource);
233
+
234
+ const accessCandidate = acRequest.candidate;
235
+
236
+ const isNewNs = !(await this.requester(AccessCandidate.clone(accessCandidate)).namespaceExists(namespace));
237
+ if (isNewNs) {
238
+ let acl = new ACL().addAccess(accessCandidate.role, accessCandidate.id, TAccessLevel.Owner).ACL;
239
+ await this.setACL(acRequest, namespace, acl);
240
+ }
241
+
242
+ return preparedSource.map((s) => s.id);
243
+ }
244
+
245
+ @SecureConnector.AccessControl
246
+ protected async delete(acRequest: AccessRequest, namespace: string, deleteTarget: DeleteTarget): Promise<void> {
247
+ const isDeleteByFilter = typeof deleteTarget === 'object';
248
+
249
+ if (isDeleteByFilter) {
250
+ // TODO: handle delete by filter logic
251
+ } else {
252
+ const _ids = Array.isArray(deleteTarget) ? deleteTarget : [deleteTarget];
253
+
254
+ const res = await this.client
255
+ .Index(this.indexName)
256
+ .namespace(this.constructNsName(acRequest.candidate as AccessCandidate, namespace))
257
+ .deleteMany(_ids);
258
+ }
259
+ }
260
+
261
+ @SecureConnector.AccessControl
262
+ protected async createDatasource(acRequest: AccessRequest, namespace: string, datasource: DatasourceDto): Promise<IStorageVectorDataSource> {
263
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
264
+ const acl = new ACL().addAccess(acRequest.candidate.role, acRequest.candidate.id, TAccessLevel.Owner);
265
+ const dsId = datasource.id || crypto.randomUUID();
266
+
267
+ const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
268
+ const chunkedText = chunkText(datasource.text, {
269
+ chunkSize: datasource.chunkSize,
270
+ chunkOverlap: datasource.chunkOverlap,
271
+ });
272
+ const label = datasource.label || 'Untitled';
273
+ const ids = Array.from({ length: chunkedText.length }, (_, i) => `${dsId}_${crypto.randomUUID()}`);
274
+ const source: IVectorDataSourceDto[] = chunkedText.map((doc, i) => {
275
+ return {
276
+ id: ids[i],
277
+ source: doc,
278
+ metadata: {
279
+ acl: acl.serializedACL,
280
+ namespaceId: formattedNs,
281
+ datasourceId: dsId,
282
+ datasourceLabel: label,
283
+ user_metadata: datasource.metadata ? jsonrepair(JSON.stringify(datasource.metadata)) : undefined,
284
+ },
285
+ };
286
+ });
287
+
288
+ const _vIds = await this.insert(acRequest, namespace, source);
289
+
290
+ const dsData: IStorageVectorDataSource = {
291
+ namespaceId: formattedNs,
292
+ candidateId: acRequest.candidate.id,
293
+ candidateRole: acRequest.candidate.role,
294
+ name: datasource.label || 'Untitled',
295
+ metadata: datasource.metadata ? jsonrepair(JSON.stringify(datasource.metadata)) : undefined,
296
+ text: datasource.text,
297
+ vectorIds: _vIds,
298
+ id: dsId,
299
+ };
300
+ // const url = `smythfs://${teamId}.team/_datasources/${dsId}.json`;
301
+ // await SmythFS.Instance.write(url, JSON.stringify(dsData), AccessCandidate.team(teamId));
302
+ await this.nkvConnector
303
+ .requester(acRequest.candidate as AccessCandidate)
304
+ .set(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`, dsId, JSON.stringify(dsData));
305
+ // return { id: dsId, vectorIds: _vIds };
306
+ return dsData;
307
+ }
308
+
309
+ @SecureConnector.AccessControl
310
+ protected async deleteDatasource(acRequest: AccessRequest, namespace: string, datasourceId: string): Promise<void> {
311
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
312
+ const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
313
+ // const url = `smythfs://${teamId}.team/_datasources/${dsId}.json`;
314
+ // await SmythFS.Instance.delete(url, AccessCandidate.team(teamId));
315
+ let ds: IStorageVectorDataSource = JSONContentHelper.create(
316
+ (
317
+ await this.nkvConnector
318
+ .requester(acRequest.candidate as AccessCandidate)
319
+ .get(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`, datasourceId)
320
+ )?.toString()
321
+ ).tryParse();
322
+
323
+ if (!ds || typeof ds !== 'object') {
324
+ throw new Error(`Data source not found with id: ${datasourceId}`);
325
+ }
326
+
327
+ await this.delete(acRequest, namespace, ds.vectorIds || []);
328
+
329
+ await this.nkvConnector
330
+ .requester(acRequest.candidate as AccessCandidate)
331
+ .delete(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`, datasourceId);
332
+ }
333
+
334
+ @SecureConnector.AccessControl
335
+ protected async listDatasources(acRequest: AccessRequest, namespace: string): Promise<IStorageVectorDataSource[]> {
336
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
337
+ const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
338
+ return (
339
+ await this.nkvConnector
340
+ .requester(acRequest.candidate as AccessCandidate)
341
+ .list(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`)
342
+ ).map((ds) => {
343
+ return JSONContentHelper.create(ds.data?.toString()).tryParse() as IStorageVectorDataSource;
344
+ });
345
+ }
346
+
347
+ @SecureConnector.AccessControl
348
+ protected async getDatasource(acRequest: AccessRequest, namespace: string, datasourceId: string): Promise<IStorageVectorDataSource> {
349
+ //const teamId = await this.accountConnector.getCandidateTeam(acRequest.candidate);
350
+ const formattedNs = this.constructNsName(acRequest.candidate as AccessCandidate, namespace);
351
+ return JSONContentHelper.create(
352
+ (
353
+ await this.nkvConnector
354
+ .requester(acRequest.candidate as AccessCandidate)
355
+ .get(`vectorDB:${this.id}:namespaces:${formattedNs}:datasources`, datasourceId)
356
+ )?.toString()
357
+ ).tryParse() as IStorageVectorDataSource;
358
+ }
359
+
360
+ private async setACL(acRequest: AccessRequest, preparedNs: string, acl: IACL): Promise<void> {
361
+ await this.cache
362
+ .requester(AccessCandidate.clone(acRequest.candidate))
363
+ .set(`vectorDB:pinecone:namespace:${preparedNs}:acl`, JSON.stringify(acl));
364
+ }
365
+
366
+ private async getACL(ac: AccessCandidate, preparedNs: string): Promise<ACL | null | undefined> {
367
+ let aclRes = await this.cache.requester(ac).get(`vectorDB:pinecone:namespace:${preparedNs}:acl`);
368
+ const acl = JSONContentHelper.create(aclRes?.toString?.()).tryParse();
369
+ return acl;
370
+ }
371
+
372
+ private async deleteACL(ac: AccessCandidate, preparedNs: string): Promise<void> {
373
+ this.cache.requester(AccessCandidate.clone(ac)).delete(`vectorDB:pinecone:namespace:${preparedNs}:acl`);
374
+ }
375
+
376
+ public constructNsName(candidate: AccessCandidate, name: string) {
377
+ const joinedName = name.trim().replace(/\s/g, '_').toLowerCase();
378
+ let prefix = candidate.id;
379
+
380
+ if (candidate.role !== TAccessRole.Team) {
381
+ //DO NOT add role prefix for teams to preserve backward compatibility
382
+ prefix = candidate.role[0] + '_' + candidate.id;
383
+ }
384
+
385
+ return `${prefix}_${joinedName}`;
386
+ }
387
+ }