Skip to content

Commit 88998cc

Browse files
authored
Update model version and stop char (#3)
1 parent 6033bcf commit 88998cc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/extension.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -258,11 +258,11 @@ export function activateOpenAI(context: vscode.ExtensionContext) {
258258
async function requestMqlTranslation(comment: string): Promise<string> {
259259
return openAIClient
260260
.createCompletion({
261-
model: "curie:ft-sublime-security-2023-04-21-20-26-44",
261+
model: "curie:ft-sublime-security-2023-08-05-00-34-40",
262262
max_tokens: 128,
263263
temperature: 0.3,
264-
stop: "END",
265-
prompt: `in mql ${comment.replace("'", "\"")} ->`,
264+
stop: ["\n"],
265+
prompt: `${comment.replace("'", "\"")} ->`,
266266
})
267267
.catch((err: any) => {
268268
vscode.window.showErrorMessage("Error fetching MQL translation: ", err);

0 commit comments

Comments
 (0)