fixed chunks

Author: deon-sanchez

src/backend/base/langflow/api/v1/knowledge_bases.py

@@ -47,6 +47,13 @@ class BulkDeleteRequest(BaseModel):
     kb_names: list[str]
 
 
+class ChunkInfo(BaseModel):
+    id: str
+    content: str
+    char_count: int
+    metadata: dict | None = None
+
+
 def get_kb_root_path() -> Path:
     """Get the knowledge bases root path."""
     return _get_knowledge_bases_dir()
@@ -385,6 +392,51 @@ async def get_knowledge_base(kb_name: str, current_user: CurrentActiveUser) -> K
         raise HTTPException(status_code=500, detail=f"Error getting knowledge base '{kb_name}': {e!s}") from e
 
 
+@router.get("/{kb_name}/chunks", status_code=HTTPStatus.OK)
+async def get_knowledge_base_chunks(kb_name: str, current_user: CurrentActiveUser) -> list[ChunkInfo]:
+    """Get all chunks from a specific knowledge base."""
+    try:
+        kb_root_path = get_kb_root_path()
+        kb_user = current_user.username
+        kb_path = kb_root_path / kb_user / kb_name
+
+        if not kb_path.exists() or not kb_path.is_dir():
+            raise HTTPException(status_code=404, detail=f"Knowledge base '{kb_name}' not found")
+
+        # Create vector store
+        chroma = Chroma(
+            persist_directory=str(kb_path),
+            collection_name=kb_name,
+        )
+
+        # Access the raw collection
+        collection = chroma._collection  # noqa: SLF001
+
+        # Fetch all documents and metadata
+        results = collection.get(include=["documents", "metadatas"])
+
+        chunks = []
+        for i, (doc_id, document, metadata) in enumerate(
+            zip(results["ids"], results["documents"], results["metadatas"], strict=False)
+        ):
+            content = document or ""
+            chunks.append(
+                ChunkInfo(
+                    id=doc_id,
+                    content=content,
+                    char_count=len(content),
+                    metadata=metadata,
+                )
+            )
+
+        return chunks
+
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error getting chunks for '{kb_name}': {e!s}") from e
+
+
 @router.delete("/{kb_name}", status_code=HTTPStatus.OK)
 async def delete_knowledge_base(kb_name: str, current_user: CurrentActiveUser) -> dict[str, str]:
     """Delete a specific knowledge base."""

src/frontend/src/controllers/API/queries/knowledge-bases/use-get-knowledge-base-chunks.ts

@@ -0,0 +1,42 @@
+import type { UseQueryResult } from "@tanstack/react-query";
+import type { useQueryFunctionType } from "@/types/api";
+import { api } from "../../api";
+import { getURL } from "../../helpers/constants";
+import { UseRequestProcessor } from "../../services/request-processor";
+
+export interface ChunkInfo {
+  id: string;
+  content: string;
+  char_count: number;
+  metadata: Record<string, unknown> | null;
+}
+
+interface GetKnowledgeBaseChunksParams {
+  kb_name: string;
+}
+
+export const useGetKnowledgeBaseChunks: useQueryFunctionType<
+  GetKnowledgeBaseChunksParams,
+  ChunkInfo[]
+> = (params, options?) => {
+  const { query } = UseRequestProcessor();
+
+  const getChunksFn = async (): Promise<ChunkInfo[]> => {
+    const res = await api.get(
+      `${getURL("KNOWLEDGE_BASES")}/${params?.kb_name}/chunks`,
+    );
+    return res.data;
+  };
+
+  const queryResult: UseQueryResult<ChunkInfo[], any> = query(
+    ["useGetKnowledgeBaseChunks", params?.kb_name],
+    getChunksFn,
+    {
+      enabled: !!params?.kb_name,
+      refetchOnWindowFocus: false,
+      ...options,
+    },
+  );
+
+  return queryResult;
+};

src/frontend/src/pages/MainPage/pages/filesPage/components/KnowledgeBasesTab.tsx

@@ -159,7 +159,10 @@ const KnowledgeBasesTab = ({
     }
   };
 
