Merge pull request #93 from openrewrite/add-timings-generative-model

add query to suggested method patterns table
openrewrite · Aug 5, 2024 · 52d8622 · 52d8622
2 parents 0288d29 + c452bad
commit 52d8622
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/main/java/io/moderne/ai/research/FindCodeThatResembles.java b/src/main/java/io/moderne/ai/research/FindCodeThatResembles.java
@@ -334,7 +334,8 @@ public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, Execu
                 if (resultGenerativeModel || resultEmbeddingModels == 1) {
                     suggestedMethodPatternsTable.insertRow(ctx, new SuggestedMethodPatterns.Row(
                             method.printTrimmed(getCursor()),
-                            methodPattern
+                            methodPattern,
+                            resembles
                     ));
                 }
 

diff --git a/src/main/java/io/moderne/ai/table/SuggestedMethodPatterns.java b/src/main/java/io/moderne/ai/table/SuggestedMethodPatterns.java
@@ -39,5 +39,9 @@ public static class Row {
         @Column(displayName = "Method Pattern",
                 description = "Method invocation pattern.")
         String methodPattern;
+
+        @Column(displayName = "Query",
+                description = "The natural language search query.")
+        String query;
     }
 }