Update ModelSelectionWidgetFactory.kt

acharneski · acharneski · commit 8f3355827da4 · 2024-03-29T02:24:06.000-04:00
diff --git a/src/main/kotlin/com/github/simiacryptus/aicoder/ui/ModelSelectionWidgetFactory.kt b/src/main/kotlin/com/github/simiacryptus/aicoder/ui/ModelSelectionWidgetFactory.kt
@@ -69,6 +69,10 @@ class ModelSelectionWidgetFactory : StatusBarWidgetFactory {
           hasFocus: Boolean
         ) {
           text = value // Here you can add more customization if needed
+         if (value != null) {
+           val model = models.find { it.modelName == value }
+           text = "${model?.provider?.name} - $value"
+         }
         }
       }
 

Original file line number	Diff line number	Diff line change
`@@ -69,6 +69,10 @@ class ModelSelectionWidgetFactory : StatusBarWidgetFactory {`
`69`	`69`	`hasFocus: Boolean`
`70`	`70`	`) {`
`71`	`71`	`text = value // Here you can add more customization if needed`
	`72`	`+ if (value != null) {`
	`73`	`+ val model = models.find { it.modelName == value }`
	`74`	`+ text = "${model?.provider?.name} - $value"`
	`75`	`+ }`
`72`	`76`	`}`
`73`	`77`	`}`
`74`	`78`