Sage commited on
Commit
3be8992
1 Parent(s): c826037
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,6 +3,7 @@ import gradio as gr
3
  import json
4
  import time
5
  from tqdm import tqdm
 
6
  import logging
7
  import requests
8
  import google
@@ -55,7 +56,7 @@ def batch_document(content, document_type, context, progress = gr.Progress()):
55
  data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
56
  combined_data.append(data)
57
  break
58
- except (openai.error.APIConnectionError, openai.error.AuthenticationError, openai.error.RateLimitError, google.api_core.exceptions.RetryError, requests.exceptions.RequestException) as e:
59
  logging.error(f'Retry {i+1} failed: {e}')
60
  if i < retries - 1:
61
  logging.error(f'Retrying in {timeout} seconds...')
@@ -76,7 +77,7 @@ def batch_document(content, document_type, context, progress = gr.Progress()):
76
  data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
77
  combined_data.append(data)
78
  break
79
- except (openai.error.APIConnectionError, openai.error.AuthenticationError, openai.error.RateLimitError, google.api_core.exceptions.RetryError, requests.exceptions.RequestException) as e:
80
  logging.error(f'Retry {i+1} failed: {e}')
81
  if i < retries - 1:
82
  logging.error(f'Retrying in {timeout} seconds...')
 
3
  import json
4
  import time
5
  from tqdm import tqdm
6
+ from azure.core.exceptions import HttpResponseError
7
  import logging
8
  import requests
9
  import google
 
56
  data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
57
  combined_data.append(data)
58
  break
59
+ except (openai.error.APIConnectionError, openai.error.AuthenticationError, openai.error.RateLimitError, HttpResponseError, requests.exceptions.RequestException) as e:
60
  logging.error(f'Retry {i+1} failed: {e}')
61
  if i < retries - 1:
62
  logging.error(f'Retrying in {timeout} seconds...')
 
77
  data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
78
  combined_data.append(data)
79
  break
80
+ except (openai.error.APIConnectionError, openai.error.AuthenticationError, openai.error.RateLimitError, HttpResponseError, requests.exceptions.RequestException) as e:
81
  logging.error(f'Retry {i+1} failed: {e}')
82
  if i < retries - 1:
83
  logging.error(f'Retrying in {timeout} seconds...')