Add toolCalls to InferenceChatCompletionEvent in LLM simulator

elastic · Dec 9, 2024 · 2355ca4 · 2355ca4
1 parent 85e4cfb
commit 2355ca4
Showing 1 changed file with 1 addition and 0 deletions.
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
@@ -63,6 +63,7 @@ function createLlmSimulator(subscriber: any) {
       subscriber.next({
         type: InferenceChatCompletionEventType.ChatCompletionMessage,
         content: msg.content,
+        toolCalls: msg.function_call ? [{ function: msg.function_call }] : [],
       });
     },
     tokenCount: async ({