@@ -45,13 +45,16 @@ import (
4545)
4646
4747var rootCmd = & cobra.Command {
48- Use : "gorse-benchmark" ,
49- Short : "Gorse Benchmarking Tool" ,
48+ Use : "gorse-cli" ,
49+ Short : "Gorse command line tool" ,
50+ Run : func (cmd * cobra.Command , args []string ) {
51+ _ = cmd .Help ()
52+ },
5053}
5154
52- var llmCmd = & cobra.Command {
53- Use : "llm" ,
54- Short : "Benchmark LLM models" ,
55+ var benchLLMCmd = & cobra.Command {
56+ Use : "bench- llm" ,
57+ Short : "Benchmark LLM models for ranking " ,
5558 Run : func (cmd * cobra.Command , args []string ) {
5659 // Load configuration
5760 configPath , _ := cmd .Flags ().GetString ("config" )
@@ -357,8 +360,8 @@ func EvaluateLLM(cfg *config.Config, train, test dataset.CFSplit, scores *sync.M
357360
358361func init () {
359362 rootCmd .PersistentFlags ().StringP ("config" , "c" , "" , "Path to configuration file" )
360- rootCmd .AddCommand (llmCmd )
361- llmCmd .PersistentFlags ().IntP ("shots" , "s" , math .MaxInt , "Number of shots for each user" )
363+ rootCmd .AddCommand (benchLLMCmd )
364+ benchLLMCmd .PersistentFlags ().IntP ("shots" , "s" , math .MaxInt , "Number of shots for each user" )
362365}
363366
364367func main () {
0 commit comments