Skip to content

Commit 7b70f4f

Browse files
committed
feat(core): add ModelCapability enum to define various model input/output capabilities
1 parent c5f51b8 commit 7b70f4f

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

packages/core/src/provider/registry.ts

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,15 @@ export enum ModelType {
145145
TEXT_EMBEDDING_MODEL = 'textEmbeddingModel'
146146
}
147147

148+
export enum ModelCapability {
149+
IMAGE_INPUT = 'imageInput',
150+
IMAGE_OUTPUT = 'imageOutput',
151+
AUDIO_INPUT = 'audioInput',
152+
AUDIO_OUTPUT = 'audioOutput',
153+
VIDEO_INPUT = 'videoInput',
154+
VIDEO_OUTPUT = 'videoOutput'
155+
}
156+
148157
export interface ModelInfo {
149158
name: string
150159

@@ -155,6 +164,8 @@ export interface ModelInfo {
155164
costPerTokenInput?: number
156165

157166
costPerTokenOutput?: number
167+
168+
capability?: ModelCapability[]
158169
}
159170

160171
export type PlatformModelInfo = Readonly<

0 commit comments

Comments
 (0)