Merge pull request #1606 from langbot-app/feat/topk_splitter

Feat/topk splitter

web/src/app/home/knowledge/components/kb-card/KBCardVO.ts

@@ -3,6 +3,7 @@ export interface IKnowledgeBaseVO {
   name: string;
   description: string;
   embeddingModelUUID: string;
+  top_k: number;
   lastUpdatedTimeAgo: string;
 }
 
@@ -11,6 +12,7 @@ export class KnowledgeBaseVO implements IKnowledgeBaseVO {
   name: string;
   description: string;
   embeddingModelUUID: string;
+  top_k: number;
   lastUpdatedTimeAgo: string;
 
   constructor(props: IKnowledgeBaseVO) {
@@ -18,6 +20,7 @@ export class KnowledgeBaseVO implements IKnowledgeBaseVO {
     this.name = props.name;
     this.description = props.description;
     this.embeddingModelUUID = props.embeddingModelUUID;
+    this.top_k = props.top_k;
     this.lastUpdatedTimeAgo = props.lastUpdatedTimeAgo;
   }
 }

web/src/app/home/knowledge/components/kb-form/KBForm.tsx

@@ -35,6 +35,10 @@ const getFormSchema = (t: (key: string) => string) =>
     embeddingModelUUID: z
       .string()
       .min(1, { message: t('knowledge.embeddingModelUUIDRequired') }),
+    top_k: z
+      .number()
+      .min(1, { message: t('knowledge.topKRequired') })
+      .max(30, { message: t('knowledge.topKMax') }),
   });
 
 export default function KBForm({
@@ -55,6 +59,7 @@ export default function KBForm({
       name: '',
       description: t('knowledge.defaultDescription'),
       embeddingModelUUID: '',
+      top_k: 5,
     },
   });
 
@@ -69,6 +74,7 @@ export default function KBForm({
           form.setValue('name', val.name);
           form.setValue('description', val.description);
           form.setValue('embeddingModelUUID', val.embeddingModelUUID);
+          form.setValue('top_k', val.top_k || 5);
         });
       }
     });
@@ -83,6 +89,7 @@ export default function KBForm({
           name: res.base.name,
           description: res.base.description,
           embeddingModelUUID: res.base.embedding_model_uuid,
+          top_k: res.base.top_k || 5,
         });
       });
     });
@@ -109,6 +116,7 @@ export default function KBForm({
         name: data.name,
         description: data.description,
         embedding_model_uuid: data.embeddingModelUUID,
+        top_k: data.top_k,
       };
       httpClient
         .updateKnowledgeBase(initKbId, updateKb)
@@ -127,6 +135,7 @@ export default function KBForm({
         name: data.name,
         description: data.description,
         embedding_model_uuid: data.embeddingModelUUID,
+        top_k: data.top_k,
       };
       httpClient
         .createKnowledgeBase(newKb)
@@ -226,6 +235,30 @@ export default function KBForm({
                 </FormItem>
               )}
             />
+            <FormField
+              control={form.control}
+              name="top_k"
+              render={({ field }) => (
+                <FormItem>
+                  <FormLabel>
+                    {t('knowledge.topK')}
+                    <span className="text-red-500">*</span>
+                  </FormLabel>
+                  <FormControl>
+                    <Input
+                      type="number"
+                      {...field}
+                      onChange={(e) => field.onChange(Number(e.target.value))}
+                      className="w-[180px] h-10 text-base appearance-none"
+                    />
+                  </FormControl>
+                  <FormDescription>
+                    {t('knowledge.topKdescription')}
+                  </FormDescription>
+                  <FormMessage />
+                </FormItem>
+              )}
+            />
           </div>
         </form>
       </Form>

web/src/app/home/knowledge/page.tsx

@@ -46,6 +46,7 @@ export default function KnowledgePage() {
           name: kb.name,
           description: kb.description,
           embeddingModelUUID: kb.embedding_model_uuid,
+          top_k: kb.top_k ?? 5,
           lastUpdatedTimeAgo: lastUpdatedTimeAgoText,
         });
       }),

web/src/app/infra/entities/api/index.ts

@@ -61,7 +61,7 @@ export interface KnowledgeBase {
   description: string;
   embedding_model_uuid: string;
   created_at?: string;
-  top_k?: number;
+  top_k: number;
 }
 
 export interface ApiRespProviderEmbeddingModels {
@@ -155,6 +155,7 @@ export interface KnowledgeBase {
   name: string;
   description: string;
   embedding_model_uuid: string;
+  top_k: number;
   created_at?: string;
   updated_at?: string;
 }