+ }
+ onChange={(e) => {
+ const value = e.target.value;
+ // Allow only integer values
+ if (value === "" || /^[0-9]+$/.test(value)) {
+ setFieldValue("token_budget", value);
+ }
+ }}
+ />
+
+ Specify the length of the time period, in hours, over
+ which the token budget will be applied.
+
- You have selected: {selectedModel}. Are you sure you want to
- update to this new embedding model?
+ You have selected: {selectedModel.model_name}. Are you sure you
+ want to update to this new embedding model?
We will re-index all your documents in the background so you will be
@@ -25,6 +28,18 @@ export function ModelSelectionConfirmaion({
normal. If you are self-hosting, we recommend that you allocate at least
16GB of RAM to Danswer during this process.
+
+ {isCustom && (
+
+ We've detected that this is a custom-specified embedding model.
+ Since we have to download the model files before verifying the
+ configuration's correctness, we won't be able to let you
+ know if the configuration is valid until after we start
+ re-indexing your documents. If there is an issue, it will show up on
+ this page as an indexing error on this page after clicking Confirm.
+
+ )}
+
{tentativeNewEmbeddingModel && (
+ model.model_name === tentativeNewEmbeddingModel.model_name
+ ) === undefined
+ }
onConfirm={() => onConfirm(tentativeNewEmbeddingModel)}
onCancel={() => setTentativeNewEmbeddingModel(null)}
/>
@@ -243,12 +247,49 @@ function Main() {
)}
+
+ Below are a curated selection of quality models that we recommend
+ you choose from.
+
+
modelOption.model_name !== currentModelName
)}
setSelectedModel={onSelect}
/>
+
+
+ Alternatively, (if you know what you're doing) you can
+ specify a{" "}
+
+ SentenceTransformers
+
+ -compatible model of your choice below. The rough list of
+ supported models can be found{" "}
+
+ here
+
+ .
+
+ NOTE: not all models listed will work with Danswer, since
+ some have unique interfaces or special requirements. If in doubt,
+ reach out to the Danswer team.
+
+
+
) : (
connectors &&
@@ -272,10 +313,10 @@ function Main() {
The table below shows the re-indexing progress of all existing
- connectors. Once all connectors have been re-indexed, the new
- model will be used for all search queries. Until then, we will
- use the old model so that no downtime is necessary during this
- transition.
+ connectors. Once all connectors have been re-indexed
+ successfully, the new model will be used for all search
+ queries. Until then, we will use the old model so that no
+ downtime is necessary during this transition.
{isLoadingOngoingReIndexingStatus ? (
diff --git a/web/src/app/admin/personas/PersonaEditor.tsx b/web/src/app/admin/personas/PersonaEditor.tsx
index 375cd6520fa..6ce77edb561 100644
--- a/web/src/app/admin/personas/PersonaEditor.tsx
+++ b/web/src/app/admin/personas/PersonaEditor.tsx
@@ -1,6 +1,6 @@
"use client";
-import { DocumentSet } from "@/lib/types";
+import { DocumentSet, UserGroup } from "@/lib/types";
import { Button, Divider, Text } from "@tremor/react";
import {
ArrayHelpers,
@@ -25,6 +25,10 @@ import {
} from "@/components/admin/connectors/Field";
import { HidableSection } from "./HidableSection";
import { FiPlus, FiX } from "react-icons/fi";
+import { EE_ENABLED } from "@/lib/constants";
+import { useUserGroups } from "@/lib/hooks";
+import { Bubble } from "@/components/Bubble";
+import { GroupsIcon } from "@/components/icons/icons";
function Label({ children }: { children: string | JSX.Element }) {
return (
@@ -50,6 +54,9 @@ export function PersonaEditor({
const router = useRouter();
const { popup, setPopup } = usePopup();
+ // EE only
+ const { data: userGroups, isLoading: userGroupsIsLoading } = useUserGroups();
+
const [finalPrompt, setFinalPrompt] = useState("");
const [finalPromptError, setFinalPromptError] = useState("");
@@ -92,6 +99,7 @@ export function PersonaEditor({
system_prompt: existingPrompt?.system_prompt ?? "",
task_prompt: existingPrompt?.task_prompt ?? "",
disable_retrieval: (existingPersona?.num_chunks ?? 10) === 0,
+ is_public: existingPersona?.is_public ?? true,
document_set_ids:
existingPersona?.document_sets?.map(
(documentSet) => documentSet.id
@@ -103,6 +111,8 @@ export function PersonaEditor({
llm_model_version_override:
existingPersona?.llm_model_version_override ?? null,
starter_messages: existingPersona?.starter_messages ?? [],
+ // EE Only
+ groups: existingPersona?.groups ?? [],
}}
validationSchema={Yup.object()
.shape({
@@ -113,6 +123,7 @@ export function PersonaEditor({
system_prompt: Yup.string(),
task_prompt: Yup.string(),
disable_retrieval: Yup.boolean().required(),
+ is_public: Yup.boolean().required(),
document_set_ids: Yup.array().of(Yup.number()),
num_chunks: Yup.number().max(20).nullable(),
include_citations: Yup.boolean().required(),
@@ -125,6 +136,8 @@ export function PersonaEditor({
message: Yup.string().required(),
})
),
+ // EE Only
+ groups: Yup.array().of(Yup.number()),
})
.test(
"system-prompt-or-task-prompt",
@@ -163,6 +176,9 @@ export function PersonaEditor({
? 0
: values.num_chunks || 10;
+ // don't set groups if marked as public
+ const groups = values.is_public ? [] : values.groups;
+
let promptResponse;
let personaResponse;
if (isUpdate) {
@@ -171,11 +187,13 @@ export function PersonaEditor({
existingPromptId: existingPrompt?.id,
...values,
num_chunks: numChunks,
+ groups,
});
} else {
[promptResponse, personaResponse] = await createPersona({
...values,
num_chunks: numChunks,
+ groups,
});
}
@@ -375,6 +393,67 @@ export function PersonaEditor({
)}
+ {EE_ENABLED && userGroups && (
+ <>
+
+ <>
+
+
+ {userGroups &&
+ userGroups.length > 0 &&
+ !values.is_public && (
+
+
+ Select which User Groups should have access to
+ this Persona.
+
+
+
+
+
+
+ Click the button below to make the chat private again.
+
+
+
+
+ ) : (
+
+
+ Ensure that all content in the chat is safe to share with the
+ whole organization. The content of the retrieved documents will
+ not be visible, but the names of cited documents as well as the
+ AI and human messages will be visible.
+
+
+
+