e7644f4b3a
<!-- .github/pull_request_template.md --> ## Description <!-- Provide a clear description of the changes in this PR --> ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin. --------- Co-authored-by: Igor Ilic <igorilic03@gmail.com>
58 lines
1.5 KiB
TypeScript
58 lines
1.5 KiB
TypeScript
import { fetch } from "@/utils";
|
|
import getDatasetGraph from "./getDatasetGraph";
|
|
import { Dataset } from "../ingestion/useDatasets";
|
|
|
|
interface GraphData {
|
|
nodes: { id: string; label: string; properties?: object }[];
|
|
edges: { source: string; target: string; label: string }[];
|
|
}
|
|
|
|
export default async function cognifyDataset(dataset: Dataset, onUpdate: (data: GraphData) => void) {
|
|
// const data = await (
|
|
return fetch("/v1/cognify", {
|
|
method: "POST",
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
},
|
|
body: JSON.stringify({
|
|
datasetIds: [dataset.id],
|
|
runInBackground: false,
|
|
}),
|
|
})
|
|
.then((response) => response.json())
|
|
.then(() => {
|
|
return getDatasetGraph(dataset)
|
|
.then((data) => {
|
|
onUpdate({
|
|
nodes: data.nodes,
|
|
edges: data.edges,
|
|
});
|
|
});
|
|
});
|
|
// )
|
|
|
|
// const websocket = new WebSocket(`ws://localhost:8000/api/v1/cognify/subscribe/${data.pipeline_run_id}`);
|
|
|
|
// let isCognifyDone = false;
|
|
|
|
// websocket.onmessage = (event) => {
|
|
// const data = JSON.parse(event.data);
|
|
// onUpdate?.({
|
|
// nodes: data.payload.nodes,
|
|
// edges: data.payload.edges,
|
|
// });
|
|
|
|
// if (data.status === "PipelineRunCompleted") {
|
|
// isCognifyDone = true;
|
|
// websocket.close();
|
|
// }
|
|
// };
|
|
|
|
// return new Promise(async (resolve) => {
|
|
// while (!isCognifyDone) {
|
|
// await new Promise(resolve => setTimeout(resolve, 1000));
|
|
// }
|
|
|
|
// resolve(true);
|
|
// });
|
|
}
|