rosemariafontana commited on
Commit
d54c79c
·
verified ·
1 Parent(s): aef1f09

Update process_data.py

Browse files
Files changed (1) hide show
  1. process_data.py +7 -3
process_data.py CHANGED
@@ -480,6 +480,10 @@ def pre_processing(processed_data):
480
 
481
  print("Preprocessing list")
482
  print(pre_processing_list)
 
 
 
 
483
 
484
  if processed_data["inputstyle"] == "individual-pieces-input-text":
485
  print("Stepwise Creation")
@@ -497,7 +501,7 @@ def pre_processing(processed_data):
497
  print(parameter_value)
498
  print(text_body)
499
  response = client.chat.completions.create(
500
- model=processed_data["parameters"]["pre_process_model_version"],
501
  messages=[
502
  {"role": "system", "content": pre_process_task},
503
  {"role": "user", "content": text_body}
@@ -528,7 +532,7 @@ def pre_processing(processed_data):
528
  print(processed_data["model_version"])
529
 
530
  response = client.chat.completions.create(
531
- model=processed_data["model_version"],
532
  messages=[
533
  {"role": "system", "content": pre_prompt},
534
  {"role": "user", "content": input_text}
@@ -670,7 +674,7 @@ def parse_survey_stack(data):
670
  processed_data["parameters"]["prompt_style"] = data[0]['data']['promptstyle']['value'] # this is a list
671
 
672
  if processed_data["parameters"]["pre_process"] == "yes":
673
- processed_data["parameters"]["pre_process_model_version"] = data[0]['data']['preprocessmodelversion']['value']
674
  processed_data["parameters"]["multiple_pre_prompts"] = data[0]['data']['multiplepreprompts']['value'] # this is a value in string words
675
  processed_data["parameters"]["preprompt_style"] = data[0]['data']['prepromptstyle']['value'] # this is a list
676
  processed_data["parameters"]["pre_prompt_first"] = data[0]['data']['preprocessingprompt1']['value']
 
480
 
481
  print("Preprocessing list")
482
  print(pre_processing_list)
483
+
484
+ print("Model Version")
485
+ model_version = processed_data["parameters"]["pre_process_model_version"]
486
+ print(model_version)
487
 
488
  if processed_data["inputstyle"] == "individual-pieces-input-text":
489
  print("Stepwise Creation")
 
501
  print(parameter_value)
502
  print(text_body)
503
  response = client.chat.completions.create(
504
+ model=model_version,
505
  messages=[
506
  {"role": "system", "content": pre_process_task},
507
  {"role": "user", "content": text_body}
 
532
  print(processed_data["model_version"])
533
 
534
  response = client.chat.completions.create(
535
+ model=model_version,
536
  messages=[
537
  {"role": "system", "content": pre_prompt},
538
  {"role": "user", "content": input_text}
 
674
  processed_data["parameters"]["prompt_style"] = data[0]['data']['promptstyle']['value'] # this is a list
675
 
676
  if processed_data["parameters"]["pre_process"] == "yes":
677
+ processed_data["parameters"]["pre_process_model_version"] = data[0]['data']['preprocessmodelversion']['value'][0]
678
  processed_data["parameters"]["multiple_pre_prompts"] = data[0]['data']['multiplepreprompts']['value'] # this is a value in string words
679
  processed_data["parameters"]["preprompt_style"] = data[0]['data']['prepromptstyle']['value'] # this is a list
680
  processed_data["parameters"]["pre_prompt_first"] = data[0]['data']['preprocessingprompt1']['value']