chatbot-ui/components/sidebar/items/models/model-item.tsx

107 lines
3.0 KiB
TypeScript

import { Input } from "@/components/ui/input"
import { Label } from "@/components/ui/label"
import { MODEL_NAME_MAX } from "@/db/limits"
import { Tables, TablesUpdate } from "@/supabase/types"
import { IconSparkles } from "@tabler/icons-react"
import { FC, useState } from "react"
import { SidebarItem } from "../all/sidebar-display-item"
import { useTranslation } from 'react-i18next'
interface ModelItemProps {
model: Tables<"models">
}
export const ModelItem: FC<ModelItemProps> = ({ model }) => {
const { t } = useTranslation()
const [isTyping, setIsTyping] = useState(false)
const [apiKey, setApiKey] = useState(model.api_key)
const [baseUrl, setBaseUrl] = useState(model.base_url)
const [description, setDescription] = useState(model.description)
const [modelId, setModelId] = useState(model.model_id)
const [name, setName] = useState(model.name)
const [contextLength, setContextLength] = useState(model.context_length)
return (
<SidebarItem
item={model}
isTyping={isTyping}
contentType="models"
icon={<IconSparkles height={30} width={30} />}
updateState={
{
api_key: apiKey,
base_url: baseUrl,
description,
context_length: contextLength,
model_id: modelId,
name
} as TablesUpdate<"models">
}
renderInputs={() => (
<>
<div className="space-y-1">
<Label>{t("side.name")}</Label>
<Input
placeholder={t("side.modelNamePlaceholder")}
value={name}
onChange={e => setName(e.target.value)}
maxLength={MODEL_NAME_MAX}
/>
</div>
<div className="space-y-1">
<Label>{t("side.modelId")}</Label>
<Input
placeholder={t("side.modelIdPlaceholder")}
value={modelId}
onChange={e => setModelId(e.target.value)}
/>
</div>
<div className="space-y-1">
<Label>{t("side.baseUrl")}</Label>
<Input
placeholder={t("side.baseUrlPlaceholder")}
value={baseUrl}
onChange={e => setBaseUrl(e.target.value)}
/>
<div className="pt-1 text-xs italic">
{t("side.apiCompatibilityNotice")}
</div>
</div>
<div className="space-y-1">
<Label>{t("side.apiKey")}</Label>
<Input
type="password"
placeholder={t("side.apiKeyPlaceholder")}
value={apiKey}
onChange={e => setApiKey(e.target.value)}
/>
</div>
<div className="space-y-1">
<Label>{t("side.maxContextLength")}</Label>
<Input
type="number"
placeholder="4096"
min={0}
value={contextLength}
onChange={e => setContextLength(parseInt(e.target.value))}
/>
</div>
</>
)}
/>
)
}