@traisetech/autopilot 2.1.0 → 2.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +60 -0
- package/bin/autopilot.js +9 -1
- package/docs/CONFIGURATION.md +12 -9
- package/docs/DESIGN_PRINCIPLES.md +62 -6
- package/package.json +1 -1
- package/src/commands/dashboard.mjs +16 -12
- package/src/commands/doctor.js +42 -4
- package/src/commands/init.js +21 -21
- package/src/commands/insights.js +33 -67
- package/src/commands/interactive.js +40 -0
- package/src/commands/leaderboard.js +9 -10
- package/src/config/defaults.js +8 -7
- package/src/config/loader.js +7 -0
- package/src/core/commit.js +19 -7
- package/src/core/events.js +16 -11
- package/src/core/gemini.js +15 -7
- package/src/core/git.js +6 -5
- package/src/core/grok.js +140 -44
- package/src/core/keys.js +77 -0
- package/src/core/watcher.js +15 -1
- package/src/index.js +70 -64
- package/src/utils/logger.js +7 -7
- package/src/utils/obfuscator.js +32 -0
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Interactive Command
|
|
3
|
+
* Toggle between Safety Mode (Interactive) and Full Autopilot (Automated)
|
|
4
|
+
*/
|
|
5
|
+
|
|
6
|
+
const logger = require('../utils/logger');
|
|
7
|
+
const { loadConfig, saveConfig } = require('../config/loader');
|
|
8
|
+
|
|
9
|
+
async function interactive(state, options) {
|
|
10
|
+
const repoPath = options?.cwd || process.cwd();
|
|
11
|
+
const isGlobal = options?.global || false;
|
|
12
|
+
|
|
13
|
+
if (!state) {
|
|
14
|
+
const config = await loadConfig(repoPath);
|
|
15
|
+
const current = config.ai?.interactive;
|
|
16
|
+
logger.info(`Current AI Mode: ${current ? '🛡️ Safety (Manual Approval)' : '🚀 Full Autopilot (Automated)'}`);
|
|
17
|
+
logger.info('Usage: autopilot interactive <on|off>');
|
|
18
|
+
return;
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
const newState = state.toLowerCase() === 'on';
|
|
22
|
+
|
|
23
|
+
// Load existing config to modify
|
|
24
|
+
const config = await loadConfig(repoPath);
|
|
25
|
+
|
|
26
|
+
// Ensure AI object exists
|
|
27
|
+
if (!config.ai) config.ai = {};
|
|
28
|
+
|
|
29
|
+
config.ai.interactive = newState;
|
|
30
|
+
|
|
31
|
+
await saveConfig(repoPath, config, isGlobal);
|
|
32
|
+
|
|
33
|
+
if (newState) {
|
|
34
|
+
logger.success(`🛡️ AI Safety Mode enabled ${isGlobal ? '(Global)' : '(Local)'}. Autopilot will ask for approval before committing.`);
|
|
35
|
+
} else {
|
|
36
|
+
logger.success(`🚀 Full Autopilot enabled ${isGlobal ? '(Global)' : '(Local)'}. Autopilot will now commit and push automatically.`);
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
module.exports = interactive;
|
|
@@ -2,11 +2,10 @@ const fs = require('fs-extra');
|
|
|
2
2
|
const path = require('path');
|
|
3
3
|
const { getGitStats, calculateMetrics } = require('./insights');
|
|
4
4
|
const logger = require('../utils/logger');
|
|
5
|
-
const open = require('open');
|
|
6
5
|
const crypto = require('crypto');
|
|
7
6
|
|
|
8
|
-
// Default API URL (can be overridden by
|
|
9
|
-
const DEFAULT_API_URL = '
|
|
7
|
+
// Default API URL (can be overridden by env)
|
|
8
|
+
const DEFAULT_API_URL = 'https://autopilot-cli.vercel.app';
|
|
10
9
|
|
|
11
10
|
async function calculateFocusTime(repoPath) {
|
|
12
11
|
const logPath = path.join(repoPath, 'autopilot.log');
|
|
@@ -16,7 +15,7 @@ async function calculateFocusTime(repoPath) {
|
|
|
16
15
|
const content = await fs.readFile(logPath, 'utf8');
|
|
17
16
|
const lines = content.split('\n').filter(l => l.trim());
|
|
18
17
|
let totalMs = 0;
|
|
19
|
-
|
|
18
|
+
|
|
20
19
|
for (const line of lines) {
|
|
21
20
|
try {
|
|
22
21
|
const entry = JSON.parse(line);
|
|
@@ -27,7 +26,7 @@ async function calculateFocusTime(repoPath) {
|
|
|
27
26
|
// ignore bad lines
|
|
28
27
|
}
|
|
29
28
|
}
|
|
30
|
-
|
|
29
|
+
|
|
31
30
|
return Math.round(totalMs / 60000); // minutes
|
|
32
31
|
} catch (error) {
|
|
33
32
|
logger.warn(`Failed to parse autopilot.log: ${error.message}`);
|
|
@@ -59,15 +58,15 @@ async function syncLeaderboard(apiUrl, options) {
|
|
|
59
58
|
}
|
|
60
59
|
|
|
61
60
|
const metrics = calculateMetrics(commits);
|
|
62
|
-
|
|
61
|
+
|
|
63
62
|
// Get user info (git config)
|
|
64
63
|
const git = require('../core/git');
|
|
65
64
|
const { stdout: username } = await git.runGit(repoPath, ['config', 'user.name']);
|
|
66
65
|
const { stdout: email } = await git.runGit(repoPath, ['config', 'user.email']);
|
|
67
|
-
|
|
66
|
+
|
|
68
67
|
const userEmail = email.trim() || 'unknown';
|
|
69
68
|
const userName = username.trim() || 'Anonymous';
|
|
70
|
-
|
|
69
|
+
|
|
71
70
|
// Anonymize ID using hash
|
|
72
71
|
const userId = crypto.createHash('sha256').update(userEmail).digest('hex').substring(0, 12);
|
|
73
72
|
|
|
@@ -87,7 +86,7 @@ async function syncLeaderboard(apiUrl, options) {
|
|
|
87
86
|
|
|
88
87
|
logger.info(`Syncing stats for ${stats.username} (ID: ${userId})...`);
|
|
89
88
|
logger.info('Note: Only metrics are shared. No code or file contents are transmitted.');
|
|
90
|
-
|
|
89
|
+
|
|
91
90
|
const response = await fetch(`${apiUrl}/api/leaderboard/sync`, {
|
|
92
91
|
method: 'POST',
|
|
93
92
|
headers: { 'Content-Type': 'application/json' },
|
|
@@ -107,4 +106,4 @@ async function syncLeaderboard(apiUrl, options) {
|
|
|
107
106
|
}
|
|
108
107
|
}
|
|
109
108
|
|
|
110
|
-
module.exports = { leaderboard };
|
|
109
|
+
module.exports = { leaderboard, syncLeaderboard };
|
package/src/config/defaults.js
CHANGED
|
@@ -7,19 +7,20 @@ const DEFAULT_CONFIG = {
|
|
|
7
7
|
debounceSeconds: 20,
|
|
8
8
|
minSecondsBetweenCommits: 180,
|
|
9
9
|
autoPush: true,
|
|
10
|
-
|
|
10
|
+
blockedBranches: ['main', 'master'],
|
|
11
11
|
requireChecks: false,
|
|
12
12
|
checks: [],
|
|
13
|
-
commitMessageMode: '
|
|
13
|
+
commitMessageMode: 'ai', // Default to AI for zero-config
|
|
14
14
|
ai: {
|
|
15
|
-
enabled:
|
|
16
|
-
provider: '
|
|
15
|
+
enabled: true, // Enabled by default
|
|
16
|
+
provider: 'grok', // Grok is the default for our system keys
|
|
17
17
|
apiKey: '',
|
|
18
18
|
grokApiKey: '',
|
|
19
|
-
model: '
|
|
20
|
-
grokModel: 'grok-beta',
|
|
21
|
-
interactive: true
|
|
19
|
+
model: 'grok-beta',
|
|
20
|
+
grokModel: 'grok-beta',
|
|
21
|
+
interactive: true // Prompt user to review AI messages by default
|
|
22
22
|
},
|
|
23
|
+
|
|
23
24
|
// Phase 1: Team Mode
|
|
24
25
|
teamMode: false,
|
|
25
26
|
pullBeforePush: true,
|
package/src/config/loader.js
CHANGED
|
@@ -33,6 +33,13 @@ const loadConfig = async (repoPath) => {
|
|
|
33
33
|
logger.warn(`Error loading local config: ${error.message}`);
|
|
34
34
|
}
|
|
35
35
|
|
|
36
|
+
// Backward compatibility: map deprecated keys
|
|
37
|
+
try {
|
|
38
|
+
if (config.blockBranches && !config.blockedBranches) {
|
|
39
|
+
config.blockedBranches = config.blockBranches;
|
|
40
|
+
}
|
|
41
|
+
} catch (_) {}
|
|
42
|
+
|
|
36
43
|
return config;
|
|
37
44
|
};
|
|
38
45
|
|
package/src/core/commit.js
CHANGED
|
@@ -18,20 +18,30 @@ const HistoryManager = require('./history');
|
|
|
18
18
|
*/
|
|
19
19
|
async function generateCommitMessage(files, diffContent, config = {}) {
|
|
20
20
|
let message = '';
|
|
21
|
+
|
|
22
|
+
// Default to smart mode if not explicitly set (preserves unit tests)
|
|
23
|
+
// Real usage will usually have a config object with defaults from defaults.js
|
|
24
|
+
const mode = config.commitMessageMode || 'smart';
|
|
25
|
+
const aiEnabled = config.ai?.enabled !== false;
|
|
26
|
+
|
|
27
|
+
|
|
21
28
|
if (!files || files.length === 0) {
|
|
22
29
|
message = 'chore: update changes';
|
|
23
|
-
} else if (
|
|
30
|
+
} else if (mode === 'simple') {
|
|
31
|
+
message = 'chore: auto-commit changes';
|
|
32
|
+
} else if (mode === 'ai' && aiEnabled) {
|
|
24
33
|
// AI Mode
|
|
25
34
|
try {
|
|
26
|
-
|
|
35
|
+
// Default to grok as it supports our internal key pool strategy
|
|
36
|
+
const provider = config.ai?.provider || 'grok';
|
|
27
37
|
logger.info(`Generating AI commit message using ${provider}...`);
|
|
28
38
|
|
|
29
39
|
if (provider === 'grok') {
|
|
30
|
-
if
|
|
31
|
-
message = await grok.generateGrokCommitMessage(diffContent, config.ai
|
|
40
|
+
// use custom key if provided, otherwise the internal pool in grok.js will handle it
|
|
41
|
+
message = await grok.generateGrokCommitMessage(diffContent, config.ai?.grokApiKey, config.ai?.grokModel);
|
|
32
42
|
} else {
|
|
33
|
-
//
|
|
34
|
-
if (!config.ai
|
|
43
|
+
// Gemini fallback (requires user key)
|
|
44
|
+
if (!config.ai?.apiKey) throw new Error('Gemini API Key not configured');
|
|
35
45
|
message = await gemini.generateAICommitMessage(diffContent, config.ai.apiKey, config.ai.model);
|
|
36
46
|
}
|
|
37
47
|
} catch (error) {
|
|
@@ -39,10 +49,12 @@ async function generateCommitMessage(files, diffContent, config = {}) {
|
|
|
39
49
|
message = generateSmartCommitMessage(files, diffContent);
|
|
40
50
|
}
|
|
41
51
|
} else {
|
|
42
|
-
// Smart Mode (
|
|
52
|
+
// Smart Mode (Fallback)
|
|
43
53
|
message = generateSmartCommitMessage(files, diffContent);
|
|
44
54
|
}
|
|
45
55
|
|
|
56
|
+
|
|
57
|
+
|
|
46
58
|
// Prepend [autopilot] tag for traceability (Phase 1 req)
|
|
47
59
|
const finalMessage = `[autopilot] ${message}`;
|
|
48
60
|
|
package/src/core/events.js
CHANGED
|
@@ -88,17 +88,22 @@ class EventSystem {
|
|
|
88
88
|
* In production, this would POST to an API
|
|
89
89
|
*/
|
|
90
90
|
async sendToBackend(event) {
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
91
|
+
const apiBase = process.env.AUTOPILOT_API_URL || 'https://autopilot-cli.vercel.app';
|
|
92
|
+
const url = `${apiBase}/api/events`;
|
|
93
|
+
const controller = new AbortController();
|
|
94
|
+
const timeout = setTimeout(() => controller.abort(), 3000);
|
|
95
|
+
try {
|
|
96
|
+
const res = await fetch(url, {
|
|
97
|
+
method: 'POST',
|
|
98
|
+
headers: { 'Content-Type': 'application/json' },
|
|
99
|
+
body: JSON.stringify(event),
|
|
100
|
+
signal: controller.signal
|
|
101
|
+
});
|
|
102
|
+
if (!res.ok) throw new Error(`HTTP ${res.status}`);
|
|
103
|
+
return true;
|
|
104
|
+
} finally {
|
|
105
|
+
clearTimeout(timeout);
|
|
106
|
+
}
|
|
102
107
|
}
|
|
103
108
|
}
|
|
104
109
|
|
package/src/core/gemini.js
CHANGED
|
@@ -41,8 +41,10 @@ ${truncatedDiff}
|
|
|
41
41
|
`;
|
|
42
42
|
|
|
43
43
|
try {
|
|
44
|
-
const url = `${BASE_API_URL}${model}:generateContent?key=${apiKey}`;
|
|
45
|
-
const
|
|
44
|
+
const url = `${BASE_API_URL}${model}:generateContent?key=${apiKey}`;
|
|
45
|
+
const controller = new AbortController();
|
|
46
|
+
const timeout = setTimeout(() => controller.abort(), 5000);
|
|
47
|
+
const response = await fetch(url, {
|
|
46
48
|
method: 'POST',
|
|
47
49
|
headers: {
|
|
48
50
|
'Content-Type': 'application/json',
|
|
@@ -57,8 +59,10 @@ ${truncatedDiff}
|
|
|
57
59
|
temperature: 0.2,
|
|
58
60
|
maxOutputTokens: 256,
|
|
59
61
|
}
|
|
60
|
-
})
|
|
62
|
+
}),
|
|
63
|
+
signal: controller.signal
|
|
61
64
|
});
|
|
65
|
+
clearTimeout(timeout);
|
|
62
66
|
|
|
63
67
|
if (!response.ok) {
|
|
64
68
|
const errorData = await response.json().catch(() => ({}));
|
|
@@ -92,16 +96,20 @@ ${truncatedDiff}
|
|
|
92
96
|
*/
|
|
93
97
|
async function validateApiKey(apiKey, model = DEFAULT_MODEL) {
|
|
94
98
|
try {
|
|
95
|
-
const url = `${BASE_API_URL}${model}:generateContent?key=${apiKey}`;
|
|
96
|
-
// Simple test call with empty prompt
|
|
97
|
-
const
|
|
99
|
+
const url = `${BASE_API_URL}${model}:generateContent?key=${apiKey}`;
|
|
100
|
+
// Simple test call with empty prompt
|
|
101
|
+
const controller = new AbortController();
|
|
102
|
+
const timeout = setTimeout(() => controller.abort(), 4000);
|
|
103
|
+
const response = await fetch(url, {
|
|
98
104
|
method: 'POST',
|
|
99
105
|
headers: { 'Content-Type': 'application/json' },
|
|
100
106
|
body: JSON.stringify({
|
|
101
107
|
contents: [{ parts: [{ text: "Hi" }] }],
|
|
102
108
|
generationConfig: { maxOutputTokens: 1 }
|
|
103
|
-
})
|
|
109
|
+
}),
|
|
110
|
+
signal: controller.signal
|
|
104
111
|
});
|
|
112
|
+
clearTimeout(timeout);
|
|
105
113
|
|
|
106
114
|
if (!response.ok) {
|
|
107
115
|
const errorData = await response.json().catch(() => ({}));
|
package/src/core/git.js
CHANGED
|
@@ -42,13 +42,14 @@ async function getPorcelainStatus(root) {
|
|
|
42
42
|
try {
|
|
43
43
|
const { stdout } = await execa('git', ['status', '--porcelain'], { cwd: root });
|
|
44
44
|
const raw = stdout.trim();
|
|
45
|
-
|
|
45
|
+
|
|
46
46
|
if (!raw) {
|
|
47
47
|
return { ok: true, files: [], raw: '' };
|
|
48
48
|
}
|
|
49
49
|
|
|
50
50
|
const files = raw
|
|
51
51
|
.split(/\r?\n/)
|
|
52
|
+
.filter(line => line.trim().length > 0)
|
|
52
53
|
.map(line => {
|
|
53
54
|
const status = line.slice(0, 2).trim();
|
|
54
55
|
const file = line.slice(3).trim();
|
|
@@ -134,7 +135,7 @@ async function isRemoteAhead(root) {
|
|
|
134
135
|
|
|
135
136
|
const { stdout } = await execa('git', ['rev-list', '--left-right', '--count', `${branch}...origin/${branch}`], { cwd: root });
|
|
136
137
|
const [aheadCount, behindCount] = stdout.trim().split(/\s+/).map(Number);
|
|
137
|
-
|
|
138
|
+
|
|
138
139
|
return {
|
|
139
140
|
ok: true,
|
|
140
141
|
ahead: aheadCount > 0,
|
|
@@ -259,10 +260,10 @@ async function isMergeInProgress(root) {
|
|
|
259
260
|
'REVERT_HEAD',
|
|
260
261
|
'BISECT_LOG'
|
|
261
262
|
];
|
|
262
|
-
|
|
263
|
+
|
|
263
264
|
// Check if .git/rebase-merge or .git/rebase-apply exists (directory check)
|
|
264
|
-
if (await fs.pathExists(path.join(gitDir, 'rebase-merge')) ||
|
|
265
|
-
|
|
265
|
+
if (await fs.pathExists(path.join(gitDir, 'rebase-merge')) ||
|
|
266
|
+
await fs.pathExists(path.join(gitDir, 'rebase-apply'))) {
|
|
266
267
|
return true;
|
|
267
268
|
}
|
|
268
269
|
|
package/src/core/grok.js
CHANGED
|
@@ -1,27 +1,78 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* xAI Grok Integration for Autopilot
|
|
3
|
-
* Generates commit messages using the Grok API
|
|
4
|
-
*/
|
|
5
|
-
|
|
6
1
|
const logger = require('../utils/logger');
|
|
2
|
+
const keys = require('./keys');
|
|
7
3
|
|
|
8
|
-
|
|
9
|
-
|
|
4
|
+
// Resolve fetch at call-time so test mocks can override it
|
|
5
|
+
function getFetch() {
|
|
6
|
+
if (typeof globalThis.fetch === 'function') {
|
|
7
|
+
return globalThis.fetch;
|
|
8
|
+
}
|
|
9
|
+
return (...args) => import('node-fetch').then(({ default: fetch }) => fetch(...args));
|
|
10
|
+
}
|
|
11
|
+
|
|
12
|
+
const GROK_API_URL = 'https://api.x.ai/v1/chat/completions';
|
|
13
|
+
const GROQ_API_URL = 'https://api.groq.com/openai/v1/chat/completions';
|
|
14
|
+
|
|
15
|
+
const DEFAULT_GROK_MODEL = 'grok-beta';
|
|
16
|
+
const DEFAULT_GROQ_MODEL = 'llama-3.3-70b-versatile';
|
|
10
17
|
|
|
11
18
|
/**
|
|
12
|
-
* Generate a commit message using Grok API
|
|
19
|
+
* Generate a commit message using Grok (or Groq) API with automatic failover
|
|
13
20
|
* @param {string} diff - The git diff content
|
|
14
|
-
* @param {string}
|
|
15
|
-
* @param {string} [model] -
|
|
21
|
+
* @param {string} [customApiKey] - Optional user-provided API Key
|
|
22
|
+
* @param {string} [model] - Model ID
|
|
16
23
|
* @returns {Promise<string>} Generated commit message
|
|
17
24
|
*/
|
|
18
|
-
async function generateGrokCommitMessage(diff,
|
|
19
|
-
if (!diff || !diff.trim())
|
|
20
|
-
|
|
25
|
+
async function generateGrokCommitMessage(diff, customApiKey, model) {
|
|
26
|
+
if (!diff || !diff.trim()) return 'chore: update changes';
|
|
27
|
+
|
|
28
|
+
// If a custom key is provided, use it directly
|
|
29
|
+
if (customApiKey) {
|
|
30
|
+
return executeRequest(diff, customApiKey, model);
|
|
21
31
|
}
|
|
22
32
|
|
|
23
|
-
//
|
|
24
|
-
|
|
33
|
+
// System Key Logic with Failover
|
|
34
|
+
let attempts = 0;
|
|
35
|
+
const maxAttempts = keys.keyCount;
|
|
36
|
+
|
|
37
|
+
while (attempts < maxAttempts) {
|
|
38
|
+
const currentKey = keys.getSystemKey();
|
|
39
|
+
|
|
40
|
+
if (!currentKey || currentKey.includes('placeholder')) {
|
|
41
|
+
throw new Error('No valid system AI keys configured.');
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
try {
|
|
45
|
+
return await executeRequest(diff, currentKey, model);
|
|
46
|
+
} catch (error) {
|
|
47
|
+
const msg = String(error?.message || error);
|
|
48
|
+
const isRateLimit = msg.includes(' 429') || msg.includes('429');
|
|
49
|
+
const isInvalid = msg.includes(' 401') || msg.includes('401') || msg.includes(' 403') || msg.includes('403');
|
|
50
|
+
|
|
51
|
+
if (isRateLimit || isInvalid) {
|
|
52
|
+
keys.markKeyAsFailed(currentKey);
|
|
53
|
+
attempts++;
|
|
54
|
+
logger.info(`Attempt ${attempts}/${maxAttempts} failed. Trying next key...`);
|
|
55
|
+
continue;
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
throw error;
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
|
|
62
|
+
throw new Error('All internal AI keys exhausted or rate-limited.');
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
/**
|
|
66
|
+
* Internal execution of the API request
|
|
67
|
+
*/
|
|
68
|
+
async function executeRequest(diff, apiKey, model) {
|
|
69
|
+
const isGroq = apiKey.startsWith('gsk_');
|
|
70
|
+
const baseUrl = isGroq ? GROQ_API_URL : GROK_API_URL;
|
|
71
|
+
const defaultModel = isGroq ? DEFAULT_GROQ_MODEL : DEFAULT_GROK_MODEL;
|
|
72
|
+
const targetModel = model || defaultModel;
|
|
73
|
+
|
|
74
|
+
const truncatedDiff =
|
|
75
|
+
diff.length > 30000 ? diff.slice(0, 30000) + '\n...(truncated)' : diff;
|
|
25
76
|
|
|
26
77
|
const systemPrompt = `You are an expert software engineer.
|
|
27
78
|
Generate a concise, standardized commit message following the Conventional Commits specification based on the provided git diff.
|
|
@@ -34,76 +85,121 @@ Rules:
|
|
|
34
85
|
5. Use types: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert.
|
|
35
86
|
6. Return ONLY the commit message, no explanations or markdown code blocks.`;
|
|
36
87
|
|
|
88
|
+
const controller = new AbortController();
|
|
89
|
+
const timeout = setTimeout(() => controller.abort(), 5000);
|
|
90
|
+
|
|
37
91
|
try {
|
|
38
|
-
const response = await
|
|
92
|
+
const response = await getFetch()(baseUrl, {
|
|
39
93
|
method: 'POST',
|
|
40
94
|
headers: {
|
|
41
95
|
'Content-Type': 'application/json',
|
|
42
|
-
|
|
96
|
+
Authorization: `Bearer ${apiKey}`,
|
|
43
97
|
},
|
|
44
98
|
body: JSON.stringify({
|
|
45
|
-
model:
|
|
99
|
+
model: targetModel,
|
|
46
100
|
messages: [
|
|
47
101
|
{ role: 'system', content: systemPrompt },
|
|
48
|
-
{ role: 'user', content: `Diff:\n${truncatedDiff}` }
|
|
102
|
+
{ role: 'user', content: `Diff:\n${truncatedDiff}` },
|
|
49
103
|
],
|
|
50
104
|
temperature: 0.2,
|
|
51
|
-
stream: false
|
|
52
|
-
})
|
|
105
|
+
stream: false,
|
|
106
|
+
}),
|
|
107
|
+
signal: controller.signal,
|
|
53
108
|
});
|
|
54
109
|
|
|
55
110
|
if (!response.ok) {
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
}
|
|
111
|
+
let msg = response.statusText;
|
|
112
|
+
let parsed = null;
|
|
59
113
|
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
114
|
+
// Prefer JSON if available
|
|
115
|
+
try {
|
|
116
|
+
if (typeof response.json === 'function') {
|
|
117
|
+
parsed = await response.json().catch(() => null);
|
|
118
|
+
}
|
|
119
|
+
} catch {}
|
|
120
|
+
|
|
121
|
+
if (parsed && typeof parsed === 'object') {
|
|
122
|
+
msg = parsed?.error?.message || msg;
|
|
123
|
+
} else {
|
|
124
|
+
// Fallback to text only if supported
|
|
125
|
+
let text = '';
|
|
126
|
+
if (typeof response.text === 'function') {
|
|
127
|
+
text = await response.text().catch(() => '');
|
|
128
|
+
try {
|
|
129
|
+
const errorData = text ? JSON.parse(text) : {};
|
|
130
|
+
msg = errorData?.error?.message || msg;
|
|
131
|
+
} catch {
|
|
132
|
+
if (text) msg = text.slice(0, 500);
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
|
|
137
|
+
throw new Error(`${isGroq ? 'Groq' : 'Grok'} API Error: ${response.status} ${msg}`);
|
|
64
138
|
}
|
|
65
139
|
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
// Cleanup markdown if present
|
|
69
|
-
message = message.replace(/^```[a-z]*\n?/, '').replace(/\n?```$/, '').trim();
|
|
140
|
+
const data = await response.json();
|
|
70
141
|
|
|
71
|
-
|
|
142
|
+
const content = data?.choices?.[0]?.message?.content;
|
|
143
|
+
if (!content) {
|
|
144
|
+
throw new Error(`No response content from ${isGroq ? 'Groq' : 'Grok'}`);
|
|
145
|
+
}
|
|
72
146
|
|
|
147
|
+
// Strip markdown fences if the model ignores instructions
|
|
148
|
+
return String(content)
|
|
149
|
+
.trim()
|
|
150
|
+
.replace(/^```[a-z]*\n?/i, '')
|
|
151
|
+
.replace(/\n?```$/i, '')
|
|
152
|
+
.trim();
|
|
73
153
|
} catch (error) {
|
|
74
|
-
|
|
154
|
+
if (error?.name === 'AbortError') {
|
|
155
|
+
throw new Error(`${isGroq ? 'Groq' : 'Grok'} API Error: request timed out`);
|
|
156
|
+
}
|
|
75
157
|
throw error;
|
|
158
|
+
} finally {
|
|
159
|
+
clearTimeout(timeout);
|
|
76
160
|
}
|
|
77
161
|
}
|
|
78
162
|
|
|
79
163
|
/**
|
|
80
|
-
* Validate
|
|
81
|
-
* @param {string} apiKey
|
|
82
|
-
* @returns {Promise<{valid: boolean, error?: string}>}
|
|
164
|
+
* Validate API Key
|
|
83
165
|
*/
|
|
84
166
|
async function validateGrokApiKey(apiKey) {
|
|
167
|
+
const isGroq = apiKey.startsWith('gsk_');
|
|
168
|
+
const baseUrl = isGroq ? GROQ_API_URL : GROK_API_URL;
|
|
169
|
+
const model = isGroq ? DEFAULT_GROQ_MODEL : DEFAULT_GROK_MODEL;
|
|
170
|
+
|
|
171
|
+
const controller = new AbortController();
|
|
172
|
+
const timeout = setTimeout(() => controller.abort(), 4000);
|
|
173
|
+
|
|
85
174
|
try {
|
|
86
|
-
const response = await
|
|
175
|
+
const response = await getFetch()(baseUrl, {
|
|
87
176
|
method: 'POST',
|
|
88
177
|
headers: {
|
|
89
178
|
'Content-Type': 'application/json',
|
|
90
|
-
|
|
179
|
+
Authorization: `Bearer ${apiKey}`,
|
|
91
180
|
},
|
|
92
181
|
body: JSON.stringify({
|
|
93
|
-
model
|
|
182
|
+
model,
|
|
94
183
|
messages: [{ role: 'user', content: 'test' }],
|
|
95
|
-
max_tokens: 1
|
|
96
|
-
|
|
184
|
+
max_tokens: 1,
|
|
185
|
+
stream: false,
|
|
186
|
+
}),
|
|
187
|
+
signal: controller.signal,
|
|
97
188
|
});
|
|
98
189
|
|
|
99
190
|
if (response.ok) return { valid: true };
|
|
191
|
+
|
|
192
|
+
// Always report status code for deterministic tests
|
|
100
193
|
return { valid: false, error: `Status: ${response.status}` };
|
|
101
194
|
} catch (error) {
|
|
102
|
-
return { valid: false, error:
|
|
195
|
+
if (error?.name === 'AbortError') return { valid: false, error: 'Request timed out' };
|
|
196
|
+
return { valid: false, error: String(error?.message || error) };
|
|
197
|
+
} finally {
|
|
198
|
+
clearTimeout(timeout);
|
|
103
199
|
}
|
|
104
200
|
}
|
|
105
201
|
|
|
106
202
|
module.exports = {
|
|
107
203
|
generateGrokCommitMessage,
|
|
108
|
-
validateGrokApiKey
|
|
204
|
+
validateGrokApiKey,
|
|
109
205
|
};
|
package/src/core/keys.js
ADDED
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Internal Key Manager for Autopilot Zero-Config AI
|
|
3
|
+
* Manages a pool of system keys with automatic rotation and failover.
|
|
4
|
+
* Keys are obfuscated to prevent automated secret scanners from revoking them.
|
|
5
|
+
*/
|
|
6
|
+
|
|
7
|
+
const logger = require('../utils/logger');
|
|
8
|
+
const { unscramble } = require('../utils/obfuscator');
|
|
9
|
+
|
|
10
|
+
// Scrambled internal system keys.
|
|
11
|
+
const SYSTEM_KEYS_SCRAMBLED = [
|
|
12
|
+
'BgYfMAkoJypBYj4hJDseHWBMBwFdaUUCZXMFDBZcNjAAWSZVSBAnJBYoaz0rJhxUfQJwdgghRB8=', // v1
|
|
13
|
+
'BgYfMEcwXwcebEEQVicCCEsjXS8yRWFzZXMFDBZcNjBYBCVaPxEzAyEOehE1EjlpBnhbWlQvR1w=', // v2
|
|
14
|
+
'BgYfMCBZARgWHz0YNzkpAFQ2AhRQS2F+ZXMFDBZcNjAOKjBuMhMjHhlXbDsXMi1OZgFeRQ8fQxg=', // v3
|
|
15
|
+
'BgYfMBcZXlZEZjECUBwdKHs5XAkPHQRaZXMFDBZcNjAjKhJbAwdXXC0BGBstEiwzXMD0dHkAUJx11WX9fXilNFh0=', // v4
|
|
16
|
+
'BgYfMDUHIDsjWTYfUAUAD2MRKBMqGWpWZXMFDBZcNjAbKAdFOjo4HyEHGBkqSjo4PxItdDodGllGdnlzcwIXQBo=', // v5
|
|
17
|
+
'BgYfMBICBTgNQUk+IlhCNBkwADUFXQVgZXMFDBZcNjA/XwZcFhhRGyI2TjNWEBJ1WnVBeS8XJyw=' // v6
|
|
18
|
+
];
|
|
19
|
+
|
|
20
|
+
|
|
21
|
+
let currentIndex = 0;
|
|
22
|
+
let failedKeys = new Set();
|
|
23
|
+
|
|
24
|
+
/**
|
|
25
|
+
* Get the current active system key (unscrambled at runtime)
|
|
26
|
+
* @returns {string|null}
|
|
27
|
+
*/
|
|
28
|
+
function getSystemKey() {
|
|
29
|
+
// If all keys failed, return null
|
|
30
|
+
if (failedKeys.size >= SYSTEM_KEYS_SCRAMBLED.length) {
|
|
31
|
+
logger.error('Sorry! All system AI keys have been exhausted or are invalid.');
|
|
32
|
+
return null;
|
|
33
|
+
}
|
|
34
|
+
|
|
35
|
+
// Find the next non-failed key
|
|
36
|
+
while (failedKeys.has(SYSTEM_KEYS_SCRAMBLED[currentIndex])) {
|
|
37
|
+
currentIndex = (currentIndex + 1) % SYSTEM_KEYS_SCRAMBLED.length;
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
const scrambled = SYSTEM_KEYS_SCRAMBLED[currentIndex];
|
|
41
|
+
return unscramble(scrambled);
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
/**
|
|
45
|
+
* Mark a key as failed (passed as unscrambled key)
|
|
46
|
+
* @param {string} unscrambledKey - The raw key that failed
|
|
47
|
+
*/
|
|
48
|
+
function markKeyAsFailed(unscrambledKey) {
|
|
49
|
+
// Find which scrambled key this belongs to
|
|
50
|
+
const index = SYSTEM_KEYS_SCRAMBLED.findIndex(s => unscramble(s) === unscrambledKey);
|
|
51
|
+
|
|
52
|
+
if (index === -1) return;
|
|
53
|
+
|
|
54
|
+
const scrambled = SYSTEM_KEYS_SCRAMBLED[index];
|
|
55
|
+
failedKeys.add(scrambled);
|
|
56
|
+
logger.warn(`AI Key ${index + 1} failed. Rotating to next available key...`);
|
|
57
|
+
|
|
58
|
+
// Advance index logic
|
|
59
|
+
if (index === currentIndex) {
|
|
60
|
+
currentIndex = (currentIndex + 1) % SYSTEM_KEYS_SCRAMBLED.length;
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
|
|
64
|
+
/**
|
|
65
|
+
* Reset all failed keys
|
|
66
|
+
*/
|
|
67
|
+
function resetPool() {
|
|
68
|
+
failedKeys.clear();
|
|
69
|
+
currentIndex = 0;
|
|
70
|
+
}
|
|
71
|
+
|
|
72
|
+
module.exports = {
|
|
73
|
+
getSystemKey,
|
|
74
|
+
markKeyAsFailed,
|
|
75
|
+
resetPool,
|
|
76
|
+
keyCount: SYSTEM_KEYS_SCRAMBLED.length
|
|
77
|
+
};
|