fix: language and footnote

pingcap · May 22, 2024 · 82a211e · 82a211e
1 parent d7d738f
commit 82a211e
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 10 deletions.
diff --git a/src/components/remark-content.tsx b/src/components/remark-content.tsx
@@ -30,19 +30,33 @@ const production: RehypeReactOptions = {
       return (
         <HoverCard openDelay={0} onOpenChange={open => {
           if (open) {
-            const a = document.querySelector(`${props.href} a:first-child`) as HTMLAnchorElement | null;
-            if (a) {
-              setLink({ title: a.textContent ?? a.href, href: a.href });
-            } else {
-              setLink(undefined);
+            const id = props.href?.replace(/^#/, '');
+            if (id) {
+              const li = document.getElementById(id);
+              if (li) {
+                const a = li.querySelector(`a:first-child`) as HTMLAnchorElement | null;
+                if (a) {
+                  setLink({ title: a.textContent ?? a.href, href: a.href });
+                  return;
+                }
+              }
             }
+            setLink(undefined);
           }
         }}>
           <HoverCardTrigger asChild>
-            <a {...props} />
+            <a
+              {...props}
+              className={cn(props.className, 'cursor-default')}
+              href={undefined}
+              onClick={event => {
+                event.preventDefault();
+                event.stopPropagation();
+              }}
+            />
           </HoverCardTrigger>
           <HoverCardPortal>
-            <HoverCardContent className="p-1 w-[200px] overflow-hidden rounded-lg border text-xs">
+            <HoverCardContent onPointerDownOutside={e => e.preventDefault()} className="p-1 w-[200px] overflow-hidden rounded-lg border text-xs">
               <HoverCardArrow className="fill-border" />
               <MessageContextSourceCard title={link?.title} href={link?.href} />
             </HoverCardContent>

diff --git a/src/core/services/llamaindex/chating.ts b/src/core/services/llamaindex/chating.ts
@@ -279,10 +279,19 @@ export class LlamaindexChatService extends AppChatService {
       trace,
     ));
 
+    const promptContext = {
+      ...kgContext,
+      sources: Array.from(allSources.values()).map((source, index) => ({
+        ordinal: index + 1,
+        ...source
+      })),
+      originalQuestion: options.userInput,
+    }
+
     // Build Query Engine.
     const { textQa, refine } = prompts;
-    const textQaPrompt = this.getPrompt(textQa, defaultTextQaPrompt, kgContext);
-    const refinePrompt = this.getPrompt(refine, defaultRefinePrompt, kgContext);
+    const textQaPrompt = this.getPrompt(textQa, defaultTextQaPrompt, promptContext);
+    const refinePrompt = this.getPrompt(refine, defaultRefinePrompt, promptContext);
     const responseBuilder = new CompactAndRefine(serviceContext, textQaPrompt, refinePrompt);
     const responseSynthesizer = new ResponseSynthesizer({
       serviceContext,
@@ -297,7 +306,7 @@ export class LlamaindexChatService extends AppChatService {
       content: message.content,
       additionalKwargs: {},
     }));
-    const condenseMessagePrompt = this.getPrompt(prompts?.condenseQuestion, defaultCondenseQuestionPrompt, kgContext);
+    const condenseMessagePrompt = this.getPrompt(prompts?.condenseQuestion, defaultCondenseQuestionPrompt, promptContext);
     const chatEngine = new CondenseQuestionChatEngine({
       serviceContext,
       queryEngine,