Skip to content

Commit 7b254e8

Browse files
committed
tweak(puerai): add model parameter
1 parent 439f52b commit 7b254e8

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

src/backend/src/modules/puterai/AIInterfaceService.js

+1
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ class AIInterfaceService extends BaseService {
3434
messages: { type: 'json' },
3535
vision: { type: 'flag' },
3636
stream: { type: 'flag' },
37+
model: { type: 'string' },
3738
},
3839
result: { type: 'json' },
3940
}

src/backend/src/modules/puterai/ClaudeService.js

+2-2
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ class ClaudeService extends BaseService {
2626

2727
static IMPLEMENTS = {
2828
['puter-chat-completion']: {
29-
async complete ({ messages, stream }) {
29+
async complete ({ messages, stream, model }) {
3030
const adapted_messages = [];
3131

3232
const system_prompts = [];
@@ -67,7 +67,7 @@ class ClaudeService extends BaseService {
6767
}, stream);
6868
(async () => {
6969
const completion = await this.anthropic.messages.stream({
70-
model: 'claude-3-5-sonnet-20240620',
70+
model: model ?? 'claude-3-5-sonnet-20240620',
7171
max_tokens: 1000,
7272
temperature: 0,
7373
system: PUTER_PROMPT + JSON.stringify(system_prompts),

src/backend/src/modules/puterai/OpenAICompletionService.js

+2-3
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ class OpenAICompletionService extends BaseService {
2323

2424
static IMPLEMENTS = {
2525
['puter-chat-completion']: {
26-
async complete ({ messages, test_mode, stream }) {
26+
async complete ({ messages, test_mode, stream, model }) {
2727
if ( test_mode ) {
2828
const { LoremIpsum } = require('lorem-ipsum');
2929
const li = new LoremIpsum({
@@ -49,9 +49,8 @@ class OpenAICompletionService extends BaseService {
4949
}
5050
}
5151

52-
const model = 'gpt-4o';
5352
return await this.complete(messages, {
54-
model,
53+
model: model ?? 'gpt-4o',
5554
moderation: true,
5655
stream,
5756
});

0 commit comments

Comments
 (0)