-  const columnDefs = createKnowledgeBaseColumns();
+  const columnDefs = createKnowledgeBaseColumns({
+    onViewChunks: onRowClick,
+    onDelete: handleDelete,
+  });
 
   if (isLoading || !knowledgeBases || !Array.isArray(knowledgeBases)) {
     return (

src/frontend/src/pages/MainPage/pages/filesPage/config/knowledgeBaseColumns.tsx

@@ -1,11 +1,27 @@
 import type { ColDef } from "ag-grid-community";
+import ForwardedIconComponent from "@/components/common/genericIconComponent";
+import { Button } from "@/components/ui/button";
+import {
+  DropdownMenu,
+  DropdownMenuContent,
+  DropdownMenuItem,
+  DropdownMenuTrigger,
+} from "@/components/ui/dropdown-menu";
+import type { KnowledgeBaseInfo } from "@/controllers/API/queries/knowledge-bases/use-get-knowledge-bases";
 import { formatFileSize } from "@/utils/stringManipulation";
 import {
   formatAverageChunkSize,
   formatNumber,
 } from "../utils/knowledgeBaseUtils";
 
-export const createKnowledgeBaseColumns = (): ColDef[] => {
+export interface KnowledgeBaseColumnsCallbacks {
+  onViewChunks?: (knowledgeBase: KnowledgeBaseInfo) => void;
+  onDelete?: (knowledgeBase: KnowledgeBaseInfo) => void;
+}
+
+export const createKnowledgeBaseColumns = (
+  callbacks?: KnowledgeBaseColumnsCallbacks,
+): ColDef[] => {
   const baseCellClass =
     "text-muted-foreground cursor-pointer select-text group-[.no-select-cells]:cursor-default group-[.no-select-cells]:select-none";
 
@@ -38,15 +54,6 @@ export const createKnowledgeBaseColumns = (): ColDef[] => {
       cellClass: baseCellClass,
       valueGetter: (params) => params.data.type || "—",
     },
-    {
-      headerName: "Owner",
-      field: "owner",
-      flex: 1,
-      sortable: false,
-      editable: false,
-      cellClass: baseCellClass,
-      valueGetter: (params) => params.data.owner || "—",
-    },
     {
       headerName: "Chunks",
       field: "chunks",
@@ -71,8 +78,57 @@ export const createKnowledgeBaseColumns = (): ColDef[] => {
       flex: 1,
       sortable: false,
       editable: false,
+      resizable: false,
       cellClass: baseCellClass,
       valueGetter: (params) => params.data.status || "—",
     },
+    {
+      headerName: "",
+      field: "actions",
+      width: 65,
+      minWidth: 65,
+      sortable: false,
+      editable: false,
+      resizable: false,
+      suppressMovable: true,
+      cellClass: "flex items-center justify-center",
+      cellRenderer: (params: { data: KnowledgeBaseInfo }) => (
+        <DropdownMenu>
+          <DropdownMenuTrigger asChild>
+            <Button
+              variant="ghost"
+              size="icon"
+              onClick={(e) => e.stopPropagation()}
+            >
+              <ForwardedIconComponent
+                name="EllipsisVertical"
+                className="h-4 w-4 text-muted-foreground"
+              />
+            </Button>
+          </DropdownMenuTrigger>
+          <DropdownMenuContent align="end">
+            <DropdownMenuItem
+              onClick={(e) => {
+                e.stopPropagation();
+                callbacks?.onViewChunks?.(params.data);
+              }}
+            >
+              <ForwardedIconComponent name="Layers" className="mr-2 h-4 w-4" />
+              View Chunks
+            </DropdownMenuItem>
+            <DropdownMenuItem
+              onClick={(e) => {
+                e.stopPropagation();
+                callbacks?.onDelete?.(params.data);
+              }}
+              className="text-destructive focus:text-destructive"
+            >
+              <ForwardedIconComponent name="Trash2" className="mr-2 h-4 w-4" />
+              Delete
+            </DropdownMenuItem>
+          </DropdownMenuContent>
+        </DropdownMenu>
+      ),
+    },
   ];
 };

src/frontend/src/pages/MainPage/pages/knowledgePage/index.tsx

@@ -2,6 +2,7 @@ import { useEffect, useRef, useState } from "react";
 import ForwardedIconComponent from "@/components/common/genericIconComponent";
 import { SidebarTrigger } from "@/components/ui/sidebar";
 import type { KnowledgeBaseInfo } from "@/controllers/API/queries/knowledge-bases/use-get-knowledge-bases";
+import { useCustomNavigate } from "@/customization/hooks/use-custom-navigate";
 import KnowledgeBaseDrawer from "../filesPage/components/KnowledgeBaseDrawer";
 import KnowledgeBasesTab from "../filesPage/components/KnowledgeBasesTab";
 
@@ -16,6 +17,7 @@ export const KnowledgePage = () => {
   const [selectedKnowledgeBase, setSelectedKnowledgeBase] =
     useState<KnowledgeBaseInfo | null>(null);
 
+  const navigate = useCustomNavigate();
   const drawerRef = useRef<HTMLDivElement>(null);
 
   useEffect(() => {
@@ -66,12 +68,7 @@ export const KnowledgePage = () => {
   }, [isDrawerOpen]);
 
   const handleKnowledgeBaseSelect = (knowledgeBase: KnowledgeBaseInfo) => {
-    if (isDrawerOpen) {
-      closeDrawer();
-    } else {
-      setSelectedKnowledgeBase(knowledgeBase);
-      // setIsDrawerOpen(true);
-    }
+    navigate(`/assets/knowledge-bases/${knowledgeBase.id}/chunks`);
   };
 
   const closeDrawer = () => {