Add argument matching

src/server.js

@@ -72,28 +72,45 @@ async function classifyIntent(text) {
 
   const INTENTS = await getIntents();
 
-  let best = {
+  let bestIntent = {
     name: "unknown",
     confidence: 0.5,
     matched: null,
+    argument: null
   };
 
   for (const intent of INTENTS) {
     for (const example of intent.examples) {
       const exampleEmbedding = await embed(example);
-      const score = cosineSimilarity(inputEmbedding, exampleEmbedding);
+      const intentScore = cosineSimilarity(inputEmbedding, exampleEmbedding);
 
-      if (score > best.confidence) {
+      if (intentScore > best.confidence) {
-        best = {
+        bestIntent = {
           name: intent.name,
-          confidence: score,
+          confidence: intentScore,
           matched: example,
+          argument: intent.arguments
         };
       }
     }
   }
 
-  return best;
+  let bestArgumentScore = 0.5
+
+  if (best.arguments) {
+    for (const argument of best.arguments) {
+      const argumentEmbedding = await embed(argument);
+      const argumentScore = cosineSimilarity(inputEmbedding, argumentEmbedding);
+
+      if (argumentScore > bestArgumentScore) {
+        bestArgumentScore = argumentScore;
+
+        bestIntent.argument = argument
+      }
+    }
+  }
+
+  return bestIntent;
 }
 
 async function handleIntent(streamKey, msg) {