Merge pull request #207 from samhvw8/feat/openai-list-models

feat list model for open ai compatible
This commit is contained in:
Matt Rubens
2024-12-27 22:44:27 -08:00
committed by GitHub
6 changed files with 406 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
@@ -61,6 +62,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) => {
@@ -118,6 +121,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
@@ -137,6 +145,7 @@ export const ExtensionStateContextProvider: React.FC<{ children: React.ReactNode
showWelcome,
theme,
openRouterModels,
openAiModels,
mcpServers,
filePaths,
soundVolume: state.soundVolume,