Update backend_utils.py

backend_utils.py

@@ -242,7 +242,7 @@ def generate_api_usage_patterns(generative_model, tokenizer, model_input, num_be
 
 def add_params(api_usage_patterns, db_params, library_id):
     patterns_cp = api_usage_patterns.copy()
-    valid = True
+    
     processed_sequences = []
     for sequence in patterns_cp:
         sequence_list = sequence.split()
@@ -251,10 +251,15 @@ def add_params(api_usage_patterns, db_params, library_id):
             continue
 
         temp_list = []
+        ref_obj = ''
+        valid = True
         for api in sequence_list:
             temp_db = db_params[(db_params.id==library_id) & (db_params.methods==api.split(".")[-1])]
-
-            if len(temp_db) > 0:
+            
+            if ref_obj == '':
+                ref_obj = api.split(".")[0]
+            
+            if len(temp_db) > 0 and ref_obj == api.split(".")[0]:
                 param = temp_db.iloc[0].params
                 new_api = api + param
                 temp_list.append(new_api)
@@ -264,8 +269,7 @@ def add_params(api_usage_patterns, db_params, library_id):
         
         if valid:
             processed_sequences.append("[API-SEP]".join(temp_list))
-        else:
-            valid = True
+
     return processed_sequences