bug fix

amazon-science · Sep 18, 2024 · 288078e · 288078e
1 parent 5f9bc39
commit 288078e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/refchecker/utils.py b/refchecker/utils.py
@@ -98,8 +98,8 @@ def get_model_batch_response(
     if not prompts or len(prompts) == 0:
         raise ValueError("Invalid input.")
 
-    if custom_func is not None:
-        return custom_func(prompts)
+    if custom_llm_api_func is not None:
+        return custom_llm_api_func(prompts)
     else:
         message_list = []
         for prompt in prompts: