@crewx/sdk 0.8.0-rc.58
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/LICENSE +201 -0
- package/README.md +1022 -0
- package/dist/adapters/MastraToolAdapter.d.ts +9 -0
- package/dist/adapters/MastraToolAdapter.js +66 -0
- package/dist/adapters/MastraToolAdapter.js.map +1 -0
- package/dist/api/index.d.ts +2 -0
- package/dist/api/index.js +8 -0
- package/dist/api/index.js.map +1 -0
- package/dist/boxing/box-storage.interface.d.ts +13 -0
- package/dist/boxing/box-storage.interface.js +3 -0
- package/dist/boxing/box-storage.interface.js.map +1 -0
- package/dist/boxing/box.service.d.ts +15 -0
- package/dist/boxing/box.service.js +70 -0
- package/dist/boxing/box.service.js.map +1 -0
- package/dist/boxing/box.types.d.ts +86 -0
- package/dist/boxing/box.types.js +3 -0
- package/dist/boxing/box.types.js.map +1 -0
- package/dist/boxing/context-builder.d.ts +8 -0
- package/dist/boxing/context-builder.js +76 -0
- package/dist/boxing/context-builder.js.map +1 -0
- package/dist/boxing/index.d.ts +6 -0
- package/dist/boxing/index.js +11 -0
- package/dist/boxing/index.js.map +1 -0
- package/dist/boxing/tokenizer.d.ts +3 -0
- package/dist/boxing/tokenizer.js +11 -0
- package/dist/boxing/tokenizer.js.map +1 -0
- package/dist/config/api-provider-parser.d.ts +58 -0
- package/dist/config/api-provider-parser.js +212 -0
- package/dist/config/api-provider-parser.js.map +1 -0
- package/dist/config/index.d.ts +3 -0
- package/dist/config/index.js +20 -0
- package/dist/config/index.js.map +1 -0
- package/dist/config/log.config.d.ts +7 -0
- package/dist/config/log.config.js +20 -0
- package/dist/config/log.config.js.map +1 -0
- package/dist/config/pricing.d.ts +10 -0
- package/dist/config/pricing.js +44 -0
- package/dist/config/pricing.js.map +1 -0
- package/dist/config/timeout.config.d.ts +14 -0
- package/dist/config/timeout.config.js +34 -0
- package/dist/config/timeout.config.js.map +1 -0
- package/dist/config/yaml-loader.d.ts +8 -0
- package/dist/config/yaml-loader.js +155 -0
- package/dist/config/yaml-loader.js.map +1 -0
- package/dist/constants/index.d.ts +4 -0
- package/dist/constants/index.js +8 -0
- package/dist/constants/index.js.map +1 -0
- package/dist/constants.d.ts +1 -0
- package/dist/constants.js +18 -0
- package/dist/constants.js.map +1 -0
- package/dist/conversation/conversation-config.d.ts +9 -0
- package/dist/conversation/conversation-config.js +22 -0
- package/dist/conversation/conversation-config.js.map +1 -0
- package/dist/conversation/conversation-history.interface.d.ts +36 -0
- package/dist/conversation/conversation-history.interface.js +3 -0
- package/dist/conversation/conversation-history.interface.js.map +1 -0
- package/dist/conversation/conversation-storage.service.d.ts +16 -0
- package/dist/conversation/conversation-storage.service.js +213 -0
- package/dist/conversation/conversation-storage.service.js.map +1 -0
- package/dist/conversation/index.d.ts +3 -0
- package/dist/conversation/index.js +20 -0
- package/dist/conversation/index.js.map +1 -0
- package/dist/core/agent/agent-factory.d.ts +37 -0
- package/dist/core/agent/agent-factory.js +68 -0
- package/dist/core/agent/agent-factory.js.map +1 -0
- package/dist/core/agent/agent-runtime.d.ts +52 -0
- package/dist/core/agent/agent-runtime.js +206 -0
- package/dist/core/agent/agent-runtime.js.map +1 -0
- package/dist/core/agent/event-bus.d.ts +44 -0
- package/dist/core/agent/event-bus.js +43 -0
- package/dist/core/agent/event-bus.js.map +1 -0
- package/dist/core/agent/index.d.ts +3 -0
- package/dist/core/agent/index.js +13 -0
- package/dist/core/agent/index.js.map +1 -0
- package/dist/core/env-defaults.d.ts +1 -0
- package/dist/core/env-defaults.js +7 -0
- package/dist/core/env-defaults.js.map +1 -0
- package/dist/core/parallel/helpers.d.ts +27 -0
- package/dist/core/parallel/helpers.js +252 -0
- package/dist/core/parallel/helpers.js.map +1 -0
- package/dist/core/parallel/index.d.ts +4 -0
- package/dist/core/parallel/index.js +11 -0
- package/dist/core/parallel/index.js.map +1 -0
- package/dist/core/parallel/parallel-runner.d.ts +16 -0
- package/dist/core/parallel/parallel-runner.js +230 -0
- package/dist/core/parallel/parallel-runner.js.map +1 -0
- package/dist/core/parallel/types.d.ts +41 -0
- package/dist/core/parallel/types.js +3 -0
- package/dist/core/parallel/types.js.map +1 -0
- package/dist/core/providers/MastraAPIProvider.d.ts +31 -0
- package/dist/core/providers/MastraAPIProvider.js +365 -0
- package/dist/core/providers/MastraAPIProvider.js.map +1 -0
- package/dist/core/providers/ai-provider.interface.d.ts +78 -0
- package/dist/core/providers/ai-provider.interface.js +23 -0
- package/dist/core/providers/ai-provider.interface.js.map +1 -0
- package/dist/core/providers/base-ai.provider.d.ts +80 -0
- package/dist/core/providers/base-ai.provider.js +1183 -0
- package/dist/core/providers/base-ai.provider.js.map +1 -0
- package/dist/core/providers/base-ai.types.d.ts +26 -0
- package/dist/core/providers/base-ai.types.js +3 -0
- package/dist/core/providers/base-ai.types.js.map +1 -0
- package/dist/core/providers/claude.provider.d.ts +19 -0
- package/dist/core/providers/claude.provider.js +170 -0
- package/dist/core/providers/claude.provider.js.map +1 -0
- package/dist/core/providers/codex.provider.d.ts +21 -0
- package/dist/core/providers/codex.provider.js +134 -0
- package/dist/core/providers/codex.provider.js.map +1 -0
- package/dist/core/providers/copilot.provider.d.ts +24 -0
- package/dist/core/providers/copilot.provider.js +146 -0
- package/dist/core/providers/copilot.provider.js.map +1 -0
- package/dist/core/providers/dynamic-provider.factory.d.ts +74 -0
- package/dist/core/providers/dynamic-provider.factory.js +643 -0
- package/dist/core/providers/dynamic-provider.factory.js.map +1 -0
- package/dist/core/providers/gemini.provider.d.ts +16 -0
- package/dist/core/providers/gemini.provider.js +101 -0
- package/dist/core/providers/gemini.provider.js.map +1 -0
- package/dist/core/providers/index.d.ts +8 -0
- package/dist/core/providers/index.js +20 -0
- package/dist/core/providers/index.js.map +1 -0
- package/dist/core/providers/mock.provider.d.ts +13 -0
- package/dist/core/providers/mock.provider.js +55 -0
- package/dist/core/providers/mock.provider.js.map +1 -0
- package/dist/core/providers/provider-factory.d.ts +3 -0
- package/dist/core/providers/provider-factory.js +65 -0
- package/dist/core/providers/provider-factory.js.map +1 -0
- package/dist/core/providers/tool-call.types.d.ts +39 -0
- package/dist/core/providers/tool-call.types.js +3 -0
- package/dist/core/providers/tool-call.types.js.map +1 -0
- package/dist/core/remote/index.d.ts +3 -0
- package/dist/core/remote/index.js +20 -0
- package/dist/core/remote/index.js.map +1 -0
- package/dist/core/remote/remote-agent-manager.d.ts +24 -0
- package/dist/core/remote/remote-agent-manager.js +195 -0
- package/dist/core/remote/remote-agent-manager.js.map +1 -0
- package/dist/core/remote/remote-transport.d.ts +15 -0
- package/dist/core/remote/remote-transport.js +70 -0
- package/dist/core/remote/remote-transport.js.map +1 -0
- package/dist/core/remote/types.d.ts +79 -0
- package/dist/core/remote/types.js +3 -0
- package/dist/core/remote/types.js.map +1 -0
- package/dist/index.d.ts +56 -0
- package/dist/index.js +133 -0
- package/dist/index.js.map +1 -0
- package/dist/internal/index.d.ts +1 -0
- package/dist/internal/index.js +6 -0
- package/dist/internal/index.js.map +1 -0
- package/dist/knowledge/DocumentManager.d.ts +4 -0
- package/dist/knowledge/DocumentManager.js +119 -0
- package/dist/knowledge/DocumentManager.js.map +1 -0
- package/dist/knowledge/index.d.ts +1 -0
- package/dist/knowledge/index.js +18 -0
- package/dist/knowledge/index.js.map +1 -0
- package/dist/parsers/agent-call.util.d.ts +3 -0
- package/dist/parsers/agent-call.util.js +17 -0
- package/dist/parsers/agent-call.util.js.map +1 -0
- package/dist/parsers/claude.parser.d.ts +2 -0
- package/dist/parsers/claude.parser.js +46 -0
- package/dist/parsers/claude.parser.js.map +1 -0
- package/dist/parsers/codex.parser.d.ts +2 -0
- package/dist/parsers/codex.parser.js +89 -0
- package/dist/parsers/codex.parser.js.map +1 -0
- package/dist/parsers/copilot.parser.d.ts +2 -0
- package/dist/parsers/copilot.parser.js +58 -0
- package/dist/parsers/copilot.parser.js.map +1 -0
- package/dist/parsers/gemini.parser.d.ts +2 -0
- package/dist/parsers/gemini.parser.js +36 -0
- package/dist/parsers/gemini.parser.js.map +1 -0
- package/dist/parsers/index.d.ts +7 -0
- package/dist/parsers/index.js +45 -0
- package/dist/parsers/index.js.map +1 -0
- package/dist/schema/skills-parser.d.ts +8 -0
- package/dist/schema/skills-parser.js +438 -0
- package/dist/schema/skills-parser.js.map +1 -0
- package/dist/schema/skills.types.d.ts +158 -0
- package/dist/schema/skills.types.js +41 -0
- package/dist/schema/skills.types.js.map +1 -0
- package/dist/schemas/api-provider.schema.d.ts +432 -0
- package/dist/schemas/api-provider.schema.js +50 -0
- package/dist/schemas/api-provider.schema.js.map +1 -0
- package/dist/services/index.d.ts +2 -0
- package/dist/services/index.js +19 -0
- package/dist/services/index.js.map +1 -0
- package/dist/services/layout-loader.service.d.ts +18 -0
- package/dist/services/layout-loader.service.js +271 -0
- package/dist/services/layout-loader.service.js.map +1 -0
- package/dist/services/layout-renderer.service.d.ts +34 -0
- package/dist/services/layout-renderer.service.js +325 -0
- package/dist/services/layout-renderer.service.js.map +1 -0
- package/dist/services/props-validator.service.d.ts +29 -0
- package/dist/services/props-validator.service.js +332 -0
- package/dist/services/props-validator.service.js.map +1 -0
- package/dist/skills/adapter/claude-skill-adapter.d.ts +11 -0
- package/dist/skills/adapter/claude-skill-adapter.js +222 -0
- package/dist/skills/adapter/claude-skill-adapter.js.map +1 -0
- package/dist/skills/index.d.ts +6 -0
- package/dist/skills/index.js +31 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/runtime/progressive-loader.d.ts +27 -0
- package/dist/skills/runtime/progressive-loader.js +186 -0
- package/dist/skills/runtime/progressive-loader.js.map +1 -0
- package/dist/skills/runtime/runtime-requirements-validator.d.ts +23 -0
- package/dist/skills/runtime/runtime-requirements-validator.js +248 -0
- package/dist/skills/runtime/runtime-requirements-validator.js.map +1 -0
- package/dist/skills/runtime/skill-runtime.service.d.ts +42 -0
- package/dist/skills/runtime/skill-runtime.service.js +434 -0
- package/dist/skills/runtime/skill-runtime.service.js.map +1 -0
- package/dist/tools/file-system.service.d.ts +10 -0
- package/dist/tools/file-system.service.js +33 -0
- package/dist/tools/file-system.service.js.map +1 -0
- package/dist/tools/find.tool.d.ts +85 -0
- package/dist/tools/find.tool.js +139 -0
- package/dist/tools/find.tool.js.map +1 -0
- package/dist/tools/glob.tool.d.ts +100 -0
- package/dist/tools/glob.tool.js +153 -0
- package/dist/tools/glob.tool.js.map +1 -0
- package/dist/tools/grep.tool.d.ts +30 -0
- package/dist/tools/grep.tool.js +137 -0
- package/dist/tools/grep.tool.js.map +1 -0
- package/dist/tools/index.d.ts +12 -0
- package/dist/tools/index.js +40 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/ls.tool.d.ts +24 -0
- package/dist/tools/ls.tool.js +94 -0
- package/dist/tools/ls.tool.js.map +1 -0
- package/dist/tools/read-file.tool.d.ts +30 -0
- package/dist/tools/read-file.tool.js +69 -0
- package/dist/tools/read-file.tool.js.map +1 -0
- package/dist/tools/replace.tool.d.ts +36 -0
- package/dist/tools/replace.tool.js +68 -0
- package/dist/tools/replace.tool.js.map +1 -0
- package/dist/tools/run-shell-command.tool.d.ts +24 -0
- package/dist/tools/run-shell-command.tool.js +64 -0
- package/dist/tools/run-shell-command.tool.js.map +1 -0
- package/dist/tools/tree.tool.d.ts +24 -0
- package/dist/tools/tree.tool.js +109 -0
- package/dist/tools/tree.tool.js.map +1 -0
- package/dist/tools/types.d.ts +42 -0
- package/dist/tools/types.js +13 -0
- package/dist/tools/types.js.map +1 -0
- package/dist/tools/utils/file-utils.d.ts +5 -0
- package/dist/tools/utils/file-utils.js +221 -0
- package/dist/tools/utils/file-utils.js.map +1 -0
- package/dist/tools/write-file.tool.d.ts +24 -0
- package/dist/tools/write-file.tool.js +55 -0
- package/dist/tools/write-file.tool.js.map +1 -0
- package/dist/types/agent.types.d.ts +134 -0
- package/dist/types/agent.types.js +16 -0
- package/dist/types/agent.types.js.map +1 -0
- package/dist/types/api-provider.types.d.ts +85 -0
- package/dist/types/api-provider.types.js +65 -0
- package/dist/types/api-provider.types.js.map +1 -0
- package/dist/types/index.d.ts +6 -0
- package/dist/types/index.js +23 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/layout.types.d.ts +115 -0
- package/dist/types/layout.types.js +22 -0
- package/dist/types/layout.types.js.map +1 -0
- package/dist/types/provider.types.d.ts +12 -0
- package/dist/types/provider.types.js +3 -0
- package/dist/types/provider.types.js.map +1 -0
- package/dist/types/skill-runtime.types.d.ts +244 -0
- package/dist/types/skill-runtime.types.js +44 -0
- package/dist/types/skill-runtime.types.js.map +1 -0
- package/dist/types/structured-payload.types.d.ts +46 -0
- package/dist/types/structured-payload.types.js +65 -0
- package/dist/types/structured-payload.types.js.map +1 -0
- package/dist/types/task-log.types.d.ts +14 -0
- package/dist/types/task-log.types.js +3 -0
- package/dist/types/task-log.types.js.map +1 -0
- package/dist/types/template.types.d.ts +38 -0
- package/dist/types/template.types.js +3 -0
- package/dist/types/template.types.js.map +1 -0
- package/dist/types.d.ts +1 -0
- package/dist/types.js +18 -0
- package/dist/types.js.map +1 -0
- package/dist/utils/api-provider-normalizer.d.ts +16 -0
- package/dist/utils/api-provider-normalizer.js +135 -0
- package/dist/utils/api-provider-normalizer.js.map +1 -0
- package/dist/utils/base-message-formatter.d.ts +32 -0
- package/dist/utils/base-message-formatter.js +170 -0
- package/dist/utils/base-message-formatter.js.map +1 -0
- package/dist/utils/error-utils.d.ts +3 -0
- package/dist/utils/error-utils.js +27 -0
- package/dist/utils/error-utils.js.map +1 -0
- package/dist/utils/index.d.ts +4 -0
- package/dist/utils/index.js +21 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/math-utils.d.ts +3 -0
- package/dist/utils/math-utils.js +10 -0
- package/dist/utils/math-utils.js.map +1 -0
- package/dist/utils/mention-parser.d.ts +18 -0
- package/dist/utils/mention-parser.js +136 -0
- package/dist/utils/mention-parser.js.map +1 -0
- package/dist/utils/string-utils.d.ts +1 -0
- package/dist/utils/string-utils.js +10 -0
- package/dist/utils/string-utils.js.map +1 -0
- package/dist/utils.d.ts +3 -0
- package/dist/utils.js +20 -0
- package/dist/utils.js.map +1 -0
- package/package.json +134 -0
- package/schema/api-provider-config.json +138 -0
- package/schema/crewx-config.json +224 -0
- package/schema/skills-config.json +306 -0
|
@@ -0,0 +1,365 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.MastraAPIProvider = void 0;
|
|
4
|
+
const core_1 = require("@mastra/core");
|
|
5
|
+
const openai_1 = require("@ai-sdk/openai");
|
|
6
|
+
const anthropic_1 = require("@ai-sdk/anthropic");
|
|
7
|
+
const google_1 = require("@ai-sdk/google");
|
|
8
|
+
const ai_sdk_provider_1 = require("@openrouter/ai-sdk-provider");
|
|
9
|
+
const openai_compatible_1 = require("@ai-sdk/openai-compatible");
|
|
10
|
+
const api_provider_types_1 = require("../../types/api-provider.types");
|
|
11
|
+
const MastraToolAdapter_1 = require("../../adapters/MastraToolAdapter");
|
|
12
|
+
const api_provider_normalizer_1 = require("../../utils/api-provider-normalizer");
|
|
13
|
+
const log_config_1 = require("../../config/log.config");
|
|
14
|
+
class MastraAPIProvider {
|
|
15
|
+
constructor(config, mode = 'query') {
|
|
16
|
+
this.filteredToolSets = {};
|
|
17
|
+
const normalizedConfig = (0, api_provider_normalizer_1.normalizeAPIProviderConfig)(config);
|
|
18
|
+
this.config = normalizedConfig.config;
|
|
19
|
+
this.permissionsByMode = normalizedConfig.permissionsByMode;
|
|
20
|
+
this.defaultMode = mode;
|
|
21
|
+
this.name = this.config.provider;
|
|
22
|
+
this.logConfig = (0, log_config_1.getLogConfig)();
|
|
23
|
+
}
|
|
24
|
+
setTaskLogHandler(handler) {
|
|
25
|
+
this.taskLogHandler = handler;
|
|
26
|
+
}
|
|
27
|
+
emitLog(taskId, level, message) {
|
|
28
|
+
if (!this.taskLogHandler)
|
|
29
|
+
return;
|
|
30
|
+
this.taskLogHandler({
|
|
31
|
+
taskId,
|
|
32
|
+
timestamp: new Date().toISOString(),
|
|
33
|
+
level,
|
|
34
|
+
message,
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
createModel(config) {
|
|
38
|
+
const { provider, model, apiKey, url } = config;
|
|
39
|
+
switch (provider) {
|
|
40
|
+
case 'api/openai': {
|
|
41
|
+
if (url && url.includes('openrouter.ai')) {
|
|
42
|
+
console.log('[MastraAPIProvider] Detected OpenRouter, using @openrouter/ai-sdk-provider');
|
|
43
|
+
const openrouter = (0, ai_sdk_provider_1.createOpenRouter)({
|
|
44
|
+
apiKey: apiKey || '',
|
|
45
|
+
});
|
|
46
|
+
return openrouter(model);
|
|
47
|
+
}
|
|
48
|
+
if (apiKey && url) {
|
|
49
|
+
const customOpenAI = (0, openai_1.createOpenAI)({ apiKey, baseURL: url });
|
|
50
|
+
return customOpenAI(model);
|
|
51
|
+
}
|
|
52
|
+
else if (apiKey) {
|
|
53
|
+
const customOpenAI = (0, openai_1.createOpenAI)({ apiKey });
|
|
54
|
+
return customOpenAI(model);
|
|
55
|
+
}
|
|
56
|
+
return (0, openai_1.openai)(model);
|
|
57
|
+
}
|
|
58
|
+
case 'api/litellm':
|
|
59
|
+
case 'api/ollama':
|
|
60
|
+
case 'api/sowonai': {
|
|
61
|
+
let defaultURL = 'http://localhost:4000';
|
|
62
|
+
let defaultKey = 'dummy';
|
|
63
|
+
if (provider === 'api/ollama') {
|
|
64
|
+
defaultURL = 'http://localhost:11434/v1';
|
|
65
|
+
defaultKey = 'ollama';
|
|
66
|
+
}
|
|
67
|
+
else if (provider === 'api/sowonai') {
|
|
68
|
+
defaultURL = 'https://api.sowon.ai/v1';
|
|
69
|
+
}
|
|
70
|
+
const providerInstance = (0, openai_compatible_1.createOpenAICompatible)({
|
|
71
|
+
name: provider.replace('api/', ''),
|
|
72
|
+
baseURL: url || defaultURL,
|
|
73
|
+
apiKey: apiKey || defaultKey,
|
|
74
|
+
});
|
|
75
|
+
return providerInstance(model);
|
|
76
|
+
}
|
|
77
|
+
case 'api/anthropic': {
|
|
78
|
+
if (apiKey && url) {
|
|
79
|
+
const customAnthropic = (0, anthropic_1.createAnthropic)({ apiKey, baseURL: url });
|
|
80
|
+
return customAnthropic(model);
|
|
81
|
+
}
|
|
82
|
+
else if (apiKey) {
|
|
83
|
+
const customAnthropic = (0, anthropic_1.createAnthropic)({ apiKey });
|
|
84
|
+
return customAnthropic(model);
|
|
85
|
+
}
|
|
86
|
+
return (0, anthropic_1.anthropic)(model);
|
|
87
|
+
}
|
|
88
|
+
case 'api/bedrock': {
|
|
89
|
+
const customAnthropic = (0, anthropic_1.createAnthropic)({
|
|
90
|
+
apiKey: apiKey || '',
|
|
91
|
+
baseURL: url || 'https://bedrock-runtime.us-east-1.amazonaws.com',
|
|
92
|
+
});
|
|
93
|
+
return customAnthropic(model);
|
|
94
|
+
}
|
|
95
|
+
case 'api/google': {
|
|
96
|
+
if (apiKey) {
|
|
97
|
+
const customGoogle = (0, google_1.createGoogleGenerativeAI)({ apiKey });
|
|
98
|
+
return customGoogle(model);
|
|
99
|
+
}
|
|
100
|
+
return (0, google_1.google)(model);
|
|
101
|
+
}
|
|
102
|
+
default:
|
|
103
|
+
throw new Error(`Unsupported provider: ${provider}`);
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
setTools(tools, context) {
|
|
107
|
+
this.toolContext = this.applyDefaultModeToContext(context);
|
|
108
|
+
this.filteredToolSets = this.buildModeAwareToolSets(tools);
|
|
109
|
+
}
|
|
110
|
+
applyDefaultModeToContext(context) {
|
|
111
|
+
if (!context) {
|
|
112
|
+
return undefined;
|
|
113
|
+
}
|
|
114
|
+
if (context.mode) {
|
|
115
|
+
return context;
|
|
116
|
+
}
|
|
117
|
+
return { ...context, mode: this.defaultMode };
|
|
118
|
+
}
|
|
119
|
+
buildModeAwareToolSets(tools) {
|
|
120
|
+
const lookup = new Map();
|
|
121
|
+
for (const tool of tools) {
|
|
122
|
+
const identifier = this.getToolIdentifier(tool);
|
|
123
|
+
if (identifier) {
|
|
124
|
+
lookup.set(identifier, tool);
|
|
125
|
+
}
|
|
126
|
+
else {
|
|
127
|
+
console.warn('[MastraAPIProvider] Ignoring tool without name/id during registration');
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
const filtered = {};
|
|
131
|
+
for (const [mode, permissions] of Object.entries(this.permissionsByMode)) {
|
|
132
|
+
if (!permissions.tools.length) {
|
|
133
|
+
filtered[mode] = [];
|
|
134
|
+
continue;
|
|
135
|
+
}
|
|
136
|
+
filtered[mode] = permissions.tools
|
|
137
|
+
.map((toolName) => {
|
|
138
|
+
if (!lookup.has(toolName)) {
|
|
139
|
+
console.warn(`[MastraAPIProvider] Tool '${toolName}' referenced in ${mode} mode but not registered`);
|
|
140
|
+
}
|
|
141
|
+
return lookup.get(toolName);
|
|
142
|
+
})
|
|
143
|
+
.filter(Boolean);
|
|
144
|
+
}
|
|
145
|
+
return filtered;
|
|
146
|
+
}
|
|
147
|
+
getToolIdentifier(tool) {
|
|
148
|
+
if (tool && typeof tool.id === 'string') {
|
|
149
|
+
return tool.id;
|
|
150
|
+
}
|
|
151
|
+
if (tool && typeof tool.name === 'string') {
|
|
152
|
+
return tool.name;
|
|
153
|
+
}
|
|
154
|
+
return undefined;
|
|
155
|
+
}
|
|
156
|
+
createMastraToolsForMode(mode) {
|
|
157
|
+
if (!this.toolContext) {
|
|
158
|
+
return {};
|
|
159
|
+
}
|
|
160
|
+
const registeredTools = this.filteredToolSets[mode] ?? [];
|
|
161
|
+
if (registeredTools.length === 0) {
|
|
162
|
+
return {};
|
|
163
|
+
}
|
|
164
|
+
const contextWithMode = { ...this.toolContext, mode };
|
|
165
|
+
return MastraToolAdapter_1.MastraToolAdapter.convertTools(registeredTools, contextWithMode);
|
|
166
|
+
}
|
|
167
|
+
async query(prompt, options = {}) {
|
|
168
|
+
return this.runWithMode('query', prompt, options);
|
|
169
|
+
}
|
|
170
|
+
async execute(prompt, options = {}) {
|
|
171
|
+
return this.runWithMode('execute', prompt, options);
|
|
172
|
+
}
|
|
173
|
+
async runWithMode(mode, prompt, options) {
|
|
174
|
+
const taskId = options.taskId || `${this.name}_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
|
|
175
|
+
try {
|
|
176
|
+
const mastraTools = this.createMastraToolsForMode(mode);
|
|
177
|
+
const toolCount = Object.keys(mastraTools).length;
|
|
178
|
+
if (toolCount > 0) {
|
|
179
|
+
console.log(`[INFO] Registered ${toolCount} tools for ${mode} mode: ${Object.keys(mastraTools).join(', ')}`);
|
|
180
|
+
}
|
|
181
|
+
else {
|
|
182
|
+
console.log(`[INFO] No tools registered for ${mode} mode`);
|
|
183
|
+
}
|
|
184
|
+
const configToUse = options.model
|
|
185
|
+
? { ...this.config, model: options.model }
|
|
186
|
+
: this.config;
|
|
187
|
+
console.log(`[INFO] Using model: ${configToUse.model}`);
|
|
188
|
+
const modelInstance = this.createModel(configToUse);
|
|
189
|
+
const agent = new core_1.Agent({
|
|
190
|
+
name: this.config.provider,
|
|
191
|
+
model: modelInstance,
|
|
192
|
+
instructions: prompt,
|
|
193
|
+
tools: mastraTools,
|
|
194
|
+
});
|
|
195
|
+
const maxSteps = Math.min(this.config.maxSteps ?? api_provider_types_1.DEFAULT_MAX_STEPS, api_provider_types_1.MAX_STEPS_LIMIT);
|
|
196
|
+
const generateOptions = { maxSteps };
|
|
197
|
+
console.log(`[INFO] Sending request to AI model (maxSteps: ${maxSteps})...`);
|
|
198
|
+
const fullOutput = await agent.generate(prompt, generateOptions);
|
|
199
|
+
console.log(`[INFO] Received response from AI model`);
|
|
200
|
+
return this.convertResponse(fullOutput, taskId);
|
|
201
|
+
}
|
|
202
|
+
catch (error) {
|
|
203
|
+
return {
|
|
204
|
+
content: '',
|
|
205
|
+
provider: this.name,
|
|
206
|
+
command: `${this.name} ${mode}`,
|
|
207
|
+
success: false,
|
|
208
|
+
error: error.message || 'Unknown error',
|
|
209
|
+
taskId,
|
|
210
|
+
};
|
|
211
|
+
}
|
|
212
|
+
}
|
|
213
|
+
extractToolName(tc) {
|
|
214
|
+
return tc.payload?.toolName || tc.toolName || 'unknown';
|
|
215
|
+
}
|
|
216
|
+
extractToolArgs(tc) {
|
|
217
|
+
return tc.payload?.args || tc.args;
|
|
218
|
+
}
|
|
219
|
+
extractToolResult(tr) {
|
|
220
|
+
return tr?.payload?.result || tr?.result;
|
|
221
|
+
}
|
|
222
|
+
convertResponse(fullOutput, taskId) {
|
|
223
|
+
let content = fullOutput.text || '';
|
|
224
|
+
const response = {
|
|
225
|
+
content,
|
|
226
|
+
provider: this.name,
|
|
227
|
+
command: `${this.name} (Mastra)`,
|
|
228
|
+
success: true,
|
|
229
|
+
taskId,
|
|
230
|
+
model: this.config.model,
|
|
231
|
+
};
|
|
232
|
+
const allToolCalls = [];
|
|
233
|
+
const steps = fullOutput.steps || [];
|
|
234
|
+
for (const step of steps) {
|
|
235
|
+
const stepToolCalls = step.toolCalls || [];
|
|
236
|
+
const stepToolResults = step.toolResults || [];
|
|
237
|
+
for (let i = 0; i < stepToolCalls.length; i++) {
|
|
238
|
+
const tc = stepToolCalls[i];
|
|
239
|
+
const tr = stepToolResults[i];
|
|
240
|
+
const toolName = this.extractToolName(tc);
|
|
241
|
+
const toolArgs = this.extractToolArgs(tc);
|
|
242
|
+
const toolResultValue = this.extractToolResult(tr);
|
|
243
|
+
console.log(`[INFO] Tool called: ${toolName}`);
|
|
244
|
+
if (toolResultValue) {
|
|
245
|
+
const resultStr = typeof toolResultValue === 'string'
|
|
246
|
+
? toolResultValue
|
|
247
|
+
: JSON.stringify(toolResultValue);
|
|
248
|
+
const preview = resultStr.substring(0, this.logConfig.toolResultMaxLength);
|
|
249
|
+
console.log(`[INFO] Tool result preview: ${preview}${resultStr.length > this.logConfig.toolResultMaxLength ? '...' : ''}`);
|
|
250
|
+
}
|
|
251
|
+
allToolCalls.push({
|
|
252
|
+
toolName,
|
|
253
|
+
toolInput: toolArgs,
|
|
254
|
+
toolResult: toolResultValue,
|
|
255
|
+
});
|
|
256
|
+
}
|
|
257
|
+
}
|
|
258
|
+
if (allToolCalls.length === 0 && fullOutput.toolCalls?.length > 0) {
|
|
259
|
+
for (let i = 0; i < fullOutput.toolCalls.length; i++) {
|
|
260
|
+
const tc = fullOutput.toolCalls[i];
|
|
261
|
+
const tr = fullOutput.toolResults?.[i];
|
|
262
|
+
const toolName = this.extractToolName(tc);
|
|
263
|
+
const toolArgs = this.extractToolArgs(tc);
|
|
264
|
+
const toolResultValue = this.extractToolResult(tr);
|
|
265
|
+
console.log(`[INFO] Tool called: ${toolName}`);
|
|
266
|
+
allToolCalls.push({
|
|
267
|
+
toolName,
|
|
268
|
+
toolInput: toolArgs,
|
|
269
|
+
toolResult: toolResultValue,
|
|
270
|
+
});
|
|
271
|
+
}
|
|
272
|
+
}
|
|
273
|
+
for (let stepIdx = 0; stepIdx < steps.length; stepIdx++) {
|
|
274
|
+
const step = steps[stepIdx];
|
|
275
|
+
const stepToolCalls = step.toolCalls || [];
|
|
276
|
+
const stepToolResults = step.toolResults || [];
|
|
277
|
+
for (let i = 0; i < stepToolCalls.length; i++) {
|
|
278
|
+
const tc = stepToolCalls[i];
|
|
279
|
+
const tr = stepToolResults[i];
|
|
280
|
+
const toolName = this.extractToolName(tc);
|
|
281
|
+
const toolArgs = this.extractToolArgs(tc);
|
|
282
|
+
const toolResult = this.extractToolResult(tr);
|
|
283
|
+
const toolUseId = `toolu_api_${stepIdx}_${i}`;
|
|
284
|
+
this.emitLog(taskId, 'STDOUT', JSON.stringify({
|
|
285
|
+
type: 'assistant',
|
|
286
|
+
message: {
|
|
287
|
+
content: [{
|
|
288
|
+
type: 'tool_use',
|
|
289
|
+
id: toolUseId,
|
|
290
|
+
name: toolName,
|
|
291
|
+
input: toolArgs,
|
|
292
|
+
}],
|
|
293
|
+
},
|
|
294
|
+
}));
|
|
295
|
+
if (toolResult !== undefined) {
|
|
296
|
+
const output = typeof toolResult === 'string'
|
|
297
|
+
? toolResult
|
|
298
|
+
: JSON.stringify(toolResult);
|
|
299
|
+
this.emitLog(taskId, 'STDOUT', JSON.stringify({
|
|
300
|
+
type: 'user',
|
|
301
|
+
message: {
|
|
302
|
+
content: [{
|
|
303
|
+
type: 'tool_result',
|
|
304
|
+
tool_use_id: toolUseId,
|
|
305
|
+
output,
|
|
306
|
+
}],
|
|
307
|
+
},
|
|
308
|
+
}));
|
|
309
|
+
}
|
|
310
|
+
}
|
|
311
|
+
}
|
|
312
|
+
if (content) {
|
|
313
|
+
this.emitLog(taskId, 'STDOUT', JSON.stringify({
|
|
314
|
+
type: 'assistant',
|
|
315
|
+
message: {
|
|
316
|
+
content: [{ type: 'text', text: content }],
|
|
317
|
+
},
|
|
318
|
+
}));
|
|
319
|
+
}
|
|
320
|
+
if (allToolCalls.length > 0) {
|
|
321
|
+
console.log(`[INFO] Total tool calls across ${steps.length} step(s): ${allToolCalls.length}`);
|
|
322
|
+
response.toolCall = allToolCalls[0];
|
|
323
|
+
response.toolCalls = allToolCalls;
|
|
324
|
+
response.steps = steps.length;
|
|
325
|
+
if (!content && allToolCalls.length > 0) {
|
|
326
|
+
const lastCall = allToolCalls[allToolCalls.length - 1];
|
|
327
|
+
if (lastCall.toolResult) {
|
|
328
|
+
content = `Tool '${lastCall.toolName}' executed successfully:\n\n${lastCall.toolResult}`;
|
|
329
|
+
response.content = content;
|
|
330
|
+
}
|
|
331
|
+
}
|
|
332
|
+
}
|
|
333
|
+
return response;
|
|
334
|
+
}
|
|
335
|
+
async isAvailable() {
|
|
336
|
+
const { provider, apiKey } = this.config;
|
|
337
|
+
if (provider === 'api/ollama') {
|
|
338
|
+
return true;
|
|
339
|
+
}
|
|
340
|
+
if (apiKey) {
|
|
341
|
+
return true;
|
|
342
|
+
}
|
|
343
|
+
switch (provider) {
|
|
344
|
+
case 'api/openai':
|
|
345
|
+
return !!process.env.OPENAI_API_KEY;
|
|
346
|
+
case 'api/anthropic':
|
|
347
|
+
return !!process.env.ANTHROPIC_API_KEY;
|
|
348
|
+
case 'api/google':
|
|
349
|
+
return !!process.env.GOOGLE_API_KEY;
|
|
350
|
+
case 'api/bedrock':
|
|
351
|
+
return !!process.env.AWS_ACCESS_KEY_ID;
|
|
352
|
+
case 'api/litellm':
|
|
353
|
+
return !!process.env.LITELLM_API_KEY;
|
|
354
|
+
case 'api/sowonai':
|
|
355
|
+
return !!process.env.SOWONAI_API_KEY;
|
|
356
|
+
default:
|
|
357
|
+
return false;
|
|
358
|
+
}
|
|
359
|
+
}
|
|
360
|
+
async getToolPath() {
|
|
361
|
+
return null;
|
|
362
|
+
}
|
|
363
|
+
}
|
|
364
|
+
exports.MastraAPIProvider = MastraAPIProvider;
|
|
365
|
+
//# sourceMappingURL=MastraAPIProvider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"MastraAPIProvider.js","sourceRoot":"","sources":["../../../src/core/providers/MastraAPIProvider.ts"],"names":[],"mappings":";;;AAgBA,uCAAqC;AACrC,2CAAsD;AACtD,iDAA+D;AAC/D,2CAAkE;AAClE,iEAA+D;AAC/D,iEAAmE;AAQnE,uEAAoF;AACpF,wEAAqE;AACrE,iFAG6C;AAC7C,wDAAuE;AAoBvE,MAAa,iBAAiB;IAU5B,YAAY,MAAyB,EAAE,OAA8B,OAAO;QANpE,qBAAgB,GAA0B,EAAE,CAAC;QAOnD,MAAM,gBAAgB,GAAG,IAAA,oDAA0B,EAAC,MAAM,CAAC,CAAC;QAC5D,IAAI,CAAC,MAAM,GAAG,gBAAgB,CAAC,MAAM,CAAC;QACtC,IAAI,CAAC,iBAAiB,GAAG,gBAAgB,CAAC,iBAAiB,CAAC;QAC5D,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,SAAS,GAAG,IAAA,yBAAY,GAAE,CAAC;IAClC,CAAC;IAMD,iBAAiB,CAAC,OAA+B;QAC/C,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;IAChC,CAAC;IAEO,OAAO,CAAC,MAAc,EAAE,KAAwB,EAAE,OAAe;QACvE,IAAI,CAAC,IAAI,CAAC,cAAc;YAAE,OAAO;QACjC,IAAI,CAAC,cAAc,CAAC;YAClB,MAAM;YACN,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,KAAK;YACL,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAiBO,WAAW,CAAC,MAAyB;QAC3C,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,MAAM,CAAC;QAEhD,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,YAAY,CAAC,CAAC,CAAC;gBAElB,IAAI,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;oBACzC,OAAO,CAAC,GAAG,CAAC,4EAA4E,CAAC,CAAC;oBAC1F,MAAM,UAAU,GAAG,IAAA,kCAAgB,EAAC;wBAClC,MAAM,EAAE,MAAM,IAAI,EAAE;qBACrB,CAAC,CAAC;oBACH,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC;gBAC3B,CAAC;gBAGD,IAAI,MAAM,IAAI,GAAG,EAAE,CAAC;oBAClB,MAAM,YAAY,GAAG,IAAA,qBAAY,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;oBAC5D,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC;gBAC7B,CAAC;qBAAM,IAAI,MAAM,EAAE,CAAC;oBAClB,MAAM,YAAY,GAAG,IAAA,qBAAY,EAAC,EAAE,MAAM,EAAE,CAAC,CAAC;oBAC9C,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC;gBAC7B,CAAC;gBACD,OAAO,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC;YACvB,CAAC;YAED,KAAK,aAAa,CAAC;YACnB,KAAK,YAAY,CAAC;YAClB,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,IAAI,UAAU,GAAG,uBAAuB,CAAC;gBACzC,IAAI,UAAU,GAAG,OAAO,CAAC;gBACzB,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;oBAC9B,UAAU,GAAG,2BAA2B,CAAC;oBACzC,UAAU,GAAG,QAAQ,CAAC;gBACxB,CAAC;qBAAM,IAAI,QAAQ,KAAK,aAAa,EAAE,CAAC;oBACtC,UAAU,GAAG,yBAAyB,CAAC;gBACzC,CAAC;gBACD,MAAM,gBAAgB,GAAG,IAAA,0CAAsB,EAAC;oBAC9C,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;oBAClC,OAAO,EAAE,GAAG,IAAI,UAAU;oBAC1B,MAAM,EAAE,MAAM,IAAI,UAAU;iBAC7B,CAAC,CAAC;gBACH,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;YACjC,CAAC;YAED,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,IAAI,MAAM,IAAI,GAAG,EAAE,CAAC;oBAClB,MAAM,eAAe,GAAG,IAAA,2BAAe,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;oBAClE,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;gBAChC,CAAC;qBAAM,IAAI,MAAM,EAAE,CAAC;oBAClB,MAAM,eAAe,GAAG,IAAA,2BAAe,EAAC,EAAE,MAAM,EAAE,CAAC,CAAC;oBACpD,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;gBAChC,CAAC;gBACD,OAAO,IAAA,qBAAS,EAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;YAED,KAAK,aAAa,CAAC,CAAC,CAAC;gBACnB,MAAM,eAAe,GAAG,IAAA,2BAAe,EAAC;oBACtC,MAAM,EAAE,MAAM,IAAI,EAAE;oBACpB,OAAO,EAAE,GAAG,IAAI,iDAAiD;iBAClE,CAAC,CAAC;gBACH,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;YAChC,CAAC;YAED,KAAK,YAAY,CAAC,CAAC,CAAC;gBAClB,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,YAAY,GAAG,IAAA,iCAAwB,EAAC,EAAE,MAAM,EAAE,CAAC,CAAC;oBAC1D,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC;gBAC7B,CAAC;gBACD,OAAO,IAAA,eAAM,EAAC,KAAK,CAAC,CAAC;YACvB,CAAC;YAED;gBACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAYD,QAAQ,CAAC,KAAY,EAAE,OAA6B;QAClD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC;QAC3D,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAC7D,CAAC;IAEO,yBAAyB,CAC/B,OAA8B;QAE9B,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACjB,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,OAAO,EAAE,GAAG,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC;IAChD,CAAC;IAEO,sBAAsB,CAAC,KAAY;QACzC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAe,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAChD,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,IAAI,CAAC,uEAAuE,CAAC,CAAC;YACxF,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAA0B,EAAE,CAAC;QAC3C,KAAK,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACzE,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;gBAC9B,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACpB,SAAS;YACX,CAAC;YAED,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,KAAK;iBAC/B,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAChB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC1B,OAAO,CAAC,IAAI,CACV,6BAA6B,QAAQ,mBAAmB,IAAI,0BAA0B,CACvF,CAAC;gBACJ,CAAC;gBACD,OAAO,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC,CAAC;iBACD,MAAM,CAAC,OAAO,CAAC,CAAC;QACrB,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,iBAAiB,CAAC,IAAS;QACjC,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO,IAAI,CAAC,EAAE,CAAC;QACjB,CAAC;QACD,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1C,OAAO,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,wBAAwB,CAAC,IAA2B;QAC1D,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAC1D,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,eAAe,GAAyB,EAAE,GAAG,IAAI,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC;QAC5E,OAAO,qCAAiB,CAAC,YAAY,CAAC,eAAe,EAAE,eAAe,CAAC,CAAC;IAC1E,CAAC;IAYD,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,UAA0B,EAAE;QACtD,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IAYD,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,UAA0B,EAAE;QACxD,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,IAA2B,EAC3B,MAAc,EACd,OAAuB;QAEvB,MAAM,MAAM,GACV,OAAO,CAAC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QAE5F,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;YAGxD,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC;YAClD,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;gBAClB,OAAO,CAAC,GAAG,CAAC,qBAAqB,SAAS,cAAc,IAAI,UAAU,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC/G,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,kCAAkC,IAAI,OAAO,CAAC,CAAC;YAC7D,CAAC;YAGD,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK;gBAC/B,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE;gBAC1C,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;YAEhB,OAAO,CAAC,GAAG,CAAC,uBAAuB,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC;YAExD,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;YAEpD,MAAM,KAAK,GAAG,IAAI,YAAK,CAAC;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC1B,KAAK,EAAE,aAAa;gBACpB,YAAY,EAAE,MAAM;gBACpB,KAAK,EAAE,WAAW;aACnB,CAAC,CAAC;YAMH,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,sCAAiB,EAAE,oCAAe,CAAC,CAAC;YACtF,MAAM,eAAe,GAAG,EAAE,QAAQ,EAAE,CAAC;YAErC,OAAO,CAAC,GAAG,CAAC,iDAAiD,QAAQ,MAAM,CAAC,CAAC;YAC7E,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;YACjE,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;YAEtD,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAClD,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,OAAO,EAAE,EAAE;gBACX,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,OAAO,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;gBAC/B,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,eAAe;gBACvC,MAAM;aACP,CAAC;QACJ,CAAC;IACH,CAAC;IAKO,eAAe,CAAC,EAAO;QAC7B,OAAO,EAAE,CAAC,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAC,QAAQ,IAAI,SAAS,CAAC;IAC1D,CAAC;IAKO,eAAe,CAAC,EAAO;QAC7B,OAAO,EAAE,CAAC,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,IAAI,CAAC;IACrC,CAAC;IAKO,iBAAiB,CAAC,EAAO;QAC/B,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,IAAI,EAAE,EAAE,MAAM,CAAC;IAC3C,CAAC;IAYO,eAAe,CAAC,UAAe,EAAE,MAAc;QACrD,IAAI,OAAO,GAAG,UAAU,CAAC,IAAI,IAAI,EAAE,CAAC;QAEpC,MAAM,QAAQ,GAAe;YAC3B,OAAO;YACP,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,OAAO,EAAE,GAAG,IAAI,CAAC,IAAI,WAAW;YAChC,OAAO,EAAE,IAAI;YACb,MAAM;YACN,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;SACzB,CAAC;QAKF,MAAM,YAAY,GAAiE,EAAE,CAAC;QACtF,MAAM,KAAK,GAAU,UAAU,CAAC,KAAK,IAAI,EAAE,CAAC;QAE5C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,aAAa,GAAU,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;YAClD,MAAM,eAAe,GAAU,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;YAEtD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9C,MAAM,EAAE,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;gBAC5B,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;gBAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;gBAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;gBAC1C,MAAM,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;gBAEnD,OAAO,CAAC,GAAG,CAAC,uBAAuB,QAAQ,EAAE,CAAC,CAAC;gBAE/C,IAAI,eAAe,EAAE,CAAC;oBACpB,MAAM,SAAS,GAAG,OAAO,eAAe,KAAK,QAAQ;wBACnD,CAAC,CAAC,eAAe;wBACjB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;oBACpC,MAAM,OAAO,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;oBAC3E,OAAO,CAAC,GAAG,CAAC,+BAA+B,OAAO,GAAG,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC7H,CAAC;gBAED,YAAY,CAAC,IAAI,CAAC;oBAChB,QAAQ;oBACR,SAAS,EAAE,QAAQ;oBACnB,UAAU,EAAE,eAAe;iBAC5B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAGD,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,CAAC,EAAE,CAAC;YAClE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACrD,MAAM,EAAE,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACnC,MAAM,EAAE,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;gBACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;gBAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;gBAC1C,MAAM,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;gBAEnD,OAAO,CAAC,GAAG,CAAC,uBAAuB,QAAQ,EAAE,CAAC,CAAC;gBAC/C,YAAY,CAAC,IAAI,CAAC;oBAChB,QAAQ;oBACR,SAAS,EAAE,QAAQ;oBACnB,UAAU,EAAE,eAAe;iBAC5B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAGD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC;YACxD,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5B,MAAM,aAAa,GAAU,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;YAClD,MAAM,eAAe,GAAU,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;YAEtD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9C,MAAM,EAAE,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;gBAC5B,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,CAAC,CAAC;gBAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;gBAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;gBAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;gBAC9C,MAAM,SAAS,GAAG,aAAa,OAAO,IAAI,CAAC,EAAE,CAAC;gBAG9C,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC;oBAC5C,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE;wBACP,OAAO,EAAE,CAAC;gCACR,IAAI,EAAE,UAAU;gCAChB,EAAE,EAAE,SAAS;gCACb,IAAI,EAAE,QAAQ;gCACd,KAAK,EAAE,QAAQ;6BAChB,CAAC;qBACH;iBACF,CAAC,CAAC,CAAC;gBAGJ,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;oBAC7B,MAAM,MAAM,GAAG,OAAO,UAAU,KAAK,QAAQ;wBAC3C,CAAC,CAAC,UAAU;wBACZ,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;oBAC/B,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC;wBAC5C,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE;4BACP,OAAO,EAAE,CAAC;oCACR,IAAI,EAAE,aAAa;oCACnB,WAAW,EAAE,SAAS;oCACtB,MAAM;iCACP,CAAC;yBACH;qBACF,CAAC,CAAC,CAAC;gBACN,CAAC;YACH,CAAC;QACH,CAAC;QAGD,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC;gBAC5C,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;iBAC3C;aACF,CAAC,CAAC,CAAC;QACN,CAAC;QAED,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,kCAAkC,KAAK,CAAC,MAAM,aAAa,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;YAG9F,QAAQ,CAAC,QAAQ,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;YAEpC,QAAQ,CAAC,SAAS,GAAG,YAAY,CAAC;YAClC,QAAQ,CAAC,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC;YAG9B,IAAI,CAAC,OAAO,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxC,MAAM,QAAQ,GAAG,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAE,CAAC;gBACxD,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;oBACxB,OAAO,GAAG,SAAS,QAAQ,CAAC,QAAQ,+BAA+B,QAAQ,CAAC,UAAU,EAAE,CAAC;oBACzF,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC;gBAC7B,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAUD,KAAK,CAAC,WAAW;QAEf,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAGzC,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC;QACd,CAAC;QAGD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,IAAI,CAAC;QACd,CAAC;QAGD,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,YAAY;gBACf,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;YACtC,KAAK,eAAe;gBAClB,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;YACzC,KAAK,YAAY;gBACf,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;YACtC,KAAK,aAAa;gBAChB,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;YACzC,KAAK,aAAa;gBAChB,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;YACvC,KAAK,aAAa;gBAChB,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;YACvC;gBACE,OAAO,KAAK,CAAC;QACjB,CAAC;IACH,CAAC;IAUD,KAAK,CAAC,WAAW;QACf,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAhhBD,8CAghBC"}
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
import type { TaskLogEntry } from '../../types/task-log.types';
|
|
2
|
+
export declare const ProviderNamespace: {
|
|
3
|
+
readonly CLI: "cli";
|
|
4
|
+
readonly PLUGIN: "plugin";
|
|
5
|
+
readonly API: "api";
|
|
6
|
+
readonly REMOTE: "remote";
|
|
7
|
+
};
|
|
8
|
+
export type ProviderNamespaceType = typeof ProviderNamespace[keyof typeof ProviderNamespace];
|
|
9
|
+
export declare const BuiltInProviders: {
|
|
10
|
+
readonly CLAUDE: "cli/claude";
|
|
11
|
+
readonly GEMINI: "cli/gemini";
|
|
12
|
+
readonly COPILOT: "cli/copilot";
|
|
13
|
+
readonly CODEX: "cli/codex";
|
|
14
|
+
};
|
|
15
|
+
export interface AIQueryOptions {
|
|
16
|
+
timeout?: number;
|
|
17
|
+
workingDirectory?: string;
|
|
18
|
+
additionalArgs?: string[];
|
|
19
|
+
taskId?: string;
|
|
20
|
+
model?: string;
|
|
21
|
+
securityKey?: string;
|
|
22
|
+
agentId?: string;
|
|
23
|
+
messages?: Array<{
|
|
24
|
+
text: string;
|
|
25
|
+
isAssistant: boolean;
|
|
26
|
+
metadata?: Record<string, any>;
|
|
27
|
+
}>;
|
|
28
|
+
pipedContext?: string;
|
|
29
|
+
env?: Record<string, string>;
|
|
30
|
+
runtime?: 'native' | 'docker' | 'podman' | {
|
|
31
|
+
type?: 'native' | 'docker' | 'podman';
|
|
32
|
+
image?: string;
|
|
33
|
+
mountPath?: string;
|
|
34
|
+
dockerArgs?: string[];
|
|
35
|
+
};
|
|
36
|
+
traceId?: string;
|
|
37
|
+
onProcessStart?: (pid: number) => void;
|
|
38
|
+
}
|
|
39
|
+
export interface AIResponse {
|
|
40
|
+
content: string;
|
|
41
|
+
provider: string;
|
|
42
|
+
command: string;
|
|
43
|
+
success: boolean;
|
|
44
|
+
error?: string;
|
|
45
|
+
taskId?: string;
|
|
46
|
+
model?: string;
|
|
47
|
+
toolCall?: {
|
|
48
|
+
toolName: string;
|
|
49
|
+
toolInput: any;
|
|
50
|
+
toolResult: any;
|
|
51
|
+
};
|
|
52
|
+
toolCalls?: Array<{
|
|
53
|
+
toolName: string;
|
|
54
|
+
toolInput: any;
|
|
55
|
+
toolResult: any;
|
|
56
|
+
}>;
|
|
57
|
+
steps?: number;
|
|
58
|
+
pid?: number;
|
|
59
|
+
exitCode?: number | null;
|
|
60
|
+
durationMs?: number;
|
|
61
|
+
timeToFirstOutputMs?: number;
|
|
62
|
+
promptLength?: number;
|
|
63
|
+
usage?: {
|
|
64
|
+
inputTokens: number;
|
|
65
|
+
outputTokens: number;
|
|
66
|
+
};
|
|
67
|
+
logs?: TaskLogEntry[];
|
|
68
|
+
}
|
|
69
|
+
export interface AIProvider {
|
|
70
|
+
readonly name: string;
|
|
71
|
+
isAvailable(): Promise<boolean>;
|
|
72
|
+
query(prompt: string, options?: AIQueryOptions): Promise<AIResponse>;
|
|
73
|
+
execute(prompt: string, options?: AIQueryOptions): Promise<AIResponse>;
|
|
74
|
+
getToolPath(): Promise<string | null>;
|
|
75
|
+
}
|
|
76
|
+
export declare class ProviderNotAvailableError extends Error {
|
|
77
|
+
constructor(providerName: string);
|
|
78
|
+
}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ProviderNotAvailableError = exports.BuiltInProviders = exports.ProviderNamespace = void 0;
|
|
4
|
+
exports.ProviderNamespace = {
|
|
5
|
+
CLI: 'cli',
|
|
6
|
+
PLUGIN: 'plugin',
|
|
7
|
+
API: 'api',
|
|
8
|
+
REMOTE: 'remote',
|
|
9
|
+
};
|
|
10
|
+
exports.BuiltInProviders = {
|
|
11
|
+
CLAUDE: 'cli/claude',
|
|
12
|
+
GEMINI: 'cli/gemini',
|
|
13
|
+
COPILOT: 'cli/copilot',
|
|
14
|
+
CODEX: 'cli/codex',
|
|
15
|
+
};
|
|
16
|
+
class ProviderNotAvailableError extends Error {
|
|
17
|
+
constructor(providerName) {
|
|
18
|
+
super(`AI Provider '${providerName}' is not available or not installed`);
|
|
19
|
+
this.name = 'ProviderNotAvailableError';
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
exports.ProviderNotAvailableError = ProviderNotAvailableError;
|
|
23
|
+
//# sourceMappingURL=ai-provider.interface.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ai-provider.interface.js","sourceRoot":"","sources":["../../../src/core/providers/ai-provider.interface.ts"],"names":[],"mappings":";;;AAIa,QAAA,iBAAiB,GAAG;IAC/B,GAAG,EAAE,KAAK;IACV,MAAM,EAAE,QAAQ;IAChB,GAAG,EAAE,KAAK;IACV,MAAM,EAAE,QAAQ;CACR,CAAC;AAIE,QAAA,gBAAgB,GAAG;IAC9B,MAAM,EAAE,YAAY;IACpB,MAAM,EAAE,YAAY;IACpB,OAAO,EAAE,aAAa;IACtB,KAAK,EAAE,WAAW;CACV,CAAC;AA+EX,MAAa,yBAA0B,SAAQ,KAAK;IAClD,YAAY,YAAoB;QAC9B,KAAK,CAAC,gBAAgB,YAAY,qCAAqC,CAAC,CAAC;QACzE,IAAI,CAAC,IAAI,GAAG,2BAA2B,CAAC;IAC1C,CAAC;CACF;AALD,8DAKC"}
|
|
@@ -0,0 +1,80 @@
|
|
|
1
|
+
import { type TimeoutConfig } from '../../config/timeout.config';
|
|
2
|
+
import { type LogConfig } from '../../config/log.config';
|
|
3
|
+
import type { AIProvider, AIQueryOptions, AIResponse } from './ai-provider.interface';
|
|
4
|
+
import type { BaseAIProviderOptions, LoggerLike } from './base-ai.types';
|
|
5
|
+
import type { ToolCallHandler } from './tool-call.types';
|
|
6
|
+
export declare class ConsoleLogger implements LoggerLike {
|
|
7
|
+
private readonly context;
|
|
8
|
+
constructor(context: string);
|
|
9
|
+
log(message: string, ...optionalParams: any[]): void;
|
|
10
|
+
warn(message: string, ...optionalParams: any[]): void;
|
|
11
|
+
error(message: string | Error, ...optionalParams: any[]): void;
|
|
12
|
+
}
|
|
13
|
+
export declare abstract class BaseAIProvider implements AIProvider {
|
|
14
|
+
abstract readonly name: string;
|
|
15
|
+
protected readonly logger: LoggerLike;
|
|
16
|
+
protected toolCallHandler?: ToolCallHandler;
|
|
17
|
+
private readonly logsDir;
|
|
18
|
+
private readonly crewxVersion;
|
|
19
|
+
private readonly taskLogHandler?;
|
|
20
|
+
private cachedPath;
|
|
21
|
+
protected readonly timeoutConfig: TimeoutConfig;
|
|
22
|
+
protected readonly logConfig: LogConfig;
|
|
23
|
+
constructor(loggerContext: string, options?: BaseAIProviderOptions);
|
|
24
|
+
protected abstract getCliCommand(): string;
|
|
25
|
+
protected abstract getDefaultArgs(): string[];
|
|
26
|
+
protected abstract getExecuteArgs(): string[];
|
|
27
|
+
protected abstract getNotInstalledMessage(): string;
|
|
28
|
+
protected getPromptInArgs(): boolean;
|
|
29
|
+
protected getDefaultQueryTimeout(): number;
|
|
30
|
+
protected getDefaultExecuteTimeout(): number;
|
|
31
|
+
protected getDefaultModel(): string | null;
|
|
32
|
+
protected getEnv(): Record<string, string>;
|
|
33
|
+
private filterClaudePrefixedEnv;
|
|
34
|
+
private resolveRuntimeMode;
|
|
35
|
+
private resolveContainerCommand;
|
|
36
|
+
private resolveDockerImage;
|
|
37
|
+
private resolveDockerMountPath;
|
|
38
|
+
private resolveDockerArgs;
|
|
39
|
+
private ensureContainerCredentials;
|
|
40
|
+
private buildDockerSpawnArgs;
|
|
41
|
+
protected substituteModelPlaceholders(args: string[], model: string): string[];
|
|
42
|
+
protected setToolCallService(toolCallHandler: ToolCallHandler): void;
|
|
43
|
+
protected setToolCallHandler(toolCallHandler: ToolCallHandler): void;
|
|
44
|
+
protected parseToolUse(content: string): {
|
|
45
|
+
isToolUse: boolean;
|
|
46
|
+
toolName?: string;
|
|
47
|
+
toolInput?: any;
|
|
48
|
+
};
|
|
49
|
+
protected parseToolUseProviderSpecific(_parsed: any): {
|
|
50
|
+
isToolUse: boolean;
|
|
51
|
+
toolName?: string;
|
|
52
|
+
toolInput?: any;
|
|
53
|
+
};
|
|
54
|
+
protected parseUsage(stdout: string): {
|
|
55
|
+
inputTokens: number;
|
|
56
|
+
outputTokens: number;
|
|
57
|
+
} | undefined;
|
|
58
|
+
protected extractAgentMessageFromJsonLines(content: string): string | null;
|
|
59
|
+
private extractAgentMessageFromJsonItem;
|
|
60
|
+
protected filterToolUseFromResponse(content: string): string;
|
|
61
|
+
protected filterRuntimeLogs(content: string): string;
|
|
62
|
+
private shouldStripPromptContainers;
|
|
63
|
+
parseProviderError(stderr: string, stdout: string, exitCode?: number | null): {
|
|
64
|
+
error: boolean;
|
|
65
|
+
message: string;
|
|
66
|
+
};
|
|
67
|
+
getToolPath(): Promise<string | null>;
|
|
68
|
+
protected wrapUserQueryWithSecurity(userQuery: string, securityKey: string): string;
|
|
69
|
+
protected extractUserQuery(wrappedQuery: string, securityKey: string): string;
|
|
70
|
+
isAvailable(): Promise<boolean>;
|
|
71
|
+
private createTaskLogFile;
|
|
72
|
+
protected appendTaskLog(taskId: string, level: 'STDOUT' | 'STDERR' | 'INFO' | 'ERROR', message: string): void;
|
|
73
|
+
query(prompt: string, options?: AIQueryOptions): Promise<AIResponse>;
|
|
74
|
+
execute(prompt: string, options?: AIQueryOptions): Promise<AIResponse>;
|
|
75
|
+
protected shouldPipeContext(_options?: AIQueryOptions): boolean;
|
|
76
|
+
private notifyProcessStart;
|
|
77
|
+
private buildPipedContext;
|
|
78
|
+
protected isStructuredPayload(value: string): boolean;
|
|
79
|
+
protected createStructuredPayload(prompt: string, context: string | null, options?: AIQueryOptions): string;
|
|
80
|
+
}
|