@smythos/sre 1.6.8 → 1.6.10

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