mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-15 18:25:18 +02:00
feat: old chat to new-chat with persistance
This commit is contained in:
parent
0c3574d049
commit
b5e20e7515
17 changed files with 490 additions and 385 deletions
|
|
@ -51,6 +51,7 @@ router = APIRouter()
|
||||||
@router.get("/threads", response_model=ThreadListResponse)
|
@router.get("/threads", response_model=ThreadListResponse)
|
||||||
async def list_threads(
|
async def list_threads(
|
||||||
search_space_id: int,
|
search_space_id: int,
|
||||||
|
limit: int | None = None,
|
||||||
session: AsyncSession = Depends(get_async_session),
|
session: AsyncSession = Depends(get_async_session),
|
||||||
user: User = Depends(current_active_user),
|
user: User = Depends(current_active_user),
|
||||||
):
|
):
|
||||||
|
|
@ -58,6 +59,10 @@ async def list_threads(
|
||||||
List all threads for the current user in a search space.
|
List all threads for the current user in a search space.
|
||||||
Returns threads and archived_threads for ThreadListPrimitive.
|
Returns threads and archived_threads for ThreadListPrimitive.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
search_space_id: The search space to list threads for
|
||||||
|
limit: Optional limit on number of threads to return (applies to active threads only)
|
||||||
|
|
||||||
Requires CHATS_READ permission.
|
Requires CHATS_READ permission.
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
|
|
@ -91,14 +96,18 @@ async def list_threads(
|
||||||
id=thread.id,
|
id=thread.id,
|
||||||
title=thread.title,
|
title=thread.title,
|
||||||
archived=thread.archived,
|
archived=thread.archived,
|
||||||
createdAt=thread.created_at,
|
created_at=thread.created_at,
|
||||||
updatedAt=thread.updated_at,
|
updated_at=thread.updated_at,
|
||||||
)
|
)
|
||||||
if thread.archived:
|
if thread.archived:
|
||||||
archived_threads.append(item)
|
archived_threads.append(item)
|
||||||
else:
|
else:
|
||||||
threads.append(item)
|
threads.append(item)
|
||||||
|
|
||||||
|
# Apply limit to active threads if specified
|
||||||
|
if limit is not None and limit > 0:
|
||||||
|
threads = threads[:limit]
|
||||||
|
|
||||||
return ThreadListResponse(threads=threads, archived_threads=archived_threads)
|
return ThreadListResponse(threads=threads, archived_threads=archived_threads)
|
||||||
|
|
||||||
except HTTPException:
|
except HTTPException:
|
||||||
|
|
@ -114,6 +123,69 @@ async def list_threads(
|
||||||
) from None
|
) from None
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/threads/search", response_model=list[ThreadListItem])
|
||||||
|
async def search_threads(
|
||||||
|
search_space_id: int,
|
||||||
|
title: str,
|
||||||
|
session: AsyncSession = Depends(get_async_session),
|
||||||
|
user: User = Depends(current_active_user),
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
Search threads by title in a search space.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
search_space_id: The search space to search in
|
||||||
|
title: The search query (case-insensitive partial match)
|
||||||
|
|
||||||
|
Requires CHATS_READ permission.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
await check_permission(
|
||||||
|
session,
|
||||||
|
user,
|
||||||
|
search_space_id,
|
||||||
|
Permission.CHATS_READ.value,
|
||||||
|
"You don't have permission to read chats in this search space",
|
||||||
|
)
|
||||||
|
|
||||||
|
# Search threads by title (case-insensitive)
|
||||||
|
query = (
|
||||||
|
select(NewChatThread)
|
||||||
|
.filter(
|
||||||
|
NewChatThread.search_space_id == search_space_id,
|
||||||
|
NewChatThread.user_id == user.id,
|
||||||
|
NewChatThread.title.ilike(f"%{title}%"),
|
||||||
|
)
|
||||||
|
.order_by(NewChatThread.updated_at.desc())
|
||||||
|
)
|
||||||
|
|
||||||
|
result = await session.execute(query)
|
||||||
|
threads = result.scalars().all()
|
||||||
|
|
||||||
|
return [
|
||||||
|
ThreadListItem(
|
||||||
|
id=thread.id,
|
||||||
|
title=thread.title,
|
||||||
|
archived=thread.archived,
|
||||||
|
created_at=thread.created_at,
|
||||||
|
updated_at=thread.updated_at,
|
||||||
|
)
|
||||||
|
for thread in threads
|
||||||
|
]
|
||||||
|
|
||||||
|
except HTTPException:
|
||||||
|
raise
|
||||||
|
except OperationalError:
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=503, detail="Database operation failed. Please try again later."
|
||||||
|
) from None
|
||||||
|
except Exception as e:
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=500,
|
||||||
|
detail=f"An unexpected error occurred while searching threads: {e!s}",
|
||||||
|
) from None
|
||||||
|
|
||||||
|
|
||||||
@router.post("/threads", response_model=NewChatThreadRead)
|
@router.post("/threads", response_model=NewChatThreadRead)
|
||||||
async def create_thread(
|
async def create_thread(
|
||||||
thread: NewChatThreadCreate,
|
thread: NewChatThreadCreate,
|
||||||
|
|
|
||||||
|
|
@ -29,7 +29,7 @@ export default function DashboardLayout({
|
||||||
const customNavMain = [
|
const customNavMain = [
|
||||||
{
|
{
|
||||||
title: "Chat",
|
title: "Chat",
|
||||||
url: `/dashboard/${search_space_id}/researcher`,
|
url: `/dashboard/${search_space_id}/new-chat`,
|
||||||
icon: "SquareTerminal",
|
icon: "SquareTerminal",
|
||||||
items: [],
|
items: [],
|
||||||
},
|
},
|
||||||
|
|
|
||||||
|
|
@ -2,26 +2,26 @@
|
||||||
|
|
||||||
import {
|
import {
|
||||||
AssistantRuntimeProvider,
|
AssistantRuntimeProvider,
|
||||||
useExternalStoreRuntime,
|
|
||||||
type ThreadMessageLike,
|
type ThreadMessageLike,
|
||||||
|
useExternalStoreRuntime,
|
||||||
} from "@assistant-ui/react";
|
} from "@assistant-ui/react";
|
||||||
import { useParams, useRouter } from "next/navigation";
|
import { useParams, useRouter } from "next/navigation";
|
||||||
import { useCallback, useEffect, useMemo, useRef, useState } from "react";
|
import { useCallback, useEffect, useMemo, useRef, useState } from "react";
|
||||||
|
import { toast } from "sonner";
|
||||||
import { Thread } from "@/components/assistant-ui/thread";
|
import { Thread } from "@/components/assistant-ui/thread";
|
||||||
import { GeneratePodcastToolUI } from "@/components/tool-ui/generate-podcast";
|
import { GeneratePodcastToolUI } from "@/components/tool-ui/generate-podcast";
|
||||||
import {
|
|
||||||
createThread,
|
|
||||||
getThreadMessages,
|
|
||||||
appendMessage,
|
|
||||||
type MessageRecord,
|
|
||||||
} from "@/lib/chat/thread-persistence";
|
|
||||||
import { getBearerToken } from "@/lib/auth-utils";
|
import { getBearerToken } from "@/lib/auth-utils";
|
||||||
import { toast } from "sonner";
|
|
||||||
import {
|
import {
|
||||||
isPodcastGenerating,
|
isPodcastGenerating,
|
||||||
looksLikePodcastRequest,
|
looksLikePodcastRequest,
|
||||||
setActivePodcastTaskId,
|
setActivePodcastTaskId,
|
||||||
} from "@/lib/chat/podcast-state";
|
} from "@/lib/chat/podcast-state";
|
||||||
|
import {
|
||||||
|
appendMessage,
|
||||||
|
createThread,
|
||||||
|
getThreadMessages,
|
||||||
|
type MessageRecord,
|
||||||
|
} from "@/lib/chat/thread-persistence";
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Convert backend message to assistant-ui ThreadMessageLike format
|
* Convert backend message to assistant-ui ThreadMessageLike format
|
||||||
|
|
@ -223,8 +223,7 @@ export default function NewChatPage() {
|
||||||
]);
|
]);
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const backendUrl =
|
const backendUrl = process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL || "http://localhost:8000";
|
||||||
process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL || "http://localhost:8000";
|
|
||||||
|
|
||||||
// Build message history for context
|
// Build message history for context
|
||||||
const messageHistory = messages
|
const messageHistory = messages
|
||||||
|
|
@ -232,11 +231,7 @@ export default function NewChatPage() {
|
||||||
.map((m) => {
|
.map((m) => {
|
||||||
let text = "";
|
let text = "";
|
||||||
for (const part of m.content) {
|
for (const part of m.content) {
|
||||||
if (
|
if (typeof part === "object" && part.type === "text" && "text" in part) {
|
||||||
typeof part === "object" &&
|
|
||||||
part.type === "text" &&
|
|
||||||
"text" in part
|
|
||||||
) {
|
|
||||||
text += part.text;
|
text += part.text;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -296,9 +291,7 @@ export default function NewChatPage() {
|
||||||
accumulatedText += parsed.delta;
|
accumulatedText += parsed.delta;
|
||||||
setMessages((prev) =>
|
setMessages((prev) =>
|
||||||
prev.map((m) =>
|
prev.map((m) =>
|
||||||
m.id === assistantMsgId
|
m.id === assistantMsgId ? { ...m, content: buildContent() } : m
|
||||||
? { ...m, content: buildContent() }
|
|
||||||
: m
|
|
||||||
)
|
)
|
||||||
);
|
);
|
||||||
break;
|
break;
|
||||||
|
|
@ -311,9 +304,7 @@ export default function NewChatPage() {
|
||||||
});
|
});
|
||||||
setMessages((prev) =>
|
setMessages((prev) =>
|
||||||
prev.map((m) =>
|
prev.map((m) =>
|
||||||
m.id === assistantMsgId
|
m.id === assistantMsgId ? { ...m, content: buildContent() } : m
|
||||||
? { ...m, content: buildContent() }
|
|
||||||
: m
|
|
||||||
)
|
)
|
||||||
);
|
);
|
||||||
break;
|
break;
|
||||||
|
|
@ -329,9 +320,7 @@ export default function NewChatPage() {
|
||||||
});
|
});
|
||||||
setMessages((prev) =>
|
setMessages((prev) =>
|
||||||
prev.map((m) =>
|
prev.map((m) =>
|
||||||
m.id === assistantMsgId
|
m.id === assistantMsgId ? { ...m, content: buildContent() } : m
|
||||||
? { ...m, content: buildContent() }
|
|
||||||
: m
|
|
||||||
)
|
)
|
||||||
);
|
);
|
||||||
break;
|
break;
|
||||||
|
|
@ -351,9 +340,7 @@ export default function NewChatPage() {
|
||||||
}
|
}
|
||||||
setMessages((prev) =>
|
setMessages((prev) =>
|
||||||
prev.map((m) =>
|
prev.map((m) =>
|
||||||
m.id === assistantMsgId
|
m.id === assistantMsgId ? { ...m, content: buildContent() } : m
|
||||||
? { ...m, content: buildContent() }
|
|
||||||
: m
|
|
||||||
)
|
)
|
||||||
);
|
);
|
||||||
break;
|
break;
|
||||||
|
|
@ -379,9 +366,7 @@ export default function NewChatPage() {
|
||||||
appendMessage(threadId, {
|
appendMessage(threadId, {
|
||||||
role: "assistant",
|
role: "assistant",
|
||||||
content: finalContent,
|
content: finalContent,
|
||||||
}).catch((err) =>
|
}).catch((err) => console.error("Failed to persist assistant message:", err));
|
||||||
console.error("Failed to persist assistant message:", err)
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
if (error instanceof Error && error.name === "AbortError") {
|
if (error instanceof Error && error.name === "AbortError") {
|
||||||
|
|
|
||||||
|
|
@ -39,4 +39,3 @@ export const InlineCitation: FC<InlineCitationProps> = ({ chunkId, citationNumbe
|
||||||
</SourceDetailPanel>
|
</SourceDetailPanel>
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -9,11 +9,10 @@ import {
|
||||||
useIsMarkdownCodeBlock,
|
useIsMarkdownCodeBlock,
|
||||||
} from "@assistant-ui/react-markdown";
|
} from "@assistant-ui/react-markdown";
|
||||||
import { CheckIcon, CopyIcon } from "lucide-react";
|
import { CheckIcon, CopyIcon } from "lucide-react";
|
||||||
import { type FC, type ReactNode, memo, useState } from "react";
|
import { type FC, memo, type ReactNode, useState } from "react";
|
||||||
import remarkGfm from "remark-gfm";
|
import remarkGfm from "remark-gfm";
|
||||||
|
|
||||||
import { TooltipIconButton } from "@/components/assistant-ui/tooltip-icon-button";
|
|
||||||
import { InlineCitation } from "@/components/assistant-ui/inline-citation";
|
import { InlineCitation } from "@/components/assistant-ui/inline-citation";
|
||||||
|
import { TooltipIconButton } from "@/components/assistant-ui/tooltip-icon-button";
|
||||||
import { cn } from "@/lib/utils";
|
import { cn } from "@/lib/utils";
|
||||||
|
|
||||||
// Citation pattern: [citation:CHUNK_ID]
|
// Citation pattern: [citation:CHUNK_ID]
|
||||||
|
|
@ -212,18 +211,12 @@ const defaultComponents = memoizeMarkdownComponents({
|
||||||
</h5>
|
</h5>
|
||||||
),
|
),
|
||||||
h6: ({ className, children, ...props }) => (
|
h6: ({ className, children, ...props }) => (
|
||||||
<h6
|
<h6 className={cn("aui-md-h6 my-4 font-semibold first:mt-0 last:mb-0", className)} {...props}>
|
||||||
className={cn("aui-md-h6 my-4 font-semibold first:mt-0 last:mb-0", className)}
|
|
||||||
{...props}
|
|
||||||
>
|
|
||||||
{processChildrenWithCitations(children)}
|
{processChildrenWithCitations(children)}
|
||||||
</h6>
|
</h6>
|
||||||
),
|
),
|
||||||
p: ({ className, children, ...props }) => (
|
p: ({ className, children, ...props }) => (
|
||||||
<p
|
<p className={cn("aui-md-p mt-5 mb-5 leading-7 first:mt-0 last:mb-0", className)} {...props}>
|
||||||
className={cn("aui-md-p mt-5 mb-5 leading-7 first:mt-0 last:mb-0", className)}
|
|
||||||
{...props}
|
|
||||||
>
|
|
||||||
{processChildrenWithCitations(children)}
|
{processChildrenWithCitations(children)}
|
||||||
</p>
|
</p>
|
||||||
),
|
),
|
||||||
|
|
@ -236,10 +229,7 @@ const defaultComponents = memoizeMarkdownComponents({
|
||||||
</a>
|
</a>
|
||||||
),
|
),
|
||||||
blockquote: ({ className, children, ...props }) => (
|
blockquote: ({ className, children, ...props }) => (
|
||||||
<blockquote
|
<blockquote className={cn("aui-md-blockquote border-l-2 pl-6 italic", className)} {...props}>
|
||||||
className={cn("aui-md-blockquote border-l-2 pl-6 italic", className)}
|
|
||||||
{...props}
|
|
||||||
>
|
|
||||||
{processChildrenWithCitations(children)}
|
{processChildrenWithCitations(children)}
|
||||||
</blockquote>
|
</blockquote>
|
||||||
),
|
),
|
||||||
|
|
|
||||||
|
|
@ -1,9 +1,15 @@
|
||||||
"use client";
|
"use client";
|
||||||
|
|
||||||
import { useCallback, useEffect, useState } from "react";
|
import {
|
||||||
|
ArchiveIcon,
|
||||||
|
MessageSquareIcon,
|
||||||
|
MoreVerticalIcon,
|
||||||
|
PlusIcon,
|
||||||
|
RotateCcwIcon,
|
||||||
|
TrashIcon,
|
||||||
|
} from "lucide-react";
|
||||||
import { useRouter } from "next/navigation";
|
import { useRouter } from "next/navigation";
|
||||||
import { ArchiveIcon, MessageSquareIcon, PlusIcon, TrashIcon, MoreVerticalIcon, RotateCcwIcon } from "lucide-react";
|
import { useCallback, useEffect, useState } from "react";
|
||||||
import { cn } from "@/lib/utils";
|
|
||||||
import { Button } from "@/components/ui/button";
|
import { Button } from "@/components/ui/button";
|
||||||
import {
|
import {
|
||||||
DropdownMenu,
|
DropdownMenu,
|
||||||
|
|
@ -13,10 +19,11 @@ import {
|
||||||
DropdownMenuTrigger,
|
DropdownMenuTrigger,
|
||||||
} from "@/components/ui/dropdown-menu";
|
} from "@/components/ui/dropdown-menu";
|
||||||
import {
|
import {
|
||||||
type ThreadListItem,
|
|
||||||
createThreadListManager,
|
createThreadListManager,
|
||||||
|
type ThreadListItem,
|
||||||
type ThreadListState,
|
type ThreadListState,
|
||||||
} from "@/lib/chat/thread-persistence";
|
} from "@/lib/chat/thread-persistence";
|
||||||
|
import { cn } from "@/lib/utils";
|
||||||
|
|
||||||
interface ThreadListProps {
|
interface ThreadListProps {
|
||||||
searchSpaceId: number;
|
searchSpaceId: number;
|
||||||
|
|
@ -123,7 +130,13 @@ export function ThreadList({ searchSpaceId, currentThreadId, className }: Thread
|
||||||
{/* Header with New Chat button */}
|
{/* Header with New Chat button */}
|
||||||
<div className="flex items-center justify-between border-b p-3">
|
<div className="flex items-center justify-between border-b p-3">
|
||||||
<h2 className="font-semibold text-sm">Conversations</h2>
|
<h2 className="font-semibold text-sm">Conversations</h2>
|
||||||
<Button variant="ghost" size="icon" className="size-8" onClick={handleNewThread} title="New Chat">
|
<Button
|
||||||
|
variant="ghost"
|
||||||
|
size="icon"
|
||||||
|
className="size-8"
|
||||||
|
onClick={handleNewThread}
|
||||||
|
title="New Chat"
|
||||||
|
>
|
||||||
<PlusIcon className="size-4" />
|
<PlusIcon className="size-4" />
|
||||||
</Button>
|
</Button>
|
||||||
</div>
|
</div>
|
||||||
|
|
|
||||||
|
|
@ -1,6 +1,6 @@
|
||||||
import Image from "next/image";
|
import Image from "next/image";
|
||||||
import { Streamdown } from "streamdown";
|
|
||||||
import type { Components } from "react-markdown";
|
import type { Components } from "react-markdown";
|
||||||
|
import { Streamdown } from "streamdown";
|
||||||
import { cn } from "@/lib/utils";
|
import { cn } from "@/lib/utils";
|
||||||
|
|
||||||
interface MarkdownViewerProps {
|
interface MarkdownViewerProps {
|
||||||
|
|
@ -68,12 +68,8 @@ export function MarkdownViewer({ content, className }: MarkdownViewerProps) {
|
||||||
<table className="min-w-full divide-y divide-border" {...props} />
|
<table className="min-w-full divide-y divide-border" {...props} />
|
||||||
</div>
|
</div>
|
||||||
),
|
),
|
||||||
th: ({ ...props }) => (
|
th: ({ ...props }) => <th className="px-3 py-2 text-left font-medium bg-muted" {...props} />,
|
||||||
<th className="px-3 py-2 text-left font-medium bg-muted" {...props} />
|
td: ({ ...props }) => <td className="px-3 py-2 border-t border-border" {...props} />,
|
||||||
),
|
|
||||||
td: ({ ...props }) => (
|
|
||||||
<td className="px-3 py-2 border-t border-border" {...props} />
|
|
||||||
),
|
|
||||||
code: ({ className, children, ...props }) => {
|
code: ({ className, children, ...props }) => {
|
||||||
const match = /language-(\w+)/.exec(className || "");
|
const match = /language-(\w+)/.exec(className || "");
|
||||||
const isInline = !match;
|
const isInline = !match;
|
||||||
|
|
@ -96,11 +92,13 @@ export function MarkdownViewer({ content, className }: MarkdownViewerProps) {
|
||||||
};
|
};
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className={cn("prose prose-sm dark:prose-invert max-w-none overflow-hidden [&_pre]:overflow-x-auto [&_code]:wrap-break-word [&_table]:block [&_table]:overflow-x-auto", className)}>
|
<div
|
||||||
<Streamdown
|
className={cn(
|
||||||
components={components}
|
"prose prose-sm dark:prose-invert max-w-none overflow-hidden [&_pre]:overflow-x-auto [&_code]:wrap-break-word [&_table]:block [&_table]:overflow-x-auto",
|
||||||
shikiTheme={["github-light", "github-dark"]}
|
className
|
||||||
>
|
)}
|
||||||
|
>
|
||||||
|
<Streamdown components={components} shikiTheme={["github-light", "github-dark"]}>
|
||||||
{content}
|
{content}
|
||||||
</Streamdown>
|
</Streamdown>
|
||||||
</div>
|
</div>
|
||||||
|
|
|
||||||
|
|
@ -1,26 +1,26 @@
|
||||||
"use client";
|
"use client";
|
||||||
|
|
||||||
import { useQuery } from "@tanstack/react-query";
|
import { useQuery } from "@tanstack/react-query";
|
||||||
import { AnimatePresence, motion, useReducedMotion } from "motion/react";
|
|
||||||
import {
|
import {
|
||||||
|
BookOpen,
|
||||||
ChevronDown,
|
ChevronDown,
|
||||||
ChevronUp,
|
ChevronUp,
|
||||||
ExternalLink,
|
ExternalLink,
|
||||||
Loader2,
|
|
||||||
X,
|
|
||||||
FileText,
|
FileText,
|
||||||
Hash,
|
Hash,
|
||||||
BookOpen,
|
Loader2,
|
||||||
Sparkles,
|
Sparkles,
|
||||||
|
X,
|
||||||
} from "lucide-react";
|
} from "lucide-react";
|
||||||
|
import { AnimatePresence, motion, useReducedMotion } from "motion/react";
|
||||||
import type React from "react";
|
import type React from "react";
|
||||||
import { type ReactNode, forwardRef, useCallback, useEffect, useRef, useState } from "react";
|
import { forwardRef, type ReactNode, useCallback, useEffect, useRef, useState } from "react";
|
||||||
import { createPortal } from "react-dom";
|
import { createPortal } from "react-dom";
|
||||||
import { MarkdownViewer } from "@/components/markdown-viewer";
|
import { MarkdownViewer } from "@/components/markdown-viewer";
|
||||||
|
import { Badge } from "@/components/ui/badge";
|
||||||
import { Button } from "@/components/ui/button";
|
import { Button } from "@/components/ui/button";
|
||||||
import { Collapsible, CollapsibleContent, CollapsibleTrigger } from "@/components/ui/collapsible";
|
import { Collapsible, CollapsibleContent, CollapsibleTrigger } from "@/components/ui/collapsible";
|
||||||
import { ScrollArea } from "@/components/ui/scroll-area";
|
import { ScrollArea } from "@/components/ui/scroll-area";
|
||||||
import { Badge } from "@/components/ui/badge";
|
|
||||||
import { documentsApiService } from "@/lib/apis/documents-api.service";
|
import { documentsApiService } from "@/lib/apis/documents-api.service";
|
||||||
import { cacheKeys } from "@/lib/query-client/cache-keys";
|
import { cacheKeys } from "@/lib/query-client/cache-keys";
|
||||||
import { cn } from "@/lib/utils";
|
import { cn } from "@/lib/utils";
|
||||||
|
|
@ -70,9 +70,7 @@ const ChunkCard = forwardRef<HTMLDivElement, ChunkCardProps>(
|
||||||
)}
|
)}
|
||||||
>
|
>
|
||||||
{/* Cited indicator glow effect */}
|
{/* Cited indicator glow effect */}
|
||||||
{isCited && (
|
{isCited && <div className="absolute inset-0 rounded-2xl bg-primary/5 blur-xl -z-10" />}
|
||||||
<div className="absolute inset-0 rounded-2xl bg-primary/5 blur-xl -z-10" />
|
|
||||||
)}
|
|
||||||
|
|
||||||
{/* Header */}
|
{/* Header */}
|
||||||
<div className="flex items-center justify-between px-5 py-4 border-b border-border/50">
|
<div className="flex items-center justify-between px-5 py-4 border-b border-border/50">
|
||||||
|
|
@ -87,9 +85,7 @@ const ChunkCard = forwardRef<HTMLDivElement, ChunkCardProps>(
|
||||||
>
|
>
|
||||||
{index + 1}
|
{index + 1}
|
||||||
</div>
|
</div>
|
||||||
<span className="text-sm text-muted-foreground">
|
<span className="text-sm text-muted-foreground">of {totalChunks} chunks</span>
|
||||||
of {totalChunks} chunks
|
|
||||||
</span>
|
|
||||||
</div>
|
</div>
|
||||||
{isCited && (
|
{isCited && (
|
||||||
<Badge variant="default" className="gap-1.5 px-3 py-1">
|
<Badge variant="default" className="gap-1.5 px-3 py-1">
|
||||||
|
|
@ -152,86 +148,97 @@ export function SourceDetailPanel({
|
||||||
const citedChunkIndex = documentData?.chunks?.findIndex((chunk) => chunk.id === chunkId) ?? -1;
|
const citedChunkIndex = documentData?.chunks?.findIndex((chunk) => chunk.id === chunkId) ?? -1;
|
||||||
|
|
||||||
// Simple scroll function that scrolls to a chunk by index
|
// Simple scroll function that scrolls to a chunk by index
|
||||||
const scrollToChunkByIndex = useCallback((chunkIndex: number, smooth = true) => {
|
const scrollToChunkByIndex = useCallback(
|
||||||
const scrollContainer = scrollAreaRef.current;
|
(chunkIndex: number, smooth = true) => {
|
||||||
if (!scrollContainer) return;
|
const scrollContainer = scrollAreaRef.current;
|
||||||
|
if (!scrollContainer) return;
|
||||||
|
|
||||||
const viewport = scrollContainer.querySelector(
|
const viewport = scrollContainer.querySelector(
|
||||||
'[data-radix-scroll-area-viewport]'
|
"[data-radix-scroll-area-viewport]"
|
||||||
) as HTMLElement | null;
|
) as HTMLElement | null;
|
||||||
if (!viewport) return;
|
if (!viewport) return;
|
||||||
|
|
||||||
const chunkElement = scrollContainer.querySelector(
|
const chunkElement = scrollContainer.querySelector(
|
||||||
`[data-chunk-index="${chunkIndex}"]`
|
`[data-chunk-index="${chunkIndex}"]`
|
||||||
) as HTMLElement | null;
|
) as HTMLElement | null;
|
||||||
if (!chunkElement) return;
|
if (!chunkElement) return;
|
||||||
|
|
||||||
// Get positions using getBoundingClientRect for accuracy
|
// Get positions using getBoundingClientRect for accuracy
|
||||||
const viewportRect = viewport.getBoundingClientRect();
|
const viewportRect = viewport.getBoundingClientRect();
|
||||||
const chunkRect = chunkElement.getBoundingClientRect();
|
const chunkRect = chunkElement.getBoundingClientRect();
|
||||||
|
|
||||||
// Calculate where to scroll to center the chunk
|
// Calculate where to scroll to center the chunk
|
||||||
const currentScrollTop = viewport.scrollTop;
|
const currentScrollTop = viewport.scrollTop;
|
||||||
const chunkTopRelativeToViewport = chunkRect.top - viewportRect.top + currentScrollTop;
|
const chunkTopRelativeToViewport = chunkRect.top - viewportRect.top + currentScrollTop;
|
||||||
const scrollTarget = chunkTopRelativeToViewport - (viewportRect.height / 2) + (chunkRect.height / 2);
|
const scrollTarget =
|
||||||
|
chunkTopRelativeToViewport - viewportRect.height / 2 + chunkRect.height / 2;
|
||||||
|
|
||||||
viewport.scrollTo({
|
viewport.scrollTo({
|
||||||
top: Math.max(0, scrollTarget),
|
top: Math.max(0, scrollTarget),
|
||||||
behavior: smooth && !shouldReduceMotion ? "smooth" : "auto",
|
behavior: smooth && !shouldReduceMotion ? "smooth" : "auto",
|
||||||
});
|
|
||||||
|
|
||||||
setActiveChunkIndex(chunkIndex);
|
|
||||||
}, [shouldReduceMotion]);
|
|
||||||
|
|
||||||
// Callback ref for the cited chunk - scrolls when the element mounts
|
|
||||||
const citedChunkRefCallback = useCallback((node: HTMLDivElement | null) => {
|
|
||||||
if (node && !hasScrolledRef.current && open) {
|
|
||||||
hasScrolledRef.current = true; // Mark immediately to prevent duplicate scrolls
|
|
||||||
|
|
||||||
// Store the node reference for the delayed scroll
|
|
||||||
const scrollToCitedChunk = () => {
|
|
||||||
const scrollContainer = scrollAreaRef.current;
|
|
||||||
if (!scrollContainer || !node.isConnected) return false;
|
|
||||||
|
|
||||||
const viewport = scrollContainer.querySelector(
|
|
||||||
'[data-radix-scroll-area-viewport]'
|
|
||||||
) as HTMLElement | null;
|
|
||||||
if (!viewport) return false;
|
|
||||||
|
|
||||||
// Get positions
|
|
||||||
const viewportRect = viewport.getBoundingClientRect();
|
|
||||||
const chunkRect = node.getBoundingClientRect();
|
|
||||||
|
|
||||||
// Calculate scroll position to center the chunk
|
|
||||||
const currentScrollTop = viewport.scrollTop;
|
|
||||||
const chunkTopRelativeToViewport = chunkRect.top - viewportRect.top + currentScrollTop;
|
|
||||||
const scrollTarget = chunkTopRelativeToViewport - (viewportRect.height / 2) + (chunkRect.height / 2);
|
|
||||||
|
|
||||||
viewport.scrollTo({
|
|
||||||
top: Math.max(0, scrollTarget),
|
|
||||||
behavior: "auto", // Instant scroll for initial positioning
|
|
||||||
});
|
|
||||||
|
|
||||||
return true;
|
|
||||||
};
|
|
||||||
|
|
||||||
// Scroll multiple times with delays to handle progressive content rendering
|
|
||||||
// Each subsequent scroll will correct for any layout shifts
|
|
||||||
const scrollAttempts = [50, 150, 300, 600, 1000];
|
|
||||||
|
|
||||||
scrollAttempts.forEach((delay) => {
|
|
||||||
setTimeout(() => {
|
|
||||||
scrollToCitedChunk();
|
|
||||||
}, delay);
|
|
||||||
});
|
});
|
||||||
|
|
||||||
// After final attempt, mark state as scrolled
|
setActiveChunkIndex(chunkIndex);
|
||||||
setTimeout(() => {
|
},
|
||||||
setHasScrolledToCited(true);
|
[shouldReduceMotion]
|
||||||
setActiveChunkIndex(citedChunkIndex);
|
);
|
||||||
}, scrollAttempts[scrollAttempts.length - 1] + 50);
|
|
||||||
}
|
// Callback ref for the cited chunk - scrolls when the element mounts
|
||||||
}, [open, citedChunkIndex]);
|
const citedChunkRefCallback = useCallback(
|
||||||
|
(node: HTMLDivElement | null) => {
|
||||||
|
if (node && !hasScrolledRef.current && open) {
|
||||||
|
hasScrolledRef.current = true; // Mark immediately to prevent duplicate scrolls
|
||||||
|
|
||||||
|
// Store the node reference for the delayed scroll
|
||||||
|
const scrollToCitedChunk = () => {
|
||||||
|
const scrollContainer = scrollAreaRef.current;
|
||||||
|
if (!scrollContainer || !node.isConnected) return false;
|
||||||
|
|
||||||
|
const viewport = scrollContainer.querySelector(
|
||||||
|
"[data-radix-scroll-area-viewport]"
|
||||||
|
) as HTMLElement | null;
|
||||||
|
if (!viewport) return false;
|
||||||
|
|
||||||
|
// Get positions
|
||||||
|
const viewportRect = viewport.getBoundingClientRect();
|
||||||
|
const chunkRect = node.getBoundingClientRect();
|
||||||
|
|
||||||
|
// Calculate scroll position to center the chunk
|
||||||
|
const currentScrollTop = viewport.scrollTop;
|
||||||
|
const chunkTopRelativeToViewport = chunkRect.top - viewportRect.top + currentScrollTop;
|
||||||
|
const scrollTarget =
|
||||||
|
chunkTopRelativeToViewport - viewportRect.height / 2 + chunkRect.height / 2;
|
||||||
|
|
||||||
|
viewport.scrollTo({
|
||||||
|
top: Math.max(0, scrollTarget),
|
||||||
|
behavior: "auto", // Instant scroll for initial positioning
|
||||||
|
});
|
||||||
|
|
||||||
|
return true;
|
||||||
|
};
|
||||||
|
|
||||||
|
// Scroll multiple times with delays to handle progressive content rendering
|
||||||
|
// Each subsequent scroll will correct for any layout shifts
|
||||||
|
const scrollAttempts = [50, 150, 300, 600, 1000];
|
||||||
|
|
||||||
|
scrollAttempts.forEach((delay) => {
|
||||||
|
setTimeout(() => {
|
||||||
|
scrollToCitedChunk();
|
||||||
|
}, delay);
|
||||||
|
});
|
||||||
|
|
||||||
|
// After final attempt, mark state as scrolled
|
||||||
|
setTimeout(
|
||||||
|
() => {
|
||||||
|
setHasScrolledToCited(true);
|
||||||
|
setActiveChunkIndex(citedChunkIndex);
|
||||||
|
},
|
||||||
|
scrollAttempts[scrollAttempts.length - 1] + 50
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[open, citedChunkIndex]
|
||||||
|
);
|
||||||
|
|
||||||
// Reset scroll state when panel closes
|
// Reset scroll state when panel closes
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
|
|
@ -271,9 +278,12 @@ export function SourceDetailPanel({
|
||||||
window.open(clickUrl, "_blank", "noopener,noreferrer");
|
window.open(clickUrl, "_blank", "noopener,noreferrer");
|
||||||
};
|
};
|
||||||
|
|
||||||
const scrollToChunk = useCallback((index: number) => {
|
const scrollToChunk = useCallback(
|
||||||
scrollToChunkByIndex(index, true);
|
(index: number) => {
|
||||||
}, [scrollToChunkByIndex]);
|
scrollToChunkByIndex(index, true);
|
||||||
|
},
|
||||||
|
[scrollToChunkByIndex]
|
||||||
|
);
|
||||||
|
|
||||||
const panelContent = (
|
const panelContent = (
|
||||||
<AnimatePresence mode="wait">
|
<AnimatePresence mode="wait">
|
||||||
|
|
@ -320,7 +330,8 @@ export function SourceDetailPanel({
|
||||||
: sourceType && formatDocumentType(sourceType)}
|
: sourceType && formatDocumentType(sourceType)}
|
||||||
{documentData?.chunks && (
|
{documentData?.chunks && (
|
||||||
<span className="ml-2">
|
<span className="ml-2">
|
||||||
• {documentData.chunks.length} chunk{documentData.chunks.length !== 1 ? "s" : ""}
|
• {documentData.chunks.length} chunk
|
||||||
|
{documentData.chunks.length !== 1 ? "s" : ""}
|
||||||
</span>
|
</span>
|
||||||
)}
|
)}
|
||||||
</p>
|
</p>
|
||||||
|
|
@ -378,9 +389,12 @@ export function SourceDetailPanel({
|
||||||
<X className="h-10 w-10 text-destructive" />
|
<X className="h-10 w-10 text-destructive" />
|
||||||
</div>
|
</div>
|
||||||
<div>
|
<div>
|
||||||
<p className="font-semibold text-destructive text-lg">Failed to load document</p>
|
<p className="font-semibold text-destructive text-lg">
|
||||||
|
Failed to load document
|
||||||
|
</p>
|
||||||
<p className="text-sm text-muted-foreground mt-2 max-w-md">
|
<p className="text-sm text-muted-foreground mt-2 max-w-md">
|
||||||
{documentByChunkFetchingError.message || "An unexpected error occurred. Please try again."}
|
{documentByChunkFetchingError.message ||
|
||||||
|
"An unexpected error occurred. Please try again."}
|
||||||
</p>
|
</p>
|
||||||
</div>
|
</div>
|
||||||
<Button variant="outline" onClick={() => onOpenChange(false)} className="mt-2">
|
<Button variant="outline" onClick={() => onOpenChange(false)} className="mt-2">
|
||||||
|
|
@ -490,18 +504,14 @@ export function SourceDetailPanel({
|
||||||
Document Information
|
Document Information
|
||||||
</h3>
|
</h3>
|
||||||
<dl className="grid grid-cols-1 sm:grid-cols-2 gap-4 text-sm">
|
<dl className="grid grid-cols-1 sm:grid-cols-2 gap-4 text-sm">
|
||||||
{Object.entries(documentData.document_metadata).map(
|
{Object.entries(documentData.document_metadata).map(([key, value]) => (
|
||||||
([key, value]) => (
|
<div key={key} className="space-y-1">
|
||||||
<div key={key} className="space-y-1">
|
<dt className="font-medium text-muted-foreground capitalize text-xs">
|
||||||
<dt className="font-medium text-muted-foreground capitalize text-xs">
|
{key.replace(/_/g, " ")}
|
||||||
{key.replace(/_/g, " ")}
|
</dt>
|
||||||
</dt>
|
<dd className="text-foreground wrap-break-word">{String(value)}</dd>
|
||||||
<dd className="text-foreground wrap-break-word">
|
</div>
|
||||||
{String(value)}
|
))}
|
||||||
</dd>
|
|
||||||
</div>
|
|
||||||
)
|
|
||||||
)}
|
|
||||||
</dl>
|
</dl>
|
||||||
</motion.div>
|
</motion.div>
|
||||||
)}
|
)}
|
||||||
|
|
|
||||||
|
|
@ -1,14 +1,11 @@
|
||||||
"use client";
|
"use client";
|
||||||
|
|
||||||
import { useQuery } from "@tanstack/react-query";
|
import { useQuery, useQueryClient } from "@tanstack/react-query";
|
||||||
import { useAtom, useAtomValue, useSetAtom } from "jotai";
|
import { useAtomValue, useSetAtom } from "jotai";
|
||||||
import { Trash2 } from "lucide-react";
|
import { Trash2 } from "lucide-react";
|
||||||
import { useRouter } from "next/navigation";
|
import { useRouter } from "next/navigation";
|
||||||
import { useTranslations } from "next-intl";
|
import { useTranslations } from "next-intl";
|
||||||
import { useCallback, useEffect, useMemo, useState } from "react";
|
import { useCallback, useMemo, useState } from "react";
|
||||||
import { deleteChatMutationAtom } from "@/atoms/chats/chat-mutation.atoms";
|
|
||||||
import { chatsAtom } from "@/atoms/chats/chat-query.atoms";
|
|
||||||
import { globalChatsQueryParamsAtom } from "@/atoms/chats/ui.atoms";
|
|
||||||
import { hasUnsavedEditorChangesAtom, pendingEditorNavigationAtom } from "@/atoms/editor/ui.atoms";
|
import { hasUnsavedEditorChangesAtom, pendingEditorNavigationAtom } from "@/atoms/editor/ui.atoms";
|
||||||
import { currentUserAtom } from "@/atoms/user/user-query.atoms";
|
import { currentUserAtom } from "@/atoms/user/user-query.atoms";
|
||||||
import { AppSidebar } from "@/components/sidebar/app-sidebar";
|
import { AppSidebar } from "@/components/sidebar/app-sidebar";
|
||||||
|
|
@ -23,6 +20,7 @@ import {
|
||||||
} from "@/components/ui/dialog";
|
} from "@/components/ui/dialog";
|
||||||
import { notesApiService } from "@/lib/apis/notes-api.service";
|
import { notesApiService } from "@/lib/apis/notes-api.service";
|
||||||
import { searchSpacesApiService } from "@/lib/apis/search-spaces-api.service";
|
import { searchSpacesApiService } from "@/lib/apis/search-spaces-api.service";
|
||||||
|
import { deleteThread, fetchThreads } from "@/lib/chat/thread-persistence";
|
||||||
import { cacheKeys } from "@/lib/query-client/cache-keys";
|
import { cacheKeys } from "@/lib/query-client/cache-keys";
|
||||||
|
|
||||||
interface AppSidebarProviderProps {
|
interface AppSidebarProviderProps {
|
||||||
|
|
@ -52,18 +50,24 @@ export function AppSidebarProvider({
|
||||||
const t = useTranslations("dashboard");
|
const t = useTranslations("dashboard");
|
||||||
const tCommon = useTranslations("common");
|
const tCommon = useTranslations("common");
|
||||||
const router = useRouter();
|
const router = useRouter();
|
||||||
const setChatsQueryParams = useSetAtom(globalChatsQueryParamsAtom);
|
const queryClient = useQueryClient();
|
||||||
const { data: chats, error: chatError, isLoading: isLoadingChats } = useAtomValue(chatsAtom);
|
const [isDeletingThread, setIsDeletingThread] = useState(false);
|
||||||
const [{ isPending: isDeletingChat, mutateAsync: deleteChat, error: deleteError }] =
|
|
||||||
useAtom(deleteChatMutationAtom);
|
|
||||||
|
|
||||||
// Editor state for handling unsaved changes
|
// Editor state for handling unsaved changes
|
||||||
const hasUnsavedEditorChanges = useAtomValue(hasUnsavedEditorChangesAtom);
|
const hasUnsavedEditorChanges = useAtomValue(hasUnsavedEditorChangesAtom);
|
||||||
const setPendingNavigation = useSetAtom(pendingEditorNavigationAtom);
|
const setPendingNavigation = useSetAtom(pendingEditorNavigationAtom);
|
||||||
|
|
||||||
useEffect(() => {
|
// Fetch new chat threads
|
||||||
setChatsQueryParams((prev) => ({ ...prev, search_space_id: searchSpaceId, skip: 0, limit: 4 }));
|
const {
|
||||||
}, [searchSpaceId, setChatsQueryParams]);
|
data: threadsData,
|
||||||
|
error: threadError,
|
||||||
|
isLoading: isLoadingThreads,
|
||||||
|
refetch: refetchThreads,
|
||||||
|
} = useQuery({
|
||||||
|
queryKey: ["threads", searchSpaceId],
|
||||||
|
queryFn: () => fetchThreads(Number(searchSpaceId), 4),
|
||||||
|
enabled: !!searchSpaceId,
|
||||||
|
});
|
||||||
|
|
||||||
const {
|
const {
|
||||||
data: searchSpace,
|
data: searchSpace,
|
||||||
|
|
@ -95,7 +99,7 @@ export function AppSidebarProvider({
|
||||||
});
|
});
|
||||||
|
|
||||||
const [showDeleteDialog, setShowDeleteDialog] = useState(false);
|
const [showDeleteDialog, setShowDeleteDialog] = useState(false);
|
||||||
const [chatToDelete, setChatToDelete] = useState<{ id: number; name: string } | null>(null);
|
const [threadToDelete, setThreadToDelete] = useState<{ id: number; name: string } | null>(null);
|
||||||
const [showDeleteNoteDialog, setShowDeleteNoteDialog] = useState(false);
|
const [showDeleteNoteDialog, setShowDeleteNoteDialog] = useState(false);
|
||||||
const [noteToDelete, setNoteToDelete] = useState<{
|
const [noteToDelete, setNoteToDelete] = useState<{
|
||||||
id: number;
|
id: number;
|
||||||
|
|
@ -103,62 +107,56 @@ export function AppSidebarProvider({
|
||||||
search_space_id: number;
|
search_space_id: number;
|
||||||
} | null>(null);
|
} | null>(null);
|
||||||
const [isDeletingNote, setIsDeletingNote] = useState(false);
|
const [isDeletingNote, setIsDeletingNote] = useState(false);
|
||||||
const [isClient, setIsClient] = useState(false);
|
|
||||||
|
|
||||||
// Set isClient to true when component mounts on the client
|
|
||||||
useEffect(() => {
|
|
||||||
setIsClient(true);
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
// Retry function
|
// Retry function
|
||||||
const retryFetch = useCallback(() => {
|
const retryFetch = useCallback(() => {
|
||||||
fetchSearchSpace();
|
fetchSearchSpace();
|
||||||
}, [fetchSearchSpace]);
|
}, [fetchSearchSpace]);
|
||||||
|
|
||||||
// Transform API response to the format expected by AppSidebar
|
// Transform threads to the format expected by AppSidebar
|
||||||
const recentChats = useMemo(() => {
|
const recentChats = useMemo(() => {
|
||||||
if (!chats) return [];
|
if (!threadsData?.threads) return [];
|
||||||
|
|
||||||
// Sort chats by created_at (most recent first)
|
// Threads are already sorted by updated_at desc from the API
|
||||||
const sortedChats = [...chats].sort((a, b) => {
|
return threadsData.threads.map((thread) => ({
|
||||||
const dateA = new Date(a.created_at).getTime();
|
name: thread.title || `Chat ${thread.id}`,
|
||||||
const dateB = new Date(b.created_at).getTime();
|
url: `/dashboard/${searchSpaceId}/new-chat/${thread.id}`,
|
||||||
return dateB - dateA; // Descending order (most recent first)
|
|
||||||
});
|
|
||||||
|
|
||||||
return sortedChats.map((chat) => ({
|
|
||||||
name: chat.title || `Chat ${chat.id}`,
|
|
||||||
url: `/dashboard/${chat.search_space_id}/researcher/${chat.id}`,
|
|
||||||
icon: "MessageCircleMore",
|
icon: "MessageCircleMore",
|
||||||
id: chat.id,
|
id: thread.id,
|
||||||
search_space_id: chat.search_space_id,
|
search_space_id: Number(searchSpaceId),
|
||||||
actions: [
|
actions: [
|
||||||
{
|
{
|
||||||
name: "Delete",
|
name: "Delete",
|
||||||
icon: "Trash2",
|
icon: "Trash2",
|
||||||
onClick: () => {
|
onClick: () => {
|
||||||
setChatToDelete({ id: chat.id, name: chat.title || `Chat ${chat.id}` });
|
setThreadToDelete({
|
||||||
|
id: thread.id,
|
||||||
|
name: thread.title || `Chat ${thread.id}`,
|
||||||
|
});
|
||||||
setShowDeleteDialog(true);
|
setShowDeleteDialog(true);
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
}));
|
}));
|
||||||
}, [chats]);
|
}, [threadsData, searchSpaceId]);
|
||||||
|
|
||||||
// Handle delete chat with better error handling
|
// Handle delete thread
|
||||||
const handleDeleteChat = useCallback(async () => {
|
const handleDeleteThread = useCallback(async () => {
|
||||||
if (!chatToDelete) return;
|
if (!threadToDelete) return;
|
||||||
|
|
||||||
|
setIsDeletingThread(true);
|
||||||
try {
|
try {
|
||||||
await deleteChat({ id: chatToDelete.id });
|
await deleteThread(threadToDelete.id);
|
||||||
|
// Invalidate threads query to refresh the list
|
||||||
|
queryClient.invalidateQueries({ queryKey: ["threads", searchSpaceId] });
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.error("Error deleting chat:", error);
|
console.error("Error deleting thread:", error);
|
||||||
// You could show a toast notification here
|
|
||||||
} finally {
|
} finally {
|
||||||
|
setIsDeletingThread(false);
|
||||||
setShowDeleteDialog(false);
|
setShowDeleteDialog(false);
|
||||||
setChatToDelete(null);
|
setThreadToDelete(null);
|
||||||
}
|
}
|
||||||
}, [chatToDelete, deleteChat]);
|
}, [threadToDelete, queryClient, searchSpaceId]);
|
||||||
|
|
||||||
// Handle delete note with confirmation
|
// Handle delete note with confirmation
|
||||||
const handleDeleteNote = useCallback(async () => {
|
const handleDeleteNote = useCallback(async () => {
|
||||||
|
|
@ -182,7 +180,7 @@ export function AppSidebarProvider({
|
||||||
|
|
||||||
// Memoized fallback chats
|
// Memoized fallback chats
|
||||||
const fallbackChats = useMemo(() => {
|
const fallbackChats = useMemo(() => {
|
||||||
if (chatError) {
|
if (threadError) {
|
||||||
return [
|
return [
|
||||||
{
|
{
|
||||||
name: t("error_loading_chats"),
|
name: t("error_loading_chats"),
|
||||||
|
|
@ -194,7 +192,7 @@ export function AppSidebarProvider({
|
||||||
{
|
{
|
||||||
name: tCommon("retry"),
|
name: tCommon("retry"),
|
||||||
icon: "RefreshCw",
|
icon: "RefreshCw",
|
||||||
onClick: retryFetch,
|
onClick: () => refetchThreads(),
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
|
|
@ -202,7 +200,7 @@ export function AppSidebarProvider({
|
||||||
}
|
}
|
||||||
|
|
||||||
return [];
|
return [];
|
||||||
}, [chatError, isLoadingChats, recentChats.length, searchSpaceId, retryFetch, t, tCommon]);
|
}, [threadError, searchSpaceId, refetchThreads, t, tCommon]);
|
||||||
|
|
||||||
// Use fallback chats if there's an error or no chats
|
// Use fallback chats if there's an error or no chats
|
||||||
const displayChats = recentChats.length > 0 ? recentChats : fallbackChats;
|
const displayChats = recentChats.length > 0 ? recentChats : fallbackChats;
|
||||||
|
|
@ -262,7 +260,7 @@ export function AppSidebarProvider({
|
||||||
// Memoized updated navSecondary
|
// Memoized updated navSecondary
|
||||||
const updatedNavSecondary = useMemo(() => {
|
const updatedNavSecondary = useMemo(() => {
|
||||||
const updated = [...navSecondary];
|
const updated = [...navSecondary];
|
||||||
if (updated.length > 0 && isClient) {
|
if (updated.length > 0) {
|
||||||
updated[0] = {
|
updated[0] = {
|
||||||
...updated[0],
|
...updated[0],
|
||||||
title:
|
title:
|
||||||
|
|
@ -275,15 +273,7 @@ export function AppSidebarProvider({
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
return updated;
|
return updated;
|
||||||
}, [
|
}, [navSecondary, searchSpace?.name, isLoadingSearchSpace, searchSpaceError, t, tCommon]);
|
||||||
navSecondary,
|
|
||||||
isClient,
|
|
||||||
searchSpace?.name,
|
|
||||||
isLoadingSearchSpace,
|
|
||||||
searchSpaceError,
|
|
||||||
t,
|
|
||||||
tCommon,
|
|
||||||
]);
|
|
||||||
|
|
||||||
// Prepare page usage data
|
// Prepare page usage data
|
||||||
const pageUsage = user
|
const pageUsage = user
|
||||||
|
|
@ -293,21 +283,6 @@ export function AppSidebarProvider({
|
||||||
}
|
}
|
||||||
: undefined;
|
: undefined;
|
||||||
|
|
||||||
// Show loading state if not client-side
|
|
||||||
if (!isClient) {
|
|
||||||
return (
|
|
||||||
<AppSidebar
|
|
||||||
searchSpaceId={searchSpaceId}
|
|
||||||
navSecondary={navSecondary}
|
|
||||||
navMain={navMain}
|
|
||||||
RecentChats={[]}
|
|
||||||
RecentNotes={[]}
|
|
||||||
onAddNote={handleAddNote}
|
|
||||||
pageUsage={pageUsage}
|
|
||||||
/>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<>
|
<>
|
||||||
<AppSidebar
|
<AppSidebar
|
||||||
|
|
@ -329,25 +304,25 @@ export function AppSidebarProvider({
|
||||||
<span>{t("delete_chat")}</span>
|
<span>{t("delete_chat")}</span>
|
||||||
</DialogTitle>
|
</DialogTitle>
|
||||||
<DialogDescription>
|
<DialogDescription>
|
||||||
{t("delete_chat_confirm")} <span className="font-medium">{chatToDelete?.name}</span>?{" "}
|
{t("delete_chat_confirm")} <span className="font-medium">{threadToDelete?.name}</span>
|
||||||
{t("action_cannot_undone")}
|
? {t("action_cannot_undone")}
|
||||||
</DialogDescription>
|
</DialogDescription>
|
||||||
</DialogHeader>
|
</DialogHeader>
|
||||||
<DialogFooter className="flex gap-2 sm:justify-end">
|
<DialogFooter className="flex gap-2 sm:justify-end">
|
||||||
<Button
|
<Button
|
||||||
variant="outline"
|
variant="outline"
|
||||||
onClick={() => setShowDeleteDialog(false)}
|
onClick={() => setShowDeleteDialog(false)}
|
||||||
disabled={isDeletingChat}
|
disabled={isDeletingThread}
|
||||||
>
|
>
|
||||||
{tCommon("cancel")}
|
{tCommon("cancel")}
|
||||||
</Button>
|
</Button>
|
||||||
<Button
|
<Button
|
||||||
variant="destructive"
|
variant="destructive"
|
||||||
onClick={handleDeleteChat}
|
onClick={handleDeleteThread}
|
||||||
disabled={isDeletingChat}
|
disabled={isDeletingThread}
|
||||||
className="gap-2"
|
className="gap-2"
|
||||||
>
|
>
|
||||||
{isDeletingChat ? (
|
{isDeletingThread ? (
|
||||||
<>
|
<>
|
||||||
<span className="h-4 w-4 animate-spin rounded-full border-2 border-current border-t-transparent" />
|
<span className="h-4 w-4 animate-spin rounded-full border-2 border-current border-t-transparent" />
|
||||||
{t("deleting")}
|
{t("deleting")}
|
||||||
|
|
|
||||||
|
|
@ -2,7 +2,16 @@
|
||||||
|
|
||||||
import { useQuery, useQueryClient } from "@tanstack/react-query";
|
import { useQuery, useQueryClient } from "@tanstack/react-query";
|
||||||
import { format } from "date-fns";
|
import { format } from "date-fns";
|
||||||
import { Loader2, MessageCircleMore, MoreHorizontal, Search, Trash2, X } from "lucide-react";
|
import {
|
||||||
|
ArchiveIcon,
|
||||||
|
Loader2,
|
||||||
|
MessageCircleMore,
|
||||||
|
MoreHorizontal,
|
||||||
|
RotateCcwIcon,
|
||||||
|
Search,
|
||||||
|
Trash2,
|
||||||
|
X,
|
||||||
|
} from "lucide-react";
|
||||||
import { useRouter } from "next/navigation";
|
import { useRouter } from "next/navigation";
|
||||||
import { useTranslations } from "next-intl";
|
import { useTranslations } from "next-intl";
|
||||||
import { useCallback, useState } from "react";
|
import { useCallback, useState } from "react";
|
||||||
|
|
@ -12,6 +21,7 @@ import {
|
||||||
DropdownMenu,
|
DropdownMenu,
|
||||||
DropdownMenuContent,
|
DropdownMenuContent,
|
||||||
DropdownMenuItem,
|
DropdownMenuItem,
|
||||||
|
DropdownMenuSeparator,
|
||||||
DropdownMenuTrigger,
|
DropdownMenuTrigger,
|
||||||
} from "@/components/ui/dropdown-menu";
|
} from "@/components/ui/dropdown-menu";
|
||||||
import { Input } from "@/components/ui/input";
|
import { Input } from "@/components/ui/input";
|
||||||
|
|
@ -25,7 +35,13 @@ import {
|
||||||
} from "@/components/ui/sheet";
|
} from "@/components/ui/sheet";
|
||||||
import { Tooltip, TooltipContent, TooltipTrigger } from "@/components/ui/tooltip";
|
import { Tooltip, TooltipContent, TooltipTrigger } from "@/components/ui/tooltip";
|
||||||
import { useDebouncedValue } from "@/hooks/use-debounced-value";
|
import { useDebouncedValue } from "@/hooks/use-debounced-value";
|
||||||
import { chatsApiService } from "@/lib/apis/chats-api.service";
|
import {
|
||||||
|
deleteThread,
|
||||||
|
fetchThreads,
|
||||||
|
searchThreads,
|
||||||
|
type ThreadListItem,
|
||||||
|
updateThread,
|
||||||
|
} from "@/lib/chat/thread-persistence";
|
||||||
import { cn } from "@/lib/utils";
|
import { cn } from "@/lib/utils";
|
||||||
|
|
||||||
interface AllChatsSidebarProps {
|
interface AllChatsSidebarProps {
|
||||||
|
|
@ -38,70 +54,85 @@ export function AllChatsSidebar({ open, onOpenChange, searchSpaceId }: AllChatsS
|
||||||
const t = useTranslations("sidebar");
|
const t = useTranslations("sidebar");
|
||||||
const router = useRouter();
|
const router = useRouter();
|
||||||
const queryClient = useQueryClient();
|
const queryClient = useQueryClient();
|
||||||
const [deletingChatId, setDeletingChatId] = useState<number | null>(null);
|
const [deletingThreadId, setDeletingThreadId] = useState<number | null>(null);
|
||||||
|
const [archivingThreadId, setArchivingThreadId] = useState<number | null>(null);
|
||||||
const [searchQuery, setSearchQuery] = useState("");
|
const [searchQuery, setSearchQuery] = useState("");
|
||||||
|
const [showArchived, setShowArchived] = useState(false);
|
||||||
const debouncedSearchQuery = useDebouncedValue(searchQuery, 300);
|
const debouncedSearchQuery = useDebouncedValue(searchQuery, 300);
|
||||||
|
|
||||||
const isSearchMode = !!debouncedSearchQuery.trim();
|
const isSearchMode = !!debouncedSearchQuery.trim();
|
||||||
|
|
||||||
// Fetch all chats (when not searching)
|
// Fetch all threads (when not searching)
|
||||||
const {
|
const {
|
||||||
data: chatsData,
|
data: threadsData,
|
||||||
error: chatsError,
|
error: threadsError,
|
||||||
isLoading: isLoadingChats,
|
isLoading: isLoadingThreads,
|
||||||
} = useQuery({
|
} = useQuery({
|
||||||
queryKey: ["all-chats", searchSpaceId],
|
queryKey: ["all-threads", searchSpaceId],
|
||||||
queryFn: () =>
|
queryFn: () => fetchThreads(Number(searchSpaceId)),
|
||||||
chatsApiService.getChats({
|
|
||||||
queryParams: {
|
|
||||||
search_space_id: Number(searchSpaceId),
|
|
||||||
},
|
|
||||||
}),
|
|
||||||
enabled: !!searchSpaceId && open && !isSearchMode,
|
enabled: !!searchSpaceId && open && !isSearchMode,
|
||||||
});
|
});
|
||||||
|
|
||||||
// Search chats (when searching)
|
// Search threads (when searching)
|
||||||
const {
|
const {
|
||||||
data: searchData,
|
data: searchData,
|
||||||
error: searchError,
|
error: searchError,
|
||||||
isLoading: isLoadingSearch,
|
isLoading: isLoadingSearch,
|
||||||
} = useQuery({
|
} = useQuery({
|
||||||
queryKey: ["search-chats", searchSpaceId, debouncedSearchQuery],
|
queryKey: ["search-threads", searchSpaceId, debouncedSearchQuery],
|
||||||
queryFn: () =>
|
queryFn: () => searchThreads(Number(searchSpaceId), debouncedSearchQuery.trim()),
|
||||||
chatsApiService.searchChats({
|
|
||||||
queryParams: {
|
|
||||||
title: debouncedSearchQuery.trim(),
|
|
||||||
search_space_id: Number(searchSpaceId),
|
|
||||||
},
|
|
||||||
}),
|
|
||||||
enabled: !!searchSpaceId && open && isSearchMode,
|
enabled: !!searchSpaceId && open && isSearchMode,
|
||||||
});
|
});
|
||||||
|
|
||||||
// Handle chat navigation
|
// Handle thread navigation
|
||||||
const handleChatClick = useCallback(
|
const handleThreadClick = useCallback(
|
||||||
(chatId: number, chatSearchSpaceId: number) => {
|
(threadId: number) => {
|
||||||
router.push(`/dashboard/${chatSearchSpaceId}/researcher/${chatId}`);
|
router.push(`/dashboard/${searchSpaceId}/new-chat/${threadId}`);
|
||||||
onOpenChange(false);
|
onOpenChange(false);
|
||||||
},
|
},
|
||||||
[router, onOpenChange]
|
[router, onOpenChange, searchSpaceId]
|
||||||
);
|
);
|
||||||
|
|
||||||
// Handle chat deletion
|
// Handle thread deletion
|
||||||
const handleDeleteChat = useCallback(
|
const handleDeleteThread = useCallback(
|
||||||
async (chatId: number) => {
|
async (threadId: number) => {
|
||||||
setDeletingChatId(chatId);
|
setDeletingThreadId(threadId);
|
||||||
try {
|
try {
|
||||||
await chatsApiService.deleteChat({ id: chatId });
|
await deleteThread(threadId);
|
||||||
toast.success(t("chat_deleted") || "Chat deleted successfully");
|
toast.success(t("chat_deleted") || "Chat deleted successfully");
|
||||||
// Invalidate queries to refresh the list
|
// Invalidate queries to refresh the list
|
||||||
queryClient.invalidateQueries({ queryKey: ["all-chats", searchSpaceId] });
|
queryClient.invalidateQueries({ queryKey: ["all-threads", searchSpaceId] });
|
||||||
queryClient.invalidateQueries({ queryKey: ["search-chats", searchSpaceId] });
|
queryClient.invalidateQueries({ queryKey: ["search-threads", searchSpaceId] });
|
||||||
queryClient.invalidateQueries({ queryKey: ["chats"] });
|
queryClient.invalidateQueries({ queryKey: ["threads", searchSpaceId] });
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.error("Error deleting chat:", error);
|
console.error("Error deleting thread:", error);
|
||||||
toast.error(t("error_deleting_chat") || "Failed to delete chat");
|
toast.error(t("error_deleting_chat") || "Failed to delete chat");
|
||||||
} finally {
|
} finally {
|
||||||
setDeletingChatId(null);
|
setDeletingThreadId(null);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[queryClient, searchSpaceId, t]
|
||||||
|
);
|
||||||
|
|
||||||
|
// Handle thread archive/unarchive
|
||||||
|
const handleToggleArchive = useCallback(
|
||||||
|
async (threadId: number, currentlyArchived: boolean) => {
|
||||||
|
setArchivingThreadId(threadId);
|
||||||
|
try {
|
||||||
|
await updateThread(threadId, { archived: !currentlyArchived });
|
||||||
|
toast.success(
|
||||||
|
currentlyArchived
|
||||||
|
? t("chat_unarchived") || "Chat restored"
|
||||||
|
: t("chat_archived") || "Chat archived"
|
||||||
|
);
|
||||||
|
queryClient.invalidateQueries({ queryKey: ["all-threads", searchSpaceId] });
|
||||||
|
queryClient.invalidateQueries({ queryKey: ["search-threads", searchSpaceId] });
|
||||||
|
queryClient.invalidateQueries({ queryKey: ["threads", searchSpaceId] });
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Error archiving thread:", error);
|
||||||
|
toast.error(t("error_archiving_chat") || "Failed to archive chat");
|
||||||
|
} finally {
|
||||||
|
setArchivingThreadId(null);
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
[queryClient, searchSpaceId, t]
|
[queryClient, searchSpaceId, t]
|
||||||
|
|
@ -112,10 +143,20 @@ export function AllChatsSidebar({ open, onOpenChange, searchSpaceId }: AllChatsS
|
||||||
setSearchQuery("");
|
setSearchQuery("");
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
// Determine which data source to use and loading/error states
|
// Determine which data source to use
|
||||||
const chats = isSearchMode ? (searchData ?? []) : (chatsData ?? []);
|
let threads: ThreadListItem[] = [];
|
||||||
const isLoading = isSearchMode ? isLoadingSearch : isLoadingChats;
|
if (isSearchMode) {
|
||||||
const error = isSearchMode ? searchError : chatsError;
|
threads = searchData ?? [];
|
||||||
|
} else if (threadsData) {
|
||||||
|
threads = showArchived ? threadsData.archived_threads : threadsData.threads;
|
||||||
|
}
|
||||||
|
|
||||||
|
const isLoading = isSearchMode ? isLoadingSearch : isLoadingThreads;
|
||||||
|
const error = isSearchMode ? searchError : threadsError;
|
||||||
|
|
||||||
|
// Get counts for tabs
|
||||||
|
const activeCount = threadsData?.threads.length ?? 0;
|
||||||
|
const archivedCount = threadsData?.archived_threads.length ?? 0;
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<Sheet open={open} onOpenChange={onOpenChange}>
|
<Sheet open={open} onOpenChange={onOpenChange}>
|
||||||
|
|
@ -150,6 +191,36 @@ export function AllChatsSidebar({ open, onOpenChange, searchSpaceId }: AllChatsS
|
||||||
</div>
|
</div>
|
||||||
</SheetHeader>
|
</SheetHeader>
|
||||||
|
|
||||||
|
{/* Tab toggle for active/archived (only show when not searching) */}
|
||||||
|
{!isSearchMode && (
|
||||||
|
<div className="flex border-b mx-3">
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={() => setShowArchived(false)}
|
||||||
|
className={cn(
|
||||||
|
"flex-1 px-3 py-2 text-center text-xs font-medium transition-colors",
|
||||||
|
!showArchived
|
||||||
|
? "border-b-2 border-primary text-primary"
|
||||||
|
: "text-muted-foreground hover:text-foreground"
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
Active ({activeCount})
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={() => setShowArchived(true)}
|
||||||
|
className={cn(
|
||||||
|
"flex-1 px-3 py-2 text-center text-xs font-medium transition-colors",
|
||||||
|
showArchived
|
||||||
|
? "border-b-2 border-primary text-primary"
|
||||||
|
: "text-muted-foreground hover:text-foreground"
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
Archived ({archivedCount})
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
<ScrollArea className="flex-1">
|
<ScrollArea className="flex-1">
|
||||||
<div className="p-2">
|
<div className="p-2">
|
||||||
{isLoading ? (
|
{isLoading ? (
|
||||||
|
|
@ -160,19 +231,21 @@ export function AllChatsSidebar({ open, onOpenChange, searchSpaceId }: AllChatsS
|
||||||
<div className="text-center py-8 text-sm text-destructive">
|
<div className="text-center py-8 text-sm text-destructive">
|
||||||
{t("error_loading_chats") || "Error loading chats"}
|
{t("error_loading_chats") || "Error loading chats"}
|
||||||
</div>
|
</div>
|
||||||
) : chats.length > 0 ? (
|
) : threads.length > 0 ? (
|
||||||
<div className="space-y-1">
|
<div className="space-y-1">
|
||||||
{chats.map((chat) => {
|
{threads.map((thread) => {
|
||||||
const isDeleting = deletingChatId === chat.id;
|
const isDeleting = deletingThreadId === thread.id;
|
||||||
|
const isArchiving = archivingThreadId === thread.id;
|
||||||
|
const isBusy = isDeleting || isArchiving;
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div
|
<div
|
||||||
key={chat.id}
|
key={thread.id}
|
||||||
className={cn(
|
className={cn(
|
||||||
"group flex items-center gap-2 rounded-md px-2 py-1.5 text-sm",
|
"group flex items-center gap-2 rounded-md px-2 py-1.5 text-sm",
|
||||||
"hover:bg-accent hover:text-accent-foreground",
|
"hover:bg-accent hover:text-accent-foreground",
|
||||||
"transition-colors cursor-pointer",
|
"transition-colors cursor-pointer",
|
||||||
isDeleting && "opacity-50 pointer-events-none"
|
isBusy && "opacity-50 pointer-events-none"
|
||||||
)}
|
)}
|
||||||
>
|
>
|
||||||
{/* Main clickable area for navigation */}
|
{/* Main clickable area for navigation */}
|
||||||
|
|
@ -180,23 +253,23 @@ export function AllChatsSidebar({ open, onOpenChange, searchSpaceId }: AllChatsS
|
||||||
<TooltipTrigger asChild>
|
<TooltipTrigger asChild>
|
||||||
<button
|
<button
|
||||||
type="button"
|
type="button"
|
||||||
onClick={() => handleChatClick(chat.id, chat.search_space_id)}
|
onClick={() => handleThreadClick(thread.id)}
|
||||||
disabled={isDeleting}
|
disabled={isBusy}
|
||||||
className="flex items-center gap-2 flex-1 min-w-0 text-left"
|
className="flex items-center gap-2 flex-1 min-w-0 text-left"
|
||||||
>
|
>
|
||||||
<MessageCircleMore className="h-4 w-4 shrink-0 text-muted-foreground" />
|
<MessageCircleMore className="h-4 w-4 shrink-0 text-muted-foreground" />
|
||||||
<span className="truncate">{chat.title}</span>
|
<span className="truncate">{thread.title || "New Chat"}</span>
|
||||||
</button>
|
</button>
|
||||||
</TooltipTrigger>
|
</TooltipTrigger>
|
||||||
<TooltipContent side="right">
|
<TooltipContent side="right">
|
||||||
<p>
|
<p>
|
||||||
{t("created") || "Created"}:{" "}
|
{t("updated") || "Updated"}:{" "}
|
||||||
{format(new Date(chat.created_at), "MMM d, yyyy 'at' h:mm a")}
|
{format(new Date(thread.updatedAt), "MMM d, yyyy 'at' h:mm a")}
|
||||||
</p>
|
</p>
|
||||||
</TooltipContent>
|
</TooltipContent>
|
||||||
</Tooltip>
|
</Tooltip>
|
||||||
|
|
||||||
{/* Actions dropdown - separate from main click area */}
|
{/* Actions dropdown */}
|
||||||
<DropdownMenu>
|
<DropdownMenu>
|
||||||
<DropdownMenuTrigger asChild>
|
<DropdownMenuTrigger asChild>
|
||||||
<Button
|
<Button
|
||||||
|
|
@ -207,7 +280,7 @@ export function AllChatsSidebar({ open, onOpenChange, searchSpaceId }: AllChatsS
|
||||||
"opacity-0 group-hover:opacity-100 focus:opacity-100",
|
"opacity-0 group-hover:opacity-100 focus:opacity-100",
|
||||||
"transition-opacity"
|
"transition-opacity"
|
||||||
)}
|
)}
|
||||||
disabled={isDeleting}
|
disabled={isBusy}
|
||||||
>
|
>
|
||||||
{isDeleting ? (
|
{isDeleting ? (
|
||||||
<Loader2 className="h-3.5 w-3.5 animate-spin" />
|
<Loader2 className="h-3.5 w-3.5 animate-spin" />
|
||||||
|
|
@ -219,7 +292,24 @@ export function AllChatsSidebar({ open, onOpenChange, searchSpaceId }: AllChatsS
|
||||||
</DropdownMenuTrigger>
|
</DropdownMenuTrigger>
|
||||||
<DropdownMenuContent align="end" className="w-40">
|
<DropdownMenuContent align="end" className="w-40">
|
||||||
<DropdownMenuItem
|
<DropdownMenuItem
|
||||||
onClick={() => handleDeleteChat(chat.id)}
|
onClick={() => handleToggleArchive(thread.id, thread.archived)}
|
||||||
|
disabled={isArchiving}
|
||||||
|
>
|
||||||
|
{thread.archived ? (
|
||||||
|
<>
|
||||||
|
<RotateCcwIcon className="mr-2 h-4 w-4" />
|
||||||
|
<span>{t("unarchive") || "Restore"}</span>
|
||||||
|
</>
|
||||||
|
) : (
|
||||||
|
<>
|
||||||
|
<ArchiveIcon className="mr-2 h-4 w-4" />
|
||||||
|
<span>{t("archive") || "Archive"}</span>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
</DropdownMenuItem>
|
||||||
|
<DropdownMenuSeparator />
|
||||||
|
<DropdownMenuItem
|
||||||
|
onClick={() => handleDeleteThread(thread.id)}
|
||||||
className="text-destructive focus:text-destructive"
|
className="text-destructive focus:text-destructive"
|
||||||
>
|
>
|
||||||
<Trash2 className="mr-2 h-4 w-4" />
|
<Trash2 className="mr-2 h-4 w-4" />
|
||||||
|
|
@ -244,10 +334,16 @@ export function AllChatsSidebar({ open, onOpenChange, searchSpaceId }: AllChatsS
|
||||||
) : (
|
) : (
|
||||||
<div className="text-center py-8">
|
<div className="text-center py-8">
|
||||||
<MessageCircleMore className="h-12 w-12 mx-auto text-muted-foreground/50 mb-3" />
|
<MessageCircleMore className="h-12 w-12 mx-auto text-muted-foreground/50 mb-3" />
|
||||||
<p className="text-sm text-muted-foreground">{t("no_chats") || "No chats yet"}</p>
|
<p className="text-sm text-muted-foreground">
|
||||||
<p className="text-xs text-muted-foreground/70 mt-1">
|
{showArchived
|
||||||
{t("start_new_chat_hint") || "Start a new chat from the researcher"}
|
? t("no_archived_chats") || "No archived chats"
|
||||||
|
: t("no_chats") || "No chats yet"}
|
||||||
</p>
|
</p>
|
||||||
|
{!showArchived && (
|
||||||
|
<p className="text-xs text-muted-foreground/70 mt-1">
|
||||||
|
{t("start_new_chat_hint") || "Start a new chat from the chat page"}
|
||||||
|
</p>
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
|
||||||
|
|
@ -25,15 +25,7 @@ function formatTime(seconds: number): string {
|
||||||
return `${mins}:${secs.toString().padStart(2, "0")}`;
|
return `${mins}:${secs.toString().padStart(2, "0")}`;
|
||||||
}
|
}
|
||||||
|
|
||||||
export function Audio({
|
export function Audio({ id, src, title, description, artwork, durationMs, className }: AudioProps) {
|
||||||
id,
|
|
||||||
src,
|
|
||||||
title,
|
|
||||||
description,
|
|
||||||
artwork,
|
|
||||||
durationMs,
|
|
||||||
className,
|
|
||||||
}: AudioProps) {
|
|
||||||
const audioRef = useRef<HTMLAudioElement>(null);
|
const audioRef = useRef<HTMLAudioElement>(null);
|
||||||
const [isPlaying, setIsPlaying] = useState(false);
|
const [isPlaying, setIsPlaying] = useState(false);
|
||||||
const [currentTime, setCurrentTime] = useState(0);
|
const [currentTime, setCurrentTime] = useState(0);
|
||||||
|
|
@ -158,7 +150,7 @@ export function Audio({
|
||||||
<div
|
<div
|
||||||
className={cn(
|
className={cn(
|
||||||
"flex items-center gap-4 rounded-xl border border-destructive/20 bg-destructive/5 p-4",
|
"flex items-center gap-4 rounded-xl border border-destructive/20 bg-destructive/5 p-4",
|
||||||
className,
|
className
|
||||||
)}
|
)}
|
||||||
>
|
>
|
||||||
<div className="flex size-16 items-center justify-center rounded-lg bg-destructive/10">
|
<div className="flex size-16 items-center justify-center rounded-lg bg-destructive/10">
|
||||||
|
|
@ -177,7 +169,7 @@ export function Audio({
|
||||||
id={id}
|
id={id}
|
||||||
className={cn(
|
className={cn(
|
||||||
"group relative overflow-hidden rounded-xl border bg-gradient-to-br from-background to-muted/30 p-4 shadow-sm transition-all hover:shadow-md",
|
"group relative overflow-hidden rounded-xl border bg-gradient-to-br from-background to-muted/30 p-4 shadow-sm transition-all hover:shadow-md",
|
||||||
className,
|
className
|
||||||
)}
|
)}
|
||||||
>
|
>
|
||||||
{/* Hidden audio element */}
|
{/* Hidden audio element */}
|
||||||
|
|
@ -190,13 +182,7 @@ export function Audio({
|
||||||
<div className="relative shrink-0">
|
<div className="relative shrink-0">
|
||||||
<div className="relative size-20 overflow-hidden rounded-lg bg-gradient-to-br from-primary/20 to-primary/5 shadow-inner">
|
<div className="relative size-20 overflow-hidden rounded-lg bg-gradient-to-br from-primary/20 to-primary/5 shadow-inner">
|
||||||
{artwork ? (
|
{artwork ? (
|
||||||
<Image
|
<Image src={artwork} alt={title} fill className="object-cover" unoptimized />
|
||||||
src={artwork}
|
|
||||||
alt={title}
|
|
||||||
fill
|
|
||||||
className="object-cover"
|
|
||||||
unoptimized
|
|
||||||
/>
|
|
||||||
) : (
|
) : (
|
||||||
<div className="flex size-full items-center justify-center">
|
<div className="flex size-full items-center justify-center">
|
||||||
<Volume2Icon className="size-8 text-primary/50" />
|
<Volume2Icon className="size-8 text-primary/50" />
|
||||||
|
|
@ -224,9 +210,7 @@ export function Audio({
|
||||||
<div className="min-w-0">
|
<div className="min-w-0">
|
||||||
<h3 className="truncate font-semibold text-foreground">{title}</h3>
|
<h3 className="truncate font-semibold text-foreground">{title}</h3>
|
||||||
{description && (
|
{description && (
|
||||||
<p className="mt-0.5 line-clamp-1 text-muted-foreground text-sm">
|
<p className="mt-0.5 line-clamp-1 text-muted-foreground text-sm">{description}</p>
|
||||||
{description}
|
|
||||||
</p>
|
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
|
@ -271,17 +255,8 @@ export function Audio({
|
||||||
|
|
||||||
{/* Volume control */}
|
{/* Volume control */}
|
||||||
<div className="flex items-center gap-2">
|
<div className="flex items-center gap-2">
|
||||||
<Button
|
<Button variant="ghost" size="icon" onClick={toggleMute} className="size-8">
|
||||||
variant="ghost"
|
{isMuted ? <VolumeXIcon className="size-4" /> : <Volume2Icon className="size-4" />}
|
||||||
size="icon"
|
|
||||||
onClick={toggleMute}
|
|
||||||
className="size-8"
|
|
||||||
>
|
|
||||||
{isMuted ? (
|
|
||||||
<VolumeXIcon className="size-4" />
|
|
||||||
) : (
|
|
||||||
<Volume2Icon className="size-4" />
|
|
||||||
)}
|
|
||||||
</Button>
|
</Button>
|
||||||
<Slider
|
<Slider
|
||||||
value={[isMuted ? 0 : volume]}
|
value={[isMuted ? 0 : volume]}
|
||||||
|
|
@ -294,12 +269,7 @@ export function Audio({
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{/* Download button */}
|
{/* Download button */}
|
||||||
<Button
|
<Button variant="outline" size="sm" onClick={handleDownload} className="gap-2">
|
||||||
variant="outline"
|
|
||||||
size="sm"
|
|
||||||
onClick={handleDownload}
|
|
||||||
className="gap-2"
|
|
||||||
>
|
|
||||||
<DownloadIcon className="size-4" />
|
<DownloadIcon className="size-4" />
|
||||||
Download
|
Download
|
||||||
</Button>
|
</Button>
|
||||||
|
|
@ -307,4 +277,3 @@ export function Audio({
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -4,13 +4,10 @@ import { makeAssistantToolUI } from "@assistant-ui/react";
|
||||||
import { AlertCircleIcon, Loader2Icon, MicIcon } from "lucide-react";
|
import { AlertCircleIcon, Loader2Icon, MicIcon } from "lucide-react";
|
||||||
import { useCallback, useEffect, useRef, useState } from "react";
|
import { useCallback, useEffect, useRef, useState } from "react";
|
||||||
import { Audio } from "@/components/tool-ui/audio";
|
import { Audio } from "@/components/tool-ui/audio";
|
||||||
|
import type { PodcastTranscriptEntry } from "@/contracts/types/podcast.types";
|
||||||
import { baseApiService } from "@/lib/apis/base-api.service";
|
import { baseApiService } from "@/lib/apis/base-api.service";
|
||||||
import { podcastsApiService } from "@/lib/apis/podcasts-api.service";
|
import { podcastsApiService } from "@/lib/apis/podcasts-api.service";
|
||||||
import {
|
import { clearActivePodcastTaskId, setActivePodcastTaskId } from "@/lib/chat/podcast-state";
|
||||||
clearActivePodcastTaskId,
|
|
||||||
setActivePodcastTaskId,
|
|
||||||
} from "@/lib/chat/podcast-state";
|
|
||||||
import type { PodcastTranscriptEntry } from "@/contracts/types/podcast.types";
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Type definitions for the generate_podcast tool
|
* Type definitions for the generate_podcast tool
|
||||||
|
|
@ -223,9 +220,7 @@ function PodcastPlayer({
|
||||||
<div className="mt-3 space-y-3 max-h-96 overflow-y-auto">
|
<div className="mt-3 space-y-3 max-h-96 overflow-y-auto">
|
||||||
{transcript.map((entry, idx) => (
|
{transcript.map((entry, idx) => (
|
||||||
<div key={`${idx}-${entry.speaker_id}`} className="text-sm">
|
<div key={`${idx}-${entry.speaker_id}`} className="text-sm">
|
||||||
<span className="font-medium text-primary">
|
<span className="font-medium text-primary">Speaker {entry.speaker_id + 1}:</span>{" "}
|
||||||
Speaker {entry.speaker_id + 1}:
|
|
||||||
</span>{" "}
|
|
||||||
<span className="text-muted-foreground">{entry.dialog}</span>
|
<span className="text-muted-foreground">{entry.dialog}</span>
|
||||||
</div>
|
</div>
|
||||||
))}
|
))}
|
||||||
|
|
@ -239,13 +234,7 @@ function PodcastPlayer({
|
||||||
/**
|
/**
|
||||||
* Polling component that checks task status and shows player when complete
|
* Polling component that checks task status and shows player when complete
|
||||||
*/
|
*/
|
||||||
function PodcastTaskPoller({
|
function PodcastTaskPoller({ taskId, title }: { taskId: string; title: string }) {
|
||||||
taskId,
|
|
||||||
title,
|
|
||||||
}: {
|
|
||||||
taskId: string;
|
|
||||||
title: string;
|
|
||||||
}) {
|
|
||||||
const [taskStatus, setTaskStatus] = useState<TaskStatusResponse>({ status: "processing" });
|
const [taskStatus, setTaskStatus] = useState<TaskStatusResponse>({ status: "processing" });
|
||||||
const pollingRef = useRef<NodeJS.Timeout | null>(null);
|
const pollingRef = useRef<NodeJS.Timeout | null>(null);
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -8,4 +8,3 @@
|
||||||
|
|
||||||
export { Audio } from "./audio";
|
export { Audio } from "./audio";
|
||||||
export { GeneratePodcastToolUI } from "./generate-podcast";
|
export { GeneratePodcastToolUI } from "./generate-podcast";
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -71,4 +71,3 @@ export function looksLikePodcastRequest(message: string): boolean {
|
||||||
|
|
||||||
return podcastPatterns.some((pattern) => pattern.test(message));
|
return podcastPatterns.some((pattern) => pattern.test(message));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -51,11 +51,26 @@ export interface ThreadHistoryLoadResponse {
|
||||||
* Fetch list of threads for a search space
|
* Fetch list of threads for a search space
|
||||||
*/
|
*/
|
||||||
export async function fetchThreads(
|
export async function fetchThreads(
|
||||||
searchSpaceId: number
|
searchSpaceId: number,
|
||||||
|
limit?: number
|
||||||
): Promise<ThreadListResponse> {
|
): Promise<ThreadListResponse> {
|
||||||
return baseApiService.get<ThreadListResponse>(
|
const params = new URLSearchParams({ search_space_id: String(searchSpaceId) });
|
||||||
`/api/v1/threads?search_space_id=${searchSpaceId}`
|
if (limit) params.append("limit", String(limit));
|
||||||
);
|
return baseApiService.get<ThreadListResponse>(`/api/v1/threads?${params}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Search threads by title
|
||||||
|
*/
|
||||||
|
export async function searchThreads(
|
||||||
|
searchSpaceId: number,
|
||||||
|
title: string
|
||||||
|
): Promise<ThreadListItem[]> {
|
||||||
|
const params = new URLSearchParams({
|
||||||
|
search_space_id: String(searchSpaceId),
|
||||||
|
title,
|
||||||
|
});
|
||||||
|
return baseApiService.get<ThreadListItem[]>(`/api/v1/threads/search?${params}`);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
@ -77,12 +92,8 @@ export async function createThread(
|
||||||
/**
|
/**
|
||||||
* Get thread messages
|
* Get thread messages
|
||||||
*/
|
*/
|
||||||
export async function getThreadMessages(
|
export async function getThreadMessages(threadId: number): Promise<ThreadHistoryLoadResponse> {
|
||||||
threadId: number
|
return baseApiService.get<ThreadHistoryLoadResponse>(`/api/v1/threads/${threadId}`);
|
||||||
): Promise<ThreadHistoryLoadResponse> {
|
|
||||||
return baseApiService.get<ThreadHistoryLoadResponse>(
|
|
||||||
`/api/v1/threads/${threadId}`
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
@ -92,11 +103,9 @@ export async function appendMessage(
|
||||||
threadId: number,
|
threadId: number,
|
||||||
message: { role: "user" | "assistant" | "system"; content: unknown }
|
message: { role: "user" | "assistant" | "system"; content: unknown }
|
||||||
): Promise<MessageRecord> {
|
): Promise<MessageRecord> {
|
||||||
return baseApiService.post<MessageRecord>(
|
return baseApiService.post<MessageRecord>(`/api/v1/threads/${threadId}/messages`, undefined, {
|
||||||
`/api/v1/threads/${threadId}/messages`,
|
body: message,
|
||||||
undefined,
|
});
|
||||||
{ body: message }
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
@ -106,11 +115,9 @@ export async function updateThread(
|
||||||
threadId: number,
|
threadId: number,
|
||||||
updates: { title?: string; archived?: boolean }
|
updates: { title?: string; archived?: boolean }
|
||||||
): Promise<ThreadRecord> {
|
): Promise<ThreadRecord> {
|
||||||
return baseApiService.put<ThreadRecord>(
|
return baseApiService.put<ThreadRecord>(`/api/v1/threads/${threadId}`, undefined, {
|
||||||
`/api/v1/threads/${threadId}`,
|
body: updates,
|
||||||
undefined,
|
});
|
||||||
{ body: updates }
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
@ -159,8 +166,7 @@ export function createThreadListManager(config: ThreadListAdapterConfig) {
|
||||||
threads: [],
|
threads: [],
|
||||||
archivedThreads: [],
|
archivedThreads: [],
|
||||||
isLoading: false,
|
isLoading: false,
|
||||||
error:
|
error: error instanceof Error ? error.message : "Failed to load threads",
|
||||||
error instanceof Error ? error.message : "Failed to load threads",
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|
|
||||||
|
|
@ -669,7 +669,13 @@
|
||||||
"more_options": "More options",
|
"more_options": "More options",
|
||||||
"clear_search": "Clear search",
|
"clear_search": "Clear search",
|
||||||
"view_all_notes": "View all notes",
|
"view_all_notes": "View all notes",
|
||||||
"add_note": "Add note"
|
"add_note": "Add note",
|
||||||
|
"archive": "Archive",
|
||||||
|
"unarchive": "Restore",
|
||||||
|
"chat_archived": "Chat archived",
|
||||||
|
"chat_unarchived": "Chat restored",
|
||||||
|
"no_archived_chats": "No archived chats",
|
||||||
|
"error_archiving_chat": "Failed to archive chat"
|
||||||
},
|
},
|
||||||
"errors": {
|
"errors": {
|
||||||
"something_went_wrong": "Something went wrong",
|
"something_went_wrong": "Something went wrong",
|
||||||
|
|
|
||||||
|
|
@ -33,4 +33,3 @@
|
||||||
],
|
],
|
||||||
"exclude": ["node_modules"]
|
"exclude": ["node_modules"]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue