fix: log dalle3 images

F33RNI · Dec 21, 2023 · 6b42912 · 6b42912
1 parent 5d1eea5
commit 6b42912
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/QueueHandler.py b/QueueHandler.py
@@ -761,7 +761,15 @@ def _collect_data(self, request_response: RequestResponseContainer, log_request=
  # Images
  for image_url in request_response.response_images:
  try:
- response = base64.b64encode(requests.get(image_url, timeout=120).content).decode("utf-8")
+ response = None
+ if isinstance(image_url, str):
+ response = base64.b64encode(requests.get(image_url, timeout=120).content).decode("utf-8")
+ else:
+ img_type, img = image_url
+ if img_type == "base64":
+ response = img
+ else:
+ raise Exception(f"Unknown image type: {img_type}")
  log_file.write(response_str_to_format.format(request_response.response_timestamp,
  request_response.id,
  request_response.user["user_name"],