Update process_data.py

process_data.py

@@ -326,6 +326,9 @@ def pre_processing(input_data, parameters):
             for text_label, text_body in input_data["input_text_pieces"].items():
                 for parameter_name, parameter_value in parameters.items():
                     if 'pre_prompt' in parameter_name and parameter_value is not None and not isinstance(parameter_value, bool):
+                        print("Prompt followed by data entered")
+                        print(parameter_value)
+                        print(text_body)
                         response = client.chat.completions.create(
                                         model=parameters["model_version"],
                                         messages=[
@@ -335,8 +338,9 @@ def pre_processing(input_data, parameters):
                                     )
                                     
                         response_text = response.choices[0].message.content
-                        text_body = response_text
-                input_data["input_text_pieces"]["pre_processed_pieces"][text_label] = response_text
+                        text_body = str(response_text.json())
+                        print(text_body)
+                input_data["input_text_pieces"]["pre_processed_pieces"][text_label] = response_text.json()
 
         return input_data