update tests

elastic · Jan 8, 2025 · 990b741 · 990b741
1 parent f14fcc2
commit 990b741
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/...lugins/elastic_assistant/server/ai_assistant_data_clients/knowledge_base/helpers.test.tsx b/...lugins/elastic_assistant/server/ai_assistant_data_clients/knowledge_base/helpers.test.tsx
@@ -180,15 +180,8 @@ describe('getStructuredToolForIndexEntry', () => {
               field1: 'value1',
               field2: 2,
             },
-            inner_hits: {
-              'test.test': {
-                hits: {
-                  hits: [
-                    { _source: { text: 'Inner text 1' } },
-                    { _source: { text: 'Inner text 2' } },
-                  ],
-                },
-              },
+            highlight: {
+              test: ['Inner text 1', 'Inner text 2'],
             },
           },
         ],

diff --git a/...rity/plugins/elastic_assistant/server/ai_assistant_data_clients/knowledge_base/helpers.ts b/...rity/plugins/elastic_assistant/server/ai_assistant_data_clients/knowledge_base/helpers.ts
@@ -214,7 +214,7 @@ export const getStructuredToolForIndexEntry = ({
           }
 
           return {
-            text: hit.highlight?.[indexEntry.field].join(' '),
+            text: hit.highlight?.[indexEntry.field].join('\n --- \n'),
           };
         });
-Original file line number
+Diff line change
@@ Expand Up / @@ -214,7 +214,7 @@ export const getStructuredToolForIndexEntry = ({ @@
               }
               return {
-                text: hit.highlight?.[indexEntry.field].join(' '),
+                text: hit.highlight?.[indexEntry.field].join('\n --- \n'),
               };
             });
@@ Expand Down @@