Add brain emoji model selector with fullscreen modal
- Create ModelSelectionModal component with list-style interface
- Add brain emoji (🧠) indicator below chat input, left-aligned
- Show current model name in lowercase or red X if not configured
- Fullscreen modal displays all available models with configuration status
- Green checkmark for configured models, red alert for unconfigured
- Integrate with existing API key management and model selection logic
- Modal handles API key setup workflow when selecting unconfigured models
- Clean separation between compact indicator and full selection interface
This commit is contained in:
parent
b33bde6b0e
commit
6a79264315
3 changed files with 194 additions and 0 deletions
|
|
@ -140,6 +140,8 @@ const ChatApp: React.FC<ChatAppProps> = () => {
|
|||
db={db as IDBPDatabase}
|
||||
selectedModel={selectedModel}
|
||||
onApiKeyUpdate={handleApiKeyUpdate}
|
||||
onModelChange={handleModelChange}
|
||||
apiKeyUpdateTrigger={apiKeyUpdateTrigger}
|
||||
mcpTools={mcpTools}
|
||||
/>
|
||||
</div>
|
||||
|
|
|
|||
|
|
@ -8,9 +8,11 @@ import { type IDBPDatabase } from 'idb'
|
|||
import { type Tool } from 'use-mcp/react'
|
||||
import ChatMessage from './ChatMessage'
|
||||
import ChatInput from './ChatInput'
|
||||
import ModelSelectionModal from './ModelSelectionModal'
|
||||
import { useAutoscroll } from '../hooks/useAutoscroll'
|
||||
import { useStreamResponse } from '../hooks/useStreamResponse'
|
||||
import { setApiKey } from '../utils/apiKeys'
|
||||
import { hasApiKey } from '../utils/apiKeys'
|
||||
import ApiKeyModal from './ApiKeyModal'
|
||||
|
||||
interface ConversationThreadProps {
|
||||
|
|
@ -21,6 +23,8 @@ interface ConversationThreadProps {
|
|||
db: IDBPDatabase
|
||||
selectedModel: Model
|
||||
onApiKeyUpdate: () => void
|
||||
onModelChange: (model: Model) => void
|
||||
apiKeyUpdateTrigger: number
|
||||
mcpTools: Tool[]
|
||||
}
|
||||
|
||||
|
|
@ -32,6 +36,8 @@ const ConversationThread: React.FC<ConversationThreadProps> = ({
|
|||
db,
|
||||
selectedModel,
|
||||
onApiKeyUpdate,
|
||||
onModelChange,
|
||||
apiKeyUpdateTrigger,
|
||||
mcpTools,
|
||||
}) => {
|
||||
const [input, setInput] = useState<string>('')
|
||||
|
|
@ -39,6 +45,7 @@ const ConversationThread: React.FC<ConversationThreadProps> = ({
|
|||
isOpen: false,
|
||||
model: null,
|
||||
})
|
||||
const [modelSelectionModal, setModelSelectionModal] = useState(false)
|
||||
|
||||
const { messagesEndRef, messagesContainerRef, scrollToBottom } = useAutoscroll()
|
||||
|
||||
|
|
@ -212,6 +219,23 @@ const ConversationThread: React.FC<ConversationThreadProps> = ({
|
|||
controller={controller}
|
||||
messagesCount={currentConversation.messages.length}
|
||||
/>
|
||||
|
||||
{/* Model selector indicator */}
|
||||
<div className="flex items-center gap-2 mt-2">
|
||||
<button
|
||||
onClick={() => setModelSelectionModal(true)}
|
||||
className="flex items-center gap-2 text-zinc-600 hover:text-zinc-900 transition-colors"
|
||||
>
|
||||
<span className="text-lg">🧠</span>
|
||||
<span className="text-sm text-zinc-500">
|
||||
{hasApiKey(selectedModel.provider.id) ? (
|
||||
selectedModel.name.toLowerCase()
|
||||
) : (
|
||||
<span className="text-red-500">✕</span>
|
||||
)}
|
||||
</span>
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
|
|
@ -221,6 +245,14 @@ const ConversationThread: React.FC<ConversationThreadProps> = ({
|
|||
provider={apiKeyModal.model?.provider ?? { id: '', name: '', baseUrl: '', apiKeyHeader: '', documentationUrl: '' }}
|
||||
onSave={handleApiKeySave}
|
||||
/>
|
||||
|
||||
<ModelSelectionModal
|
||||
isOpen={modelSelectionModal}
|
||||
onClose={() => setModelSelectionModal(false)}
|
||||
selectedModel={selectedModel}
|
||||
onModelChange={onModelChange}
|
||||
apiKeyUpdateTrigger={apiKeyUpdateTrigger}
|
||||
/>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
|
|
|
|||
160
examples/chat-ui/src/components/ModelSelectionModal.tsx
Normal file
160
examples/chat-ui/src/components/ModelSelectionModal.tsx
Normal file
|
|
@ -0,0 +1,160 @@
|
|||
import React, { useState, useEffect } from 'react'
|
||||
import { X, AlertCircle, CheckCircle } from 'lucide-react'
|
||||
import { type Model, availableModels } from '../types/models'
|
||||
import { hasApiKey, setApiKey, clearApiKey } from '../utils/apiKeys'
|
||||
import ApiKeyModal from './ApiKeyModal'
|
||||
|
||||
interface ModelSelectionModalProps {
|
||||
isOpen: boolean
|
||||
onClose: () => void
|
||||
selectedModel: Model
|
||||
onModelChange: (model: Model) => void
|
||||
apiKeyUpdateTrigger: number
|
||||
}
|
||||
|
||||
const ModelSelectionModal: React.FC<ModelSelectionModalProps> = ({
|
||||
isOpen,
|
||||
onClose,
|
||||
selectedModel,
|
||||
onModelChange,
|
||||
apiKeyUpdateTrigger,
|
||||
}) => {
|
||||
const [apiKeyModal, setApiKeyModal] = useState<{ isOpen: boolean; model: Model | null }>({
|
||||
isOpen: false,
|
||||
model: null,
|
||||
})
|
||||
const [apiKeyStatuses, setApiKeyStatuses] = useState<Record<string, boolean>>({})
|
||||
|
||||
useEffect(() => {
|
||||
const statuses: Record<string, boolean> = {}
|
||||
availableModels.forEach(model => {
|
||||
statuses[model.provider.id] = hasApiKey(model.provider.id)
|
||||
})
|
||||
setApiKeyStatuses(statuses)
|
||||
}, [apiKeyUpdateTrigger])
|
||||
|
||||
useEffect(() => {
|
||||
if (isOpen) {
|
||||
document.body.style.overflow = 'hidden'
|
||||
} else {
|
||||
document.body.style.overflow = 'unset'
|
||||
}
|
||||
|
||||
return () => {
|
||||
document.body.style.overflow = 'unset'
|
||||
}
|
||||
}, [isOpen])
|
||||
|
||||
const handleModelSelect = (model: Model) => {
|
||||
const hasKey = hasApiKey(model.provider.id)
|
||||
|
||||
if (!hasKey) {
|
||||
setApiKeyModal({ isOpen: true, model })
|
||||
} else {
|
||||
onModelChange(model)
|
||||
onClose()
|
||||
}
|
||||
}
|
||||
|
||||
const handleApiKeySave = (apiKey: string) => {
|
||||
if (apiKeyModal.model) {
|
||||
setApiKey(apiKeyModal.model.provider.id, apiKey)
|
||||
setApiKeyStatuses(prev => ({ ...prev, [apiKeyModal.model!.provider.id]: true }))
|
||||
onModelChange(apiKeyModal.model)
|
||||
setApiKeyModal({ isOpen: false, model: null })
|
||||
onClose()
|
||||
}
|
||||
}
|
||||
|
||||
const handleClearApiKey = (providerId: string, e: React.MouseEvent) => {
|
||||
e.stopPropagation()
|
||||
clearApiKey(providerId)
|
||||
setApiKeyStatuses(prev => ({ ...prev, [providerId]: false }))
|
||||
}
|
||||
|
||||
const getStatusIcon = (providerId: string) => {
|
||||
const hasKey = apiKeyStatuses[providerId]
|
||||
|
||||
if (hasKey) {
|
||||
return <CheckCircle size={20} className="text-green-500" />
|
||||
} else {
|
||||
return <AlertCircle size={20} className="text-red-500" />
|
||||
}
|
||||
}
|
||||
|
||||
if (!isOpen) return null
|
||||
|
||||
return (
|
||||
<>
|
||||
<div className="fixed inset-0 bg-black bg-opacity-50 z-50 flex items-center justify-center p-4">
|
||||
<div className="bg-white rounded-lg shadow-xl w-full max-w-2xl max-h-[80vh] overflow-hidden">
|
||||
<div className="flex items-center justify-between p-6 border-b border-zinc-200">
|
||||
<h2 className="text-xl font-semibold text-zinc-900">Select Model</h2>
|
||||
<button
|
||||
onClick={onClose}
|
||||
className="text-zinc-400 hover:text-zinc-600 p-1"
|
||||
>
|
||||
<X size={24} />
|
||||
</button>
|
||||
</div>
|
||||
|
||||
<div className="p-6 overflow-y-auto">
|
||||
<div className="space-y-3">
|
||||
{availableModels.map((model) => {
|
||||
const isSelected = model.id === selectedModel.id
|
||||
const hasKey = apiKeyStatuses[model.provider.id]
|
||||
|
||||
return (
|
||||
<div
|
||||
key={model.id}
|
||||
className={`border rounded-lg p-4 cursor-pointer transition-colors ${
|
||||
isSelected
|
||||
? 'border-blue-500 bg-blue-50'
|
||||
: 'border-zinc-200 hover:border-zinc-300 hover:bg-zinc-50'
|
||||
}`}
|
||||
onClick={() => handleModelSelect(model)}
|
||||
>
|
||||
<div className="flex items-center justify-between">
|
||||
<div className="flex items-center gap-3">
|
||||
<span className="text-2xl">{model.provider.logo}</span>
|
||||
<div>
|
||||
<div className="font-medium text-zinc-900">{model.name}</div>
|
||||
<div className="text-sm text-zinc-500">{model.provider.name}</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="flex items-center gap-2">
|
||||
{getStatusIcon(model.provider.id)}
|
||||
<span className={`text-sm ${hasKey ? 'text-green-600' : 'text-red-600'}`}>
|
||||
{hasKey ? 'Configured' : 'Not configured'}
|
||||
</span>
|
||||
{hasKey && (
|
||||
<button
|
||||
onClick={(e) => handleClearApiKey(model.provider.id, e)}
|
||||
className="text-zinc-400 hover:text-red-500 p-1 ml-2"
|
||||
title="Clear API key"
|
||||
>
|
||||
<X size={16} />
|
||||
</button>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
)
|
||||
})}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<ApiKeyModal
|
||||
isOpen={apiKeyModal.isOpen}
|
||||
onClose={() => setApiKeyModal({ isOpen: false, model: null })}
|
||||
provider={apiKeyModal.model?.provider ?? { id: '', name: '', baseUrl: '', apiKeyHeader: '', documentationUrl: '' }}
|
||||
onSave={handleApiKeySave}
|
||||
/>
|
||||
</>
|
||||
)
|
||||
}
|
||||
|
||||
export default ModelSelectionModal
|
||||
Loading…
Reference in a new issue