Skip to content

Commit 2452140

Browse files
committed
update import
1 parent f42e486 commit 2452140

File tree

1 file changed

+6
-10
lines changed
  • backend/src/build-system/handlers/frontend-code-generate

1 file changed

+6
-10
lines changed

backend/src/build-system/handlers/frontend-code-generate/index.ts

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,7 @@ import * as path from 'path';
1212
import { readFile } from 'fs/promises';
1313

1414
// Utility functions (similar to your parseGenerateTag, removeCodeBlockFences)
15-
import {
16-
parseGenerateTag,
17-
removeCodeBlockFences,
18-
} from 'src/build-system/utils/database-utils';
15+
import { parseGenerateTag } from 'src/build-system/utils/strings';
1916

2017
// The function from step #1
2118
import { generateFrontEndCodePrompt, generateCSSPrompt } from './prompt';
@@ -127,14 +124,13 @@ export class FrontendCodeHandler implements BuildHandler<string> {
127124
this.logger.debug('Generated frontend code prompt.');
128125

129126
let generatedCode = '';
127+
const model = 'gpt-4o-mini';
130128
try {
131129
// Call the model
132-
const modelResponse = await context.model.chatSync(
133-
{
134-
content: frontendCodePrompt,
135-
},
136-
'gpt-4o-mini', // or whichever model you need
137-
);
130+
const modelResponse = await context.model.chatSync({
131+
model,
132+
messages: [{ content: frontendCodePrompt, role: 'system' }],
133+
});
138134

139135
// Parse the output
140136
generatedCode = parseGenerateTag(modelResponse);

0 commit comments

Comments
 (0)