ros/automaker
1
0
Fork 0
mirror of https://github.com/AutoMaker-Org/automaker.git synced 2026-02-03 08:53:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(server): Extract JSON extraction utility to shared module

Browse Source 
Created libs/server/src/lib/json-extractor.ts with reusable JSON
extraction utilities for parsing AI responses:

- extractJson<T>(): Multi-strategy JSON extraction
- extractJsonWithKey<T>(): Extract with required key validation
- extractJsonWithArray<T>(): Extract with array property validation

Strategies (tried in order):
1. JSON in ```json code block
2. JSON in ``` code block
3. Find JSON object by matching braces (with optional required key)
4. Find any JSON object by matching braces
5. First { to last }
6. Parse entire response

Updated:
- generate-suggestions.ts: Use extractJsonWithArray('suggestions')
- validate-issue.ts: Use extractJson()

Both files now use the shared utility instead of local implementations,
following DRY principle.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Kacper
2025-12-30 15:18:45 +01:00
parent 26e4ac0d2f
commit 19016f03d7
3 changed files with 221 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

105
apps/server/src/routes/suggestions/generate-suggestions.ts
View File

@@ -11,6 +11,7 @@ import { createLogger } from '@automaker/utils';
import { DEFAULT_PHASE_MODELS, isCursorModel } from '@automaker/types';
import { resolveModelString } from '@automaker/model-resolver';
import { createSuggestionsOptions } from '../../lib/sdk-options.js';
import { extractJsonWithArray } from '../../lib/json-extractor.js';
import { ProviderFactory } from '../../providers/provider-factory.js';
import { FeatureLoader } from '../../services/feature-loader.js';
import { getAppSpecPath } from '@automaker/platform';
@@ -289,9 +290,13 @@ ${JSON.stringify(suggestionsSchema, null, 2)}`;
})),
});
} else {
// Fallback: try to parse from text using multiple strategies
// Fallback: try to parse from text using shared extraction utility
logger.warn('No structured output received, attempting to parse from text');
const parsed = extractSuggestionsJson(responseText);
const parsed = extractJsonWithArray<{ suggestions: Array<Record<string, unknown>> }>(
responseText,
'suggestions',
{ logger }
);
if (parsed && parsed.suggestions) {
events.emit('suggestions:event', {
type: 'suggestions_complete',
@@ -322,99 +327,3 @@ ${JSON.stringify(suggestionsSchema, null, 2)}`;
});
}
}
/**
* Extract suggestions JSON from response text using multiple strategies.
* Handles various formats: markdown code blocks, raw JSON, etc.
*/
function extractSuggestionsJson(
responseText: string
): { suggestions: Array<Record<string, unknown>> } | null {
const strategies = [
// Strategy 1: JSON in ```json code block
() => {
const match = responseText.match(/```json\s*([\s\S]*?)```/);
if (match) {
return JSON.parse(match[1].trim());
}
return null;
},
// Strategy 2: JSON in ``` code block (no language specified)
() => {
const match = responseText.match(/```\s*([\s\S]*?)```/);
if (match) {
const content = match[1].trim();
if (content.startsWith('{') && content.includes('"suggestions"')) {
return JSON.parse(content);
}
}
return null;
},
// Strategy 3: Find JSON object containing "suggestions" array
() => {
// Find the start of the JSON object
const startIdx = responseText.indexOf('{"suggestions"');
if (startIdx === -1) return null;
// Find matching closing brace by counting brackets
let depth = 0;
let endIdx = -1;
for (let i = startIdx; i < responseText.length; i++) {
if (responseText[i] === '{') depth++;
if (responseText[i] === '}') {
depth--;
if (depth === 0) {
endIdx = i + 1;
break;
}
}
}
if (endIdx > startIdx) {
return JSON.parse(responseText.slice(startIdx, endIdx));
}
return null;
},
// Strategy 4: Find any JSON object with suggestions
() => {
const startIdx = responseText.indexOf('{');
if (startIdx === -1) return null;
// Find matching closing brace
let depth = 0;
let endIdx = -1;
for (let i = startIdx; i < responseText.length; i++) {
if (responseText[i] === '{') depth++;
if (responseText[i] === '}') {
depth--;
if (depth === 0) {
endIdx = i + 1;
break;
}
}
}
if (endIdx > startIdx) {
const parsed = JSON.parse(responseText.slice(startIdx, endIdx));
if (parsed.suggestions && Array.isArray(parsed.suggestions)) {
return parsed;
}
}
return null;
},
];
for (const strategy of strategies) {
try {
const result = strategy();
if (result && result.suggestions && Array.isArray(result.suggestions)) {
logger.debug('Successfully extracted suggestions JSON');
return result;
}
} catch {
// Strategy failed, try next
}
}
return null;
}