@su-record/vibe 2.7.16 → 2.7.18
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/.env.example +37 -37
- package/CLAUDE.md +153 -134
- package/LICENSE +21 -21
- package/README.md +449 -449
- package/agents/architect-low.md +41 -41
- package/agents/architect-medium.md +59 -59
- package/agents/architect.md +80 -80
- package/agents/build-error-resolver.md +115 -115
- package/agents/compounder.md +261 -261
- package/agents/diagrammer.md +178 -178
- package/agents/docs/api-documenter.md +99 -99
- package/agents/docs/changelog-writer.md +93 -93
- package/agents/e2e-tester.md +294 -294
- package/agents/explorer-low.md +42 -42
- package/agents/explorer-medium.md +59 -59
- package/agents/explorer.md +48 -48
- package/agents/implementer-low.md +43 -43
- package/agents/implementer-medium.md +52 -52
- package/agents/implementer.md +54 -54
- package/agents/junior-mentor.md +141 -141
- package/agents/planning/requirements-analyst.md +84 -84
- package/agents/planning/ux-advisor.md +83 -83
- package/agents/qa/acceptance-tester.md +86 -86
- package/agents/qa/edge-case-finder.md +93 -93
- package/agents/refactor-cleaner.md +143 -143
- package/agents/research/best-practices-agent.md +199 -199
- package/agents/research/codebase-patterns-agent.md +157 -157
- package/agents/research/framework-docs-agent.md +188 -188
- package/agents/research/security-advisory-agent.md +213 -213
- package/agents/review/architecture-reviewer.md +107 -107
- package/agents/review/complexity-reviewer.md +116 -116
- package/agents/review/data-integrity-reviewer.md +88 -88
- package/agents/review/git-history-reviewer.md +103 -103
- package/agents/review/performance-reviewer.md +86 -86
- package/agents/review/python-reviewer.md +150 -150
- package/agents/review/rails-reviewer.md +139 -139
- package/agents/review/react-reviewer.md +144 -144
- package/agents/review/security-reviewer.md +80 -80
- package/agents/review/simplicity-reviewer.md +140 -140
- package/agents/review/test-coverage-reviewer.md +116 -116
- package/agents/review/typescript-reviewer.md +127 -127
- package/agents/searcher.md +54 -54
- package/agents/simplifier.md +120 -120
- package/agents/tester.md +49 -49
- package/agents/ui/ui-a11y-auditor.md +93 -93
- package/agents/ui/ui-antipattern-detector.md +94 -94
- package/agents/ui/ui-dataviz-advisor.md +69 -69
- package/agents/ui/ui-design-system-gen.md +57 -57
- package/agents/ui/ui-industry-analyzer.md +49 -49
- package/agents/ui/ui-layout-architect.md +65 -65
- package/agents/ui/ui-stack-implementer.md +68 -68
- package/agents/ui/ux-compliance-reviewer.md +81 -81
- package/agents/ui-previewer.md +258 -258
- package/commands/vibe.analyze.md +379 -379
- package/commands/vibe.review.md +607 -607
- package/commands/vibe.run.md +2124 -2124
- package/commands/vibe.spec.md +1195 -1195
- package/commands/vibe.spec.review.md +569 -569
- package/commands/vibe.utils.md +413 -413
- package/commands/vibe.verify.md +484 -484
- package/dist/cli/collaborator.js +52 -52
- package/dist/cli/commands/evolution.js +12 -12
- package/dist/cli/commands/info.js +51 -51
- package/dist/cli/commands/init.js +5 -5
- package/dist/cli/commands/remove.js +14 -14
- package/dist/cli/commands/sentinel.js +27 -27
- package/dist/cli/commands/skills.js +5 -5
- package/dist/cli/commands/slack.js +10 -10
- package/dist/cli/commands/telegram.js +12 -12
- package/dist/cli/commands/upgrade.d.ts +3 -3
- package/dist/cli/commands/upgrade.d.ts.map +1 -1
- package/dist/cli/commands/upgrade.js +24 -3
- package/dist/cli/commands/upgrade.js.map +1 -1
- package/dist/cli/detect.js +32 -32
- package/dist/cli/index.js +51 -51
- package/dist/cli/llm/claude-commands.js +16 -16
- package/dist/cli/llm/config.js +18 -18
- package/dist/cli/llm/gemini-commands.js +16 -16
- package/dist/cli/llm/gpt-commands.js +19 -19
- package/dist/cli/llm/help.js +21 -21
- package/dist/cli/postinstall/cursor-agents.js +32 -32
- package/dist/cli/postinstall/cursor-rules.js +83 -83
- package/dist/cli/postinstall/cursor-skills.js +743 -743
- package/dist/cli/setup/Provisioner.js +42 -42
- package/dist/infra/lib/DeepInit.js +24 -24
- package/dist/infra/lib/IterationTracker.js +11 -11
- package/dist/infra/lib/PythonParser.js +108 -108
- package/dist/infra/lib/ReviewRace.js +96 -96
- package/dist/infra/lib/SkillFrontmatter.js +28 -28
- package/dist/infra/lib/SkillQualityGate.js +9 -9
- package/dist/infra/lib/SkillRepository.js +159 -159
- package/dist/infra/lib/UltraQA.js +99 -99
- package/dist/infra/lib/autonomy/AuditStore.js +41 -41
- package/dist/infra/lib/autonomy/ConfirmationStore.js +30 -30
- package/dist/infra/lib/autonomy/EventOutbox.js +38 -38
- package/dist/infra/lib/autonomy/PolicyEngine.js +18 -18
- package/dist/infra/lib/autonomy/SecuritySentinel.js +1 -1
- package/dist/infra/lib/autonomy/SuggestionStore.js +33 -33
- package/dist/infra/lib/embedding/VectorStore.js +22 -22
- package/dist/infra/lib/evolution/AgentAnalyzer.js +10 -10
- package/dist/infra/lib/evolution/DescriptionOptimizer.js +21 -21
- package/dist/infra/lib/evolution/GenerationRegistry.js +36 -36
- package/dist/infra/lib/evolution/InsightStore.js +90 -90
- package/dist/infra/lib/evolution/RollbackManager.js +5 -5
- package/dist/infra/lib/evolution/SkillBenchmark.js +23 -23
- package/dist/infra/lib/evolution/SkillEvalRunner.js +50 -50
- package/dist/infra/lib/evolution/SkillGapDetector.js +10 -10
- package/dist/infra/lib/evolution/UsageTracker.js +28 -28
- package/dist/infra/lib/gemini/orchestration.js +5 -5
- package/dist/infra/lib/gpt/orchestration.js +4 -4
- package/dist/infra/lib/memory/KnowledgeGraph.js +4 -4
- package/dist/infra/lib/memory/MemorySearch.js +57 -57
- package/dist/infra/lib/memory/MemoryStorage.js +181 -181
- package/dist/infra/lib/memory/ObservationStore.js +28 -28
- package/dist/infra/lib/memory/ReflectionStore.js +30 -30
- package/dist/infra/lib/memory/SessionRAGRetriever.js +7 -7
- package/dist/infra/lib/memory/SessionRAGStore.js +225 -225
- package/dist/infra/lib/memory/SessionSummarizer.js +9 -9
- package/dist/infra/lib/telemetry/SkillTelemetry.d.ts +52 -0
- package/dist/infra/lib/telemetry/SkillTelemetry.d.ts.map +1 -0
- package/dist/infra/lib/telemetry/SkillTelemetry.js +117 -0
- package/dist/infra/lib/telemetry/SkillTelemetry.js.map +1 -0
- package/dist/infra/lib/telemetry/SkillTelemetry.test.d.ts +2 -0
- package/dist/infra/lib/telemetry/SkillTelemetry.test.d.ts.map +1 -0
- package/dist/infra/lib/telemetry/SkillTelemetry.test.js +91 -0
- package/dist/infra/lib/telemetry/SkillTelemetry.test.js.map +1 -0
- package/dist/infra/orchestrator/AgentManager.js +12 -12
- package/dist/infra/orchestrator/AgentRegistry.js +65 -65
- package/dist/infra/orchestrator/MultiLlmResearch.js +8 -8
- package/dist/infra/orchestrator/SwarmOrchestrator.test.js +16 -16
- package/dist/infra/orchestrator/parallelResearch.js +24 -24
- package/dist/test-helpers/index.d.ts +36 -0
- package/dist/test-helpers/index.d.ts.map +1 -0
- package/dist/test-helpers/index.js +85 -0
- package/dist/test-helpers/index.js.map +1 -0
- package/dist/test-helpers/index.test.d.ts +2 -0
- package/dist/test-helpers/index.test.d.ts.map +1 -0
- package/dist/test-helpers/index.test.js +92 -0
- package/dist/test-helpers/index.test.js.map +1 -0
- package/dist/tools/convention/analyzeComplexity.test.js +115 -115
- package/dist/tools/convention/validateCodeQuality.test.js +104 -104
- package/dist/tools/memory/createMemoryTimeline.js +10 -10
- package/dist/tools/memory/getMemoryGraph.js +12 -12
- package/dist/tools/memory/getSessionContext.js +9 -9
- package/dist/tools/memory/linkMemories.js +14 -14
- package/dist/tools/memory/listMemories.js +4 -4
- package/dist/tools/memory/recallMemory.js +4 -4
- package/dist/tools/memory/saveMemory.js +4 -4
- package/dist/tools/memory/searchMemoriesAdvanced.js +23 -23
- package/dist/tools/semantic/analyzeDependencyGraph.js +12 -12
- package/dist/tools/semantic/astGrep.test.js +6 -6
- package/dist/tools/spec/prdParser.test.js +171 -171
- package/dist/tools/spec/specGenerator.js +169 -169
- package/dist/tools/spec/traceabilityMatrix.js +64 -64
- package/dist/tools/spec/traceabilityMatrix.test.js +28 -28
- package/hooks/gemini-hooks.json +73 -73
- package/hooks/hooks.json +137 -137
- package/hooks/scripts/code-check.js +77 -77
- package/hooks/scripts/context-save.js +212 -212
- package/hooks/scripts/hud-status.js +291 -291
- package/hooks/scripts/keyword-detector.js +214 -214
- package/hooks/scripts/llm-orchestrate.js +475 -475
- package/hooks/scripts/post-edit.js +32 -32
- package/hooks/scripts/pre-tool-guard.js +125 -125
- package/hooks/scripts/prompt-dispatcher.js +185 -185
- package/hooks/scripts/sentinel-guard.js +104 -104
- package/hooks/scripts/session-start.js +106 -106
- package/hooks/scripts/stop-notify.js +209 -209
- package/hooks/scripts/utils.js +100 -100
- package/languages/csharp-unity.md +515 -515
- package/languages/gdscript-godot.md +470 -470
- package/languages/ruby-rails.md +489 -489
- package/languages/typescript-angular.md +433 -433
- package/languages/typescript-astro.md +416 -416
- package/languages/typescript-electron.md +406 -406
- package/languages/typescript-nestjs.md +524 -524
- package/languages/typescript-svelte.md +407 -407
- package/languages/typescript-tauri.md +365 -365
- package/package.json +123 -121
- package/skills/agents-md/SKILL.md +120 -120
- package/skills/arch-guard/SKILL.md +180 -180
- package/skills/brand-assets/SKILL.md +146 -146
- package/skills/capability-loop/SKILL.md +167 -167
- package/skills/characterization-test/SKILL.md +206 -206
- package/skills/commerce-patterns/SKILL.md +63 -63
- package/skills/commit-push-pr/SKILL.md +75 -75
- package/skills/context7-usage/SKILL.md +105 -105
- package/skills/core-capabilities/SKILL.md +13 -13
- package/skills/e2e-commerce/SKILL.md +61 -61
- package/skills/exec-plan/SKILL.md +147 -147
- package/skills/frontend-design/SKILL.md +12 -12
- package/skills/git-worktree/SKILL.md +72 -72
- package/skills/handoff/SKILL.md +109 -109
- package/skills/parallel-research/SKILL.md +87 -87
- package/skills/priority-todos/SKILL.md +63 -63
- package/skills/seo-checklist/SKILL.md +57 -57
- package/skills/techdebt/SKILL.md +122 -122
- package/skills/tool-fallback/SKILL.md +103 -103
- package/skills/typescript-advanced-types/SKILL.md +66 -66
- package/skills/ui-ux-pro-max/SKILL.md +221 -221
- package/skills/vercel-react-best-practices/SKILL.md +59 -59
- package/skills/video-production/SKILL.md +51 -51
- package/vibe/config.json +29 -29
- package/vibe/constitution.md +227 -227
- package/vibe/rules/principles/communication-guide.md +98 -98
- package/vibe/rules/principles/development-philosophy.md +52 -52
- package/vibe/rules/principles/quick-start.md +102 -102
- package/vibe/rules/quality/bdd-contract-testing.md +393 -393
- package/vibe/rules/quality/checklist.md +276 -276
- package/vibe/rules/quality/performance.md +236 -236
- package/vibe/rules/quality/testing-strategy.md +440 -440
- package/vibe/rules/standards/anti-patterns.md +541 -541
- package/vibe/rules/standards/code-structure.md +291 -291
- package/vibe/rules/standards/complexity-metrics.md +313 -313
- package/vibe/rules/standards/git-workflow.md +237 -237
- package/vibe/rules/standards/naming-conventions.md +198 -198
- package/vibe/rules/standards/security.md +305 -305
- package/vibe/rules/writing/document-style.md +74 -74
- package/vibe/setup.sh +31 -31
- package/vibe/templates/constitution-template.md +252 -252
- package/vibe/templates/contract-backend-template.md +526 -526
- package/vibe/templates/contract-frontend-template.md +599 -599
- package/vibe/templates/feature-template.md +96 -96
- package/vibe/templates/spec-template.md +221 -221
- package/vibe/ui-ux-data/charts.csv +26 -26
- package/vibe/ui-ux-data/colors.csv +97 -97
- package/vibe/ui-ux-data/icons.csv +101 -101
- package/vibe/ui-ux-data/landing.csv +31 -31
- package/vibe/ui-ux-data/products.csv +96 -96
- package/vibe/ui-ux-data/react-performance.csv +45 -45
- package/vibe/ui-ux-data/stacks/astro.csv +54 -54
- package/vibe/ui-ux-data/stacks/flutter.csv +53 -53
- package/vibe/ui-ux-data/stacks/html-tailwind.csv +56 -56
- package/vibe/ui-ux-data/stacks/jetpack-compose.csv +53 -53
- package/vibe/ui-ux-data/stacks/nextjs.csv +53 -53
- package/vibe/ui-ux-data/stacks/nuxt-ui.csv +51 -51
- package/vibe/ui-ux-data/stacks/nuxtjs.csv +59 -59
- package/vibe/ui-ux-data/stacks/react-native.csv +52 -52
- package/vibe/ui-ux-data/stacks/react.csv +54 -54
- package/vibe/ui-ux-data/stacks/shadcn.csv +61 -61
- package/vibe/ui-ux-data/stacks/svelte.csv +54 -54
- package/vibe/ui-ux-data/stacks/swiftui.csv +51 -51
- package/vibe/ui-ux-data/stacks/vue.csv +50 -50
- package/vibe/ui-ux-data/styles.csv +68 -68
- package/vibe/ui-ux-data/typography.csv +57 -57
- package/vibe/ui-ux-data/ui-reasoning.csv +101 -101
- package/vibe/ui-ux-data/ux-guidelines.csv +99 -99
- package/vibe/ui-ux-data/version.json +31 -31
- package/vibe/ui-ux-data/web-interface.csv +31 -31
|
@@ -1,475 +1,475 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* UserPromptSubmit Hook - LLM 오케스트레이션 (GPT/Gemini)
|
|
3
|
-
*
|
|
4
|
-
* Usage:
|
|
5
|
-
* node llm-orchestrate.js <provider> <mode> "prompt"
|
|
6
|
-
* node llm-orchestrate.js <provider> <mode> "systemPrompt" "prompt"
|
|
7
|
-
*
|
|
8
|
-
* provider: gpt | gemini
|
|
9
|
-
* mode: orchestrate | orchestrate-json | image | analyze-image
|
|
10
|
-
*
|
|
11
|
-
* Image Mode:
|
|
12
|
-
* node llm-orchestrate.js gemini image "prompt" --output "./image.png"
|
|
13
|
-
* node llm-orchestrate.js gemini image "prompt" --output "./image.png" --size "1920x1080"
|
|
14
|
-
*
|
|
15
|
-
* Features:
|
|
16
|
-
* - CLI-based: GPT → codex exec, Gemini → gemini -p
|
|
17
|
-
* - Exponential backoff retry (3 attempts)
|
|
18
|
-
* - Auto fallback: gpt ↔ gemini
|
|
19
|
-
* - Overload/rate-limit detection
|
|
20
|
-
* - Image generation (Gemini API, Gemini Flash Image model)
|
|
21
|
-
* - Image analysis (Gemini API multimodal)
|
|
22
|
-
*
|
|
23
|
-
* Analyze-Image Mode:
|
|
24
|
-
* node llm-orchestrate.js gemini analyze-image "./image.png" "Analyze this UI"
|
|
25
|
-
* node llm-orchestrate.js gemini analyze-image "./image.png" "prompt" --system "system prompt"
|
|
26
|
-
*
|
|
27
|
-
* Input: JSON from stdin with { prompt: string } (when no CLI args)
|
|
28
|
-
*/
|
|
29
|
-
import { getLibBaseUrl, readVibeConfig } from './utils.js';
|
|
30
|
-
import fs from 'fs';
|
|
31
|
-
import path from 'path';
|
|
32
|
-
import os from 'os';
|
|
33
|
-
import crypto from 'crypto';
|
|
34
|
-
import { execSync, spawn } from 'child_process';
|
|
35
|
-
|
|
36
|
-
const LIB_URL = getLibBaseUrl();
|
|
37
|
-
const DEFAULT_SYSTEM_PROMPT = 'You are a helpful assistant.';
|
|
38
|
-
|
|
39
|
-
const provider = process.argv[2] || 'gemini';
|
|
40
|
-
const mode = process.argv[3] || 'orchestrate';
|
|
41
|
-
|
|
42
|
-
// Retry configuration
|
|
43
|
-
const MAX_RETRIES = 3;
|
|
44
|
-
const INITIAL_DELAY_MS = 2000;
|
|
45
|
-
|
|
46
|
-
// Errors that should skip retry and go to fallback immediately
|
|
47
|
-
const SKIP_RETRY_PATTERNS = [
|
|
48
|
-
/rate.?limit/i,
|
|
49
|
-
/quota/i,
|
|
50
|
-
/unauthorized/i,
|
|
51
|
-
/forbidden/i,
|
|
52
|
-
/authentication/i,
|
|
53
|
-
/401/,
|
|
54
|
-
/403/,
|
|
55
|
-
/429/,
|
|
56
|
-
];
|
|
57
|
-
|
|
58
|
-
// Errors that should trigger retry with backoff
|
|
59
|
-
const RETRY_PATTERNS = [
|
|
60
|
-
/overload/i,
|
|
61
|
-
/503/,
|
|
62
|
-
/5\d\d/,
|
|
63
|
-
/network/i,
|
|
64
|
-
/timeout/i,
|
|
65
|
-
/ECONNRESET/i,
|
|
66
|
-
/ETIMEDOUT/i,
|
|
67
|
-
/spawn error/i,
|
|
68
|
-
];
|
|
69
|
-
|
|
70
|
-
function shouldSkipRetry(errorMsg) {
|
|
71
|
-
return SKIP_RETRY_PATTERNS.some(pattern => pattern.test(errorMsg));
|
|
72
|
-
}
|
|
73
|
-
|
|
74
|
-
function shouldRetry(errorMsg) {
|
|
75
|
-
return RETRY_PATTERNS.some(pattern => pattern.test(errorMsg));
|
|
76
|
-
}
|
|
77
|
-
|
|
78
|
-
function sleep(ms) {
|
|
79
|
-
return new Promise(resolve => setTimeout(resolve, ms));
|
|
80
|
-
}
|
|
81
|
-
|
|
82
|
-
// ============================================
|
|
83
|
-
// Image Generation (delegates to gemini-api)
|
|
84
|
-
// ============================================
|
|
85
|
-
|
|
86
|
-
function parseImageArgs(args) {
|
|
87
|
-
const result = { prompt: null, output: './generated-image.png', size: '1024x1024', model: 'nano-banana' };
|
|
88
|
-
for (let i = 0; i < args.length; i++) {
|
|
89
|
-
if (args[i] === '--output' || args[i] === '-o') {
|
|
90
|
-
result.output = args[++i];
|
|
91
|
-
} else if (args[i] === '--size' || args[i] === '-s') {
|
|
92
|
-
result.size = args[++i];
|
|
93
|
-
} else if (args[i] === '--pro') {
|
|
94
|
-
result.model = 'nano-banana-pro';
|
|
95
|
-
} else if (!args[i].startsWith('-') && !result.prompt) {
|
|
96
|
-
result.prompt = args[i];
|
|
97
|
-
}
|
|
98
|
-
}
|
|
99
|
-
return result;
|
|
100
|
-
}
|
|
101
|
-
|
|
102
|
-
// ============================================
|
|
103
|
-
// Image Analysis (delegates to gemini-api)
|
|
104
|
-
// ============================================
|
|
105
|
-
|
|
106
|
-
function parseAnalyzeImageArgs(args) {
|
|
107
|
-
const result = { imagePath: null, prompt: null, systemPrompt: null };
|
|
108
|
-
for (let i = 0; i < args.length; i++) {
|
|
109
|
-
if (args[i] === '--system' || args[i] === '-s') {
|
|
110
|
-
result.systemPrompt = args[++i];
|
|
111
|
-
} else if (!args[i].startsWith('-')) {
|
|
112
|
-
if (!result.imagePath) {
|
|
113
|
-
result.imagePath = args[i];
|
|
114
|
-
} else if (!result.prompt) {
|
|
115
|
-
result.prompt = args[i];
|
|
116
|
-
}
|
|
117
|
-
}
|
|
118
|
-
}
|
|
119
|
-
return result;
|
|
120
|
-
}
|
|
121
|
-
|
|
122
|
-
// ============================================
|
|
123
|
-
// CLI Provider Functions
|
|
124
|
-
// ============================================
|
|
125
|
-
|
|
126
|
-
const CLI_TIMEOUT_MS = 120000;
|
|
127
|
-
const IS_WINDOWS = os.platform() === 'win32';
|
|
128
|
-
|
|
129
|
-
function spawnCli(cmd, args, options) {
|
|
130
|
-
// Windows: .cmd shim은 spawn 직접 실행 불가 → cmd.exe /c 로 실행
|
|
131
|
-
if (IS_WINDOWS) {
|
|
132
|
-
return spawn(process.env.ComSpec || 'cmd.exe', ['/c', cmd, ...args], options);
|
|
133
|
-
}
|
|
134
|
-
return spawn(cmd, args, options);
|
|
135
|
-
}
|
|
136
|
-
|
|
137
|
-
function buildCliPrompt(prompt, sysPrompt, jsonMode) {
|
|
138
|
-
let fullPrompt = sysPrompt && sysPrompt !== DEFAULT_SYSTEM_PROMPT
|
|
139
|
-
? `[System]\n${sysPrompt}\n\n[User]\n${prompt}`
|
|
140
|
-
: prompt;
|
|
141
|
-
if (jsonMode) {
|
|
142
|
-
fullPrompt += '\n\nIMPORTANT: You must respond with valid JSON only.';
|
|
143
|
-
}
|
|
144
|
-
return fullPrompt;
|
|
145
|
-
}
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
function callCodexCli(prompt, sysPrompt, jsonMode, model) {
|
|
149
|
-
const fullPrompt = buildCliPrompt(prompt, sysPrompt, jsonMode);
|
|
150
|
-
const outputFile = path.join(os.tmpdir(), `vibe-codex-${crypto.randomUUID()}.txt`);
|
|
151
|
-
// stdin pipe로 프롬프트 전달 (shell escaping 이슈 회피)
|
|
152
|
-
const args = ['exec', '-m', model, '--sandbox', 'read-only', '--ephemeral', '-o', outputFile, '-'];
|
|
153
|
-
|
|
154
|
-
// config에 저장된 API key를 환경변수로 전달 (임베딩과 동일한 키 사용)
|
|
155
|
-
const vibeConfig = readVibeConfig();
|
|
156
|
-
const apiKey = vibeConfig.credentials?.gpt?.apiKey || process.env.OPENAI_API_KEY;
|
|
157
|
-
const env = apiKey ? { ...process.env, OPENAI_API_KEY: apiKey } : process.env;
|
|
158
|
-
|
|
159
|
-
return new Promise((resolve, reject) => {
|
|
160
|
-
const proc = spawnCli('codex', args, {
|
|
161
|
-
stdio: ['pipe', 'pipe', 'pipe'],
|
|
162
|
-
timeout: CLI_TIMEOUT_MS,
|
|
163
|
-
env,
|
|
164
|
-
});
|
|
165
|
-
proc.stdin.write(fullPrompt);
|
|
166
|
-
proc.stdin.end();
|
|
167
|
-
|
|
168
|
-
let stderr = '';
|
|
169
|
-
proc.stderr.on('data', (d) => { stderr += d.toString(); });
|
|
170
|
-
|
|
171
|
-
proc.on('close', (code) => {
|
|
172
|
-
if (code === 0 && fs.existsSync(outputFile)) {
|
|
173
|
-
const result = fs.readFileSync(outputFile, 'utf8');
|
|
174
|
-
try { fs.unlinkSync(outputFile); } catch { /* ignore */ }
|
|
175
|
-
resolve(result);
|
|
176
|
-
} else {
|
|
177
|
-
try { fs.unlinkSync(outputFile); } catch { /* ignore */ }
|
|
178
|
-
reject(new Error(`codex exec failed (code ${code}): ${stderr.slice(0, 500)}`));
|
|
179
|
-
}
|
|
180
|
-
});
|
|
181
|
-
|
|
182
|
-
proc.on('error', (err) => {
|
|
183
|
-
try { fs.unlinkSync(outputFile); } catch { /* ignore */ }
|
|
184
|
-
reject(new Error(`codex exec spawn error: ${err.message}`));
|
|
185
|
-
});
|
|
186
|
-
});
|
|
187
|
-
}
|
|
188
|
-
|
|
189
|
-
function callGeminiCli(prompt, sysPrompt, jsonMode, model) {
|
|
190
|
-
const fullPrompt = buildCliPrompt(prompt, sysPrompt, jsonMode);
|
|
191
|
-
// -p 로 headless 모드, stdin으로 프롬프트 전달 (stdin is appended to -p value)
|
|
192
|
-
const args = ['-p', '.', '-o', 'text'];
|
|
193
|
-
if (model) args.push('-m', model);
|
|
194
|
-
|
|
195
|
-
return new Promise((resolve, reject) => {
|
|
196
|
-
const proc = spawnCli('gemini', args, {
|
|
197
|
-
stdio: ['pipe', 'pipe', 'pipe'],
|
|
198
|
-
timeout: CLI_TIMEOUT_MS,
|
|
199
|
-
});
|
|
200
|
-
proc.stdin.write(fullPrompt);
|
|
201
|
-
proc.stdin.end();
|
|
202
|
-
|
|
203
|
-
let stdout = '';
|
|
204
|
-
let stderr = '';
|
|
205
|
-
proc.stdout.on('data', (d) => { stdout += d.toString(); });
|
|
206
|
-
proc.stderr.on('data', (d) => { stderr += d.toString(); });
|
|
207
|
-
|
|
208
|
-
proc.on('close', (code) => {
|
|
209
|
-
if (code === 0 && stdout.trim()) {
|
|
210
|
-
resolve(stdout.trim());
|
|
211
|
-
} else {
|
|
212
|
-
reject(new Error(`gemini cli failed (code ${code}): ${(stderr || stdout).slice(0, 500)}`));
|
|
213
|
-
}
|
|
214
|
-
});
|
|
215
|
-
|
|
216
|
-
proc.on('error', (err) => {
|
|
217
|
-
reject(new Error(`gemini cli spawn error: ${err.message}`));
|
|
218
|
-
});
|
|
219
|
-
});
|
|
220
|
-
}
|
|
221
|
-
|
|
222
|
-
async function callProvider(providerName, prompt, sysPrompt, jsonMode) {
|
|
223
|
-
const vibeConfig = readVibeConfig();
|
|
224
|
-
|
|
225
|
-
if (providerName === 'gpt' || providerName === 'gpt-codex') {
|
|
226
|
-
const isCodex = providerName === 'gpt-codex';
|
|
227
|
-
const model = isCodex
|
|
228
|
-
? (vibeConfig.models?.gptCodex || process.env.GPT_CODEX_MODEL || 'gpt-5.3-codex')
|
|
229
|
-
: (vibeConfig.models?.gpt || process.env.GPT_MODEL || 'gpt-5.4');
|
|
230
|
-
return await callCodexCli(prompt, sysPrompt, jsonMode, model);
|
|
231
|
-
}
|
|
232
|
-
|
|
233
|
-
if (providerName === 'gemini') {
|
|
234
|
-
const model = vibeConfig.models?.gemini || process.env.GEMINI_MODEL || 'gemini-3.1-pro-preview';
|
|
235
|
-
return await callGeminiCli(prompt, sysPrompt, jsonMode, model);
|
|
236
|
-
}
|
|
237
|
-
|
|
238
|
-
throw new Error(`Unknown provider: ${providerName}`);
|
|
239
|
-
}
|
|
240
|
-
|
|
241
|
-
async function callWithRetry(providerName, prompt, sysPrompt, jsonMode) {
|
|
242
|
-
let lastError;
|
|
243
|
-
|
|
244
|
-
for (let attempt = 1; attempt <= MAX_RETRIES; attempt++) {
|
|
245
|
-
try {
|
|
246
|
-
return { success: true, result: await callProvider(providerName, prompt, sysPrompt, jsonMode) };
|
|
247
|
-
} catch (e) {
|
|
248
|
-
lastError = e;
|
|
249
|
-
const errorMsg = e.message || String(e);
|
|
250
|
-
|
|
251
|
-
// Skip retry for auth/quota errors - go to fallback immediately
|
|
252
|
-
if (shouldSkipRetry(errorMsg)) {
|
|
253
|
-
return { success: false, error: errorMsg, skipToFallback: true };
|
|
254
|
-
}
|
|
255
|
-
|
|
256
|
-
// Retry with backoff for transient errors
|
|
257
|
-
if (shouldRetry(errorMsg) && attempt < MAX_RETRIES) {
|
|
258
|
-
const delay = INITIAL_DELAY_MS * Math.pow(2, attempt - 1);
|
|
259
|
-
console.error(`[${providerName.toUpperCase()}] Retry ${attempt}/${MAX_RETRIES} after ${delay}ms...`);
|
|
260
|
-
await sleep(delay);
|
|
261
|
-
continue;
|
|
262
|
-
}
|
|
263
|
-
|
|
264
|
-
// Unknown error or max retries reached
|
|
265
|
-
return { success: false, error: errorMsg, skipToFallback: false };
|
|
266
|
-
}
|
|
267
|
-
}
|
|
268
|
-
|
|
269
|
-
return { success: false, error: lastError?.message || 'Max retries exceeded', skipToFallback: false };
|
|
270
|
-
}
|
|
271
|
-
|
|
272
|
-
async function main() {
|
|
273
|
-
// Image mode handling
|
|
274
|
-
if (mode === 'image') {
|
|
275
|
-
if (provider !== 'gemini') {
|
|
276
|
-
console.log('[IMAGE] Error: Image generation only supports gemini provider');
|
|
277
|
-
return;
|
|
278
|
-
}
|
|
279
|
-
|
|
280
|
-
const imageArgs = parseImageArgs(process.argv.slice(4));
|
|
281
|
-
if (!imageArgs.prompt) {
|
|
282
|
-
console.log('[IMAGE] Error: --prompt is required');
|
|
283
|
-
return;
|
|
284
|
-
}
|
|
285
|
-
|
|
286
|
-
// Ensure output directory exists
|
|
287
|
-
const outputDir = path.dirname(imageArgs.output);
|
|
288
|
-
if (outputDir && outputDir !== '.' && !fs.existsSync(outputDir)) {
|
|
289
|
-
fs.mkdirSync(outputDir, { recursive: true });
|
|
290
|
-
}
|
|
291
|
-
|
|
292
|
-
const modelLabel = imageArgs.model === 'nano-banana-pro' ? 'Gemini Pro Image' : 'Gemini Flash Image';
|
|
293
|
-
console.error(`[IMAGE] Generating with ${modelLabel}...`);
|
|
294
|
-
console.error(` Prompt: ${imageArgs.prompt}`);
|
|
295
|
-
console.error(` Size: ${imageArgs.size}`);
|
|
296
|
-
console.error(` Model: ${modelLabel}`);
|
|
297
|
-
console.error(` Output: ${imageArgs.output}`);
|
|
298
|
-
|
|
299
|
-
try {
|
|
300
|
-
const geminiApi = await import(`${LIB_URL}gemini-api.js`);
|
|
301
|
-
const image = await geminiApi.generateImage(imageArgs.prompt, { size: imageArgs.size, model: imageArgs.model });
|
|
302
|
-
fs.writeFileSync(imageArgs.output, image.data);
|
|
303
|
-
const stats = fs.statSync(imageArgs.output);
|
|
304
|
-
const sizeKB = (stats.size / 1024).toFixed(1);
|
|
305
|
-
|
|
306
|
-
console.log(JSON.stringify({
|
|
307
|
-
success: true,
|
|
308
|
-
path: path.resolve(imageArgs.output),
|
|
309
|
-
size: stats.size,
|
|
310
|
-
sizeKB: `${sizeKB} KB`,
|
|
311
|
-
prompt: imageArgs.prompt
|
|
312
|
-
}));
|
|
313
|
-
} catch (err) {
|
|
314
|
-
console.log(JSON.stringify({
|
|
315
|
-
success: false,
|
|
316
|
-
error: err.message,
|
|
317
|
-
prompt: imageArgs.prompt
|
|
318
|
-
}));
|
|
319
|
-
}
|
|
320
|
-
return;
|
|
321
|
-
}
|
|
322
|
-
|
|
323
|
-
// Analyze-image mode handling
|
|
324
|
-
if (mode === 'analyze-image') {
|
|
325
|
-
if (provider !== 'gemini') {
|
|
326
|
-
console.log('[ANALYZE-IMAGE] Error: Image analysis only supports gemini provider');
|
|
327
|
-
return;
|
|
328
|
-
}
|
|
329
|
-
|
|
330
|
-
const analyzeArgs = parseAnalyzeImageArgs(process.argv.slice(4));
|
|
331
|
-
if (!analyzeArgs.imagePath) {
|
|
332
|
-
console.log('[ANALYZE-IMAGE] Error: image path is required');
|
|
333
|
-
return;
|
|
334
|
-
}
|
|
335
|
-
if (!analyzeArgs.prompt) {
|
|
336
|
-
console.log('[ANALYZE-IMAGE] Error: analysis prompt is required');
|
|
337
|
-
return;
|
|
338
|
-
}
|
|
339
|
-
|
|
340
|
-
const absolutePath = path.resolve(analyzeArgs.imagePath);
|
|
341
|
-
if (!fs.existsSync(absolutePath)) {
|
|
342
|
-
console.log(JSON.stringify({
|
|
343
|
-
success: false,
|
|
344
|
-
error: `Image file not found: ${absolutePath}`,
|
|
345
|
-
imagePath: analyzeArgs.imagePath,
|
|
346
|
-
}));
|
|
347
|
-
return;
|
|
348
|
-
}
|
|
349
|
-
|
|
350
|
-
console.error(`[ANALYZE-IMAGE] Analyzing image with Gemini...`);
|
|
351
|
-
console.error(` Image: ${absolutePath}`);
|
|
352
|
-
console.error(` Prompt: ${analyzeArgs.prompt}`);
|
|
353
|
-
|
|
354
|
-
try {
|
|
355
|
-
const geminiApi = await import(`${LIB_URL}gemini-api.js`);
|
|
356
|
-
const options = {};
|
|
357
|
-
if (analyzeArgs.systemPrompt) {
|
|
358
|
-
options.systemPrompt = analyzeArgs.systemPrompt;
|
|
359
|
-
}
|
|
360
|
-
const analysis = await geminiApi.analyzeImage(absolutePath, analyzeArgs.prompt, options);
|
|
361
|
-
|
|
362
|
-
console.log(JSON.stringify({
|
|
363
|
-
success: true,
|
|
364
|
-
analysis,
|
|
365
|
-
imagePath: absolutePath,
|
|
366
|
-
}));
|
|
367
|
-
} catch (err) {
|
|
368
|
-
console.log(JSON.stringify({
|
|
369
|
-
success: false,
|
|
370
|
-
error: err.message,
|
|
371
|
-
imagePath: absolutePath,
|
|
372
|
-
}));
|
|
373
|
-
}
|
|
374
|
-
return;
|
|
375
|
-
}
|
|
376
|
-
|
|
377
|
-
// Text generation mode (orchestrate / orchestrate-json)
|
|
378
|
-
let prompt;
|
|
379
|
-
let systemPrompt = DEFAULT_SYSTEM_PROMPT;
|
|
380
|
-
|
|
381
|
-
// --input <file> 플래그를 먼저 확인 (위치 인자보다 우선)
|
|
382
|
-
const inputFlagIdx = process.argv.indexOf('--input');
|
|
383
|
-
if (inputFlagIdx !== -1 && process.argv[inputFlagIdx + 1]) {
|
|
384
|
-
const inputFile = process.argv[inputFlagIdx + 1];
|
|
385
|
-
|
|
386
|
-
// Path traversal 방지: 절대경로로 resolve 후 임시 디렉토리 또는 프로젝트 내 확인
|
|
387
|
-
const resolvedInput = path.resolve(inputFile);
|
|
388
|
-
const allowedDirs = [os.tmpdir(), process.cwd()];
|
|
389
|
-
const isAllowed = allowedDirs.some(dir => resolvedInput.startsWith(path.resolve(dir)));
|
|
390
|
-
if (!isAllowed) {
|
|
391
|
-
console.log(`[${provider.toUpperCase()}] Error: Input file must be in temp directory or project directory`);
|
|
392
|
-
return;
|
|
393
|
-
}
|
|
394
|
-
|
|
395
|
-
try {
|
|
396
|
-
const inputData = fs.readFileSync(resolvedInput, 'utf8');
|
|
397
|
-
const parsed = JSON.parse(inputData);
|
|
398
|
-
prompt = parsed.prompt;
|
|
399
|
-
if (parsed.systemPrompt) {
|
|
400
|
-
systemPrompt = parsed.systemPrompt;
|
|
401
|
-
}
|
|
402
|
-
} catch (err) {
|
|
403
|
-
console.log(`[${provider.toUpperCase()}] Error: Failed to read input file: ${err.message}`);
|
|
404
|
-
return;
|
|
405
|
-
}
|
|
406
|
-
} else {
|
|
407
|
-
// CLI argument 사용
|
|
408
|
-
// Usage 1: node script.js gpt orchestrate "system prompt" "user prompt"
|
|
409
|
-
// Usage 2: node script.js gpt orchestrate "user prompt" (uses default system prompt)
|
|
410
|
-
const arg4 = process.argv[4]?.trim();
|
|
411
|
-
const arg5 = process.argv.slice(5).join(' ').trim();
|
|
412
|
-
|
|
413
|
-
if (arg5) {
|
|
414
|
-
// 5번째 인자가 있으면: arg4=시스템 프롬프트, arg5=사용자 프롬프트
|
|
415
|
-
systemPrompt = arg4;
|
|
416
|
-
prompt = arg5;
|
|
417
|
-
} else if (arg4) {
|
|
418
|
-
// 4번째 인자만 있으면: arg4=사용자 프롬프트 (시스템 프롬프트는 기본값)
|
|
419
|
-
prompt = arg4;
|
|
420
|
-
} else {
|
|
421
|
-
// Hook에서 호출: stdin으로 JSON 입력 (fallback)
|
|
422
|
-
let inputData = '';
|
|
423
|
-
for await (const chunk of process.stdin) {
|
|
424
|
-
inputData += chunk;
|
|
425
|
-
}
|
|
426
|
-
|
|
427
|
-
try {
|
|
428
|
-
const parsed = JSON.parse(inputData);
|
|
429
|
-
prompt = parsed.prompt;
|
|
430
|
-
} catch {
|
|
431
|
-
console.log(`[${provider.toUpperCase()}] Error: Invalid JSON input`);
|
|
432
|
-
return;
|
|
433
|
-
}
|
|
434
|
-
}
|
|
435
|
-
}
|
|
436
|
-
|
|
437
|
-
// 접두사 제거
|
|
438
|
-
const prefixPatterns = {
|
|
439
|
-
gpt: /^(gpt[-.\s]|지피티-|vibe-gpt-)\s*/i,
|
|
440
|
-
'gpt-codex': /^(gpt[-.\s]|지피티-|vibe-gpt-)\s*/i,
|
|
441
|
-
gemini: /^(gemini[-.\s]|제미나이-|vibe-gemini-)\s*/i,
|
|
442
|
-
};
|
|
443
|
-
const cleanPrompt = prompt.replace(prefixPatterns[provider] || /^/, '').trim();
|
|
444
|
-
const jsonMode = mode === 'orchestrate-json';
|
|
445
|
-
|
|
446
|
-
// Provider chain: primary → cross fallback
|
|
447
|
-
// gpt-codex fallback: gpt-codex → gemini (codex는 full gpt로 fallback하지 않음)
|
|
448
|
-
const providerLabels = { gpt: 'GPT', 'gpt-codex': 'GPT Codex', gemini: 'Gemini' };
|
|
449
|
-
const isGpt = provider === 'gpt' || provider === 'gpt-codex';
|
|
450
|
-
const providerChain = isGpt
|
|
451
|
-
? [provider, 'gemini']
|
|
452
|
-
: ['gemini', 'gpt'];
|
|
453
|
-
|
|
454
|
-
for (const currentProvider of providerChain) {
|
|
455
|
-
const label = providerLabels[currentProvider] || currentProvider.toUpperCase();
|
|
456
|
-
const result = await callWithRetry(currentProvider, cleanPrompt, systemPrompt, jsonMode);
|
|
457
|
-
|
|
458
|
-
if (result.success) {
|
|
459
|
-
console.log(`${label} response: ${result.result}`);
|
|
460
|
-
return;
|
|
461
|
-
}
|
|
462
|
-
|
|
463
|
-
// Log failure and try fallback
|
|
464
|
-
if (currentProvider !== providerChain[providerChain.length - 1]) {
|
|
465
|
-
const nextProvider = providerChain[providerChain.indexOf(currentProvider) + 1];
|
|
466
|
-
const nextLabel = providerLabels[nextProvider] || nextProvider.toUpperCase();
|
|
467
|
-
console.error(`[${currentProvider.toUpperCase()}] Failed: ${result.error}. Falling back to ${nextLabel}...`);
|
|
468
|
-
} else {
|
|
469
|
-
// All providers failed
|
|
470
|
-
console.log(`[LLM] Error: All providers failed. Last error: ${result.error}`);
|
|
471
|
-
}
|
|
472
|
-
}
|
|
473
|
-
}
|
|
474
|
-
|
|
475
|
-
main();
|
|
1
|
+
/**
|
|
2
|
+
* UserPromptSubmit Hook - LLM 오케스트레이션 (GPT/Gemini)
|
|
3
|
+
*
|
|
4
|
+
* Usage:
|
|
5
|
+
* node llm-orchestrate.js <provider> <mode> "prompt"
|
|
6
|
+
* node llm-orchestrate.js <provider> <mode> "systemPrompt" "prompt"
|
|
7
|
+
*
|
|
8
|
+
* provider: gpt | gemini
|
|
9
|
+
* mode: orchestrate | orchestrate-json | image | analyze-image
|
|
10
|
+
*
|
|
11
|
+
* Image Mode:
|
|
12
|
+
* node llm-orchestrate.js gemini image "prompt" --output "./image.png"
|
|
13
|
+
* node llm-orchestrate.js gemini image "prompt" --output "./image.png" --size "1920x1080"
|
|
14
|
+
*
|
|
15
|
+
* Features:
|
|
16
|
+
* - CLI-based: GPT → codex exec, Gemini → gemini -p
|
|
17
|
+
* - Exponential backoff retry (3 attempts)
|
|
18
|
+
* - Auto fallback: gpt ↔ gemini
|
|
19
|
+
* - Overload/rate-limit detection
|
|
20
|
+
* - Image generation (Gemini API, Gemini Flash Image model)
|
|
21
|
+
* - Image analysis (Gemini API multimodal)
|
|
22
|
+
*
|
|
23
|
+
* Analyze-Image Mode:
|
|
24
|
+
* node llm-orchestrate.js gemini analyze-image "./image.png" "Analyze this UI"
|
|
25
|
+
* node llm-orchestrate.js gemini analyze-image "./image.png" "prompt" --system "system prompt"
|
|
26
|
+
*
|
|
27
|
+
* Input: JSON from stdin with { prompt: string } (when no CLI args)
|
|
28
|
+
*/
|
|
29
|
+
import { getLibBaseUrl, readVibeConfig } from './utils.js';
|
|
30
|
+
import fs from 'fs';
|
|
31
|
+
import path from 'path';
|
|
32
|
+
import os from 'os';
|
|
33
|
+
import crypto from 'crypto';
|
|
34
|
+
import { execSync, spawn } from 'child_process';
|
|
35
|
+
|
|
36
|
+
const LIB_URL = getLibBaseUrl();
|
|
37
|
+
const DEFAULT_SYSTEM_PROMPT = 'You are a helpful assistant.';
|
|
38
|
+
|
|
39
|
+
const provider = process.argv[2] || 'gemini';
|
|
40
|
+
const mode = process.argv[3] || 'orchestrate';
|
|
41
|
+
|
|
42
|
+
// Retry configuration
|
|
43
|
+
const MAX_RETRIES = 3;
|
|
44
|
+
const INITIAL_DELAY_MS = 2000;
|
|
45
|
+
|
|
46
|
+
// Errors that should skip retry and go to fallback immediately
|
|
47
|
+
const SKIP_RETRY_PATTERNS = [
|
|
48
|
+
/rate.?limit/i,
|
|
49
|
+
/quota/i,
|
|
50
|
+
/unauthorized/i,
|
|
51
|
+
/forbidden/i,
|
|
52
|
+
/authentication/i,
|
|
53
|
+
/401/,
|
|
54
|
+
/403/,
|
|
55
|
+
/429/,
|
|
56
|
+
];
|
|
57
|
+
|
|
58
|
+
// Errors that should trigger retry with backoff
|
|
59
|
+
const RETRY_PATTERNS = [
|
|
60
|
+
/overload/i,
|
|
61
|
+
/503/,
|
|
62
|
+
/5\d\d/,
|
|
63
|
+
/network/i,
|
|
64
|
+
/timeout/i,
|
|
65
|
+
/ECONNRESET/i,
|
|
66
|
+
/ETIMEDOUT/i,
|
|
67
|
+
/spawn error/i,
|
|
68
|
+
];
|
|
69
|
+
|
|
70
|
+
function shouldSkipRetry(errorMsg) {
|
|
71
|
+
return SKIP_RETRY_PATTERNS.some(pattern => pattern.test(errorMsg));
|
|
72
|
+
}
|
|
73
|
+
|
|
74
|
+
function shouldRetry(errorMsg) {
|
|
75
|
+
return RETRY_PATTERNS.some(pattern => pattern.test(errorMsg));
|
|
76
|
+
}
|
|
77
|
+
|
|
78
|
+
function sleep(ms) {
|
|
79
|
+
return new Promise(resolve => setTimeout(resolve, ms));
|
|
80
|
+
}
|
|
81
|
+
|
|
82
|
+
// ============================================
|
|
83
|
+
// Image Generation (delegates to gemini-api)
|
|
84
|
+
// ============================================
|
|
85
|
+
|
|
86
|
+
function parseImageArgs(args) {
|
|
87
|
+
const result = { prompt: null, output: './generated-image.png', size: '1024x1024', model: 'nano-banana' };
|
|
88
|
+
for (let i = 0; i < args.length; i++) {
|
|
89
|
+
if (args[i] === '--output' || args[i] === '-o') {
|
|
90
|
+
result.output = args[++i];
|
|
91
|
+
} else if (args[i] === '--size' || args[i] === '-s') {
|
|
92
|
+
result.size = args[++i];
|
|
93
|
+
} else if (args[i] === '--pro') {
|
|
94
|
+
result.model = 'nano-banana-pro';
|
|
95
|
+
} else if (!args[i].startsWith('-') && !result.prompt) {
|
|
96
|
+
result.prompt = args[i];
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
return result;
|
|
100
|
+
}
|
|
101
|
+
|
|
102
|
+
// ============================================
|
|
103
|
+
// Image Analysis (delegates to gemini-api)
|
|
104
|
+
// ============================================
|
|
105
|
+
|
|
106
|
+
function parseAnalyzeImageArgs(args) {
|
|
107
|
+
const result = { imagePath: null, prompt: null, systemPrompt: null };
|
|
108
|
+
for (let i = 0; i < args.length; i++) {
|
|
109
|
+
if (args[i] === '--system' || args[i] === '-s') {
|
|
110
|
+
result.systemPrompt = args[++i];
|
|
111
|
+
} else if (!args[i].startsWith('-')) {
|
|
112
|
+
if (!result.imagePath) {
|
|
113
|
+
result.imagePath = args[i];
|
|
114
|
+
} else if (!result.prompt) {
|
|
115
|
+
result.prompt = args[i];
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
return result;
|
|
120
|
+
}
|
|
121
|
+
|
|
122
|
+
// ============================================
|
|
123
|
+
// CLI Provider Functions
|
|
124
|
+
// ============================================
|
|
125
|
+
|
|
126
|
+
const CLI_TIMEOUT_MS = 120000;
|
|
127
|
+
const IS_WINDOWS = os.platform() === 'win32';
|
|
128
|
+
|
|
129
|
+
function spawnCli(cmd, args, options) {
|
|
130
|
+
// Windows: .cmd shim은 spawn 직접 실행 불가 → cmd.exe /c 로 실행
|
|
131
|
+
if (IS_WINDOWS) {
|
|
132
|
+
return spawn(process.env.ComSpec || 'cmd.exe', ['/c', cmd, ...args], options);
|
|
133
|
+
}
|
|
134
|
+
return spawn(cmd, args, options);
|
|
135
|
+
}
|
|
136
|
+
|
|
137
|
+
function buildCliPrompt(prompt, sysPrompt, jsonMode) {
|
|
138
|
+
let fullPrompt = sysPrompt && sysPrompt !== DEFAULT_SYSTEM_PROMPT
|
|
139
|
+
? `[System]\n${sysPrompt}\n\n[User]\n${prompt}`
|
|
140
|
+
: prompt;
|
|
141
|
+
if (jsonMode) {
|
|
142
|
+
fullPrompt += '\n\nIMPORTANT: You must respond with valid JSON only.';
|
|
143
|
+
}
|
|
144
|
+
return fullPrompt;
|
|
145
|
+
}
|
|
146
|
+
|
|
147
|
+
|
|
148
|
+
function callCodexCli(prompt, sysPrompt, jsonMode, model) {
|
|
149
|
+
const fullPrompt = buildCliPrompt(prompt, sysPrompt, jsonMode);
|
|
150
|
+
const outputFile = path.join(os.tmpdir(), `vibe-codex-${crypto.randomUUID()}.txt`);
|
|
151
|
+
// stdin pipe로 프롬프트 전달 (shell escaping 이슈 회피)
|
|
152
|
+
const args = ['exec', '-m', model, '--sandbox', 'read-only', '--ephemeral', '-o', outputFile, '-'];
|
|
153
|
+
|
|
154
|
+
// config에 저장된 API key를 환경변수로 전달 (임베딩과 동일한 키 사용)
|
|
155
|
+
const vibeConfig = readVibeConfig();
|
|
156
|
+
const apiKey = vibeConfig.credentials?.gpt?.apiKey || process.env.OPENAI_API_KEY;
|
|
157
|
+
const env = apiKey ? { ...process.env, OPENAI_API_KEY: apiKey } : process.env;
|
|
158
|
+
|
|
159
|
+
return new Promise((resolve, reject) => {
|
|
160
|
+
const proc = spawnCli('codex', args, {
|
|
161
|
+
stdio: ['pipe', 'pipe', 'pipe'],
|
|
162
|
+
timeout: CLI_TIMEOUT_MS,
|
|
163
|
+
env,
|
|
164
|
+
});
|
|
165
|
+
proc.stdin.write(fullPrompt);
|
|
166
|
+
proc.stdin.end();
|
|
167
|
+
|
|
168
|
+
let stderr = '';
|
|
169
|
+
proc.stderr.on('data', (d) => { stderr += d.toString(); });
|
|
170
|
+
|
|
171
|
+
proc.on('close', (code) => {
|
|
172
|
+
if (code === 0 && fs.existsSync(outputFile)) {
|
|
173
|
+
const result = fs.readFileSync(outputFile, 'utf8');
|
|
174
|
+
try { fs.unlinkSync(outputFile); } catch { /* ignore */ }
|
|
175
|
+
resolve(result);
|
|
176
|
+
} else {
|
|
177
|
+
try { fs.unlinkSync(outputFile); } catch { /* ignore */ }
|
|
178
|
+
reject(new Error(`codex exec failed (code ${code}): ${stderr.slice(0, 500)}`));
|
|
179
|
+
}
|
|
180
|
+
});
|
|
181
|
+
|
|
182
|
+
proc.on('error', (err) => {
|
|
183
|
+
try { fs.unlinkSync(outputFile); } catch { /* ignore */ }
|
|
184
|
+
reject(new Error(`codex exec spawn error: ${err.message}`));
|
|
185
|
+
});
|
|
186
|
+
});
|
|
187
|
+
}
|
|
188
|
+
|
|
189
|
+
function callGeminiCli(prompt, sysPrompt, jsonMode, model) {
|
|
190
|
+
const fullPrompt = buildCliPrompt(prompt, sysPrompt, jsonMode);
|
|
191
|
+
// -p 로 headless 모드, stdin으로 프롬프트 전달 (stdin is appended to -p value)
|
|
192
|
+
const args = ['-p', '.', '-o', 'text'];
|
|
193
|
+
if (model) args.push('-m', model);
|
|
194
|
+
|
|
195
|
+
return new Promise((resolve, reject) => {
|
|
196
|
+
const proc = spawnCli('gemini', args, {
|
|
197
|
+
stdio: ['pipe', 'pipe', 'pipe'],
|
|
198
|
+
timeout: CLI_TIMEOUT_MS,
|
|
199
|
+
});
|
|
200
|
+
proc.stdin.write(fullPrompt);
|
|
201
|
+
proc.stdin.end();
|
|
202
|
+
|
|
203
|
+
let stdout = '';
|
|
204
|
+
let stderr = '';
|
|
205
|
+
proc.stdout.on('data', (d) => { stdout += d.toString(); });
|
|
206
|
+
proc.stderr.on('data', (d) => { stderr += d.toString(); });
|
|
207
|
+
|
|
208
|
+
proc.on('close', (code) => {
|
|
209
|
+
if (code === 0 && stdout.trim()) {
|
|
210
|
+
resolve(stdout.trim());
|
|
211
|
+
} else {
|
|
212
|
+
reject(new Error(`gemini cli failed (code ${code}): ${(stderr || stdout).slice(0, 500)}`));
|
|
213
|
+
}
|
|
214
|
+
});
|
|
215
|
+
|
|
216
|
+
proc.on('error', (err) => {
|
|
217
|
+
reject(new Error(`gemini cli spawn error: ${err.message}`));
|
|
218
|
+
});
|
|
219
|
+
});
|
|
220
|
+
}
|
|
221
|
+
|
|
222
|
+
async function callProvider(providerName, prompt, sysPrompt, jsonMode) {
|
|
223
|
+
const vibeConfig = readVibeConfig();
|
|
224
|
+
|
|
225
|
+
if (providerName === 'gpt' || providerName === 'gpt-codex') {
|
|
226
|
+
const isCodex = providerName === 'gpt-codex';
|
|
227
|
+
const model = isCodex
|
|
228
|
+
? (vibeConfig.models?.gptCodex || process.env.GPT_CODEX_MODEL || 'gpt-5.3-codex')
|
|
229
|
+
: (vibeConfig.models?.gpt || process.env.GPT_MODEL || 'gpt-5.4');
|
|
230
|
+
return await callCodexCli(prompt, sysPrompt, jsonMode, model);
|
|
231
|
+
}
|
|
232
|
+
|
|
233
|
+
if (providerName === 'gemini') {
|
|
234
|
+
const model = vibeConfig.models?.gemini || process.env.GEMINI_MODEL || 'gemini-3.1-pro-preview';
|
|
235
|
+
return await callGeminiCli(prompt, sysPrompt, jsonMode, model);
|
|
236
|
+
}
|
|
237
|
+
|
|
238
|
+
throw new Error(`Unknown provider: ${providerName}`);
|
|
239
|
+
}
|
|
240
|
+
|
|
241
|
+
async function callWithRetry(providerName, prompt, sysPrompt, jsonMode) {
|
|
242
|
+
let lastError;
|
|
243
|
+
|
|
244
|
+
for (let attempt = 1; attempt <= MAX_RETRIES; attempt++) {
|
|
245
|
+
try {
|
|
246
|
+
return { success: true, result: await callProvider(providerName, prompt, sysPrompt, jsonMode) };
|
|
247
|
+
} catch (e) {
|
|
248
|
+
lastError = e;
|
|
249
|
+
const errorMsg = e.message || String(e);
|
|
250
|
+
|
|
251
|
+
// Skip retry for auth/quota errors - go to fallback immediately
|
|
252
|
+
if (shouldSkipRetry(errorMsg)) {
|
|
253
|
+
return { success: false, error: errorMsg, skipToFallback: true };
|
|
254
|
+
}
|
|
255
|
+
|
|
256
|
+
// Retry with backoff for transient errors
|
|
257
|
+
if (shouldRetry(errorMsg) && attempt < MAX_RETRIES) {
|
|
258
|
+
const delay = INITIAL_DELAY_MS * Math.pow(2, attempt - 1);
|
|
259
|
+
console.error(`[${providerName.toUpperCase()}] Retry ${attempt}/${MAX_RETRIES} after ${delay}ms...`);
|
|
260
|
+
await sleep(delay);
|
|
261
|
+
continue;
|
|
262
|
+
}
|
|
263
|
+
|
|
264
|
+
// Unknown error or max retries reached
|
|
265
|
+
return { success: false, error: errorMsg, skipToFallback: false };
|
|
266
|
+
}
|
|
267
|
+
}
|
|
268
|
+
|
|
269
|
+
return { success: false, error: lastError?.message || 'Max retries exceeded', skipToFallback: false };
|
|
270
|
+
}
|
|
271
|
+
|
|
272
|
+
async function main() {
|
|
273
|
+
// Image mode handling
|
|
274
|
+
if (mode === 'image') {
|
|
275
|
+
if (provider !== 'gemini') {
|
|
276
|
+
console.log('[IMAGE] Error: Image generation only supports gemini provider');
|
|
277
|
+
return;
|
|
278
|
+
}
|
|
279
|
+
|
|
280
|
+
const imageArgs = parseImageArgs(process.argv.slice(4));
|
|
281
|
+
if (!imageArgs.prompt) {
|
|
282
|
+
console.log('[IMAGE] Error: --prompt is required');
|
|
283
|
+
return;
|
|
284
|
+
}
|
|
285
|
+
|
|
286
|
+
// Ensure output directory exists
|
|
287
|
+
const outputDir = path.dirname(imageArgs.output);
|
|
288
|
+
if (outputDir && outputDir !== '.' && !fs.existsSync(outputDir)) {
|
|
289
|
+
fs.mkdirSync(outputDir, { recursive: true });
|
|
290
|
+
}
|
|
291
|
+
|
|
292
|
+
const modelLabel = imageArgs.model === 'nano-banana-pro' ? 'Gemini Pro Image' : 'Gemini Flash Image';
|
|
293
|
+
console.error(`[IMAGE] Generating with ${modelLabel}...`);
|
|
294
|
+
console.error(` Prompt: ${imageArgs.prompt}`);
|
|
295
|
+
console.error(` Size: ${imageArgs.size}`);
|
|
296
|
+
console.error(` Model: ${modelLabel}`);
|
|
297
|
+
console.error(` Output: ${imageArgs.output}`);
|
|
298
|
+
|
|
299
|
+
try {
|
|
300
|
+
const geminiApi = await import(`${LIB_URL}gemini-api.js`);
|
|
301
|
+
const image = await geminiApi.generateImage(imageArgs.prompt, { size: imageArgs.size, model: imageArgs.model });
|
|
302
|
+
fs.writeFileSync(imageArgs.output, image.data);
|
|
303
|
+
const stats = fs.statSync(imageArgs.output);
|
|
304
|
+
const sizeKB = (stats.size / 1024).toFixed(1);
|
|
305
|
+
|
|
306
|
+
console.log(JSON.stringify({
|
|
307
|
+
success: true,
|
|
308
|
+
path: path.resolve(imageArgs.output),
|
|
309
|
+
size: stats.size,
|
|
310
|
+
sizeKB: `${sizeKB} KB`,
|
|
311
|
+
prompt: imageArgs.prompt
|
|
312
|
+
}));
|
|
313
|
+
} catch (err) {
|
|
314
|
+
console.log(JSON.stringify({
|
|
315
|
+
success: false,
|
|
316
|
+
error: err.message,
|
|
317
|
+
prompt: imageArgs.prompt
|
|
318
|
+
}));
|
|
319
|
+
}
|
|
320
|
+
return;
|
|
321
|
+
}
|
|
322
|
+
|
|
323
|
+
// Analyze-image mode handling
|
|
324
|
+
if (mode === 'analyze-image') {
|
|
325
|
+
if (provider !== 'gemini') {
|
|
326
|
+
console.log('[ANALYZE-IMAGE] Error: Image analysis only supports gemini provider');
|
|
327
|
+
return;
|
|
328
|
+
}
|
|
329
|
+
|
|
330
|
+
const analyzeArgs = parseAnalyzeImageArgs(process.argv.slice(4));
|
|
331
|
+
if (!analyzeArgs.imagePath) {
|
|
332
|
+
console.log('[ANALYZE-IMAGE] Error: image path is required');
|
|
333
|
+
return;
|
|
334
|
+
}
|
|
335
|
+
if (!analyzeArgs.prompt) {
|
|
336
|
+
console.log('[ANALYZE-IMAGE] Error: analysis prompt is required');
|
|
337
|
+
return;
|
|
338
|
+
}
|
|
339
|
+
|
|
340
|
+
const absolutePath = path.resolve(analyzeArgs.imagePath);
|
|
341
|
+
if (!fs.existsSync(absolutePath)) {
|
|
342
|
+
console.log(JSON.stringify({
|
|
343
|
+
success: false,
|
|
344
|
+
error: `Image file not found: ${absolutePath}`,
|
|
345
|
+
imagePath: analyzeArgs.imagePath,
|
|
346
|
+
}));
|
|
347
|
+
return;
|
|
348
|
+
}
|
|
349
|
+
|
|
350
|
+
console.error(`[ANALYZE-IMAGE] Analyzing image with Gemini...`);
|
|
351
|
+
console.error(` Image: ${absolutePath}`);
|
|
352
|
+
console.error(` Prompt: ${analyzeArgs.prompt}`);
|
|
353
|
+
|
|
354
|
+
try {
|
|
355
|
+
const geminiApi = await import(`${LIB_URL}gemini-api.js`);
|
|
356
|
+
const options = {};
|
|
357
|
+
if (analyzeArgs.systemPrompt) {
|
|
358
|
+
options.systemPrompt = analyzeArgs.systemPrompt;
|
|
359
|
+
}
|
|
360
|
+
const analysis = await geminiApi.analyzeImage(absolutePath, analyzeArgs.prompt, options);
|
|
361
|
+
|
|
362
|
+
console.log(JSON.stringify({
|
|
363
|
+
success: true,
|
|
364
|
+
analysis,
|
|
365
|
+
imagePath: absolutePath,
|
|
366
|
+
}));
|
|
367
|
+
} catch (err) {
|
|
368
|
+
console.log(JSON.stringify({
|
|
369
|
+
success: false,
|
|
370
|
+
error: err.message,
|
|
371
|
+
imagePath: absolutePath,
|
|
372
|
+
}));
|
|
373
|
+
}
|
|
374
|
+
return;
|
|
375
|
+
}
|
|
376
|
+
|
|
377
|
+
// Text generation mode (orchestrate / orchestrate-json)
|
|
378
|
+
let prompt;
|
|
379
|
+
let systemPrompt = DEFAULT_SYSTEM_PROMPT;
|
|
380
|
+
|
|
381
|
+
// --input <file> 플래그를 먼저 확인 (위치 인자보다 우선)
|
|
382
|
+
const inputFlagIdx = process.argv.indexOf('--input');
|
|
383
|
+
if (inputFlagIdx !== -1 && process.argv[inputFlagIdx + 1]) {
|
|
384
|
+
const inputFile = process.argv[inputFlagIdx + 1];
|
|
385
|
+
|
|
386
|
+
// Path traversal 방지: 절대경로로 resolve 후 임시 디렉토리 또는 프로젝트 내 확인
|
|
387
|
+
const resolvedInput = path.resolve(inputFile);
|
|
388
|
+
const allowedDirs = [os.tmpdir(), process.cwd()];
|
|
389
|
+
const isAllowed = allowedDirs.some(dir => resolvedInput.startsWith(path.resolve(dir)));
|
|
390
|
+
if (!isAllowed) {
|
|
391
|
+
console.log(`[${provider.toUpperCase()}] Error: Input file must be in temp directory or project directory`);
|
|
392
|
+
return;
|
|
393
|
+
}
|
|
394
|
+
|
|
395
|
+
try {
|
|
396
|
+
const inputData = fs.readFileSync(resolvedInput, 'utf8');
|
|
397
|
+
const parsed = JSON.parse(inputData);
|
|
398
|
+
prompt = parsed.prompt;
|
|
399
|
+
if (parsed.systemPrompt) {
|
|
400
|
+
systemPrompt = parsed.systemPrompt;
|
|
401
|
+
}
|
|
402
|
+
} catch (err) {
|
|
403
|
+
console.log(`[${provider.toUpperCase()}] Error: Failed to read input file: ${err.message}`);
|
|
404
|
+
return;
|
|
405
|
+
}
|
|
406
|
+
} else {
|
|
407
|
+
// CLI argument 사용
|
|
408
|
+
// Usage 1: node script.js gpt orchestrate "system prompt" "user prompt"
|
|
409
|
+
// Usage 2: node script.js gpt orchestrate "user prompt" (uses default system prompt)
|
|
410
|
+
const arg4 = process.argv[4]?.trim();
|
|
411
|
+
const arg5 = process.argv.slice(5).join(' ').trim();
|
|
412
|
+
|
|
413
|
+
if (arg5) {
|
|
414
|
+
// 5번째 인자가 있으면: arg4=시스템 프롬프트, arg5=사용자 프롬프트
|
|
415
|
+
systemPrompt = arg4;
|
|
416
|
+
prompt = arg5;
|
|
417
|
+
} else if (arg4) {
|
|
418
|
+
// 4번째 인자만 있으면: arg4=사용자 프롬프트 (시스템 프롬프트는 기본값)
|
|
419
|
+
prompt = arg4;
|
|
420
|
+
} else {
|
|
421
|
+
// Hook에서 호출: stdin으로 JSON 입력 (fallback)
|
|
422
|
+
let inputData = '';
|
|
423
|
+
for await (const chunk of process.stdin) {
|
|
424
|
+
inputData += chunk;
|
|
425
|
+
}
|
|
426
|
+
|
|
427
|
+
try {
|
|
428
|
+
const parsed = JSON.parse(inputData);
|
|
429
|
+
prompt = parsed.prompt;
|
|
430
|
+
} catch {
|
|
431
|
+
console.log(`[${provider.toUpperCase()}] Error: Invalid JSON input`);
|
|
432
|
+
return;
|
|
433
|
+
}
|
|
434
|
+
}
|
|
435
|
+
}
|
|
436
|
+
|
|
437
|
+
// 접두사 제거
|
|
438
|
+
const prefixPatterns = {
|
|
439
|
+
gpt: /^(gpt[-.\s]|지피티-|vibe-gpt-)\s*/i,
|
|
440
|
+
'gpt-codex': /^(gpt[-.\s]|지피티-|vibe-gpt-)\s*/i,
|
|
441
|
+
gemini: /^(gemini[-.\s]|제미나이-|vibe-gemini-)\s*/i,
|
|
442
|
+
};
|
|
443
|
+
const cleanPrompt = prompt.replace(prefixPatterns[provider] || /^/, '').trim();
|
|
444
|
+
const jsonMode = mode === 'orchestrate-json';
|
|
445
|
+
|
|
446
|
+
// Provider chain: primary → cross fallback
|
|
447
|
+
// gpt-codex fallback: gpt-codex → gemini (codex는 full gpt로 fallback하지 않음)
|
|
448
|
+
const providerLabels = { gpt: 'GPT', 'gpt-codex': 'GPT Codex', gemini: 'Gemini' };
|
|
449
|
+
const isGpt = provider === 'gpt' || provider === 'gpt-codex';
|
|
450
|
+
const providerChain = isGpt
|
|
451
|
+
? [provider, 'gemini']
|
|
452
|
+
: ['gemini', 'gpt'];
|
|
453
|
+
|
|
454
|
+
for (const currentProvider of providerChain) {
|
|
455
|
+
const label = providerLabels[currentProvider] || currentProvider.toUpperCase();
|
|
456
|
+
const result = await callWithRetry(currentProvider, cleanPrompt, systemPrompt, jsonMode);
|
|
457
|
+
|
|
458
|
+
if (result.success) {
|
|
459
|
+
console.log(`${label} response: ${result.result}`);
|
|
460
|
+
return;
|
|
461
|
+
}
|
|
462
|
+
|
|
463
|
+
// Log failure and try fallback
|
|
464
|
+
if (currentProvider !== providerChain[providerChain.length - 1]) {
|
|
465
|
+
const nextProvider = providerChain[providerChain.indexOf(currentProvider) + 1];
|
|
466
|
+
const nextLabel = providerLabels[nextProvider] || nextProvider.toUpperCase();
|
|
467
|
+
console.error(`[${currentProvider.toUpperCase()}] Failed: ${result.error}. Falling back to ${nextLabel}...`);
|
|
468
|
+
} else {
|
|
469
|
+
// All providers failed
|
|
470
|
+
console.log(`[LLM] Error: All providers failed. Last error: ${result.error}`);
|
|
471
|
+
}
|
|
472
|
+
}
|
|
473
|
+
}
|
|
474
|
+
|
|
475
|
+
main();
|