Feat: native model capabilities (#14787)
This commit is contained in:
@@ -19,6 +19,15 @@ registerEnumType(ModelProvider, {
|
||||
name: 'ModelProvider',
|
||||
});
|
||||
|
||||
@ObjectType()
|
||||
class NativeModelCapabilities {
|
||||
@Field(() => Boolean, { nullable: true })
|
||||
webSearch?: boolean;
|
||||
|
||||
@Field(() => Boolean, { nullable: true })
|
||||
twitterSearch?: boolean;
|
||||
}
|
||||
|
||||
@ObjectType()
|
||||
export class ClientAIModelConfig {
|
||||
@Field(() => String)
|
||||
@@ -35,6 +44,9 @@ export class ClientAIModelConfig {
|
||||
|
||||
@Field(() => Number)
|
||||
outputCostPer1kTokensInCredits: number;
|
||||
|
||||
@Field(() => NativeModelCapabilities, { nullable: true })
|
||||
nativeCapabilities?: NativeModelCapabilities;
|
||||
}
|
||||
|
||||
@ObjectType()
|
||||
|
||||
+1
@@ -46,6 +46,7 @@ export class ClientConfigService {
|
||||
modelId: registeredModel.modelId,
|
||||
label: builtInModel?.label || registeredModel.modelId,
|
||||
provider: registeredModel.provider,
|
||||
nativeCapabilities: builtInModel?.nativeCapabilities,
|
||||
inputCostPer1kTokensInCredits: builtInModel
|
||||
? convertCentsToBillingCredits(
|
||||
builtInModel.inputCostPer1kTokensInCents,
|
||||
|
||||
Reference in New Issue
Block a user