[Bug]: Failed to Visualize Graph in the Frontend UI Due to Component Missing

[Ao-Last]

Bug Description

The UI failed to show the graph visualization with frontend component missing error.

Steps to Reproduce

1. Add knowledge to a dataset with cognee.add("Cognee is a knowledge database ingestion engine", dataset_name="my_knowledge_dataset")
2. Start the UI by congnee-cli -ui, there is a dependency issue need to be solve as mentioned in COG-4329
3. Open the UI in http://localhost:3000
4. Click the three dots icon showing next to the dataset name and then click the visualize option available for the dataset.
5. Get the error.

Expected Behavior

It shows the graph visualization.

Actual Behavior

Get a Next.js error.

Environment

• OS: macos 26.3.1
• Python version: 3.12
• Cognee version: 0.5.5

Logs/Error Messages

## Error Type
Build Error

## Error Message
Module not found: Can't resolve '@/lib/adaptCogneeGraphData'

## Build Output
./src/app/visualize/[datasetId]/page.tsx:5:1
Module not found: Can't resolve '@/lib/adaptCogneeGraphData'
  3 | import { useEffect, useState } from "react";
  4 | import { fetch } from "@/utils";
> 5 | import { adaptCogneeGraphData, validateCogneeGraphResponse } from "@/lib/adaptCogneeGraphData";
    | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  6 | import { CogneeGraphResponse } from "@/types/CogneeAPI";
  7 | import MemoryGraphVisualization from "@/ui/elements/MemoryGraphVisualization";
  8 | import { Edge, Node } from "@/ui/rendering/graph/types";

Import map: aliased to relative './src/lib/adaptCogneeGraphData' inside of [project]/

Import traces:
  Client Component Browser:
    ./src/app/visualize/[datasetId]/page.tsx [Client Component Browser]
    ./src/app/visualize/[datasetId]/page.tsx [Server Component]

  Client Component SSR:
    ./src/app/visualize/[datasetId]/page.tsx [Client Component SSR]
    ./src/app/visualize/[datasetId]/page.tsx [Server Component]

https://nextjs.org/docs/messages/module-not-found

Next.js version: 16.1.6 (Turbopack)

Additional Context

No response

Pre-submission Checklist

• I have searched existing issues to ensure this bug hasn't been reported already
• I have provided a clear and detailed description of the bug
• I have included steps to reproduce the issue
• I have included my environment details

[github-actions]

Hello @Ao-Last, thank you for your interest in our work!

If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

[coderabbitai]

🔗 Related PRs

#101 - add_cl_graphs() fixed - config change - graph model -> KnowledgeGraph [merged]
#493 - feat: implements graph visualization method for cognee [merged]
#665 - docs: add graph visualization [merged]

---

📝 Issue Planner

_{Check the box below or use the @coderabbitai plan command to generate an implementation plan and prompts that you can use with your favorite coding assistant.}

• Create Plan

---

🧪 Issue enrichment is currently in open beta.

You can configure auto-planning by selecting labels in the issue_enrichment configuration.

To disable automatic issue enrichment, add the following to your .coderabbit.yaml:

issue_enrichment:
  auto_enrich:
    enabled: false

💬 Have feedback or questions? Drop into our discord!

[dexters1]

We are working on refactoring the whole UI at the moment, it will take a bit before we are ready to release it.

[morvenhuang]

any update?