frontend: support chat engine's is_public field

Author: 634750802

frontend/app/src/api/chat-engines.ts

@@ -13,6 +13,7 @@ export interface ChatEngine {
   fast_llm_id: number | null;
   reranker_id: number | null;
   is_default: boolean;
+  is_public: boolean;
 }
 
 export interface CreateChatEngineParams {
@@ -134,6 +135,7 @@ const chatEngineSchema = z.object({
   fast_llm_id: z.number().nullable(),
   reranker_id: z.number().nullable(),
   is_default: z.boolean(),
+  is_public: z.boolean(),
 }) satisfies ZodType<ChatEngine, any, any>;
 
 export async function getDefaultChatEngineOptions (): Promise<ChatEngineOptions> {

frontend/app/src/components/chat-engine/create-chat-engine-form.tsx

@@ -23,6 +23,7 @@ import { z } from 'zod';
 
 const schema = z.object({
   name: z.string().min(1),
+  is_public: z.boolean().optional(),
   llm_id: z.number().optional(),
   fast_llm_id: z.number().optional(),
   reranker_id: z.number().optional(),
@@ -42,6 +43,7 @@ const schema = z.object({
 const field = formFieldLayout<typeof schema>();
 
 const nameSchema = z.string().min(1);
+const isPublicSchema = z.boolean();
 const kbSchema = z.object({ id: z.number() }).array().min(1);
 const kgGraphDepthSchema = z.number().min(1).optional();
 
@@ -53,6 +55,9 @@ export function CreateChatEngineForm ({ defaultChatEngineOptions }: { defaultCha
 
   const form = useForm({
     onSubmit: onSubmitHelper(schema, async data => {
+      if (data.is_public == null) {
+        data.is_public = true;
+      }
       const ce = await createChatEngine(data);
       startTransition(() => {
         router.push(`/chat-engines/${ce.id}`);
@@ -85,6 +90,9 @@ export function CreateChatEngineForm ({ defaultChatEngineOptions }: { defaultCha
               <field.Basic required name="name" label="Name" defaultValue="" validators={{ onSubmit: nameSchema, onBlur: nameSchema }}>
                 <FormInput placeholder="Enter chat engine name" />
               </field.Basic>
+              <field.Contained name='is_public' label="Is Public" defaultValue={true}>
+                <FormSwitch />
+              </field.Contained>
               <SubSection title="Models">
                 <field.Basic name="llm_id" label="LLM">
                   <LLMSelect />
@@ -279,4 +287,4 @@ const llmPromptDescriptions: { [P in typeof llmPromptFields[number]]: string } =
   'clarifying_question_prompt': 'Prompt template for generating clarifying questions when the user\'s input needs more context or specificity',
   'generate_goal_prompt': 'Prompt template for generating conversation goals and objectives based on user input',
   'further_questions_prompt': 'Prompt template for generating follow-up questions to continue the conversation',
-}; 
+};

frontend/app/src/components/chat-engine/update-chat-engine-form.tsx

@@ -74,6 +74,11 @@ export function UpdateChatEngineForm ({ chatEngine, defaultChatEngineOptions }:
               <FormSwitch />
             </field.Contained>
           </GeneralSettingsField>
+          <GeneralSettingsField accessor={isPublicAccessor} schema={isPublicSchema}>
+            <field.Contained unimportant name="value" label="Is Public" fallbackValue={chatEngine.is_public}>
+              <FormSwitch />
+            </field.Contained>
+          </GeneralSettingsField>
           <SubSection title="Models">
             <GeneralSettingsField accessor={llmIdAccessor} schema={idSchema}>
               <field.Basic name="value" label="LLM">
@@ -311,6 +316,9 @@ const clarifyAccessorSchema = z.boolean().nullable().optional();
 const isDefaultAccessor = fieldAccessor<ChatEngine, 'is_default'>('is_default');
 const isDefaultSchema = z.boolean();
 
+const isPublicAccessor = fieldAccessor<ChatEngine, 'is_public'>('is_public');
+const isPublicSchema = z.boolean();
+
 const getIdAccessor = (id: KeyOfType<ChatEngine, number | null>) => fieldAccessor<ChatEngine, KeyOfType<ChatEngine, number | null>>(id);
 const idSchema = z.number().nullable();
 const llmIdAccessor = getIdAccessor('llm_id');

frontend/app/src/components/chat/message-input.tsx

@@ -41,8 +41,8 @@ export function MessageInput ({
     onChangeRef.current?.(ev);
   }, []);
 
-  const showShowSelectChatEngine = !!auth.me?.is_superuser && !!onEngineChange;
   const { data, isLoading } = useAllChatEngines();
+  const showShowSelectChatEngine = !!data?.length && !!onEngineChange;
 
   return (
     <div className={cn('bg-background border p-2 rounded-lg', className)}>