jketreno/backstory
1
0
Fork 0
Code Issues 12 Actions Releases Activity

Fix 2D / 3D error message on update

Browse Source
This commit is contained in:
James Ketr 2025-04-24 12:04:42 -07:00
parent ca9dd950b3
commit 7059d5ef24
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
frontend/src/VectorVisualizer.tsx
View File

@ -10,7 +10,7 @@ import SendIcon from '@mui/icons-material/Send';
import FormControlLabel from '@mui/material/FormControlLabel';
import Switch from '@mui/material/Switch';
import { SeverityType } from './Snack';
import { SetSnackType } from './Snack';
interface Metadata {
doc_type?: string;
@ -18,7 +18,7 @@ interface Metadata {
}
interface ResultData {
embeddings: number[][] | number[][][];
embeddings: (number[])[];
documents: string[];
metadatas: Metadata[];
ids: string[];
@ -42,7 +42,7 @@ interface PlotData {
interface VectorVisualizerProps {
connectionBase: string;
sessionId?: string;
setSnack: (message: string, severity: SeverityType) => void;
setSnack: SetSnackType;
inline?: boolean;
rag?: any;
}
@ -141,11 +141,19 @@ const VectorVisualizer: React.FC<VectorVisualizerProps> = ({ setSnack, rag, inli
if (!result || !result.embeddings) return;
if (result.embeddings.length === 0) return;
const vectors: number[][] = [...result.embeddings as number[][]];
const vectors: (number[])[] = [...result.embeddings];
const documents = [...result.documents || []];
const metadatas = [...result.metadatas || []];
const ids = [...result.ids || []];
let is2D = vectors.every((v: number[]) => v.length === 2);
let is3D = vectors.every((v: number[]) => v.length === 3);
console.log(`Embeddings are ${is2D ? '2D' : is3D ? '3D' : 'invaalid'} and view2D is ${view2D}`);
if ((view2D && !is2D) || (!view2D && !is3D)) {
return;
}
if (view2D && rag && rag.umap_embedding_2d) {
metadatas.unshift({ doc_type: 'query' });
documents.unshift('Query');
@ -169,10 +177,11 @@ const VectorVisualizer: React.FC<VectorVisualizerProps> = ({ setSnack, rag, inli
}
}
const is2D = vectors.every((v: number[]) => v.length === 2);
const is3D = vectors.every((v: number[]) => v.length === 3);
is2D = vectors.every((v: number[]) => v.length === 2);
is3D = vectors.every((v: number[]) => v.length === 3);
if (!is2D && !is3D) {
console.error('Vectors are neither 2D nor 3D');
console.warn('Modified vectors are neither 2D nor 3D');
return;
}