Feat: native model capabilities (#14787)

This commit is contained in:
Abdul Rahman
2025-10-01 22:07:21 +05:30
committed by GitHub
parent d715533a90
commit 20403664e3
29 changed files with 523 additions and 59 deletions
@@ -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()
@@ -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,