Update process_data.py

process_data.py

@@ -46,9 +46,6 @@ def generate_json(input_data, parameters):
             response_format=FarmActivities,
         )
 
-        if 'error' in farm_info_response:
-            raise ValueError(f"API error: {interactions_response['error']['message']}")
-            
         farm_generated_json = farm_info_response.choices[0].message.parsed
 
         
@@ -65,9 +62,6 @@ def generate_json(input_data, parameters):
             ],
             response_format=Interactions,
         )
-
-        if 'error' in interactions_response:
-            raise ValueError(f"API error: {interactions_response['error']['message']}")
             
         interactions_generated_json = interactions_response.choices[0].message.parsed
 
@@ -85,9 +79,6 @@ def generate_json(input_data, parameters):
             response_format=Trial,
         )
         
-        if 'error' in trial_response:
-            raise ValueError(f"API error: {trial_response['error']['message']}")
-        
         trial_generated_json = trial_response.choices[0].message.parsed
 
         print("TRIALS JSON: ")
@@ -175,14 +166,8 @@ def generate_json_pieces(specification, model_version, additional_json_creation_
         print(combined_json) # debugging
 
         pretty_json = combined_json.json()
-
-        if 'error' in response:
-            raise ValueError(f"API error: {response['error']['message']}")
         
         return pretty_json
-
-    except ValidationError as e:
-        return {"error": str(e)}
     except Exception as e:
         return {"error": "Failed to generate valid JSON. " + str(e)}
     
@@ -222,7 +207,7 @@ def pre_processing(input_data, parameters):
                         ]
                     )
                 
-                    response_text = response['choices'][0]['message']['content']
+                    response_text = response.choices[0].message.content
                     partially_processed_input = json.loads(response_text)
                     input_text = partially_processed_input