@juspay/neurolink 7.33.4 → 7.34.0
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.md +9 -0
- package/README.md +37 -0
- package/dist/cli/errorHandler.d.ts +1 -0
- package/dist/cli/errorHandler.js +28 -0
- package/dist/cli/factories/commandFactory.d.ts +23 -0
- package/dist/cli/factories/commandFactory.js +375 -60
- package/dist/cli/factories/ollamaCommandFactory.js +7 -1
- package/dist/cli/index.d.ts +1 -1
- package/dist/cli/index.js +9 -164
- package/dist/cli/loop/optionsSchema.d.ts +15 -0
- package/dist/cli/loop/optionsSchema.js +59 -0
- package/dist/cli/loop/session.d.ts +15 -0
- package/dist/cli/loop/session.js +252 -0
- package/dist/cli/parser.d.ts +1 -0
- package/dist/cli/parser.js +158 -0
- package/dist/cli/utils/ollamaUtils.js +6 -0
- package/dist/config/{conversationMemoryConfig.d.ts → conversationMemory.d.ts} +1 -1
- package/dist/core/baseProvider.js +17 -3
- package/dist/core/conversationMemoryFactory.d.ts +23 -0
- package/dist/core/conversationMemoryFactory.js +144 -0
- package/dist/core/conversationMemoryInitializer.d.ts +14 -0
- package/dist/core/conversationMemoryInitializer.js +127 -0
- package/dist/core/conversationMemoryManager.d.ts +3 -2
- package/dist/core/conversationMemoryManager.js +4 -3
- package/dist/core/redisConversationMemoryManager.d.ts +73 -0
- package/dist/core/redisConversationMemoryManager.js +483 -0
- package/dist/core/types.d.ts +1 -1
- package/dist/lib/config/{conversationMemoryConfig.d.ts → conversationMemory.d.ts} +1 -1
- package/dist/lib/core/baseProvider.js +17 -3
- package/dist/lib/core/conversationMemoryFactory.d.ts +23 -0
- package/dist/lib/core/conversationMemoryFactory.js +144 -0
- package/dist/lib/core/conversationMemoryInitializer.d.ts +14 -0
- package/dist/lib/core/conversationMemoryInitializer.js +127 -0
- package/dist/lib/core/conversationMemoryManager.d.ts +3 -2
- package/dist/lib/core/conversationMemoryManager.js +4 -3
- package/dist/lib/core/redisConversationMemoryManager.d.ts +73 -0
- package/dist/lib/core/redisConversationMemoryManager.js +483 -0
- package/dist/lib/core/types.d.ts +1 -1
- package/dist/lib/neurolink.d.ts +15 -9
- package/dist/lib/neurolink.js +218 -67
- package/dist/lib/providers/amazonBedrock.d.ts +4 -4
- package/dist/lib/providers/anthropic.d.ts +4 -4
- package/dist/lib/providers/azureOpenai.d.ts +4 -4
- package/dist/lib/providers/googleAiStudio.d.ts +4 -4
- package/dist/lib/providers/googleAiStudio.js +1 -1
- package/dist/lib/providers/huggingFace.d.ts +4 -4
- package/dist/lib/providers/litellm.d.ts +1 -1
- package/dist/lib/providers/mistral.d.ts +4 -4
- package/dist/lib/providers/mistral.js +2 -2
- package/dist/lib/providers/openAI.d.ts +4 -4
- package/dist/lib/session/globalSessionState.d.ts +27 -0
- package/dist/lib/session/globalSessionState.js +77 -0
- package/dist/lib/types/{conversationTypes.d.ts → conversation.d.ts} +32 -0
- package/dist/lib/types/generateTypes.d.ts +1 -1
- package/dist/lib/types/streamTypes.d.ts +1 -1
- package/dist/lib/utils/conversationMemory.d.ts +22 -0
- package/dist/lib/utils/conversationMemory.js +121 -0
- package/dist/lib/utils/conversationMemoryUtils.d.ts +1 -1
- package/dist/lib/utils/conversationMemoryUtils.js +2 -2
- package/dist/lib/utils/messageBuilder.d.ts +1 -1
- package/dist/lib/utils/messageBuilder.js +1 -1
- package/dist/lib/utils/redis.d.ts +42 -0
- package/dist/lib/utils/redis.js +263 -0
- package/dist/neurolink.d.ts +15 -9
- package/dist/neurolink.js +218 -67
- package/dist/providers/amazonBedrock.d.ts +4 -4
- package/dist/providers/anthropic.d.ts +4 -4
- package/dist/providers/azureOpenai.d.ts +4 -4
- package/dist/providers/googleAiStudio.d.ts +4 -4
- package/dist/providers/googleAiStudio.js +1 -1
- package/dist/providers/huggingFace.d.ts +4 -4
- package/dist/providers/litellm.d.ts +1 -1
- package/dist/providers/mistral.d.ts +4 -4
- package/dist/providers/mistral.js +2 -2
- package/dist/providers/openAI.d.ts +4 -4
- package/dist/session/globalSessionState.d.ts +27 -0
- package/dist/session/globalSessionState.js +77 -0
- package/dist/types/{conversationTypes.d.ts → conversation.d.ts} +32 -0
- package/dist/types/generateTypes.d.ts +1 -1
- package/dist/types/streamTypes.d.ts +1 -1
- package/dist/utils/conversationMemory.d.ts +22 -0
- package/dist/utils/conversationMemory.js +121 -0
- package/dist/utils/conversationMemoryUtils.d.ts +1 -1
- package/dist/utils/conversationMemoryUtils.js +2 -2
- package/dist/utils/messageBuilder.d.ts +1 -1
- package/dist/utils/messageBuilder.js +1 -1
- package/dist/utils/redis.d.ts +42 -0
- package/dist/utils/redis.js +263 -0
- package/package.json +3 -1
- /package/dist/config/{conversationMemoryConfig.js → conversationMemory.js} +0 -0
- /package/dist/lib/config/{conversationMemoryConfig.js → conversationMemory.js} +0 -0
- /package/dist/lib/types/{conversationTypes.js → conversation.js} +0 -0
- /package/dist/types/{conversationTypes.js → conversation.js} +0 -0
@@ -0,0 +1,144 @@
|
|
1
|
+
/**
|
2
|
+
* Conversation Memory Factory for NeuroLink
|
3
|
+
* Creates appropriate conversation memory manager based on configuration
|
4
|
+
*/
|
5
|
+
import { ConversationMemoryManager } from "./conversationMemoryManager.js";
|
6
|
+
import { RedisConversationMemoryManager } from "./redisConversationMemoryManager.js";
|
7
|
+
import { logger } from "../utils/logger.js";
|
8
|
+
/**
|
9
|
+
* Creates a conversation memory manager based on configuration
|
10
|
+
*/
|
11
|
+
export function createConversationMemoryManager(config, storageType = "memory", redisConfig) {
|
12
|
+
logger.debug("[conversationMemoryFactory] Creating conversation memory manager", {
|
13
|
+
storageType,
|
14
|
+
config: {
|
15
|
+
enabled: config.enabled,
|
16
|
+
maxSessions: config.maxSessions,
|
17
|
+
maxTurnsPerSession: config.maxTurnsPerSession,
|
18
|
+
enableSummarization: config.enableSummarization,
|
19
|
+
summarizationThresholdTurns: config.summarizationThresholdTurns,
|
20
|
+
summarizationTargetTurns: config.summarizationTargetTurns,
|
21
|
+
summarizationProvider: config.summarizationProvider,
|
22
|
+
summarizationModel: config.summarizationModel,
|
23
|
+
},
|
24
|
+
hasRedisConfig: !!redisConfig,
|
25
|
+
});
|
26
|
+
// Default to memory storage
|
27
|
+
if (storageType === "memory" || !storageType) {
|
28
|
+
logger.debug("[conversationMemoryFactory] Creating in-memory conversation manager");
|
29
|
+
const memoryManager = new ConversationMemoryManager(config);
|
30
|
+
logger.debug("[conversationMemoryFactory] In-memory conversation manager created successfully", {
|
31
|
+
managerType: memoryManager.constructor.name,
|
32
|
+
});
|
33
|
+
return memoryManager;
|
34
|
+
}
|
35
|
+
// Redis storage
|
36
|
+
if (storageType === "redis") {
|
37
|
+
logger.debug("[conversationMemoryFactory] Creating Redis conversation manager", {
|
38
|
+
host: redisConfig?.host || "localhost",
|
39
|
+
port: redisConfig?.port || 6379,
|
40
|
+
keyPrefix: redisConfig?.keyPrefix || "neurolink:conversation:",
|
41
|
+
ttl: redisConfig?.ttl || 86400,
|
42
|
+
hasConnectionOptions: !!redisConfig?.connectionOptions,
|
43
|
+
});
|
44
|
+
const redisManager = new RedisConversationMemoryManager(config, redisConfig);
|
45
|
+
logger.debug("[conversationMemoryFactory] Redis conversation manager created successfully", {
|
46
|
+
managerType: redisManager.constructor.name,
|
47
|
+
config: {
|
48
|
+
maxSessions: config.maxSessions,
|
49
|
+
maxTurnsPerSession: config.maxTurnsPerSession,
|
50
|
+
},
|
51
|
+
});
|
52
|
+
return redisManager;
|
53
|
+
}
|
54
|
+
// Fallback to memory storage for unknown types
|
55
|
+
logger.warn(`[conversationMemoryFactory] Unknown storage type: ${storageType}, falling back to memory storage`);
|
56
|
+
const fallbackManager = new ConversationMemoryManager(config);
|
57
|
+
logger.debug("[conversationMemoryFactory] Fallback memory manager created successfully", {
|
58
|
+
managerType: fallbackManager.constructor.name,
|
59
|
+
});
|
60
|
+
return fallbackManager;
|
61
|
+
}
|
62
|
+
/**
|
63
|
+
* Get storage type from environment variable or configuration
|
64
|
+
*/
|
65
|
+
export function getStorageType() {
|
66
|
+
// Get the raw value from environment, or use default
|
67
|
+
const rawStorageType = process.env.STORAGE_TYPE;
|
68
|
+
// Default to "memory" if not set
|
69
|
+
if (!rawStorageType) {
|
70
|
+
logger.debug("[conversationMemoryFactory] No storage type configured, using default", {
|
71
|
+
storageType: "memory",
|
72
|
+
fromEnv: false,
|
73
|
+
});
|
74
|
+
return "memory";
|
75
|
+
}
|
76
|
+
// Normalize: trim and convert to lowercase
|
77
|
+
const normalizedStorageType = rawStorageType.trim().toLowerCase();
|
78
|
+
// Validate against allowed StorageType values
|
79
|
+
const validStorageTypes = ["memory", "redis"];
|
80
|
+
if (validStorageTypes.includes(normalizedStorageType)) {
|
81
|
+
logger.debug("[conversationMemoryFactory] Determined storage type", {
|
82
|
+
storageType: normalizedStorageType,
|
83
|
+
fromEnv: true,
|
84
|
+
envValue: rawStorageType,
|
85
|
+
normalized: normalizedStorageType !== rawStorageType,
|
86
|
+
});
|
87
|
+
return normalizedStorageType;
|
88
|
+
}
|
89
|
+
else {
|
90
|
+
// Invalid storage type, log warning and return default
|
91
|
+
logger.warn(`[conversationMemoryFactory] Unrecognized storage type in environment: "${rawStorageType}", falling back to "memory"`, {
|
92
|
+
providedValue: rawStorageType,
|
93
|
+
normalizedValue: normalizedStorageType,
|
94
|
+
validValues: validStorageTypes,
|
95
|
+
usingDefault: true,
|
96
|
+
});
|
97
|
+
return "memory";
|
98
|
+
}
|
99
|
+
}
|
100
|
+
/**
|
101
|
+
* Get Redis configuration from environment variables
|
102
|
+
*/
|
103
|
+
export function getRedisConfigFromEnv() {
|
104
|
+
logger.debug("[conversationMemoryFactory] Reading Redis configuration from environment", {
|
105
|
+
REDIS_HOST: process.env.REDIS_HOST || "(not set)",
|
106
|
+
REDIS_PORT: process.env.REDIS_PORT || "(not set)",
|
107
|
+
REDIS_PASSWORD: process.env.REDIS_PASSWORD ? "******" : "(not set)",
|
108
|
+
REDIS_DB: process.env.REDIS_DB || "(not set)",
|
109
|
+
REDIS_KEY_PREFIX: process.env.REDIS_KEY_PREFIX || "(not set)",
|
110
|
+
REDIS_TTL: process.env.REDIS_TTL || "(not set)",
|
111
|
+
REDIS_CONNECT_TIMEOUT: process.env.REDIS_CONNECT_TIMEOUT || "(not set)",
|
112
|
+
REDIS_MAX_RETRIES: process.env.REDIS_MAX_RETRIES || "(not set)",
|
113
|
+
REDIS_RETRY_DELAY: process.env.REDIS_RETRY_DELAY || "(not set)",
|
114
|
+
});
|
115
|
+
const config = {
|
116
|
+
host: process.env.REDIS_HOST,
|
117
|
+
port: process.env.REDIS_PORT ? Number(process.env.REDIS_PORT) : undefined,
|
118
|
+
password: process.env.REDIS_PASSWORD,
|
119
|
+
db: process.env.REDIS_DB ? Number(process.env.REDIS_DB) : undefined,
|
120
|
+
keyPrefix: process.env.REDIS_KEY_PREFIX,
|
121
|
+
ttl: process.env.REDIS_TTL ? Number(process.env.REDIS_TTL) : undefined,
|
122
|
+
connectionOptions: {
|
123
|
+
connectTimeout: process.env.REDIS_CONNECT_TIMEOUT
|
124
|
+
? Number(process.env.REDIS_CONNECT_TIMEOUT)
|
125
|
+
: undefined,
|
126
|
+
maxRetriesPerRequest: process.env.REDIS_MAX_RETRIES
|
127
|
+
? Number(process.env.REDIS_MAX_RETRIES)
|
128
|
+
: undefined,
|
129
|
+
retryDelayOnFailover: process.env.REDIS_RETRY_DELAY
|
130
|
+
? Number(process.env.REDIS_RETRY_DELAY)
|
131
|
+
: undefined,
|
132
|
+
},
|
133
|
+
};
|
134
|
+
logger.debug("[conversationMemoryFactory] Redis configuration normalized", {
|
135
|
+
host: config.host || "localhost",
|
136
|
+
port: config.port || 6379,
|
137
|
+
hasPassword: !!config.password,
|
138
|
+
db: config.db || 0,
|
139
|
+
keyPrefix: config.keyPrefix || "neurolink:conversation:",
|
140
|
+
ttl: config.ttl || 86400,
|
141
|
+
hasConnectionOptions: !!config.connectionOptions,
|
142
|
+
});
|
143
|
+
return config;
|
144
|
+
}
|
@@ -0,0 +1,14 @@
|
|
1
|
+
/**
|
2
|
+
* Conversation Memory Initializer
|
3
|
+
* Provides integration with Redis storage for conversation memory
|
4
|
+
*/
|
5
|
+
import type { ConversationMemoryConfig } from "../types/conversation.js";
|
6
|
+
import type { ConversationMemoryManager } from "./conversationMemoryManager.js";
|
7
|
+
import type { RedisConversationMemoryManager } from "./redisConversationMemoryManager.js";
|
8
|
+
/**
|
9
|
+
* Initialize conversation memory for NeuroLink
|
10
|
+
* This function decides whether to use in-memory or Redis storage
|
11
|
+
*/
|
12
|
+
export declare function initializeConversationMemory(config?: {
|
13
|
+
conversationMemory?: Partial<ConversationMemoryConfig>;
|
14
|
+
}): Promise<ConversationMemoryManager | RedisConversationMemoryManager | null>;
|
@@ -0,0 +1,127 @@
|
|
1
|
+
/**
|
2
|
+
* Conversation Memory Initializer
|
3
|
+
* Provides integration with Redis storage for conversation memory
|
4
|
+
*/
|
5
|
+
import { createConversationMemoryManager, getStorageType, getRedisConfigFromEnv, } from "./conversationMemoryFactory.js";
|
6
|
+
import { applyConversationMemoryDefaults } from "../utils/conversationMemory.js";
|
7
|
+
import { logger } from "../utils/logger.js";
|
8
|
+
/**
|
9
|
+
* Initialize conversation memory for NeuroLink
|
10
|
+
* This function decides whether to use in-memory or Redis storage
|
11
|
+
*/
|
12
|
+
export async function initializeConversationMemory(config) {
|
13
|
+
logger.debug("[conversationMemoryInitializer] Initialize conversation memory called", {
|
14
|
+
hasConfig: !!config,
|
15
|
+
hasMemoryConfig: !!config?.conversationMemory,
|
16
|
+
memoryEnabled: config?.conversationMemory?.enabled || false,
|
17
|
+
storageType: process.env.STORAGE_TYPE || "memory",
|
18
|
+
});
|
19
|
+
if (!config?.conversationMemory?.enabled) {
|
20
|
+
logger.debug("[conversationMemoryInitializer] Conversation memory not enabled - skipping initialization");
|
21
|
+
return null;
|
22
|
+
}
|
23
|
+
try {
|
24
|
+
// Apply default configuration
|
25
|
+
logger.debug("[conversationMemoryInitializer] Applying conversation memory defaults");
|
26
|
+
const memoryConfig = applyConversationMemoryDefaults(config.conversationMemory);
|
27
|
+
logger.debug("[conversationMemoryInitializer] Memory configuration processed", {
|
28
|
+
enabled: memoryConfig.enabled,
|
29
|
+
maxSessions: memoryConfig.maxSessions,
|
30
|
+
maxTurnsPerSession: memoryConfig.maxTurnsPerSession,
|
31
|
+
enableSummarization: memoryConfig.enableSummarization,
|
32
|
+
});
|
33
|
+
// Determine storage type from environment
|
34
|
+
const storageType = getStorageType();
|
35
|
+
logger.debug("[conversationMemoryInitializer] Storage type determined", {
|
36
|
+
storageType,
|
37
|
+
fromEnv: !!process.env.STORAGE_TYPE,
|
38
|
+
});
|
39
|
+
if (storageType === "redis") {
|
40
|
+
logger.info("[conversationMemoryInitializer] Initializing Redis-based conversation memory manager");
|
41
|
+
// Get Redis configuration from environment
|
42
|
+
logger.debug("[conversationMemoryInitializer] Getting Redis configuration from environment");
|
43
|
+
const redisConfig = getRedisConfigFromEnv();
|
44
|
+
logger.debug("[conversationMemoryInitializer] Redis configuration retrieved", {
|
45
|
+
host: redisConfig.host || "localhost",
|
46
|
+
port: redisConfig.port || 6379,
|
47
|
+
hasPassword: !!redisConfig.password,
|
48
|
+
db: redisConfig.db || 0,
|
49
|
+
keyPrefix: redisConfig.keyPrefix || "neurolink:conversation:",
|
50
|
+
ttl: redisConfig.ttl || 86400,
|
51
|
+
});
|
52
|
+
// Create Redis-based conversation memory manager
|
53
|
+
logger.debug("[conversationMemoryInitializer] Creating Redis conversation memory manager");
|
54
|
+
const redisMemoryManager = createConversationMemoryManager(memoryConfig, "redis", redisConfig);
|
55
|
+
logger.debug("[conversationMemoryInitializer] Checking Redis manager creation result", {
|
56
|
+
managerType: redisMemoryManager?.constructor?.name || "unknown",
|
57
|
+
isRedisType: redisMemoryManager?.constructor?.name ===
|
58
|
+
"RedisConversationMemoryManager",
|
59
|
+
hasConfig: !!redisMemoryManager?.config,
|
60
|
+
});
|
61
|
+
logger.info("[conversationMemoryInitializer] Redis conversation memory manager created successfully", {
|
62
|
+
host: redisConfig.host || "localhost",
|
63
|
+
port: redisConfig.port || 6379,
|
64
|
+
maxSessions: memoryConfig.maxSessions,
|
65
|
+
maxTurnsPerSession: memoryConfig.maxTurnsPerSession,
|
66
|
+
managerType: redisMemoryManager?.constructor?.name,
|
67
|
+
});
|
68
|
+
// Perform basic validation
|
69
|
+
if (redisMemoryManager?.constructor?.name !==
|
70
|
+
"RedisConversationMemoryManager") {
|
71
|
+
logger.warn("[conversationMemoryInitializer] Created manager is not of RedisConversationMemoryManager type", {
|
72
|
+
actualType: redisMemoryManager?.constructor?.name,
|
73
|
+
});
|
74
|
+
}
|
75
|
+
return redisMemoryManager;
|
76
|
+
}
|
77
|
+
else {
|
78
|
+
logger.info("[conversationMemoryInitializer] Initializing in-memory conversation memory manager");
|
79
|
+
// Create in-memory conversation memory manager
|
80
|
+
logger.debug("[conversationMemoryInitializer] Creating in-memory conversation memory manager");
|
81
|
+
const memoryManager = createConversationMemoryManager(memoryConfig);
|
82
|
+
logger.debug("[conversationMemoryInitializer] Checking memory manager creation result", {
|
83
|
+
managerType: memoryManager?.constructor?.name || "unknown",
|
84
|
+
isInMemoryType: memoryManager?.constructor?.name === "ConversationMemoryManager",
|
85
|
+
hasConfig: !!memoryManager?.config,
|
86
|
+
});
|
87
|
+
logger.info("[conversationMemoryInitializer] In-memory conversation memory manager created successfully", {
|
88
|
+
maxSessions: memoryConfig.maxSessions,
|
89
|
+
maxTurnsPerSession: memoryConfig.maxTurnsPerSession,
|
90
|
+
managerType: memoryManager?.constructor?.name,
|
91
|
+
});
|
92
|
+
return memoryManager;
|
93
|
+
}
|
94
|
+
}
|
95
|
+
catch (error) {
|
96
|
+
logger.error("[conversationMemoryInitializer] Failed to initialize conversation memory", {
|
97
|
+
error: error instanceof Error ? error.message : String(error),
|
98
|
+
errorName: error instanceof Error ? error.name : "UnknownError",
|
99
|
+
errorStack: error instanceof Error ? error.stack : undefined,
|
100
|
+
storageType: process.env.STORAGE_TYPE || "memory",
|
101
|
+
memoryConfig: {
|
102
|
+
enabled: config?.conversationMemory?.enabled,
|
103
|
+
maxSessions: config?.conversationMemory?.maxSessions,
|
104
|
+
maxTurnsPerSession: config?.conversationMemory?.maxTurnsPerSession,
|
105
|
+
},
|
106
|
+
redisConfig: {
|
107
|
+
host: process.env.REDIS_HOST || "(not set)",
|
108
|
+
port: process.env.REDIS_PORT || "(not set)",
|
109
|
+
hasPassword: !!process.env.REDIS_PASSWORD,
|
110
|
+
keyPrefix: process.env.REDIS_KEY_PREFIX || "(not set)",
|
111
|
+
},
|
112
|
+
});
|
113
|
+
// Log additional diagnostics for redis errors
|
114
|
+
if (process.env.STORAGE_TYPE === "redis") {
|
115
|
+
logger.error("[conversationMemoryInitializer] Redis configuration error details", {
|
116
|
+
REDIS_HOST: process.env.REDIS_HOST || "(not set)",
|
117
|
+
REDIS_PORT: process.env.REDIS_PORT || "(not set)",
|
118
|
+
REDIS_PASSWORD: process.env.REDIS_PASSWORD ? "******" : "(not set)",
|
119
|
+
REDIS_DB: process.env.REDIS_DB || "(not set)",
|
120
|
+
REDIS_KEY_PREFIX: process.env.REDIS_KEY_PREFIX || "(not set)",
|
121
|
+
REDIS_TTL: process.env.REDIS_TTL || "(not set)",
|
122
|
+
errorMessage: error instanceof Error ? error.message : String(error),
|
123
|
+
});
|
124
|
+
}
|
125
|
+
throw error;
|
126
|
+
}
|
127
|
+
}
|
@@ -2,7 +2,7 @@
|
|
2
2
|
* Conversation Memory Manager for NeuroLink
|
3
3
|
* Handles in-memory conversation storage, session management, and context injection
|
4
4
|
*/
|
5
|
-
import type { ConversationMemoryConfig, SessionMemory, ConversationMemoryStats, ChatMessage } from "../types/
|
5
|
+
import type { ConversationMemoryConfig, SessionMemory, ConversationMemoryStats, ChatMessage } from "../types/conversation.js";
|
6
6
|
export declare class ConversationMemoryManager {
|
7
7
|
private sessions;
|
8
8
|
config: ConversationMemoryConfig;
|
@@ -20,8 +20,9 @@ export declare class ConversationMemoryManager {
|
|
20
20
|
/**
|
21
21
|
* Build context messages for AI prompt injection (ULTRA-OPTIMIZED)
|
22
22
|
* Returns pre-stored message array with zero conversion overhead
|
23
|
+
* Now consistently async to match Redis implementation
|
23
24
|
*/
|
24
|
-
buildContextMessages(sessionId: string): ChatMessage[]
|
25
|
+
buildContextMessages(sessionId: string): Promise<ChatMessage[]>;
|
25
26
|
getSession(sessionId: string): SessionMemory | undefined;
|
26
27
|
createSummarySystemMessage(content: string): ChatMessage;
|
27
28
|
private _summarizeSession;
|
@@ -2,8 +2,8 @@
|
|
2
2
|
* Conversation Memory Manager for NeuroLink
|
3
3
|
* Handles in-memory conversation storage, session management, and context injection
|
4
4
|
*/
|
5
|
-
import { ConversationMemoryError } from "../types/
|
6
|
-
import { DEFAULT_MAX_TURNS_PER_SESSION, DEFAULT_MAX_SESSIONS, MESSAGES_PER_TURN, } from "../config/
|
5
|
+
import { ConversationMemoryError } from "../types/conversation.js";
|
6
|
+
import { DEFAULT_MAX_TURNS_PER_SESSION, DEFAULT_MAX_SESSIONS, MESSAGES_PER_TURN, } from "../config/conversationMemory.js";
|
7
7
|
import { logger } from "../utils/logger.js";
|
8
8
|
import { NeuroLink } from "../neurolink.js";
|
9
9
|
export class ConversationMemoryManager {
|
@@ -74,8 +74,9 @@ export class ConversationMemoryManager {
|
|
74
74
|
/**
|
75
75
|
* Build context messages for AI prompt injection (ULTRA-OPTIMIZED)
|
76
76
|
* Returns pre-stored message array with zero conversion overhead
|
77
|
+
* Now consistently async to match Redis implementation
|
77
78
|
*/
|
78
|
-
buildContextMessages(sessionId) {
|
79
|
+
async buildContextMessages(sessionId) {
|
79
80
|
const session = this.sessions.get(sessionId);
|
80
81
|
return session ? session.messages : [];
|
81
82
|
}
|
@@ -0,0 +1,73 @@
|
|
1
|
+
/**
|
2
|
+
* Redis Conversation Memory Manager for NeuroLink
|
3
|
+
* Redis-based implementation of conversation storage with same interface as ConversationMemoryManager
|
4
|
+
*/
|
5
|
+
import type { ConversationMemoryConfig, SessionMemory, ConversationMemoryStats, ChatMessage, RedisStorageConfig } from "../types/conversation.js";
|
6
|
+
/**
|
7
|
+
* Redis-based implementation of the ConversationMemoryManager
|
8
|
+
* Uses the same interface but stores data in Redis
|
9
|
+
*/
|
10
|
+
export declare class RedisConversationMemoryManager {
|
11
|
+
config: ConversationMemoryConfig;
|
12
|
+
private isInitialized;
|
13
|
+
private redisConfig;
|
14
|
+
private redisClient;
|
15
|
+
constructor(config: ConversationMemoryConfig, redisConfig?: RedisStorageConfig);
|
16
|
+
/**
|
17
|
+
* Initialize the memory manager with Redis connection
|
18
|
+
*/
|
19
|
+
initialize(): Promise<void>;
|
20
|
+
/**
|
21
|
+
* Store a conversation turn for a session
|
22
|
+
*/
|
23
|
+
storeConversationTurn(sessionId: string, userId: string | undefined, userMessage: string, aiResponse: string): Promise<void>;
|
24
|
+
/**
|
25
|
+
* Build context messages for AI prompt injection
|
26
|
+
*/
|
27
|
+
buildContextMessages(sessionId: string): Promise<ChatMessage[]>;
|
28
|
+
/**
|
29
|
+
* Get session data
|
30
|
+
*/
|
31
|
+
getSession(sessionId: string): Promise<SessionMemory | undefined>;
|
32
|
+
/**
|
33
|
+
* Create summary system message
|
34
|
+
*/
|
35
|
+
createSummarySystemMessage(content: string): ChatMessage;
|
36
|
+
/**
|
37
|
+
* Close Redis connection
|
38
|
+
*/
|
39
|
+
close(): Promise<void>;
|
40
|
+
/**
|
41
|
+
* Get statistics about conversation storage
|
42
|
+
*/
|
43
|
+
getStats(): Promise<ConversationMemoryStats>;
|
44
|
+
/**
|
45
|
+
* Clear a specific session
|
46
|
+
*/
|
47
|
+
clearSession(sessionId: string): Promise<boolean>;
|
48
|
+
/**
|
49
|
+
* Clear all sessions
|
50
|
+
*/
|
51
|
+
clearAllSessions(): Promise<void>;
|
52
|
+
/**
|
53
|
+
* Summarize messages for a session
|
54
|
+
*/
|
55
|
+
private _summarizeMessages;
|
56
|
+
/**
|
57
|
+
* Create summarization prompt
|
58
|
+
*/
|
59
|
+
private _createSummarizationPrompt;
|
60
|
+
/**
|
61
|
+
* Ensure Redis client is initialized
|
62
|
+
*/
|
63
|
+
private ensureInitialized;
|
64
|
+
/**
|
65
|
+
* Enforce session limit
|
66
|
+
*
|
67
|
+
* NOTE: This function is maintained only for backward compatibility with the
|
68
|
+
* in-memory implementation. In Redis, we do not actually delete sessions based on
|
69
|
+
* a global limit, as this could cause race conditions in a distributed environment.
|
70
|
+
* Each session is managed independently with its own TTL.
|
71
|
+
*/
|
72
|
+
private enforceSessionLimit;
|
73
|
+
}
|