From 3fed713ae6499d6cf71475ec5c65f74de6217c09 Mon Sep 17 00:00:00 2001
From: Anthony <brnaba@amazon.fr>
Date: Tue, 23 Jul 2024 15:42:07 +0200
Subject: [PATCH] Fixed Classifier test

---
 tests/Classifier.test.ts | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/Classifier.test.ts b/tests/Classifier.test.ts
index e66422c..a5d766e 100644
--- a/tests/Classifier.test.ts
+++ b/tests/Classifier.test.ts
@@ -45,16 +45,16 @@ describe("Classifier", () => {
 
     test("Classify test output", async () => {
 
-        let response = await orchestrator.classifyIntent("What is aws lambda", "userId", "sessionId");    
+        let response = await orchestrator.classifier.classify("What is aws lambda", []);    
         expect(response.selectedAgent?.name).toBe('Tech Agent');
 
-        response = await orchestrator.classifyIntent("What is an aspirin", "userId", "sessionId");    
+        response = await orchestrator.classifier.classify("What is an aspirin", []);    
         expect(response.selectedAgent?.name).toBe('Health Agent');
 
-        response = await orchestrator.classifyIntent("book a flight", "userId", "sessionId");   
+        response = await orchestrator.classifier.classify("book a flight", []);   
         expect(response.selectedAgent?.name).toBe('AirlinesBot'); 
 
-        response = await orchestrator.classifyIntent("3+5", "userId", "sessionId");   
+        response = await orchestrator.classifier.classify("3+5", []);   
         expect(response.selectedAgent?.name).toBe('Math Agent'); 
       }, 15000);