Sage
commited on
Commit
•
3be8992
1
Parent(s):
c826037
test
Browse files
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,
|
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,
|
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...')
|