added: fragment as a response

arakoodev · Nov 17, 2023 · ec5fd9c · ec5fd9c
1 parent 56a70cf
commit ec5fd9c
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 2 deletions.
diff --git a/JS/edgechains/examples/src/routes/hydeSearch.route.ts b/JS/edgechains/examples/src/routes/hydeSearch.route.ts
@@ -1,6 +1,6 @@
 import { Hono } from 'hono';
 import { hydeSearchAdaEmbedding } from '../service/HydeSearchService';
-
+import { HydeFragmentData } from '../types/HydeFragmentData';
 const HydeSearch = new Hono();
 
 HydeSearch.get('/hyde-search/query-rrf', (c) => {
@@ -295,7 +295,35 @@ HydeSearch.get('/search', async (c) => {
         orderRRF: query.orderRRF,
       };
     const answer = await hydeSearchAdaEmbedding(arkRequest);
-    return c.text(answer.finalAnswer);
+    const final_answer = answer.finalAnswer;
+    const responses = answer.wordEmbeddings;
+    const data: HydeFragmentData = { responses, final_answer}
+    return c.html(`
+    <html lang="en">
+    <div>
+        <div class="card card-active">
+            <div class="card-body">${data.final_answer}</div>
+        </div>
+            <ul class="list-unstyled mb-0">
+              ${data.responses.map(
+                (item) => `
+                  <li>
+                    <div class="card">
+                      <div class="card-body">
+                        ${item.rawText != null
+                          ? `<div class="card card-body">${item.rawText}</div>`
+                          : `<div class="card card-body">${item.metadata}</div>`}
+                        ${item.filename != null ? `<div class="card card-body" style="color: blue;">${item.filename}</div>` : ''}
+                        ${item.titleMetadata != null ? `<div class="card card-body" style="color: blue;">${item.titleMetadata}</div>` : ''}
+                        ${item.documentDate != null ? `<div class="card card-body" style="color: blue;">${item.documentDate}</div>` : ''}
+                      </div>
+                    </div>
+                  </li>
+                `
+              )}
+            </ul>
+  </html>
+    `);
 })
 
 export default HydeSearch;
diff --git a/JS/edgechains/examples/src/types/HydeFragmentData.ts b/JS/edgechains/examples/src/types/HydeFragmentData.ts
@@ -0,0 +1,10 @@
+export interface HydeFragmentData {
+    responses: Array<{
+      rawText?: string;
+      metadata?: string;
+      filename?: string;
+      titleMetadata?: string;
+      documentDate?: string;
+    }>;
+    final_answer?: string;
+  }