@@ -53,6 +53,8 @@ class AIChatService extends BaseService {
53
53
54
54
const svc_event = this . services . get ( 'event' ) ;
55
55
svc_event . on ( 'ai.prompt.report-usage' , async ( _ , details ) => {
56
+ if ( details . service_used === 'fake-chat' ) return ;
57
+
56
58
const values = {
57
59
user_id : details . actor ?. type ?. user ?. id ,
58
60
app_id : details . actor ?. type ?. app ?. id ?? null ,
@@ -289,7 +291,9 @@ class AIChatService extends BaseService {
289
291
const svc_driver = this . services . get ( 'driver' ) ;
290
292
let ret , error , errors = [ ] ;
291
293
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
+ } ) ;
293
297
await this . check_usage_ ( {
294
298
actor : Context . get ( 'actor' ) ,
295
299
service : service_used ,
@@ -594,13 +598,21 @@ class AIChatService extends BaseService {
594
598
} ) ;
595
599
}
596
600
597
- get_model_from_request ( parameters ) {
601
+ get_model_from_request ( parameters , modified_context = { } ) {
598
602
const client_driver_call = Context . get ( 'client_driver_call' ) ;
599
603
let { intended_service } = client_driver_call ;
604
+
605
+ if ( modified_context . intended_service ) {
606
+ intended_service = modified_context . intended_service ;
607
+ }
600
608
601
609
let model = parameters . model ;
602
610
if ( ! model ) {
603
611
const service = this . services . get ( intended_service ) ;
612
+ console . log ( {
613
+ what : intended_service ,
614
+ w : service . get_default_model
615
+ } ) ;
604
616
if ( ! service . get_default_model ) {
605
617
throw new Error ( 'could not infer model from service' ) ;
606
618
}
0 commit comments