Skip to content

Commit 4fa116b

Browse files
Fix : Prompt service error handling improvement (#35)
* Handle errors for json response * Version bump 0.18.1 -> 0.18.2
1 parent 1dd4525 commit 4fa116b

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

src/unstract/sdk/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "0.18.1"
1+
__version__ = "0.18.2"
22

33

44
def get_sdk_version():

src/unstract/sdk/prompt.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,16 @@ def _post_call(
7575
result["status"] = "OK"
7676
result["structure_output"] = response.text
7777
except RequestException as e:
78-
result["error"] = f"Error occurred: {e}"
78+
# Extract error information from the response if available
79+
error_message = str(e)
80+
content_type = response.headers.get("Content-Type", "").lower()
81+
if "application/json" in content_type:
82+
response_json = response.json()
83+
if "error" in response_json:
84+
error_message = response_json["error"]
85+
elif response.text:
86+
error_message = response.text
87+
result["error"] = error_message
7988
self.tool.stream_log(
8089
f"Error while fetching response for prompt: {result['error']}",
8190
level=LogLevel.ERROR,

0 commit comments

Comments
 (0)