-
-
-
- setQueuedChatSeed(null)}
- emptyStateSuggestions={highlightsData?.questions}
- />
-
+
-type DocumentWithMemories = DocumentsResponse["documents"][0]
-
+/**
+ * Graph Dialog component
+ */
export function GraphDialog() {
const { user } = useAuth()
const { documentIds: allHighlightDocumentIds } = useGraphHighlights()
const { selectedProject } = useProject()
- const { isOpen } = useChatOpen()
+ const { isOpen: isChatOpen } = useChatOpen()
const { isOpen: showGraphModal, setIsOpen: setShowGraphModal } =
useGraphModal()
- const [injectedDocs, setInjectedDocs] = useState([])
const [showAddMemoryView, setShowAddMemoryView] = useState(false)
const [showConnectAIModal, setShowConnectAIModal] = useState(false)
const isMobile = useIsMobile()
- const IS_DEV = process.env.NODE_ENV === "development"
- const PAGE_SIZE = IS_DEV ? 100 : 100
- const MAX_TOTAL = 1000
-
- const {
- data,
- error,
- isPending,
- isFetchingNextPage,
- hasNextPage,
- fetchNextPage,
- } = useInfiniteQuery({
- queryKey: ["documents-with-memories", selectedProject],
- initialPageParam: 1,
- queryFn: async ({ pageParam }) => {
- const response = await $fetch("@post/documents/documents", {
- body: {
- page: pageParam as number,
- limit: (pageParam as number) === 1 ? (IS_DEV ? 500 : 500) : PAGE_SIZE,
- sort: "createdAt",
- order: "desc",
- containerTags: selectedProject ? [selectedProject] : undefined,
- },
- disableValidation: true,
- })
-
- if (response.error) {
- throw new Error(response.error?.message || "Failed to fetch documents")
- }
-
- return response.data
- },
- getNextPageParam: (lastPage, allPages) => {
- if (!lastPage || !lastPage.pagination) return undefined
- if (!Array.isArray(allPages)) return undefined
-
- const loaded = allPages.reduce(
- (acc, p) => acc + (p.documents?.length ?? 0),
- 0,
- )
- if (loaded >= MAX_TOTAL) return undefined
-
- const { currentPage, totalPages } = lastPage.pagination
- if (currentPage < totalPages) {
- return currentPage + 1
- }
- return undefined
- },
- staleTime: 5 * 60 * 1000,
- enabled: !!user, // Only run query if user is authenticated
- })
-
- const baseDocuments = useMemo(() => {
- return (
- data?.pages.flatMap((p: DocumentsResponse) => p.documents ?? []) ?? []
- )
- }, [data])
-
- const allDocuments = useMemo(() => {
- if (injectedDocs.length === 0) return baseDocuments
- const byId = new Map()
- for (const d of injectedDocs) byId.set(d.id, d)
- for (const d of baseDocuments) if (!byId.has(d.id)) byId.set(d.id, d)
- return Array.from(byId.values())
- }, [baseDocuments, injectedDocs])
-
- const totalLoaded = allDocuments.length
- const hasMore = hasNextPage
- const isLoadingMore = isFetchingNextPage
-
- const loadMoreDocuments = useCallback(async (): Promise => {
- if (hasNextPage && !isFetchingNextPage) {
- await fetchNextPage()
- return
- }
- return
- }, [hasNextPage, isFetchingNextPage, fetchNextPage])
-
- // Handle highlighted documents injection for chat
- useEffect(() => {
- if (!isOpen) return
- if (!allHighlightDocumentIds || allHighlightDocumentIds.length === 0) return
- const present = new Set()
- for (const d of [...baseDocuments, ...injectedDocs]) {
- if (d.id) present.add(d.id)
- if (d.customId) present.add(d.customId as string)
- }
- const missing = allHighlightDocumentIds.filter(
- (id: string) => !present.has(id),
- )
- if (missing.length === 0) return
- let cancelled = false
- const run = async () => {
- try {
- const resp = await $fetch("@post/documents/documents/by-ids", {
- body: {
- ids: missing,
- by: "customId",
- containerTags: selectedProject ? [selectedProject] : undefined,
- },
- disableValidation: true,
- })
- if (cancelled || resp?.error) return
- const extraDocs = resp?.data?.documents as
- | DocumentWithMemories[]
- | undefined
- if (!extraDocs || extraDocs.length === 0) return
- setInjectedDocs((prev) => {
- const seen = new Set([
- ...prev.map((d) => d.id),
- ...baseDocuments.map((d) => d.id),
- ])
- const merged = [...prev]
- for (const doc of extraDocs) {
- if (!seen.has(doc.id)) {
- merged.push(doc)
- seen.add(doc.id)
- }
- }
- return merged
- })
- } catch {}
- }
- void run()
- return () => {
- cancelled = true
- }
- }, [
- isOpen,
- allHighlightDocumentIds,
- baseDocuments,
- injectedDocs,
- selectedProject,
- ])
-
if (!user) return null
+ // Convert selectedProject to containerTags array
+ const containerTags = selectedProject ? [selectedProject] : undefined
+
return (
<>
+ {viewMode === "graph" && !isMobile ? (
+
diff --git a/apps/web/components/graph-dialog.tsx b/apps/web/components/graph-dialog.tsx
index ef291d69..3f75cdee 100644
--- a/apps/web/components/graph-dialog.tsx
+++ b/apps/web/components/graph-dialog.tsx
@@ -1,12 +1,8 @@
"use client"
import { useAuth } from "@lib/auth-context"
-import { $fetch } from "@repo/lib/api"
-import type { DocumentsWithMemoriesResponseSchema } from "@repo/validation/api"
-import { useInfiniteQuery } from "@tanstack/react-query"
-import { useCallback, useEffect, useMemo, useState } from "react"
-import type { z } from "zod"
-import { MemoryGraph } from "@repo/ui/memory-graph"
+import { useState } from "react"
+import { MemoryGraph } from "@/components/new/memory-graph/memory-graph"
import { Dialog, DialogContent, DialogTitle } from "@repo/ui/components/dialog"
import { ConnectAIModal } from "@/components/connect-ai-modal"
import { AddMemoryView } from "@/components/views/add-memory"
@@ -14,179 +10,39 @@ import { useChatOpen, useProject, useGraphModal } from "@/stores"
import { useGraphHighlights } from "@/stores/highlights"
import { useIsMobile } from "@hooks/use-mobile"
-type DocumentsResponse = z.infer
+
+ ) : (
+
+
+ )}
+
+
+ setQueuedChatSeed(null)}
+ emptyStateSuggestions={highlightsData?.questions}
+ />
+
+