Skip to content

Commit 13a895b

Browse files
committed
fix: fake_chat default model and usage errors
1 parent 1e2f9df commit 13a895b

File tree

2 files changed

+17
-2
lines changed

2 files changed

+17
-2
lines changed

src/backend/src/modules/puterai/AIChatService.js

+14-2
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,8 @@ class AIChatService extends BaseService {
5353

5454
const svc_event = this.services.get('event');
5555
svc_event.on('ai.prompt.report-usage', async (_, details) => {
56+
if ( details.service_used === 'fake-chat' ) return;
57+
5658
const values = {
5759
user_id: details.actor?.type?.user?.id,
5860
app_id: details.actor?.type?.app?.id ?? null,
@@ -289,7 +291,9 @@ class AIChatService extends BaseService {
289291
const svc_driver = this.services.get('driver');
290292
let ret, error, errors = [];
291293
let service_used = intended_service;
292-
let model_used = this.get_model_from_request(parameters);
294+
let model_used = this.get_model_from_request(parameters, {
295+
intended_service
296+
});
293297
await this.check_usage_({
294298
actor: Context.get('actor'),
295299
service: service_used,
@@ -594,13 +598,21 @@ class AIChatService extends BaseService {
594598
});
595599
}
596600

597-
get_model_from_request (parameters) {
601+
get_model_from_request (parameters, modified_context = {}) {
598602
const client_driver_call = Context.get('client_driver_call');
599603
let { intended_service } = client_driver_call;
604+
605+
if ( modified_context.intended_service ) {
606+
intended_service = modified_context.intended_service;
607+
}
600608

601609
let model = parameters.model;
602610
if ( ! model ) {
603611
const service = this.services.get(intended_service);
612+
console.log({
613+
what: intended_service,
614+
w: service.get_default_model
615+
});
604616
if ( ! service.get_default_model ) {
605617
throw new Error('could not infer model from service');
606618
}

src/backend/src/modules/puterai/FakeChatService.js

+3
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,9 @@ const BaseService = require("../../services/BaseService");
99
* Implements the 'puter-chat-completion' interface with list() and complete() methods.
1010
*/
1111
class FakeChatService extends BaseService {
12+
get_default_model () {
13+
return 'fake';
14+
}
1215
static IMPLEMENTS = {
1316
['puter-chat-completion']: {
1417
/**

0 commit comments

Comments
 (0)