@multiplayer-app/ai-agent-node 0.1.0-beta.4 → 0.1.0-beta.41
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/dist/cjs/config.cjs +88 -38
- package/dist/cjs/config.cjs.map +1 -1
- package/dist/cjs/config.d.ts +57 -23
- package/dist/cjs/config.d.ts.map +1 -1
- package/dist/cjs/helpers/AIHelper.cjs +160 -101
- package/dist/cjs/helpers/AIHelper.cjs.map +1 -1
- package/dist/cjs/helpers/AIHelper.d.ts +21 -13
- package/dist/cjs/helpers/AIHelper.d.ts.map +1 -1
- package/dist/cjs/helpers/AIHelper.test.cjs +22 -15
- package/dist/cjs/helpers/AIHelper.test.cjs.map +1 -1
- package/dist/cjs/helpers/ConfigHelper.cjs +19 -11
- package/dist/cjs/helpers/ConfigHelper.cjs.map +1 -1
- package/dist/cjs/helpers/ConfigHelper.d.ts.map +1 -1
- package/dist/cjs/helpers/ContextLimiter.cjs +2 -3
- package/dist/cjs/helpers/ContextLimiter.cjs.map +1 -1
- package/dist/cjs/helpers/FileHelper.cjs +132 -154
- package/dist/cjs/helpers/FileHelper.cjs.map +1 -1
- package/dist/cjs/helpers/FileHelper.d.ts +19 -25
- package/dist/cjs/helpers/FileHelper.d.ts.map +1 -1
- package/dist/cjs/helpers/index.cjs +4 -5
- package/dist/cjs/helpers/index.cjs.map +1 -1
- package/dist/cjs/helpers/index.d.ts +4 -5
- package/dist/cjs/helpers/index.d.ts.map +1 -1
- package/dist/cjs/index.cjs +118 -31
- package/dist/cjs/index.cjs.map +1 -1
- package/dist/cjs/index.d.ts +42 -13
- package/dist/cjs/index.d.ts.map +1 -1
- package/dist/cjs/libs/index.cjs +2 -3
- package/dist/cjs/libs/index.cjs.map +1 -1
- package/dist/cjs/libs/index.d.ts +2 -3
- package/dist/cjs/libs/index.d.ts.map +1 -1
- package/dist/cjs/libs/logger/index.cjs +5 -5
- package/dist/cjs/libs/logger/index.cjs.map +1 -1
- package/dist/cjs/libs/logger/index.d.ts +2 -2
- package/dist/cjs/libs/logger/index.d.ts.map +1 -1
- package/dist/cjs/libs/logger/kafkajs-logger-creator.cjs +2 -2
- package/dist/cjs/libs/logger/kafkajs-logger-creator.cjs.map +1 -1
- package/dist/cjs/libs/s3/index.cjs +3 -39
- package/dist/cjs/libs/s3/index.cjs.map +1 -1
- package/dist/cjs/libs/s3/index.d.ts +1 -2
- package/dist/cjs/libs/s3/index.d.ts.map +1 -1
- package/dist/cjs/libs/s3/s3.lib.cjs +174 -191
- package/dist/cjs/libs/s3/s3.lib.cjs.map +1 -1
- package/dist/cjs/libs/s3/s3.lib.d.ts +29 -22
- package/dist/cjs/libs/s3/s3.lib.d.ts.map +1 -1
- package/dist/cjs/processors/ActivityProcessor.cjs +36 -0
- package/dist/cjs/processors/ActivityProcessor.cjs.map +1 -0
- package/dist/cjs/processors/ActivityProcessor.d.ts +27 -0
- package/dist/cjs/processors/ActivityProcessor.d.ts.map +1 -0
- package/dist/cjs/processors/ActivityProcessor.test.cjs +84 -0
- package/dist/cjs/processors/ActivityProcessor.test.cjs.map +1 -0
- package/dist/cjs/processors/ActivityProcessor.test.d.ts +2 -0
- package/dist/cjs/processors/ActivityProcessor.test.d.ts.map +1 -0
- package/dist/cjs/processors/ChatProcessor.cjs +335 -163
- package/dist/cjs/processors/ChatProcessor.cjs.map +1 -1
- package/dist/cjs/processors/ChatProcessor.d.ts +56 -9
- package/dist/cjs/processors/ChatProcessor.d.ts.map +1 -1
- package/dist/cjs/processors/ChatProcessor.test.cjs +450 -0
- package/dist/cjs/processors/ChatProcessor.test.cjs.map +1 -0
- package/dist/cjs/processors/ChatProcessor.test.d.ts +2 -0
- package/dist/cjs/processors/ChatProcessor.test.d.ts.map +1 -0
- package/dist/cjs/processors/ModelsProcessor.cjs +2 -2
- package/dist/cjs/processors/ModelsProcessor.cjs.map +1 -1
- package/dist/cjs/processors/index.cjs +3 -2
- package/dist/cjs/processors/index.cjs.map +1 -1
- package/dist/cjs/processors/index.d.ts +3 -2
- package/dist/cjs/processors/index.d.ts.map +1 -1
- package/dist/cjs/services/AIService.cjs +50 -31
- package/dist/cjs/services/AIService.cjs.map +1 -1
- package/dist/cjs/services/AIService.d.ts +16 -5
- package/dist/cjs/services/AIService.d.ts.map +1 -1
- package/dist/cjs/services/InternalEventsHandler.cjs +5 -5
- package/dist/cjs/services/InternalEventsHandler.cjs.map +1 -1
- package/dist/cjs/services/InternalEventsHandler.d.ts +3 -1
- package/dist/cjs/services/InternalEventsHandler.d.ts.map +1 -1
- package/dist/cjs/services/ModelFetcher.cjs +9 -15
- package/dist/cjs/services/ModelFetcher.cjs.map +1 -1
- package/dist/cjs/services/ModelFetcher.d.ts +2 -7
- package/dist/cjs/services/ModelFetcher.d.ts.map +1 -1
- package/dist/cjs/services/RedisService.cjs +35 -32
- package/dist/cjs/services/RedisService.cjs.map +1 -1
- package/dist/cjs/services/RedisService.d.ts +5 -2
- package/dist/cjs/services/RedisService.d.ts.map +1 -1
- package/dist/cjs/services/SocketService.cjs +19 -20
- package/dist/cjs/services/SocketService.cjs.map +1 -1
- package/dist/cjs/services/SocketService.d.ts +9 -6
- package/dist/cjs/services/SocketService.d.ts.map +1 -1
- package/dist/cjs/services/index.cjs +5 -6
- package/dist/cjs/services/index.cjs.map +1 -1
- package/dist/cjs/services/index.d.ts +5 -6
- package/dist/cjs/services/index.d.ts.map +1 -1
- package/dist/cjs/store/AgentStore.cjs +4 -5
- package/dist/cjs/store/AgentStore.cjs.map +1 -1
- package/dist/cjs/store/AgentStore.d.ts +2 -1
- package/dist/cjs/store/AgentStore.d.ts.map +1 -1
- package/dist/cjs/store/ArtifactStore.cjs +2 -4
- package/dist/cjs/store/ArtifactStore.cjs.map +1 -1
- package/dist/cjs/store/ConfigStore.cjs +8 -11
- package/dist/cjs/store/ConfigStore.cjs.map +1 -1
- package/dist/cjs/store/ConfigStore.d.ts.map +1 -1
- package/dist/cjs/store/ConfigStore.test.cjs +2 -2
- package/dist/cjs/store/ConfigStore.test.cjs.map +1 -1
- package/dist/cjs/store/ModelStore.cjs +4 -4
- package/dist/cjs/store/ModelStore.cjs.map +1 -1
- package/dist/cjs/store/ModelStore.test.cjs +17 -17
- package/dist/cjs/store/ModelStore.test.cjs.map +1 -1
- package/dist/cjs/store/index.cjs +4 -4
- package/dist/cjs/store/index.cjs.map +1 -1
- package/dist/cjs/store/index.d.ts +4 -4
- package/dist/cjs/store/index.d.ts.map +1 -1
- package/dist/cjs/tools/proposeFormValuesTool.d.ts +2 -2
- package/dist/cjs/tsconfig.cjs.tsbuildinfo +1 -0
- package/dist/esm/config.d.ts +57 -23
- package/dist/esm/config.d.ts.map +1 -1
- package/dist/esm/config.js +88 -36
- package/dist/esm/config.js.map +1 -1
- package/dist/esm/helpers/AIHelper.d.ts +21 -13
- package/dist/esm/helpers/AIHelper.d.ts.map +1 -1
- package/dist/esm/helpers/AIHelper.js +159 -98
- package/dist/esm/helpers/AIHelper.js.map +1 -1
- package/dist/esm/helpers/AIHelper.test.js +21 -14
- package/dist/esm/helpers/AIHelper.test.js.map +1 -1
- package/dist/esm/helpers/ConfigHelper.d.ts.map +1 -1
- package/dist/esm/helpers/ConfigHelper.js +17 -9
- package/dist/esm/helpers/ConfigHelper.js.map +1 -1
- package/dist/esm/helpers/ContextLimiter.js +2 -3
- package/dist/esm/helpers/ContextLimiter.js.map +1 -1
- package/dist/esm/helpers/FileHelper.d.ts +19 -25
- package/dist/esm/helpers/FileHelper.d.ts.map +1 -1
- package/dist/esm/helpers/FileHelper.js +132 -149
- package/dist/esm/helpers/FileHelper.js.map +1 -1
- package/dist/esm/helpers/index.d.ts +4 -5
- package/dist/esm/helpers/index.d.ts.map +1 -1
- package/dist/esm/helpers/index.js +4 -5
- package/dist/esm/helpers/index.js.map +1 -1
- package/dist/esm/index.d.ts +42 -13
- package/dist/esm/index.d.ts.map +1 -1
- package/dist/esm/index.js +87 -12
- package/dist/esm/index.js.map +1 -1
- package/dist/esm/libs/index.d.ts +2 -3
- package/dist/esm/libs/index.d.ts.map +1 -1
- package/dist/esm/libs/index.js +2 -3
- package/dist/esm/libs/index.js.map +1 -1
- package/dist/esm/libs/logger/index.d.ts +2 -2
- package/dist/esm/libs/logger/index.d.ts.map +1 -1
- package/dist/esm/libs/logger/index.js +2 -2
- package/dist/esm/libs/logger/index.js.map +1 -1
- package/dist/esm/libs/logger/kafkajs-logger-creator.js +1 -1
- package/dist/esm/libs/logger/kafkajs-logger-creator.js.map +1 -1
- package/dist/esm/libs/s3/index.d.ts +1 -2
- package/dist/esm/libs/s3/index.d.ts.map +1 -1
- package/dist/esm/libs/s3/index.js +1 -2
- package/dist/esm/libs/s3/index.js.map +1 -1
- package/dist/esm/libs/s3/s3.lib.d.ts +29 -22
- package/dist/esm/libs/s3/s3.lib.d.ts.map +1 -1
- package/dist/esm/libs/s3/s3.lib.js +178 -177
- package/dist/esm/libs/s3/s3.lib.js.map +1 -1
- package/dist/esm/processors/ActivityProcessor.d.ts +27 -0
- package/dist/esm/processors/ActivityProcessor.d.ts.map +1 -0
- package/dist/esm/processors/ActivityProcessor.js +33 -0
- package/dist/esm/processors/ActivityProcessor.js.map +1 -0
- package/dist/esm/processors/ActivityProcessor.test.d.ts +2 -0
- package/dist/esm/processors/ActivityProcessor.test.d.ts.map +1 -0
- package/dist/esm/processors/ActivityProcessor.test.js +82 -0
- package/dist/esm/processors/ActivityProcessor.test.js.map +1 -0
- package/dist/esm/processors/ChatProcessor.d.ts +56 -9
- package/dist/esm/processors/ChatProcessor.d.ts.map +1 -1
- package/dist/esm/processors/ChatProcessor.js +334 -153
- package/dist/esm/processors/ChatProcessor.js.map +1 -1
- package/dist/esm/processors/ChatProcessor.test.d.ts +2 -0
- package/dist/esm/processors/ChatProcessor.test.d.ts.map +1 -0
- package/dist/esm/processors/ChatProcessor.test.js +448 -0
- package/dist/esm/processors/ChatProcessor.test.js.map +1 -0
- package/dist/esm/processors/ModelsProcessor.js +1 -1
- package/dist/esm/processors/ModelsProcessor.js.map +1 -1
- package/dist/esm/processors/index.d.ts +3 -2
- package/dist/esm/processors/index.d.ts.map +1 -1
- package/dist/esm/processors/index.js +3 -2
- package/dist/esm/processors/index.js.map +1 -1
- package/dist/esm/services/AIService.d.ts +16 -5
- package/dist/esm/services/AIService.d.ts.map +1 -1
- package/dist/esm/services/AIService.js +53 -32
- package/dist/esm/services/AIService.js.map +1 -1
- package/dist/esm/services/InternalEventsHandler.d.ts +3 -1
- package/dist/esm/services/InternalEventsHandler.d.ts.map +1 -1
- package/dist/esm/services/InternalEventsHandler.js +10 -8
- package/dist/esm/services/InternalEventsHandler.js.map +1 -1
- package/dist/esm/services/ModelFetcher.d.ts +2 -7
- package/dist/esm/services/ModelFetcher.d.ts.map +1 -1
- package/dist/esm/services/ModelFetcher.js +4 -9
- package/dist/esm/services/ModelFetcher.js.map +1 -1
- package/dist/esm/services/RedisService.d.ts +5 -2
- package/dist/esm/services/RedisService.d.ts.map +1 -1
- package/dist/esm/services/RedisService.js +25 -21
- package/dist/esm/services/RedisService.js.map +1 -1
- package/dist/esm/services/SocketService.d.ts +9 -6
- package/dist/esm/services/SocketService.d.ts.map +1 -1
- package/dist/esm/services/SocketService.js +15 -13
- package/dist/esm/services/SocketService.js.map +1 -1
- package/dist/esm/services/index.d.ts +5 -6
- package/dist/esm/services/index.d.ts.map +1 -1
- package/dist/esm/services/index.js +5 -6
- package/dist/esm/services/index.js.map +1 -1
- package/dist/esm/store/AgentStore.d.ts +2 -1
- package/dist/esm/store/AgentStore.d.ts.map +1 -1
- package/dist/esm/store/AgentStore.js +5 -5
- package/dist/esm/store/AgentStore.js.map +1 -1
- package/dist/esm/store/ArtifactStore.js +3 -7
- package/dist/esm/store/ArtifactStore.js.map +1 -1
- package/dist/esm/store/ConfigStore.d.ts.map +1 -1
- package/dist/esm/store/ConfigStore.js +16 -19
- package/dist/esm/store/ConfigStore.js.map +1 -1
- package/dist/esm/store/ConfigStore.test.js +1 -1
- package/dist/esm/store/ConfigStore.test.js.map +1 -1
- package/dist/esm/store/ModelStore.js +6 -6
- package/dist/esm/store/ModelStore.js.map +1 -1
- package/dist/esm/store/ModelStore.test.js +12 -12
- package/dist/esm/store/ModelStore.test.js.map +1 -1
- package/dist/esm/store/index.d.ts +4 -4
- package/dist/esm/store/index.d.ts.map +1 -1
- package/dist/esm/store/index.js +4 -4
- package/dist/esm/store/index.js.map +1 -1
- package/dist/esm/tools/proposeFormValuesTool.d.ts +2 -2
- package/dist/esm/tsconfig.esm.tsbuildinfo +1 -0
- package/package.json +15 -15
- package/dist/cjs/helpers/SetupHelper.cjs +0 -37
- package/dist/cjs/helpers/SetupHelper.cjs.map +0 -1
- package/dist/cjs/helpers/SetupHelper.d.ts +0 -5
- package/dist/cjs/helpers/SetupHelper.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/config.cjs +0 -8
- package/dist/cjs/libs/kafka/config.cjs.map +0 -1
- package/dist/cjs/libs/kafka/config.d.ts +0 -5
- package/dist/cjs/libs/kafka/config.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/consumer.cjs +0 -133
- package/dist/cjs/libs/kafka/consumer.cjs.map +0 -1
- package/dist/cjs/libs/kafka/consumer.d.ts +0 -16
- package/dist/cjs/libs/kafka/consumer.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/index.cjs +0 -19
- package/dist/cjs/libs/kafka/index.cjs.map +0 -1
- package/dist/cjs/libs/kafka/index.d.ts +0 -3
- package/dist/cjs/libs/kafka/index.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/kafka.cjs +0 -27
- package/dist/cjs/libs/kafka/kafka.cjs.map +0 -1
- package/dist/cjs/libs/kafka/kafka.d.ts +0 -3
- package/dist/cjs/libs/kafka/kafka.d.ts.map +0 -1
- package/dist/cjs/libs/kafka/producer.cjs +0 -48
- package/dist/cjs/libs/kafka/producer.cjs.map +0 -1
- package/dist/cjs/libs/kafka/producer.d.ts +0 -11
- package/dist/cjs/libs/kafka/producer.d.ts.map +0 -1
- package/dist/cjs/libs/logger/config.cjs +0 -9
- package/dist/cjs/libs/logger/config.cjs.map +0 -1
- package/dist/cjs/libs/logger/config.d.ts +0 -5
- package/dist/cjs/libs/logger/config.d.ts.map +0 -1
- package/dist/cjs/libs/s3/config.cjs +0 -10
- package/dist/cjs/libs/s3/config.cjs.map +0 -1
- package/dist/cjs/libs/s3/config.d.ts +0 -7
- package/dist/cjs/libs/s3/config.d.ts.map +0 -1
- package/dist/cjs/services/KafkaService.cjs +0 -123
- package/dist/cjs/services/KafkaService.cjs.map +0 -1
- package/dist/cjs/services/KafkaService.d.ts +0 -35
- package/dist/cjs/services/KafkaService.d.ts.map +0 -1
- package/dist/esm/helpers/SetupHelper.d.ts +0 -5
- package/dist/esm/helpers/SetupHelper.d.ts.map +0 -1
- package/dist/esm/helpers/SetupHelper.js +0 -32
- package/dist/esm/helpers/SetupHelper.js.map +0 -1
- package/dist/esm/libs/kafka/config.d.ts +0 -5
- package/dist/esm/libs/kafka/config.d.ts.map +0 -1
- package/dist/esm/libs/kafka/config.js +0 -5
- package/dist/esm/libs/kafka/config.js.map +0 -1
- package/dist/esm/libs/kafka/consumer.d.ts +0 -16
- package/dist/esm/libs/kafka/consumer.d.ts.map +0 -1
- package/dist/esm/libs/kafka/consumer.js +0 -126
- package/dist/esm/libs/kafka/consumer.js.map +0 -1
- package/dist/esm/libs/kafka/index.d.ts +0 -3
- package/dist/esm/libs/kafka/index.d.ts.map +0 -1
- package/dist/esm/libs/kafka/index.js +0 -3
- package/dist/esm/libs/kafka/index.js.map +0 -1
- package/dist/esm/libs/kafka/kafka.d.ts +0 -3
- package/dist/esm/libs/kafka/kafka.d.ts.map +0 -1
- package/dist/esm/libs/kafka/kafka.js +0 -24
- package/dist/esm/libs/kafka/kafka.js.map +0 -1
- package/dist/esm/libs/kafka/producer.d.ts +0 -11
- package/dist/esm/libs/kafka/producer.d.ts.map +0 -1
- package/dist/esm/libs/kafka/producer.js +0 -44
- package/dist/esm/libs/kafka/producer.js.map +0 -1
- package/dist/esm/libs/logger/config.d.ts +0 -5
- package/dist/esm/libs/logger/config.d.ts.map +0 -1
- package/dist/esm/libs/logger/config.js +0 -6
- package/dist/esm/libs/logger/config.js.map +0 -1
- package/dist/esm/libs/s3/config.d.ts +0 -7
- package/dist/esm/libs/s3/config.d.ts.map +0 -1
- package/dist/esm/libs/s3/config.js +0 -7
- package/dist/esm/libs/s3/config.js.map +0 -1
- package/dist/esm/services/KafkaService.d.ts +0 -35
- package/dist/esm/services/KafkaService.d.ts.map +0 -1
- package/dist/esm/services/KafkaService.js +0 -120
- package/dist/esm/services/KafkaService.js.map +0 -1
- package/dist/tsconfig.cjs.tsbuildinfo +0 -1
- package/dist/tsconfig.esm.tsbuildinfo +0 -1
|
@@ -8,20 +8,24 @@ const zod_1 = require("zod");
|
|
|
8
8
|
const ai_1 = require("ai");
|
|
9
9
|
const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
|
|
10
10
|
const ai_sdk_provider_1 = require("@openrouter/ai-sdk-provider");
|
|
11
|
-
const
|
|
12
|
-
const index_js_1 = require("../libs/logger/index.js");
|
|
11
|
+
const logger_1 = require("../libs/logger/index.cjs");
|
|
13
12
|
class AIService {
|
|
14
|
-
constructor(modelStore) {
|
|
13
|
+
constructor(modelStore, config) {
|
|
15
14
|
this.modelStore = modelStore;
|
|
15
|
+
this.config = config;
|
|
16
16
|
}
|
|
17
|
-
|
|
18
|
-
const selectedModel = modelId ||
|
|
17
|
+
getLanguageModelId(modelId) {
|
|
18
|
+
const selectedModel = modelId || this.config.defaultModel || this.modelStore.getModels()[0].id;
|
|
19
19
|
const model = selectedModel ?
|
|
20
20
|
this.modelStore.getModelById(selectedModel) :
|
|
21
21
|
this.modelStore.getModels()[0]; //get first model if no model is selected
|
|
22
22
|
if (!model) {
|
|
23
23
|
throw new Error(`Model with id "${selectedModel}" is not available. Ensure you have appropriate key in the environment variables.`);
|
|
24
24
|
}
|
|
25
|
+
return model;
|
|
26
|
+
}
|
|
27
|
+
getLanguageModel(modelId, websearchEnabled) {
|
|
28
|
+
const model = this.getLanguageModelId(modelId);
|
|
25
29
|
return this.createLanguageModel(model, websearchEnabled);
|
|
26
30
|
}
|
|
27
31
|
/**
|
|
@@ -31,21 +35,21 @@ class AIService {
|
|
|
31
35
|
const provider = model.provider || 'openai'; // Default to openai if provider not specified
|
|
32
36
|
switch (provider) {
|
|
33
37
|
case 'openai':
|
|
34
|
-
return (0, openai_1.
|
|
38
|
+
return (0, openai_1.createOpenAI)({ apiKey: this.config.openaiApiKey })(model.id);
|
|
35
39
|
case 'anthropic':
|
|
36
|
-
return (0, anthropic_1.
|
|
40
|
+
return (0, anthropic_1.createAnthropic)({ apiKey: this.config.anthropicApiKey })(model.id);
|
|
37
41
|
case 'google':
|
|
38
|
-
return (0, google_1.
|
|
42
|
+
return (0, google_1.createGoogleGenerativeAI)({ apiKey: this.config.googleApiKey })(model.id);
|
|
39
43
|
case 'openrouter':
|
|
40
44
|
return (0, ai_sdk_provider_1.createOpenRouter)({
|
|
41
|
-
apiKey:
|
|
45
|
+
apiKey: this.config.openrouterApiKey,
|
|
42
46
|
})(model.id, { plugins: websearchEnabled ? [{ id: "web", engine: undefined, max_results: 2 }] : [] });
|
|
43
47
|
default:
|
|
44
48
|
throw new Error(`Unsupported provider: ${provider}`);
|
|
45
49
|
}
|
|
46
50
|
}
|
|
47
51
|
getWebSearchTool(modelId) {
|
|
48
|
-
const selectedModel = modelId ||
|
|
52
|
+
const selectedModel = modelId || this.config.defaultModel || this.modelStore.getModels()[0].id;
|
|
49
53
|
const model = selectedModel ?
|
|
50
54
|
this.modelStore.getModelById(selectedModel) :
|
|
51
55
|
this.modelStore.getModels()[0]; //get first model if no model is selected
|
|
@@ -59,7 +63,7 @@ class AIService {
|
|
|
59
63
|
return anthropic_1.anthropic.tools.webSearch_20250305();
|
|
60
64
|
case 'google':
|
|
61
65
|
return google_1.google.tools.googleSearch({});
|
|
62
|
-
case 'openrouter':
|
|
66
|
+
case 'openrouter':
|
|
63
67
|
return undefined;
|
|
64
68
|
default:
|
|
65
69
|
return undefined;
|
|
@@ -88,8 +92,7 @@ class AIService {
|
|
|
88
92
|
}
|
|
89
93
|
if (data.headersToPass) {
|
|
90
94
|
params.headers = data.headersToPass.reduce((acc, headerName) => {
|
|
91
|
-
|
|
92
|
-
acc[headerName] = (_b = (_a = context === null || context === void 0 ? void 0 : context.headers) === null || _a === void 0 ? void 0 : _a[headerName]) !== null && _b !== void 0 ? _b : '';
|
|
95
|
+
acc[headerName] = context?.headers?.[headerName] ?? '';
|
|
93
96
|
return acc;
|
|
94
97
|
}, {});
|
|
95
98
|
}
|
|
@@ -137,12 +140,15 @@ class AIService {
|
|
|
137
140
|
// });
|
|
138
141
|
// }
|
|
139
142
|
getAgentOptions(config, modelId, context) {
|
|
143
|
+
let websearchEnabled = false;
|
|
140
144
|
const tools = config.tools.reduce((acc, toolConfig) => {
|
|
141
|
-
var _a;
|
|
142
145
|
const tool = this.getTool(toolConfig, modelId, context);
|
|
143
|
-
if (tool &&
|
|
146
|
+
if (tool && toolConfig.data?.title) {
|
|
144
147
|
acc[toolConfig.data.title] = tool;
|
|
145
148
|
}
|
|
149
|
+
if (toolConfig.type === ai_agent_types_1.AgentToolType.WEB_SEARCH) {
|
|
150
|
+
websearchEnabled = true;
|
|
151
|
+
}
|
|
146
152
|
return acc;
|
|
147
153
|
}, {});
|
|
148
154
|
return {
|
|
@@ -151,44 +157,57 @@ class AIService {
|
|
|
151
157
|
system: config.systemPrompt,
|
|
152
158
|
toolChoice: config.toolChoice,
|
|
153
159
|
name: config.name,
|
|
160
|
+
stopWhen: config.stopWhen,
|
|
161
|
+
prepareStep: config.prepareStep,
|
|
162
|
+
maxOutputTokens: config.maxOutputTokens,
|
|
163
|
+
temperature: config.temperature,
|
|
164
|
+
topP: config.topP,
|
|
165
|
+
topK: config.topK,
|
|
166
|
+
presencePenalty: config.presencePenalty,
|
|
167
|
+
frequencyPenalty: config.frequencyPenalty,
|
|
168
|
+
stopSequences: config.stopSequences,
|
|
169
|
+
seed: config.seed,
|
|
170
|
+
websearchEnabled
|
|
154
171
|
};
|
|
155
172
|
}
|
|
156
173
|
async generateText(options) {
|
|
157
|
-
var _a;
|
|
158
|
-
const websearchEnabled = !!((_a = options.tools) === null || _a === void 0 ? void 0 : _a[ai_agent_types_1.AgentToolType.WEB_SEARCH]);
|
|
159
174
|
try {
|
|
175
|
+
const { stopWhen, model, websearchEnabled, ...restOptions } = options;
|
|
160
176
|
const { text } = await (0, ai_1.generateText)({
|
|
161
|
-
...
|
|
162
|
-
model: this.getLanguageModel(
|
|
177
|
+
...restOptions,
|
|
178
|
+
model: this.getLanguageModel(model, !!websearchEnabled),
|
|
179
|
+
...(stopWhen !== undefined ? { stopWhen } : {}),
|
|
163
180
|
});
|
|
164
181
|
return text;
|
|
165
182
|
}
|
|
166
183
|
catch (error) {
|
|
167
|
-
|
|
184
|
+
logger_1.logger.error(error);
|
|
168
185
|
throw new Error(`Failed to generate text: ${error instanceof Error ? error.message : 'Unknown error'}`);
|
|
169
186
|
}
|
|
170
187
|
}
|
|
171
188
|
streamText(options) {
|
|
172
|
-
|
|
173
|
-
const { onError, onChunk, onFinish, onStepFinish } = options;
|
|
174
|
-
const websearchEnabled = !!((_a = options.tools) === null || _a === void 0 ? void 0 : _a[ai_agent_types_1.AgentToolType.WEB_SEARCH]);
|
|
189
|
+
const { onError, onChunk, onFinish, onStepFinish, onAbort, signal, websearchEnabled, model, ...restOptions } = options;
|
|
175
190
|
try {
|
|
176
191
|
return (0, ai_1.streamText)({
|
|
177
|
-
...
|
|
178
|
-
|
|
192
|
+
...restOptions,
|
|
193
|
+
abortSignal: signal,
|
|
194
|
+
model: this.getLanguageModel(model, !!websearchEnabled),
|
|
179
195
|
onError({ error }) {
|
|
180
|
-
|
|
181
|
-
onError
|
|
196
|
+
logger_1.logger.error(error);
|
|
197
|
+
onError?.(error);
|
|
182
198
|
},
|
|
183
199
|
onChunk(chunk) {
|
|
184
|
-
onChunk
|
|
200
|
+
onChunk?.(chunk);
|
|
185
201
|
},
|
|
186
202
|
onFinish({ text, finishReason, usage, response, steps, totalUsage }) {
|
|
187
|
-
onFinish
|
|
203
|
+
onFinish?.({ text, finishReason, usage, response, steps, totalUsage });
|
|
188
204
|
},
|
|
189
205
|
onStepFinish(step) {
|
|
190
|
-
onStepFinish
|
|
191
|
-
}
|
|
206
|
+
onStepFinish?.(step);
|
|
207
|
+
},
|
|
208
|
+
onAbort(event) {
|
|
209
|
+
onAbort?.(event);
|
|
210
|
+
},
|
|
192
211
|
});
|
|
193
212
|
}
|
|
194
213
|
catch (error) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AIService.js","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":";;;AAAA,
|
|
1
|
+
{"version":3,"file":"AIService.js","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":";;;AAAA,2CAAoD;AACpD,iDAA6D;AAC7D,2CAAgE;AAChE,6BAAsB;AACtB,2BAWY;AACZ,oEASyC;AAEzC,iEAA6D;AAC7D,2CAAsC;AAmCtC,MAAa,SAAS;IAGpB,YAAY,UAAsB,EAAE,MAAoB;QACtD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,kBAAkB,CAAC,OAA2B;QAC5C,MAAM,aAAa,GAAG,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/F,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC;YAC3B,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;YAC7C,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,yCAAyC;QAC3E,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,kBAAkB,aAAa,mFAAmF,CAAC,CAAC;QACtI,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gBAAgB,CAAC,OAA2B,EAAE,gBAAyB;QACrE,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC;IAC3D,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,KAAkB,EAAE,gBAAgB,GAAG,KAAK;QACtE,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,IAAI,QAAQ,CAAC,CAAC,8CAA8C;QAC3F,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,QAAQ;gBACX,OAAO,IAAA,qBAAY,EAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YACnE,KAAK,WAAW;gBACd,OAAO,IAAA,2BAAe,EAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,EAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YACzE,KAAK,QAAQ;gBACX,OAAO,IAAA,iCAAwB,EAAC,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YAC/E,KAAK,YAAY;gBACf,OAAO,IAAA,kCAAgB,EAAC;oBACtB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;iBACrC,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,gBAAgB,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAA6B,CAAC;YACpI;gBACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,OAAgB;QACvC,MAAM,aAAa,GAAG,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/F,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC;YAC3B,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;YAC7C,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,yCAAyC;QAC3E,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,kBAAkB,aAAa,mFAAmF,CAAC,CAAC;QACtI,CAAC;QACD,QAAQ,KAAK,CAAC,QAAQ,EAAE,CAAC;YACvB,KAAK,QAAQ;gBACX,OAAO,eAAM,CAAC,KAAK,CAAC,SAAS,EAAU,CAAA;YACzC,KAAK,WAAW;gBACd,OAAO,qBAAS,CAAC,KAAK,CAAC,kBAAkB,EAAU,CAAA;YACrD,KAAK,QAAQ;gBACX,OAAO,eAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAS,CAAA;YAC9C,KAAK,YAAY;gBACf,OAAO,SAAS,CAAA;YAClB;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,UAAwC,EAAE,OAAgB,EAAE,OAA2B;QACrG,IAAI,UAAU,CAAC,IAAI,KAAK,8BAAa,CAAC,QAAQ,EAAE,CAAC;YAC/C,MAAM,IAAI,GAAG,UAAU,CAAC,IAAoB,CAAC;YAC7C,OAAO,IAAA,SAAI,EAAuE;gBAChF,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;oBACpB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;oBAChD,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;iBACtE,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,KAAsE,EAAE,EAAE;oBACxF,IAAI,CAAC;wBACH,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;wBACnB,MAAM,MAAM,GAAgB,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAA;wBACnD,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;4BACd,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC3C,CAAC;wBACD,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;4BACrB,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,WAAqC,CAAC,CAAC,QAAQ,EAAE,CAAC;wBAChG,CAAC;wBACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;4BACvB,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE;gCAC7D,GAAG,CAAC,UAAU,CAAC,GAAG,OAAO,EAAE,OAAO,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;gCACvD,OAAO,GAAG,CAAC;4BACb,CAAC,EAAE,EAAE,CAAC,CAAA;wBACR,CAAC;wBAEC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;wBAC1C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;wBACrC,OAAO,MAAM,CAAC;oBAChB,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,OAAO,KAAK,CAAC;oBACf,CAAC;gBACH,CAAC;aACJ,CAAC,CAAA;QACJ,CAAC;QAED,IAAI,UAAU,CAAC,IAAI,KAAK,8BAAa,CAAC,UAAU,EAAE,CAAC;YACjD,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,UAAU,CAAC,IAAI,KAAK,8BAAa,CAAC,cAAc,EAAE,CAAC;YACrD,MAAM,IAAI,GAAG,UAAU,CAAC,IAAoC,CAAC;YAC7D,iFAAiF;YACjF,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;gBAE/C,OAAO,IAAA,SAAI,EAAC;oBACV,GAAG,IAAI;oBACP,OAAO,EAAE,KAAK,EAAE,KAAmB,EAAE,OAA8B,EAAE,EAAE;wBACrE,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;oBAC9C,CAAC;iBACF,CAAC,CAAC;YACL,CAAC;YACD,OAAO,IAAA,SAAI,EAAC,IAAY,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAGD,0FAA0F;IAC1F,sEAAsE;IACtE,sDAAsD;IACtD,2CAA2C;IAC3C,yCAAyC;IACzC,QAAQ;IACR,kBAAkB;IAClB,uBAAuB;IAEvB,uBAAuB;IACvB,oEAAoE;IACpE,mCAAmC;IACnC,aAAa;IACb,+FAA+F;IAC/F,QAAQ;IACR,IAAI;IAEJ,eAAe,CAAC,MAAsC,EAAE,OAAgB,EAAE,OAA2B;QACnG,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAC7B,MAAM,KAAK,GAAY,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE;YAC7D,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YACxD,IAAI,IAAI,IAAI,UAAU,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC;gBACnC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAA;YACnC,CAAC;YACD,IAAI,UAAU,CAAC,IAAI,KAAK,8BAAa,CAAC,UAAU,EAAE,CAAC;gBACjD,gBAAgB,GAAG,IAAI,CAAA;YACzB,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAa,CAAC,CAAC;QAClB,OAAO;YACL,KAAK,EAAE,OAAO,IAAI,MAAM,CAAC,YAAY;YACrC,KAAK;YACL,MAAM,EAAE,MAAM,CAAC,YAAY;YAC3B,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,eAAe,EAAE,MAAM,CAAC,eAAe;YACvC,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,eAAe,EAAE,MAAM,CAAC,eAAe;YACvC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,aAAa,EAAE,MAAM,CAAC,aAAa;YACnC,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,gBAAgB;SACjB,CAAC;IACJ,CAAC;IAGD,KAAK,CAAC,YAAY,CAAC,OAA4B;QAC7C,IAAI,CAAC;YACH,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,gBAAgB,EAAE,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC;YACtE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAA,iBAAY,EAAC;gBAClC,GAAG,WAAW;gBACd,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,gBAAgB,CAAC;gBACvD,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAChD,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QAC1G,CAAC;IACH,CAAC;IAED,UAAU,CAAC,OAA0B;QACnC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE,KAAK,EAAE,GAAG,WAAW,EAAE,GAAG,OAAO,CAAC;QAEvH,IAAI,CAAC;YACH,OAAO,IAAA,eAAU,EAAC;gBAChB,GAAG,WAAW;gBACd,WAAW,EAAE,MAAM;gBACnB,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,gBAAgB,CAAC;gBACvD,OAAO,CAAC,EAAE,KAAK,EAAE;oBACf,eAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACpB,OAAO,EAAE,CAAC,KAAc,CAAC,CAAC;gBAC5B,CAAC;gBACD,OAAO,CAAC,KAAK;oBACX,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;gBACnB,CAAC;gBACD,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE;oBACjE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBACzE,CAAC;gBACD,YAAY,CAAC,IAAI;oBACf,YAAY,EAAE,CAAC,IAAI,CAAC,CAAC;gBACvB,CAAC;gBACD,OAAO,CAAC,KAAK;oBACX,OAAO,EAAE,CAAC,KAAK,CAAC,CAAA;gBAClB,CAAC;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAAC,CAAC;QACxG,CAAC;IACH,CAAC;CACF;AA7ND,8BA6NC"}
|
|
@@ -1,16 +1,26 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import
|
|
3
|
-
import { ModelStore } from '../store/ModelStore
|
|
1
|
+
import { type LanguageModel, ModelMessage, type PrepareStepFunction, streamText, Tool, ToolSet, type StopCondition, StepResult } from 'ai';
|
|
2
|
+
import { AdditionalContext, AgentConfig, ModelOption, ToolChoice } from '@multiplayer-app/ai-agent-types';
|
|
3
|
+
import { ModelStore } from '../store/ModelStore';
|
|
4
|
+
import { Config } from '../config';
|
|
4
5
|
export interface GenerateTextOptions {
|
|
5
6
|
messages: ModelMessage[];
|
|
6
7
|
system?: string;
|
|
7
8
|
temperature?: number;
|
|
8
9
|
maxOutputTokens?: number;
|
|
9
10
|
topP?: number;
|
|
11
|
+
topK?: number;
|
|
12
|
+
presencePenalty?: number;
|
|
13
|
+
frequencyPenalty?: number;
|
|
14
|
+
stopSequences?: string[];
|
|
15
|
+
seed?: number;
|
|
10
16
|
toolChoice?: ToolChoice<ToolSet>;
|
|
11
17
|
tools?: ToolSet;
|
|
12
18
|
model?: string;
|
|
13
19
|
signal?: AbortSignal;
|
|
20
|
+
websearchEnabled?: boolean;
|
|
21
|
+
stopWhen?: StopCondition<ToolSet> | Array<StopCondition<ToolSet>>;
|
|
22
|
+
prepareStep?: PrepareStepFunction<ToolSet>;
|
|
23
|
+
onStepFinish?: (stepResult: StepResult<ToolSet>) => Promise<void> | void;
|
|
14
24
|
}
|
|
15
25
|
export interface StreamTextOptions extends GenerateTextOptions {
|
|
16
26
|
onChunk?: (chunk: any) => void;
|
|
@@ -24,14 +34,15 @@ export interface StreamTextOptions extends GenerateTextOptions {
|
|
|
24
34
|
totalUsage: any;
|
|
25
35
|
}) => void;
|
|
26
36
|
onAbort?: (data: any) => void;
|
|
27
|
-
onStepFinish?: (step: any) => void;
|
|
28
37
|
}
|
|
29
38
|
export interface LocalFunctionTool extends Omit<Tool, 'title'> {
|
|
30
39
|
title: string;
|
|
31
40
|
}
|
|
32
41
|
export declare class AIService {
|
|
33
42
|
private modelStore;
|
|
34
|
-
|
|
43
|
+
private config;
|
|
44
|
+
constructor(modelStore: ModelStore, config: Config["ai"]);
|
|
45
|
+
getLanguageModelId(modelId: string | undefined): ModelOption;
|
|
35
46
|
getLanguageModel(modelId: string | undefined, websearchEnabled: boolean): LanguageModel;
|
|
36
47
|
/**
|
|
37
48
|
* Create a language model instance based on the provider
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AIService.d.ts","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":"AAIA,OAAO,EAEL,
|
|
1
|
+
{"version":3,"file":"AIService.d.ts","sourceRoot":"","sources":["../../../src/services/AIService.ts"],"names":[],"mappings":"AAIA,OAAO,EAEL,KAAK,aAAa,EAClB,YAAY,EACZ,KAAK,mBAAmB,EACxB,UAAU,EAEV,IAAI,EAEJ,OAAO,EACP,KAAK,aAAa,EAAkC,UAAU,EAC/D,MAAM,IAAI,CAAC;AACZ,OAAO,EACL,iBAAiB,EAEjB,WAAW,EAIX,WAAW,EACX,UAAU,EACX,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAC,UAAU,EAAC,MAAM,qBAAqB,CAAC;AAG/C,OAAO,EAAC,MAAM,EAAC,MAAM,WAAW,CAAC;AAEjC,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IACjC,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,EAAE,aAAa,CAAC,OAAO,CAAC,GAAG,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;IAClE,WAAW,CAAC,EAAE,mBAAmB,CAAC,OAAO,CAAC,CAAC;IAC3C,YAAY,CAAC,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;CAC1E;AAED,MAAM,WAAW,iBAAkB,SAAQ,mBAAmB;IAC5D,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACjC,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,YAAY,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,GAAG,CAAC;QAAC,QAAQ,EAAE,GAAG,CAAC;QAAC,KAAK,EAAE,GAAG,CAAC;QAAC,UAAU,EAAE,GAAG,CAAA;KAAE,KAAK,IAAI,CAAC;IAC1H,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,IAAI,CAAC;CAC/B;AAED,MAAM,WAAW,iBAAkB,SAAQ,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC;IAC5D,KAAK,EAAE,MAAM,CAAC;CACf;AAED,qBAAa,SAAS;IACpB,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,MAAM,CAAe;gBACjB,UAAU,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;IAKxD,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,WAAW;IAW5D,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,EAAE,gBAAgB,EAAE,OAAO,GAAG,aAAa;IAKvF;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAkB3B,OAAO,CAAC,gBAAgB;IAsBxB,OAAO,CAAC,OAAO;IA4Ef,eAAe,CAAC,MAAM,EAAE,WAAW,CAAC,iBAAiB,CAAC,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,IAAI,CAAC,mBAAmB,EAAE,UAAU,CAAC,GAAG;QAAC,IAAI,EAAE,MAAM,CAAA;KAAC;IAiCxJ,YAAY,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC;IAgBjE,UAAU,CAAC,OAAO,EAAE,iBAAiB,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC;CA6BtE"}
|
|
@@ -1,14 +1,14 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.InternalEventsHandler = void 0;
|
|
4
|
-
const
|
|
5
|
-
const index_js_1 = require("../libs/logger/index.js");
|
|
4
|
+
const logger_1 = require("../libs/logger/index.cjs");
|
|
6
5
|
/**
|
|
7
6
|
* Lightweight Redis pub/sub helper to share events across service replicas.
|
|
8
7
|
*/
|
|
9
8
|
class InternalEventsHandler {
|
|
10
|
-
constructor(channel) {
|
|
9
|
+
constructor(channel, redisService) {
|
|
11
10
|
this.channel = channel;
|
|
11
|
+
this.redisService = redisService;
|
|
12
12
|
this.pubClient = null;
|
|
13
13
|
this.subClient = null;
|
|
14
14
|
this.initialized = false;
|
|
@@ -17,7 +17,7 @@ class InternalEventsHandler {
|
|
|
17
17
|
async ensureClients() {
|
|
18
18
|
if (this.initialized)
|
|
19
19
|
return;
|
|
20
|
-
const { pubClient, subClient } = await
|
|
20
|
+
const { pubClient, subClient } = await this.redisService.getPubSubClients(this.channel);
|
|
21
21
|
this.pubClient = pubClient;
|
|
22
22
|
this.subClient = subClient;
|
|
23
23
|
this.initialized = true;
|
|
@@ -51,7 +51,7 @@ class InternalEventsHandler {
|
|
|
51
51
|
await handler(parsed);
|
|
52
52
|
}
|
|
53
53
|
catch (error) {
|
|
54
|
-
|
|
54
|
+
logger_1.logger.error(`InternalEventsHandler failed to process message on ${this.channel}:`, error);
|
|
55
55
|
}
|
|
56
56
|
});
|
|
57
57
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"InternalEventsHandler.js","sourceRoot":"","sources":["../../../src/services/InternalEventsHandler.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"InternalEventsHandler.js","sourceRoot":"","sources":["../../../src/services/InternalEventsHandler.ts"],"names":[],"mappings":";;;AACA,2CAAwC;AAExC;;GAEG;AACH,MAAa,qBAAqB;IAChC,YACmB,OAAe,EACf,YAA0B;QAD1B,YAAO,GAAP,OAAO,CAAQ;QACf,iBAAY,GAAZ,YAAY,CAAc;QAGrC,cAAS,GAA6B,IAAI,CAAC;QAC3C,cAAS,GAA6B,IAAI,CAAC;QAC3C,gBAAW,GAAG,KAAK,CAAC;QACpB,eAAU,GAAG,KAAK,CAAC;IALxB,CAAC;IAOI,KAAK,CAAC,aAAa;QACzB,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAE7B,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxF,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAI,OAAU;QACzB,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAI,OAA6C;QAC3D,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,MAAM,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC7D,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAM,CAAC;gBACxC,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;YACxB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,eAAM,CAAC,KAAK,CAAC,sDAAsD,IAAI,CAAC,OAAO,GAAG,EAAE,KAAK,CAAC,CAAC;YAC7F,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAvDD,sDAuDC"}
|
|
@@ -1,9 +1,11 @@
|
|
|
1
|
+
import { RedisService } from './RedisService';
|
|
1
2
|
/**
|
|
2
3
|
* Lightweight Redis pub/sub helper to share events across service replicas.
|
|
3
4
|
*/
|
|
4
5
|
export declare class InternalEventsHandler<T = unknown> {
|
|
5
6
|
private readonly channel;
|
|
6
|
-
|
|
7
|
+
private readonly redisService;
|
|
8
|
+
constructor(channel: string, redisService: RedisService);
|
|
7
9
|
private pubClient;
|
|
8
10
|
private subClient;
|
|
9
11
|
private initialized;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"InternalEventsHandler.d.ts","sourceRoot":"","sources":["../../../src/services/InternalEventsHandler.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"InternalEventsHandler.d.ts","sourceRoot":"","sources":["../../../src/services/InternalEventsHandler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAqB,MAAM,gBAAgB,CAAC;AAGjE;;GAEG;AACH,qBAAa,qBAAqB,CAAC,CAAC,GAAG,OAAO;IAE1C,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,YAAY;gBADZ,OAAO,EAAE,MAAM,EACf,YAAY,EAAE,YAAY;IAG7C,OAAO,CAAC,SAAS,CAAkC;IACnD,OAAO,CAAC,SAAS,CAAkC;IACnD,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,UAAU,CAAS;YAEb,aAAa;IAS3B;;OAEG;IACG,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAS7C;;OAEG;IACG,MAAM,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;CAoB9E"}
|
|
@@ -1,22 +1,16 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.ModelFetcher = void 0;
|
|
4
|
-
const
|
|
5
|
-
const index_js_1 = require("../libs/logger/index.js");
|
|
4
|
+
const logger_1 = require("../libs/logger/index.cjs");
|
|
6
5
|
class ModelFetcher {
|
|
7
|
-
constructor(config
|
|
8
|
-
this.config =
|
|
9
|
-
openaiApiKey: config.openaiApiKey || config_js_1.config.ai.openaiApiKey,
|
|
10
|
-
anthropicApiKey: config.anthropicApiKey || config_js_1.config.ai.anthropicApiKey,
|
|
11
|
-
googleApiKey: config.googleApiKey || config_js_1.config.ai.googleApiKey,
|
|
12
|
-
openrouterApiKey: config.openrouterApiKey || config_js_1.config.ai.openrouterApiKey
|
|
13
|
-
};
|
|
6
|
+
constructor(config) {
|
|
7
|
+
this.config = config;
|
|
14
8
|
}
|
|
15
9
|
/**
|
|
16
10
|
* Analyze environment and fetch models for all available providers
|
|
17
11
|
*/
|
|
18
12
|
async fetchAllModels() {
|
|
19
|
-
|
|
13
|
+
logger_1.logger.info('Fetching available AI models...');
|
|
20
14
|
const allModels = [];
|
|
21
15
|
const providers = this.detectAvailableProviders();
|
|
22
16
|
for (const provider of providers) {
|
|
@@ -25,7 +19,7 @@ class ModelFetcher {
|
|
|
25
19
|
allModels.push(...models);
|
|
26
20
|
}
|
|
27
21
|
catch (error) {
|
|
28
|
-
|
|
22
|
+
logger_1.logger.warn(`Failed to fetch models for provider ${provider}:`, error instanceof Error ? error.message : 'Unknown error');
|
|
29
23
|
}
|
|
30
24
|
}
|
|
31
25
|
return allModels;
|
|
@@ -96,7 +90,7 @@ class ModelFetcher {
|
|
|
96
90
|
return models;
|
|
97
91
|
}
|
|
98
92
|
catch (error) {
|
|
99
|
-
|
|
93
|
+
logger_1.logger.error('Failed to fetch OpenAI models:', error);
|
|
100
94
|
return [];
|
|
101
95
|
}
|
|
102
96
|
}
|
|
@@ -130,7 +124,7 @@ class ModelFetcher {
|
|
|
130
124
|
return models;
|
|
131
125
|
}
|
|
132
126
|
catch (error) {
|
|
133
|
-
|
|
127
|
+
logger_1.logger.error('Failed to fetch Anthropic models:', error);
|
|
134
128
|
return [];
|
|
135
129
|
}
|
|
136
130
|
}
|
|
@@ -172,7 +166,7 @@ class ModelFetcher {
|
|
|
172
166
|
];
|
|
173
167
|
}
|
|
174
168
|
catch (error) {
|
|
175
|
-
|
|
169
|
+
logger_1.logger.error('Failed to fetch Google models:', error);
|
|
176
170
|
return [];
|
|
177
171
|
}
|
|
178
172
|
}
|
|
@@ -210,7 +204,7 @@ class ModelFetcher {
|
|
|
210
204
|
return models;
|
|
211
205
|
}
|
|
212
206
|
catch (error) {
|
|
213
|
-
|
|
207
|
+
logger_1.logger.error('Failed to fetch OpenRouter models:', error);
|
|
214
208
|
return [];
|
|
215
209
|
}
|
|
216
210
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ModelFetcher.js","sourceRoot":"","sources":["../../../src/services/ModelFetcher.ts"],"names":[],"mappings":";;;AACA,
|
|
1
|
+
{"version":3,"file":"ModelFetcher.js","sourceRoot":"","sources":["../../../src/services/ModelFetcher.ts"],"names":[],"mappings":";;;AACA,2CAAwC;AASxC,MAAa,YAAY;IAGvB,YAAY,MAAoB;QAC9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc;QAClB,eAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;QAC/C,MAAM,SAAS,GAAkB,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAElD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;gBAC3D,SAAS,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;YAC5B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,eAAM,CAAC,IAAI,CAAC,uCAAuC,QAAQ,GAAG,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;YAC5H,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,wBAAwB;QAC9B,MAAM,SAAS,GAAa,EAAE,CAAC;QAE/B,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;YACjC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC7B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC7B,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3B,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;YAChC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9B,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC7B,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,sBAAsB,CAAC,QAAgB;QAC3C,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAClC,KAAK,WAAW;gBACd,OAAO,IAAI,CAAC,oBAAoB,EAAE,CAAC;YACrC,KAAK,QAAQ;gBACX,OAAO,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAClC,KAAK,YAAY;gBACf,OAAO,IAAI,CAAC,qBAAqB,EAAE,CAAC;YACtC;gBACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB;QAC7B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC9B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,kCAAkC,EAAE;gBAC/D,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE;oBACrD,cAAc,EAAE,kBAAkB;iBACnC;aACF,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,uBAAuB,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACpF,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,MAAM,GAAkB,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;iBAC5C,MAAM,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;iBAChF,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,CAAC;gBACpB,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,QAAQ,EAAE,QAAQ;gBAClB,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACtC,WAAW,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,EAAE,EAAE,QAAQ,CAAC;aAC1D,CAAC,CAAC,CAAC;YAEN,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;YACtD,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,oBAAoB;QAChC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;YACjC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,qCAAqC,EAAE;gBAClE,OAAO,EAAE;oBACP,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;oBACxC,mBAAmB,EAAE,YAAY;oBACjC,cAAc,EAAE,kBAAkB;iBACnC;aACF,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACvF,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,MAAM,GAAkB,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;iBAC5C,MAAM,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;iBAClE,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,CAAC;gBACpB,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACtC,WAAW,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,EAAE,EAAE,WAAW,CAAC;aAC7D,CAAC,CAAC,CAAC;YAEN,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;YACzD,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB;QAC7B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC9B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC;YACH,mDAAmD;YACnD,6BAA6B;YAC7B,OAAO;gBACL;oBACE,EAAE,EAAE,sBAAsB;oBAC1B,QAAQ,EAAE,QAAQ;oBAClB,KAAK,EAAE,kBAAkB;oBACzB,WAAW,EAAE,2CAA2C;iBACzD;gBACD;oBACE,EAAE,EAAE,gBAAgB;oBACpB,QAAQ,EAAE,QAAQ;oBAClB,KAAK,EAAE,gBAAgB;oBACvB,WAAW,EAAE,qCAAqC;iBACnD;gBACD;oBACE,EAAE,EAAE,kBAAkB;oBACtB,QAAQ,EAAE,QAAQ;oBAClB,KAAK,EAAE,kBAAkB;oBACzB,WAAW,EAAE,iCAAiC;iBAC/C;gBACD;oBACE,EAAE,EAAE,YAAY;oBAChB,QAAQ,EAAE,QAAQ;oBAClB,KAAK,EAAE,YAAY;oBACnB,WAAW,EAAE,yBAAyB;iBACvC;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;YACtD,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,qBAAqB;QACjC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAClC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,gEAAgE,EAAE;gBAC7F,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE;oBACzD,cAAc,EAAE,kBAAkB;iBACnC;aACF,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACxF,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,MAAM,GAAkB,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,CAAC;gBACnE,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,QAAQ,EAAE,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAChC,KAAK,EAAE,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACpD,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,qBAAqB,KAAK,CAAC,EAAE,EAAE;aAClE,CAAC,CAAC;iBACF,MAAM,CAAC,CAAC,KAAkB,EAAE,EAAE,CAC7B,KAAK,CAAC,QAAQ,KAAK,QAAQ;gBAC3B,KAAK,CAAC,QAAQ,KAAK,WAAW;gBAC9B,KAAK,CAAC,QAAQ,KAAK,QAAQ,CAAC;iBAC7B,GAAG,CAAC,CAAC,KAAkB,EAAE,EAAE,CAAC,CAAC;gBAC5B,GAAG,KAAK;gBACR,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC,CAAC;YACJ,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC1D,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,OAAe;QACtC,+BAA+B;QAC/B,OAAO,OAAO;aACX,KAAK,CAAC,GAAG,CAAC;aACV,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;aAC3D,IAAI,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,OAAe,EAAE,QAAgB;QAC3D,MAAM,YAAY,GAA2B;YAC3C,gBAAgB;YAChB,QAAQ,EAAE,0CAA0C;YACpD,aAAa,EAAE,2CAA2C;YAC1D,aAAa,EAAE,gCAAgC;YAC/C,OAAO,EAAE,oBAAoB;YAC7B,eAAe,EAAE,kCAAkC;YACnD,YAAY,EAAE,2BAA2B;YACzC,SAAS,EAAE,gCAAgC;YAC3C,mBAAmB;YACnB,4BAA4B,EAAE,4CAA4C;YAC1E,4BAA4B,EAAE,4CAA4C;YAC1E,wBAAwB,EAAE,kCAAkC;YAC5D,0BAA0B,EAAE,gCAAgC;YAC5D,yBAAyB,EAAE,0CAA0C;SACtE,CAAC;QAEF,OAAO,YAAY,CAAC,OAAO,CAAC,IAAI,GAAG,QAAQ,WAAW,OAAO,EAAE,CAAC;IAClE,CAAC;CACF;AAtQD,oCAsQC"}
|
|
@@ -1,10 +1,5 @@
|
|
|
1
1
|
import type { ModelOption } from '@multiplayer-app/ai-agent-types';
|
|
2
|
-
|
|
3
|
-
openaiApiKey?: string;
|
|
4
|
-
anthropicApiKey?: string;
|
|
5
|
-
googleApiKey?: string;
|
|
6
|
-
openrouterApiKey?: string;
|
|
7
|
-
}
|
|
2
|
+
import { Config } from '../config';
|
|
8
3
|
export interface FetchedModel {
|
|
9
4
|
id: string;
|
|
10
5
|
provider: string;
|
|
@@ -13,7 +8,7 @@ export interface FetchedModel {
|
|
|
13
8
|
}
|
|
14
9
|
export declare class ModelFetcher {
|
|
15
10
|
private config;
|
|
16
|
-
constructor(config
|
|
11
|
+
constructor(config: Config["ai"]);
|
|
17
12
|
/**
|
|
18
13
|
* Analyze environment and fetch models for all available providers
|
|
19
14
|
*/
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ModelFetcher.d.ts","sourceRoot":"","sources":["../../../src/services/ModelFetcher.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;
|
|
1
|
+
{"version":3,"file":"ModelFetcher.d.ts","sourceRoot":"","sources":["../../../src/services/ModelFetcher.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAEnE,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACnC,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,MAAM,CAAe;gBAEjB,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;IAIhC;;OAEG;IACG,cAAc,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;IAiB9C;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAoBhC;;OAEG;IACG,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAetE;;OAEG;YACW,iBAAiB;IAkC/B;;OAEG;YACW,oBAAoB;IAmClC;;OAEG;YACW,iBAAiB;IAwC/B;;OAEG;YACW,qBAAqB;IAuCnC;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAQxB;;OAEG;IACH,OAAO,CAAC,mBAAmB;CAoB5B"}
|
|
@@ -1,61 +1,66 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.
|
|
3
|
+
exports.RedisService = void 0;
|
|
4
4
|
const redis_1 = require("redis");
|
|
5
|
-
const
|
|
6
|
-
const index_js_1 = require("../libs/logger/index.js");
|
|
5
|
+
const logger_1 = require("../libs/logger/index.cjs");
|
|
7
6
|
class RedisService {
|
|
8
|
-
constructor() {
|
|
7
|
+
constructor(config) {
|
|
9
8
|
this.client = null;
|
|
10
9
|
this.pubSubClients = new Map();
|
|
11
10
|
this.isConnected = false;
|
|
11
|
+
this.config = config;
|
|
12
|
+
}
|
|
13
|
+
getRedisUrl() {
|
|
14
|
+
if (this.config?.url) {
|
|
15
|
+
return this.config.url;
|
|
16
|
+
}
|
|
17
|
+
return `redis://${this.config?.host}:${this.config?.port}`;
|
|
12
18
|
}
|
|
13
19
|
/**
|
|
14
20
|
* Initialize Redis client connection
|
|
15
21
|
*/
|
|
16
22
|
async initialize(redisConfigOverride) {
|
|
17
|
-
var _a, _b;
|
|
18
23
|
if (this.client && this.isConnected) {
|
|
19
|
-
|
|
24
|
+
logger_1.logger.info('Redis client already initialized');
|
|
20
25
|
return;
|
|
21
26
|
}
|
|
22
|
-
const redisUrl =
|
|
23
|
-
|
|
24
|
-
`redis://${
|
|
27
|
+
const redisUrl = redisConfigOverride?.url ||
|
|
28
|
+
this.config.url ||
|
|
29
|
+
`redis://${redisConfigOverride?.host || this.config.host}:${redisConfigOverride?.port || this.config.port}`;
|
|
25
30
|
const redisConfig = {
|
|
26
31
|
url: redisUrl
|
|
27
32
|
};
|
|
28
|
-
if (
|
|
29
|
-
redisConfig.password =
|
|
33
|
+
if (redisConfigOverride?.password || this.config.password) {
|
|
34
|
+
redisConfig.password = redisConfigOverride?.password || this.config.password;
|
|
30
35
|
}
|
|
31
|
-
if (
|
|
32
|
-
redisConfig.database =
|
|
36
|
+
if (redisConfigOverride?.database !== undefined || this.config.database !== undefined) {
|
|
37
|
+
redisConfig.database = redisConfigOverride?.database ?? this.config.database ?? 0;
|
|
33
38
|
}
|
|
34
39
|
this.client = (0, redis_1.createClient)(redisConfig);
|
|
35
40
|
this.client.on('error', (err) => {
|
|
36
|
-
|
|
41
|
+
logger_1.logger.error('Redis Client Error:', err);
|
|
37
42
|
this.isConnected = false;
|
|
38
43
|
});
|
|
39
44
|
this.client.on('connect', () => {
|
|
40
|
-
|
|
45
|
+
logger_1.logger.info('Redis Client connecting...');
|
|
41
46
|
});
|
|
42
47
|
this.client.on('ready', () => {
|
|
43
|
-
|
|
48
|
+
logger_1.logger.info('Redis Client ready');
|
|
44
49
|
this.isConnected = true;
|
|
45
50
|
});
|
|
46
51
|
this.client.on('reconnecting', () => {
|
|
47
|
-
|
|
52
|
+
logger_1.logger.info('Redis Client reconnecting...');
|
|
48
53
|
});
|
|
49
54
|
this.client.on('end', () => {
|
|
50
|
-
|
|
55
|
+
logger_1.logger.info('Redis Client connection ended');
|
|
51
56
|
this.isConnected = false;
|
|
52
57
|
});
|
|
53
58
|
try {
|
|
54
59
|
await this.client.connect();
|
|
55
|
-
|
|
60
|
+
logger_1.logger.info('Redis client connected successfully');
|
|
56
61
|
}
|
|
57
62
|
catch (error) {
|
|
58
|
-
|
|
63
|
+
logger_1.logger.error('Failed to connect to Redis:', error);
|
|
59
64
|
throw error;
|
|
60
65
|
}
|
|
61
66
|
}
|
|
@@ -174,15 +179,15 @@ class RedisService {
|
|
|
174
179
|
if (existingPair) {
|
|
175
180
|
return existingPair;
|
|
176
181
|
}
|
|
177
|
-
const redisUrl =
|
|
182
|
+
const redisUrl = this.getRedisUrl();
|
|
178
183
|
const redisConfig = {
|
|
179
184
|
url: redisUrl
|
|
180
185
|
};
|
|
181
|
-
if (
|
|
182
|
-
redisConfig.password =
|
|
186
|
+
if (this.config.password) {
|
|
187
|
+
redisConfig.password = this.config.password;
|
|
183
188
|
}
|
|
184
|
-
if (
|
|
185
|
-
redisConfig.database =
|
|
189
|
+
if (this.config.database !== undefined) {
|
|
190
|
+
redisConfig.database = this.config.database;
|
|
186
191
|
}
|
|
187
192
|
// Create pub client
|
|
188
193
|
const pubClient = (0, redis_1.createClient)(redisConfig);
|
|
@@ -190,21 +195,21 @@ class RedisService {
|
|
|
190
195
|
const subClient = pubClient.duplicate();
|
|
191
196
|
// Set up error handlers
|
|
192
197
|
pubClient.on('error', (err) => {
|
|
193
|
-
|
|
198
|
+
logger_1.logger.error('Redis Pub Client Error:', err);
|
|
194
199
|
});
|
|
195
200
|
subClient.on('error', (err) => {
|
|
196
|
-
|
|
201
|
+
logger_1.logger.error('Redis Sub Client Error:', err);
|
|
197
202
|
});
|
|
198
203
|
// Connect both clients
|
|
199
204
|
try {
|
|
200
205
|
await Promise.all([pubClient.connect(), subClient.connect()]);
|
|
201
|
-
|
|
206
|
+
logger_1.logger.info(`Redis pub/sub clients connected successfully for ${name}`);
|
|
202
207
|
const pair = { pubClient, subClient };
|
|
203
208
|
this.pubSubClients.set(name, pair);
|
|
204
209
|
return pair;
|
|
205
210
|
}
|
|
206
211
|
catch (error) {
|
|
207
|
-
|
|
212
|
+
logger_1.logger.error(`Failed to connect Redis pub/sub clients for ${name}:`, error);
|
|
208
213
|
throw error;
|
|
209
214
|
}
|
|
210
215
|
}
|
|
@@ -231,10 +236,8 @@ class RedisService {
|
|
|
231
236
|
this.pubSubClients.clear();
|
|
232
237
|
await Promise.all(disconnectPromises);
|
|
233
238
|
this.isConnected = false;
|
|
234
|
-
|
|
239
|
+
logger_1.logger.info('Redis clients disconnected');
|
|
235
240
|
}
|
|
236
241
|
}
|
|
237
242
|
exports.RedisService = RedisService;
|
|
238
|
-
// Singleton instance
|
|
239
|
-
exports.redisService = new RedisService();
|
|
240
243
|
//# sourceMappingURL=RedisService.js.map
|