Skip to content

Commit 2c84446

Browse files
committed
feat(sdk): Update LLM functions to return PromptResult type
1 parent 2ffb5a3 commit 2c84446

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

sdk/src/impl/llm.ts

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import { AnalyticsEvent } from '@codebuff/common/constants/analytics-events'
22
import { models, PROFIT_MARGIN } from '@codebuff/common/old-constants'
33
import { buildArray } from '@codebuff/common/util/array'
4-
import { getErrorObject } from '@codebuff/common/util/error'
4+
import { getErrorObject, promptAborted, promptSuccess } from '@codebuff/common/util/error'
55
import { convertCbToModelMessages } from '@codebuff/common/util/messages'
66
import { isExplicitlyDefinedModel } from '@codebuff/common/util/model-utils'
77
import { 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

575575
export 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

633633
export 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

Comments
 (0)