diff --git a/backend/models/workspaceDocument.js b/backend/models/workspaceDocument.js index 08118c3..39a6813 100644 --- a/backend/models/workspaceDocument.js +++ b/backend/models/workspaceDocument.js @@ -170,12 +170,13 @@ const WorkspaceDocument = { include: { organization: true }, }); + const connector = await OrganizationConnection.get({ organization_id: workspace.organization.id }); const vectorDb = selectConnector(connector); - const dimensions = await vectorDb.indexDimensions(); + const dimensions = await vectorDb.indexDimensions(workspace.fname); return dimensions; } catch (e) { diff --git a/backend/utils/vectordatabases/providers/chroma/index.js b/backend/utils/vectordatabases/providers/chroma/index.js index 67bd455..76f8072 100644 --- a/backend/utils/vectordatabases/providers/chroma/index.js +++ b/backend/utils/vectordatabases/providers/chroma/index.js @@ -72,6 +72,11 @@ class Chroma { return { result: totalVectors, error: null }; } + async indexDimensions() { + // Chroma does not support this, defaulting to openai's 1536 + return 1536; + } + // Collections === namespaces for Chroma to normalize interfaces async collections() { return await this.namespaces(); diff --git a/frontend/src/pages/WorkspaceDashboard/Statistics/index.tsx b/frontend/src/pages/WorkspaceDashboard/Statistics/index.tsx index 4553ce9..ced36ba 100644 --- a/frontend/src/pages/WorkspaceDashboard/Statistics/index.tsx +++ b/frontend/src/pages/WorkspaceDashboard/Statistics/index.tsx @@ -27,7 +27,7 @@ const Statistics = ({ }); const [dimensions, setDimensions] = useState({ status: 'loading', - value: 0, + value: '-', }); const [clipboardMessage, setClipboardMessage] = useState('');