feat openai-list-models

This commit is contained in:
sam hoang
2024-12-25 01:02:33 +07:00
parent bc7dee6097
commit 198e1965e9
6 changed files with 407 additions and 7 deletions

View File

@@ -17,6 +17,7 @@ export interface ExtensionStateContextType extends ExtensionState {
showWelcome: boolean
theme: any
openRouterModels: Record<string, ModelInfo>
openAiModels: string[],
mcpServers: McpServer[]
filePaths: string[]
setApiConfiguration: (config: ApiConfiguration) => void
@@ -59,6 +60,8 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
const [openRouterModels, setOpenRouterModels] = useState<Record<string, ModelInfo>>({
[openRouterDefaultModelId]: openRouterDefaultModelInfo,
})
const [openAiModels, setOpenAiModels] = useState<string[]>([])
const [mcpServers, setMcpServers] = useState<McpServer[]>([])
const handleMessage = useCallback((event: MessageEvent) => {
@@ -116,6 +119,11 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
})
break
}
case "openAiModels": {
const updatedModels = message.openAiModels ?? []
setOpenAiModels(updatedModels)
break
}
case "mcpServers": {
setMcpServers(message.mcpServers ?? [])
break
@@ -135,6 +143,7 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
showWelcome,
theme,
openRouterModels,
openAiModels,
mcpServers,
filePaths,
soundVolume: state.soundVolume,