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