JVM: Fix bug on cross reference code retrieval (#424)

This PR fixes a bug in the wrong source code retrieval logic for function cross reference. Signed-off-by: Arthur Chan <[email protected]>
google · Jul 3, 2024 · c99f52b · c99f52b
1 parent 739d89b
commit c99f52b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/llm_toolkit/prompt_builder.py b/llm_toolkit/prompt_builder.py
@@ -724,10 +724,8 @@ def _format_source_reference(self, signature: str) -> Tuple[str, str]:
 
     # Query for source code of target method callsites
     xref_source_list = []
-    for xref in introspector.query_introspector_cross_references(
+    for xref_source in introspector.query_introspector_cross_references(
         self.benchmark.project, signature):
-      xref_source = introspector.query_introspector_function_source(
-          self.benchmark.project, xref)
       if xref_source:
         xref_source_list.append(xref_source)