feat: Support a way to define default model by adding DEFAULT_MODEL env.

This commit is contained in:
Wayland Zhan
2024-04-19 06:57:15 +00:00
parent 9b2cb1e1c3
commit c96e4b7966
6 changed files with 81 additions and 15 deletions

View File

@@ -1,14 +1,15 @@
import { useMemo } from "react";
import { useAccessStore, useAppConfig } from "../store";
import { collectModels } from "./model";
import { collectModels, collectModelsWithDefaultModel } from "./model";
export function useAllModels() {
const accessStore = useAccessStore();
const configStore = useAppConfig();
const models = useMemo(() => {
return collectModels(
return collectModelsWithDefaultModel(
configStore.models,
[configStore.customModels, accessStore.customModels].join(","),
accessStore.defaultModel,
);
}, [accessStore.customModels, configStore.customModels, configStore.models]);