Browse Source

fix llm_sensitivity typo

StrayWarrior 4 months ago
parent
commit
b283c92253
1 changed files with 4 additions and 4 deletions
  1. 4 4
      applications/llm_sensitivity.py

+ 4 - 4
applications/llm_sensitivity.py

@@ -80,7 +80,7 @@ def do_check_titles(text):
 """
     return request_llm_api(text_prompt, text)
 
-def check_titles(titles, retun_map=False):
+def check_titles(titles, return_map=False):
     n_titles = len(titles)
 
     batch_size = 20
@@ -102,9 +102,9 @@ def check_titles(titles, retun_map=False):
                 try:
                     res = do_check_titles(title)
                     json_res = json.loads(res)
-                    if isinstance(res, list):
+                    if isinstance(json_res, list):
                         json_data.append(json_res[0])
-                    elif isinstance(res, dict):
+                    elif isinstance(json_res, dict):
                         json_data.append(json_res)
                     else:
                         raise Exception('Invalid Response')
@@ -122,7 +122,7 @@ def check_titles(titles, retun_map=False):
                 item['hit_rule'] = 99
         else:
             item['hit_rule'] = 0
-    if retun_map:
+    if return_map:
         result_map = {}
         for item in json_data:
             if item['title'] not in result_map: