add default model + minor fixes (#2638)

* add default model + minor fixes

* fix build

* minor additional fix

* build fix
This commit is contained in:
pablodanswer
2024-10-01 10:43:43 -07:00
committed by GitHub
parent 2f2fc08553
commit 9a4e51a18e
5 changed files with 20 additions and 7 deletions

View File

@@ -119,7 +119,7 @@ export function PersonasTable({
id: persona.id.toString(),
cells: [
<div key="name" className="flex">
{!persona.is_default_persona && (
{!persona.builtin_persona && (
<FiEdit2
className="mr-1 my-auto cursor-pointer"
onClick={() =>

View File

@@ -208,7 +208,7 @@ export function ChatPage({
};
const llmOverrideManager = useLlmOverride(
user?.preferences.default_model,
user?.preferences.default_model ?? null,
selectedChatSession,
defaultTemperature
);
@@ -1779,6 +1779,7 @@ export function ChatPage({
{settingsToggled && (
<SetDefaultModelModal
setPopup={setPopup}
setLlmOverride={llmOverrideManager.setGlobalDefault}
defaultModel={user?.preferences.default_model!}
refreshUser={refreshUser}

View File

@@ -7,22 +7,23 @@ import { LLMProviderDescriptor } from "@/app/admin/configuration/llm/interfaces"
import { destructureValue, structureValue } from "@/lib/llm/utils";
import { setUserDefaultModel } from "@/lib/users/UserSettings";
import { useRouter } from "next/navigation";
import { usePopup } from "@/components/admin/connectors/Popup";
import { PopupSpec } from "@/components/admin/connectors/Popup";
export function SetDefaultModelModal({
setPopup,
llmProviders,
onClose,
setLlmOverride,
defaultModel,
refreshUser,
}: {
setPopup: (popupSpec: PopupSpec | null) => void;
llmProviders: LLMProviderDescriptor[];
setLlmOverride: Dispatch<SetStateAction<LlmOverride>>;
onClose: () => void;
defaultModel: string | null;
refreshUser: () => void;
}) {
const { popup, setPopup } = usePopup();
const containerRef = useRef<HTMLDivElement>(null);
const messageRef = useRef<HTMLDivElement>(null);
@@ -127,7 +128,6 @@ export function SetDefaultModelModal({
modalClassName="rounded-lg bg-white max-w-xl"
>
<>
{popup}
<div className="flex mb-4">
<h2 className="text-2xl text-emphasis font-bold flex my-auto">
Set Default Model

View File

@@ -151,7 +151,7 @@ export function useLlmOverride(
defaultTemperature?: number
): LlmOverrideManager {
const [globalDefault, setGlobalDefault] = useState<LlmOverride>(
globalModel
globalModel != null
? destructureValue(globalModel)
: {
name: "",
@@ -182,6 +182,18 @@ export function useLlmOverride(
defaultTemperature != undefined ? defaultTemperature : 0
);
useEffect(() => {
setGlobalDefault(
globalModel != null
? destructureValue(globalModel)
: {
name: "",
provider: "",
modelName: "",
}
);
}, [globalModel]);
useEffect(() => {
setTemperature(defaultTemperature !== undefined ? defaultTemperature : 0);
}, [defaultTemperature]);