11import { AnalyticsEvent } from '@codebuff/common/constants/analytics-events'
22import { models , PROFIT_MARGIN } from '@codebuff/common/old-constants'
33import { buildArray } from '@codebuff/common/util/array'
4- import { getErrorObject } from '@codebuff/common/util/error'
4+ import { getErrorObject , promptAborted , promptSuccess } from '@codebuff/common/util/error'
55import { convertCbToModelMessages } from '@codebuff/common/util/messages'
66import { isExplicitlyDefinedModel } from '@codebuff/common/util/model-utils'
77import { StopSequenceHandler } from '@codebuff/common/util/stop-sequence'
@@ -199,7 +199,7 @@ export async function* promptAiSdkStream(
199199 } ,
200200 'Skipping stream due to canceled user input' ,
201201 )
202- return null
202+ return promptAborted ( 'User cancelled input' )
203203 }
204204
205205 const modelParams : ModelRequestParams = {
@@ -569,7 +569,7 @@ export async function* promptAiSdkStream(
569569 }
570570 }
571571
572- return messageId
572+ return promptSuccess ( messageId )
573573}
574574
575575export async function promptAiSdk (
@@ -585,7 +585,7 @@ export async function promptAiSdk(
585585 } ,
586586 'Skipping prompt due to canceled user input' ,
587587 )
588- return ''
588+ return promptAborted ( 'User cancelled input' )
589589 }
590590
591591 const modelParams : ModelRequestParams = {
@@ -627,7 +627,7 @@ export async function promptAiSdk(
627627 )
628628 }
629629
630- return content
630+ return promptSuccess ( content )
631631}
632632
633633export async function promptAiSdkStructured < T > (
@@ -643,7 +643,7 @@ export async function promptAiSdkStructured<T>(
643643 } ,
644644 'Skipping structured prompt due to canceled user input' ,
645645 )
646- throw new Error ( 'Request aborted ')
646+ return promptAborted ( 'User cancelled input ')
647647 }
648648 const modelParams : ModelRequestParams = {
649649 apiKey : params . apiKey ,
@@ -686,5 +686,5 @@ export async function promptAiSdkStructured<T>(
686686 )
687687 }
688688
689- return content
689+ return promptSuccess ( content )
690690}
0 commit comments