@smythos/sre 1.6.1 → 1.6.8

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