Skip to content
This repository was archived by the owner on Jul 4, 2025. It is now read-only.

Commit 5d02f22

Browse files
committed
chore: add benchmark's model config argument
1 parent 03072aa commit 5d02f22

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

cortex-js/src/infrastructure/commanders/benchmark.command.ts

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,10 @@ import { BenchmarkConfig } from './types/benchmark-config.interface';
44

55
@SubCommand({
66
name: 'benchmark',
7-
subCommands: [],
7+
arguments: '[model_id]',
8+
argsDescription: {
9+
model_id: 'Model to benchmark with',
10+
},
811
description:
912
'Benchmark and analyze the performance of a specific AI model using a variety of system resources',
1013
})
@@ -17,7 +20,10 @@ export class BenchmarkCommand extends CommandRunner {
1720
passedParams: string[],
1821
options?: Partial<BenchmarkConfig>,
1922
): Promise<void> {
20-
return this.benchmarkUsecases.benchmark(options ?? {});
23+
return this.benchmarkUsecases.benchmark({
24+
...options,
25+
...(passedParams[0] ? { modelId: passedParams[0] } : {}),
26+
});
2127
}
2228

2329
@Option({

0 commit comments

Comments
 (0)