Remove safety_settings from all calls

vertex_api_invoice_extractor.py

@@ -84,13 +84,14 @@ def extract_invoice_numbers_with_vertex_ai_single_hop(base64_image: str) -> list
  "temperature": 0.1,
  "top_p": 0.95,
  }
-
+ 
  safety_settings = {
  generative_models.HarmCategory.HARM_CATEGORY_HATE_SPEECH: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  }
+ safety_settings = {}
 
  responses = model.generate_content(
  [image_part, text_prompt],
@@ -141,6 +142,7 @@ def extract_column_headers(base64_image: str) -> list[str]:
  generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  }
+ safety_settings = {}
 
  responses = model.generate_content(
  [image_part, text_prompt],
@@ -191,6 +193,7 @@ def extract_invoice_numbers_for_column(base64_image: str, column_name: str) -> I
  generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  }
+ safety_settings = {}
 
  responses = model.generate_content(
  [image_part, text_prompt],
@@ -355,6 +358,7 @@ def extract_invoice_numbers_from_text_with_vertex_ai_single_hop(text: str) -> li
  generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  }
+ safety_settings = {}
 
  responses = model.generate_content(
  prompt,
@@ -430,6 +434,7 @@ def extract_column_headers_from_text(text: str) -> list[str]:
  generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  }
+ safety_settings = {}
 
  response = model.generate_content(
  prompt,
@@ -474,6 +479,7 @@ def extract_invoice_numbers_for_column_from_text(text: str, column_name: str) ->
  generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  }
+ safety_settings = {}
 
  response = model.generate_content(
  prompt,
@@ -510,6 +516,7 @@ def extract_payment_amounts_with_vertex_ai(base64_image: str) -> list[PaymentAmo
  generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  }
+ safety_settings = {}
 
  responses = model.generate_content(
  [image_part, text_prompt],
@@ -553,13 +560,14 @@ def extract_payment_amounts_from_text_with_vertex_ai(text: str) -> list[PaymentA
  "temperature": 0.1,
  "top_p": 0.95,
  }
-
+ 
  safety_settings = {
  generative_models.HarmCategory.HARM_CATEGORY_HATE_SPEECH: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
  generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
- }
+ }
+ safety_settings = {}
 
  response = model.generate_content(
  prompt,