Spaces:
Paused
Paused
Commit ·
5f311b3
1
Parent(s): 4e966cd
Adding error handling
Browse files- backend/query_llm.py +10 -2
backend/query_llm.py
CHANGED
|
@@ -11,9 +11,17 @@ headers = {
|
|
| 11 |
"Content-Type": "application/json"
|
| 12 |
}
|
| 13 |
|
|
|
|
| 14 |
def call_jais(payload):
|
| 15 |
-
|
| 16 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
def generate(prompt: str):
|
| 19 |
payload = {'inputs': '', 'prompt':prompt}
|
|
|
|
| 11 |
"Content-Type": "application/json"
|
| 12 |
}
|
| 13 |
|
| 14 |
+
|
| 15 |
def call_jais(payload):
|
| 16 |
+
try:
|
| 17 |
+
response = requests.post(API_URL, headers=headers, json=payload)
|
| 18 |
+
response.raise_for_status() # This will raise an exception for HTTP error codes
|
| 19 |
+
return response.json()
|
| 20 |
+
except requests.exceptions.HTTPError as http_err:
|
| 21 |
+
raise gr.Error(f"An error occurred while processing the request. {http_err}")
|
| 22 |
+
except Exception as err:
|
| 23 |
+
raise gr.Error(f"An error occurred while processing the request. {err}")
|
| 24 |
+
|
| 25 |
|
| 26 |
def generate(prompt: str):
|
| 27 |
payload = {'inputs': '', 'prompt':prompt}
|