33import type { UseChatHelpers } from "@ai-sdk/react" ;
44import type { UIMessage } from "ai" ;
55import equal from "fast-deep-equal" ;
6- import { ArrowUpIcon , BrainIcon , EyeIcon , LockIcon , WrenchIcon } from "lucide-react" ;
6+ import {
7+ ArrowUpIcon ,
8+ BrainIcon ,
9+ EyeIcon ,
10+ LockIcon ,
11+ WrenchIcon ,
12+ } from "lucide-react" ;
713import {
814 type ChangeEvent ,
915 type Dispatch ,
@@ -28,11 +34,6 @@ import {
2834 ModelSelectorName ,
2935 ModelSelectorTrigger ,
3036} from "@/components/ai-elements/model-selector" ;
31- import {
32- Tooltip ,
33- TooltipContent ,
34- TooltipTrigger ,
35- } from "@/components/ui/tooltip" ;
3637import {
3738 type ChatModel ,
3839 chatModels ,
@@ -547,16 +548,27 @@ function PureModelSelectorCompact({
547548 ]
548549 : chatModels ;
549550
550- const grouped : Record < string , { model : ChatModel ; curated : boolean } [ ] > = { } ;
551+ const grouped : Record <
552+ string ,
553+ { model : ChatModel ; curated : boolean } [ ]
554+ > = { } ;
551555 for ( const model of allModels ) {
552- const key = curatedIds . has ( model . id ) ? "_available" : model . provider ;
553- if ( ! grouped [ key ] ) grouped [ key ] = [ ] ;
556+ const key = curatedIds . has ( model . id )
557+ ? "_available"
558+ : model . provider ;
559+ if ( ! grouped [ key ] ) {
560+ grouped [ key ] = [ ] ;
561+ }
554562 grouped [ key ] . push ( { model, curated : curatedIds . has ( model . id ) } ) ;
555563 }
556564
557565 const sortedKeys = Object . keys ( grouped ) . sort ( ( a , b ) => {
558- if ( a === "_available" ) return - 1 ;
559- if ( b === "_available" ) return 1 ;
566+ if ( a === "_available" ) {
567+ return - 1 ;
568+ }
569+ if ( b === "_available" ) {
570+ return 1 ;
571+ }
560572 return a . localeCompare ( b ) ;
561573 } ) ;
562574
@@ -587,7 +599,11 @@ function PureModelSelectorCompact({
587599
588600 return sortedKeys . map ( ( key ) => (
589601 < ModelSelectorGroup
590- heading = { key === "_available" ? "Available" : ( providerNames [ key ] ?? key ) }
602+ heading = {
603+ key === "_available"
604+ ? "Available"
605+ : ( providerNames [ key ] ?? key )
606+ }
591607 key = { key }
592608 >
593609 { grouped [ key ] . map ( ( { model, curated } ) => {
@@ -602,7 +618,9 @@ function PureModelSelectorCompact({
602618 ) }
603619 key = { model . id }
604620 onSelect = { ( ) => {
605- if ( ! curated ) return ;
621+ if ( ! curated ) {
622+ return ;
623+ }
606624 onModelChange ?.( model . id ) ;
607625 setCookie ( "chat-model" , model . id ) ;
608626 setOpen ( false ) ;
0 commit comments