lee/puter
1
0
Fork 0
mirror of https://github.com/HeyPuter/puter.git synced 2026-08-12 01:01:47 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

tweak(puerai): add model parameter

Browse Source
This commit is contained in:
KernelDeimos
2024-08-21 23:15:53 -04:00
parent 439f52b5a3
commit 7b254e8676
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/backend/src/modules/puterai/AIInterfaceService.js
+1
View File
@@ -34,6 +34,7 @@ class AIInterfaceService extends BaseService {
messages: { type: 'json' },
vision: { type: 'flag' },
stream: { type: 'flag' },
model: { type: 'string' },
},
result: { type: 'json' },
}
src/backend/src/modules/puterai/ClaudeService.js
+2 -2
View File
@@ -26,7 +26,7 @@ class ClaudeService extends BaseService {
static IMPLEMENTS = {
['puter-chat-completion']: {
async complete ({ messages, stream }) {
async complete ({ messages, stream, model }) {
const adapted_messages = [];
const system_prompts = [];
@@ -67,7 +67,7 @@ class ClaudeService extends BaseService {
}, stream);
(async () => {
const completion = await this.anthropic.messages.stream({
model: 'claude-3-5-sonnet-20240620',
model: model ?? 'claude-3-5-sonnet-20240620',
max_tokens: 1000,
temperature: 0,
system: PUTER_PROMPT + JSON.stringify(system_prompts),
src/backend/src/modules/puterai/OpenAICompletionService.js
+2 -3
View File
@@ -23,7 +23,7 @@ class OpenAICompletionService extends BaseService {
static IMPLEMENTS = {
['puter-chat-completion']: {
async complete ({ messages, test_mode, stream }) {
async complete ({ messages, test_mode, stream, model }) {
if ( test_mode ) {
const { LoremIpsum } = require('lorem-ipsum');
const li = new LoremIpsum({
@@ -49,9 +49,8 @@ class OpenAICompletionService extends BaseService {
}
}
const model = 'gpt-4o';
return await this.complete(messages, {
model,
model: model ?? 'gpt-4o',
moderation: true,
stream,
});