diff --git a/app/ui/src/components/Common/BotForm.tsx b/app/ui/src/components/Common/BotForm.tsx
index 90898e4d..a2837662 100644
--- a/app/ui/src/components/Common/BotForm.tsx
+++ b/app/ui/src/components/Common/BotForm.tsx
@@ -480,9 +480,11 @@ export const BotForm = ({
onFinish={createBot}
form={form}
className="space-y-6"
- initialValues={{
- embedding: "dialoqbase_eb_text-embedding-ada-002",
- model: "gpt-3.5-turbo-dbase",
+ initialValues={{
+ embedding:
+ botConfig?.defaultEmbeddingModel ||
+ "dialoqbase_eb_text-embedding-ada-002",
+ model: botConfig?.defaultChatModel || "gpt-3.5-turbo-dbase",
maxDepth: 2,
maxLinks: 10,
options: {
@@ -646,7 +648,7 @@ export const BotForm = ({
.toLowerCase()
.localeCompare((optionB?.label ?? "").toLowerCase())
}
- placeholder="Select a chat model"
+ placeholder="Select a Chat Model"
options={botConfig.chatModel}
/>
@@ -675,6 +677,7 @@ export const BotForm = ({
.localeCompare((optionB?.label ?? "").toLowerCase())
}
options={botConfig.embeddingModel}
+ placeholder="Select an Embedding Model"
/>
diff --git a/app/ui/src/hooks/useCreateConfig.tsx b/app/ui/src/hooks/useCreateConfig.tsx
index 1a090407..7f1493c9 100644
--- a/app/ui/src/hooks/useCreateConfig.tsx
+++ b/app/ui/src/hooks/useCreateConfig.tsx
@@ -16,6 +16,8 @@ export const useCreateConfig = () => {
label: string;
value: string;
}[];
+ defaultChatModel?: string;
+ defaultEmbeddingModel?: string;
};
}
);
diff --git a/app/ui/src/routes/settings/application.tsx b/app/ui/src/routes/settings/application.tsx
index c38c9c13..ff70e79f 100644
--- a/app/ui/src/routes/settings/application.tsx
+++ b/app/ui/src/routes/settings/application.tsx
@@ -1,16 +1,19 @@
-import { Form, InputNumber, Switch, notification } from "antd";
+import { Form, InputNumber, Switch, notification, Select } from "antd";
import React from "react";
import api from "../../services/api";
-import { useMutation, useQuery } from "@tanstack/react-query";
+import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
import { SettingsLayout } from "../../Layout/SettingsLayout";
import { SkeletonLoading } from "../../components/Common/SkeletonLoading";
import { useNavigate } from "react-router-dom";
import { ApplicationCard } from "../../components/Settings/Application/ApplicationCard";
+import { useCreateConfig } from "../../hooks/useCreateConfig";
export default function SettingsApplicationRoot() {
const [form] = Form.useForm();
const navigate = useNavigate();
+ const { data: models, status: modeStatus } = useCreateConfig();
+ const queryClient = useQueryClient();
const { data, status } = useQuery(["fetchApplicationSettings"], async () => {
const response = await api.get("/admin/dialoqbase-settings");
@@ -20,6 +23,8 @@ export default function SettingsApplicationRoot() {
allowUserToRegister: boolean;
defaultChunkSize: number;
defaultChunkOverlap: number;
+ defaultChatModel: string;
+ defaultEmbeddingModel: string;
};
});
@@ -43,6 +48,7 @@ export default function SettingsApplicationRoot() {
onUpdateApplicatoon,
{
onSuccess: (data) => {
+ queryClient.invalidateQueries(["fetchBotCreateConfig"]);
notification.success({
message: "Success",
description: data.message,
@@ -129,6 +135,66 @@ export default function SettingsApplicationRoot() {
>