implement dialog for selected docs

langchain-ai · Oct 17, 2024 · 950268b · 950268b
1 parent 52567c0
commit 950268b
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 58 deletions.
diff --git a/frontend/app/components/DocumentDialog.tsx b/frontend/app/components/DocumentDialog.tsx
@@ -11,20 +11,23 @@ import { TooltipIconButton } from "./ui/assistant-ui/tooltip-icon-button";
 
 interface DocumentDialogProps {
   document: Record<string, any>;
+  trigger?: React.ReactNode;
 }
 
 export function DocumentDialog(props: DocumentDialogProps) {
+  const trigger = props.trigger || (
+    <TooltipIconButton
+      tooltip={props.document.metadata.title}
+      variant="outline"
+      className="w-6 h-6 z-50 transition-colors ease-in-out bg-transparent hover:bg-gray-500 border-gray-400 text-gray-300"
+    >
+      <File />
+    </TooltipIconButton>
+  );
+
   return (
     <Dialog>
-      <DialogTrigger asChild>
-        <TooltipIconButton
-          tooltip={props.document.metadata.title}
-          variant="outline"
-          className="w-6 h-6 z-50 transition-colors ease-in-out bg-transparent hover:bg-gray-500 border-gray-400 text-gray-300"
-        >
-          <File />
-        </TooltipIconButton>
-      </DialogTrigger>
+      <DialogTrigger asChild={!props.trigger}>{trigger}</DialogTrigger>
       <DialogContent className="max-w-3xl max-h-[80vh] overflow-y-auto bg-gray-700 text-gray-200">
         <DialogHeader>
           <DialogTitle className="flex items-center justify-start gap-4">

diff --git a/frontend/app/components/SelectedDocumentsToolUI.tsx b/frontend/app/components/SelectedDocumentsToolUI.tsx
@@ -1,13 +1,8 @@
 import { useAssistantToolUI } from "@assistant-ui/react";
 import { Card, CardContent, CardHeader, CardTitle } from "./ui/card";
-import { BookOpenText, Plus, SquareArrowOutUpRight } from "lucide-react";
-import {
-  TooltipProvider,
-  TooltipTrigger,
-  TooltipContent,
-  Tooltip,
-} from "./ui/tooltip";
+import { BookOpenText, Plus } from "lucide-react";
 import { Sheet, SheetContent, SheetTrigger } from "./ui/sheet";
+import { DocumentDialog } from "./DocumentDialog";
 
 type Document = {
   page_content: string;
@@ -36,42 +31,6 @@ const DocumentCard = ({ document }: { document: Document }) => {
   );
 };
 
-const DocumentCardTooltip = ({ document }: { document: Document }) => {
-  const description =
-    document.metadata.description && document.metadata.description !== ""
-      ? document.metadata.description
-      : document.page_content.slice(0, 250);
-
-  return (
-    <TooltipProvider>
-      <Tooltip defaultOpen delayDuration={0}>
-        <TooltipTrigger asChild>
-          <DocumentCard document={document} />
-        </TooltipTrigger>
-        <TooltipContent className="flex flex-col max-w-[600px] whitespace-pre-wrap">
-          <div className="flex flex-col gap-1">
-            <p className="font-medium text-gray-300">
-              {document.metadata.title}
-            </p>
-            <div className="flex flex-wrap justify-start">
-              <a
-                href={document.metadata.source}
-                target="_blank"
-                rel="noopener noreferrer"
-                className="flex items-center text-blue-400 hover:text-blue-300 transition-colors duration-200 break-all"
-              >
-                Source{" "}
-                <SquareArrowOutUpRight className="ml-1 h-4 w-4 flex-shrink-0" />
-              </a>
-            </div>
-            <p className="text-xs font-light text-gray-400">{description}</p>
-          </div>
-        </TooltipContent>
-      </Tooltip>
-    </TooltipProvider>
-  );
-};
-
 export const useSelectedDocumentsUI = () =>
   useAssistantToolUI({
     toolName: "selected_documents",
@@ -92,9 +51,10 @@ export const useSelectedDocumentsUI = () =>
           </span>
           <div className="flex flex-wrap items-center justify-start gap-2">
             {displayedDocuments.map((document, docIndex) => (
-              <DocumentCardTooltip
-                key={`final-document-${docIndex}`}
+              <DocumentDialog
+                key={`all-documents-${docIndex}`}
                 document={document}
+                trigger={<DocumentCard document={document} />}
               />
             ))}
             {remainingDocuments.length > 0 && (
@@ -114,9 +74,10 @@ export const useSelectedDocumentsUI = () =>
                     </h2>
                     <div className="flex flex-wrap gap-2">
                       {documents.map((document, docIndex) => (
-                        <DocumentCardTooltip
+                        <DocumentDialog
                           key={`all-documents-${docIndex}`}
                           document={document}
+                          trigger={<DocumentCard document={document} />}
                         />
                       ))}
                     </div>

diff --git a/frontend/app/hooks/useThreads.tsx b/frontend/app/hooks/useThreads.tsx
@@ -117,7 +117,6 @@ export function useThreads(userId: string | undefined) {
       return newThreads;
     });
     if (id === threadId) {
-      console.log("Deleted current thread. Creating a new one.");
       clearMessages();
       // Create a new thread. Use .then to avoid blocking the UI.
       // Once completed re-fetch threads to update UI.

diff --git a/frontend/app/page.tsx b/frontend/app/page.tsx
@@ -1,6 +1,6 @@
 "use client";
 
-import React, { useEffect, useState } from "react";
+import React, { useState } from "react";
 import {
   AppendMessage,
   AssistantRuntimeProvider,
@@ -100,7 +100,7 @@ export default function ContentComposerChatInterface(): React.ReactElement {
     isRunning,
     onNew,
   });
-  console.log("messages.length", messages.length);
+
   return (
     <div className="overflow-hidden w-full flex md:flex-row flex-col relative">
       {messages.length > 0 ? (