Update process_data.py

process_data.py

@@ -324,26 +324,27 @@ def pre_processing(input_data, parameters):
         if parameters["chaining"]:
             print("Chaining")
             for text_label, text_body in input_data["input_text_pieces"].items():
-                for parameter_name, parameter_value in parameters.items():
-                    if 'data_input' in parameter_name and parameter_value and not isinstance(parameter_value, bool) and text_body:
-                        print("Text Label")
-                        print(text_label)
-                        print("Prompt followed by data entered")
-                        print(parameter_value)
-                        print(text_body)
-                        response = client.chat.completions.create(
-                                        model=parameters["model_version"],
-                                        messages=[
-                                            {"role": "system", "content": parameter_value},
-                                            {"role": "user", "content": text_body}
-                                        ]
-                                    )
-                                    
-                        response_text = response.choices[0].message.content
-                        print("Response text")
-                        print(response_text)
-                        
-                        input_data["input_text_pieces"]["pre_processed_pieces"][text_label] = response_text
+                if 'data_input' in text_label:
+                    for parameter_name, parameter_value in parameters.items():
+                        if 'pre_prompt' in parameter_name and parameter_value and not isinstance(parameter_value, bool) and text_body:
+                            print("Text Label")
+                            print(text_label)
+                            print("Prompt followed by data entered")
+                            print(parameter_value)
+                            print(text_body)
+                            response = client.chat.completions.create(
+                                            model=parameters["model_version"],
+                                            messages=[
+                                                {"role": "system", "content": parameter_value},
+                                                {"role": "user", "content": text_body}
+                                            ]
+                                        )
+                                        
+                            response_text = response.choices[0].message.content
+                            print("Response text")
+                            print(response_text)
+                            
+                            input_data["input_text_pieces"]["pre_processed_pieces"][text_label] = response_text
 
         return input_data