Remove categories/queries where needed

elastic · Jun 7, 2024 · 28ce486 · 28ce486
1 parent aabe6f7
commit 28ce486
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 13 deletions.
diff --git a/...lugins/observability_solution/observability_ai_assistant/public/service/create_service.ts b/...lugins/observability_solution/observability_ai_assistant/public/service/create_service.ts
@@ -72,10 +72,6 @@ export function createService({
       return of(
         createFunctionRequestMessage({
           name: 'context',
-          args: {
-            queries: [],
-            categories: [],
-          },
         }),
         createFunctionResponseMessage({
           name: 'context',

diff --git a/x-pack/plugins/observability_solution/observability_ai_assistant/server/functions/context.ts b/x-pack/plugins/observability_solution/observability_ai_assistant/server/functions/context.ts
@@ -151,7 +151,6 @@ async function retrieveSuggestions({
 }) {
   const recallResponse = await client.recall({
     queries,
-    categories: [],
   });
 
   return recallResponse.entries.map((entry) => omit(entry, 'labels', 'is_correction'));

diff --git a/...bservability_ai_assistant/server/service/client/get_context_function_request_if_needed.ts b/...bservability_ai_assistant/server/service/client/get_context_function_request_if_needed.ts
@@ -28,9 +28,5 @@ export function getContextFunctionRequestIfNeeded(
 
   return createFunctionRequestMessage({
     name: CONTEXT_FUNCTION_NAME,
-    args: {
-      queries: [],
-      categories: [],
-    },
   });
 }
diff --git a/...ins/observability_solution/observability_ai_assistant/server/service/client/index.test.ts b/...ins/observability_solution/observability_ai_assistant/server/service/client/index.test.ts
@@ -1232,7 +1232,7 @@ describe('Observability AI Assistant client', () => {
             role: MessageRole.Assistant,
             function_call: {
               name: CONTEXT_FUNCTION_NAME,
-              arguments: JSON.stringify({ queries: [], categories: [] }),
+              arguments: JSON.stringify({}),
               trigger: MessageRole.Assistant,
             },
           },
@@ -1456,7 +1456,7 @@ describe('Observability AI Assistant client', () => {
             role: MessageRole.Assistant,
             function_call: {
               name: CONTEXT_FUNCTION_NAME,
-              arguments: JSON.stringify({ queries: [], categories: [] }),
+              arguments: JSON.stringify({}),
               trigger: MessageRole.Assistant,
             },
           },

diff --git a/...ability_solution/observability_ai_assistant_app/public/components/chat/chat_body.test.tsx b/...ability_solution/observability_ai_assistant_app/public/components/chat/chat_body.test.tsx
@@ -40,7 +40,7 @@ describe('<ChatBody>', () => {
           role: 'assistant',
           function_call: {
             name: CONTEXT_FUNCTION_NAME,
-            arguments: '{"queries":[],"categories":[]}',
+            arguments: '{}',
             trigger: 'assistant',
           },
           content: '',
@@ -88,7 +88,7 @@ describe('<ChatBody>', () => {
           role: 'assistant',
           function_call: {
             name: CONTEXT_FUNCTION_NAME,
-            arguments: '{"queries":[],"categories":[]}',
+            arguments: '{}',
             trigger: 'assistant',
           },
           content: '',