hhzorgand朱豪豪
95b7b8b9a3
Milvus supports multiple databases, but the sidebar only ever showed collections under the hardcoded "default" database. The backend returned ["default"] for every vector store and the frontend loaded collections for a single database and flattened them under the connection node. - Backend (vector_driver): add list_databases() that calls Milvus /v2/vectordb/databases/list (falling back to "default" for older Milvus and for non-Milvus vector stores), and thread a database parameter through find_documents() instead of hardcoding "default". - document_ops: list_databases_core delegates VectorDb to list_databases(); find_documents_core forwards the selected database. - Frontend: add a "vector-database" tree node type. For Milvus, connections expand into database nodes, which expand into their own collections. Collection node ids are namespaced by database to avoid collisions across databases, and the collection id is stored in meta so activation no longer parses the node id. Fixes #2693 Co-authored-by: 朱豪豪 <zhu.hh.sz@belle.com.cn>
80 lines
3.5 KiB
TypeScript
80 lines
3.5 KiB
TypeScript
import type { DatabaseType, TreeNode } from "@/types/database";
|
|
|
|
const sidebarTreeCollator = new Intl.Collator(undefined, { numeric: true, sensitivity: "base" });
|
|
|
|
function sortByLabel(nodes: readonly TreeNode[]): TreeNode[] {
|
|
return [...nodes].sort((left, right) => sidebarTreeCollator.compare(left.label, right.label));
|
|
}
|
|
|
|
function sortRecursive(node: TreeNode, databaseType?: DatabaseType): TreeNode {
|
|
const children = node.children ? sortSidebarTreeChildrenForParent(node, node.children, databaseType) : node.children;
|
|
const hiddenChildren = node.hiddenChildren ? sortSidebarTreeChildrenForParent(node, node.hiddenChildren, databaseType) : node.hiddenChildren;
|
|
if (children === node.children && hiddenChildren === node.hiddenChildren) return node;
|
|
return {
|
|
...node,
|
|
children,
|
|
hiddenChildren,
|
|
};
|
|
}
|
|
|
|
export function sortSidebarTreeChildrenForParent(parent: Pick<TreeNode, "type">, children: readonly TreeNode[], databaseType?: DatabaseType): TreeNode[] {
|
|
const normalized = children.map((child) => sortRecursive(child, databaseType));
|
|
|
|
if (parent.type === "mongo-db") {
|
|
const gridFsNodes = normalized.filter((child) => child.type === "mongo-gridfs");
|
|
const collections = normalized.filter((child) => child.type !== "mongo-gridfs");
|
|
return [...gridFsNodes, ...sortByLabel(collections)];
|
|
}
|
|
|
|
if (parent.type === "vector-database") {
|
|
return sortByLabel(normalized);
|
|
}
|
|
|
|
if (parent.type === "mongo-buckets") {
|
|
return sortByLabel(normalized);
|
|
}
|
|
|
|
if (parent.type === "connection") {
|
|
const savedSqlNodes = normalized.filter((child) => child.type === "saved-sql-root");
|
|
const userAdminNodes = normalized.filter((child) => child.type === "user-admin");
|
|
const regularChildren = normalized.filter((child) => child.type !== "user-admin" && child.type !== "saved-sql-root");
|
|
const withConnectionUtilityOrder = (children: TreeNode[]) => [...savedSqlNodes, ...children, ...userAdminNodes];
|
|
|
|
if (databaseType === "mongodb" || databaseType === "elasticsearch" || databaseType === "qdrant" || databaseType === "milvus" || databaseType === "weaviate" || databaseType === "chromadb") {
|
|
return withConnectionUtilityOrder(sortByLabel(regularChildren));
|
|
}
|
|
|
|
if (databaseType === "duckdb") {
|
|
const schemas = sortByLabel(regularChildren.filter((child) => child.type === "schema"));
|
|
const databases = sortByLabel(regularChildren.filter((child) => child.type === "database"));
|
|
const rest = regularChildren.filter((child) => child.type !== "schema" && child.type !== "database");
|
|
return withConnectionUtilityOrder([...schemas, ...databases, ...rest]);
|
|
}
|
|
|
|
if (regularChildren.every((child) => child.type === "database")) {
|
|
return withConnectionUtilityOrder(sortByLabel(regularChildren));
|
|
}
|
|
|
|
if (regularChildren.every((child) => child.type === "schema")) {
|
|
return withConnectionUtilityOrder(sortByLabel(regularChildren));
|
|
}
|
|
|
|
return withConnectionUtilityOrder(regularChildren);
|
|
}
|
|
|
|
if (parent.type === "database") {
|
|
if (databaseType === "sqlserver") {
|
|
const objectGroups = normalized.filter((child) => child.type.startsWith("group-"));
|
|
const schemas = sortByLabel(normalized.filter((child) => child.type === "schema"));
|
|
const rest = normalized.filter((child) => !child.type.startsWith("group-") && child.type !== "schema");
|
|
return [...objectGroups, ...schemas, ...rest];
|
|
}
|
|
|
|
if (normalized.every((child) => child.type === "schema")) {
|
|
return sortByLabel(normalized);
|
|
}
|
|
}
|
|
|
|
return normalized;
|
|
}
|