feat: Add ModelSelect component to SidepanelForm

This commit is contained in:
n4ze3m
2024-05-08 10:44:51 +05:30
parent f630addefc
commit 88ad1fcab7
3 changed files with 70 additions and 2 deletions

View File

@@ -1,4 +1,4 @@
import { useQuery } from "@tanstack/react-query"
import { useQuery, useQueryClient } from "@tanstack/react-query"
import { Select } from "antd"
import { RotateCcw } from "lucide-react"
import { useEffect, useState } from "react"
@@ -15,6 +15,7 @@ import {
export const EmptySidePanel = () => {
const [ollamaURL, setOllamaURL] = useState<string>("")
const { t } = useTranslation(["playground", "common"])
const queryClient = useQueryClient()
const {
data: ollamaInfo,
status: ollamaStatus,
@@ -26,7 +27,9 @@ export const EmptySidePanel = () => {
const ollamaURL = await getOllamaURL()
const isOk = await isOllamaRunning()
const models = await fetchChatModels({ returnEmpty: false })
queryClient.invalidateQueries({
queryKey: ["getAllModelsForSelect"]
})
return {
isOk,
models,