hivetrace 1.3.14__py3-none-any.whl → 1.3.16__py3-none-any.whl

hivetrace/handlers/response_builder.py

@@ -47,8 +47,23 @@ class ResponseBuilder:
 
         Automatically determines the response type and creates the corresponding model.
         """
+        excluded_keys = {
+            "message_id",
+            "trace_id",
+            "request_id",
+            "success",
+            "timestamp",
+            "blocked",
+        }
+
+        status = api_response.get("status")
+        status_is_success = (
+            isinstance(status, str) and "success" in status.lower()
+        )
+
         if (
             api_response.get("success")
+            or status_is_success
             or "message_id" in api_response
             or "trace_id" in api_response
         ):
@@ -59,7 +74,7 @@ class ResponseBuilder:
                 additional_data={
                     k: v
                     for k, v in api_response.items()
-                    if k not in ["message_id", "trace_id", "request_id"]
+                    if k not in excluded_keys
                 },
             )
 

hivetrace/models/responses.py

@@ -7,7 +7,7 @@ class BaseResponse(BaseModel):
     """Base response model."""
 
     class Config:
-        extra = "forbid"
+        extra = "allow"
         use_enum_values = True
 
 

{hivetrace-1.3.14.dist-info → hivetrace-1.3.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hivetrace
-Version: 1.3.14
+Version: 1.3.16
 Summary: Hivetrace SDK for monitoring LLM applications
 Home-page: http://hivetrace.ai
 Author: Raft

{hivetrace-1.3.14.dist-info → hivetrace-1.3.16.dist-info}/RECORD

@@ -31,15 +31,15 @@ hivetrace/errors/network.py,sha256=4GlFpUU9BV5Q1yoizadRQjDPLlIn7lh4Nyn8e7ZAQQg,6
 hivetrace/errors/validation.py,sha256=2XFgxumDAU8dZnkDiF_eFqANiqsc8lIsbwjrzcAva98,1020
 hivetrace/handlers/__init__.py,sha256=9edrtdJyvx8UkANmrjjVhg9f0kWMxfXVrmURTUrW6mg,435
 hivetrace/handlers/error_handler.py,sha256=aWLL--HKBm4h8AO4oQSpsFObq2TjHXntYXBDXWp-Q30,3774
-hivetrace/handlers/response_builder.py,sha256=p3mFHqetd2FYrbJc4j8PV6zI8OF9KtSyjifpPy8asjA,2590
+hivetrace/handlers/response_builder.py,sha256=PaIT7DXMTLK5dXYDzyyWJHa-lxtloTGDWsqXhF2Ax-I,2933
 hivetrace/models/__init__.py,sha256=qQvtDkI0Awlch6c_kedH5Jq8aWn3XhDXbxbch5nf-RI,1020
 hivetrace/models/requests.py,sha256=iv7NXm5q-tVzs-54o9DfydgE8aMV-AjLa0vgzjGaKCQ,2588
-hivetrace/models/responses.py,sha256=U06Gc4ZU2WmaPxjy1Dg5BtTDad6MOBURiDPTeLyDtcE,3209
+hivetrace/models/responses.py,sha256=F2IECVxZMAm5wfzRlsLdn9sdTT0qHEmPEnv0nJJ1G-U,3208
 hivetrace/utils/__init__.py,sha256=BNYbeSuUbrZL7RradjE_OFAxam3L6eexbL2IMfjImv0,747
 hivetrace/utils/error_helpers.py,sha256=egVQpENputLR8exNpV1cui2LSHqbf8pI6SHRbLdxOX8,2661
 hivetrace/utils/uuid_generator.py,sha256=W4i2tUSyClNKNgm4O-bk_Qkkmw3cWIuf29DjwXftx0c,344
-hivetrace-1.3.14.dist-info/licenses/LICENSE,sha256=8d3g3prbWPDLQ5AV0dtyWfYTj5QPl8MJ_wlr2l8pjEU,11333
-hivetrace-1.3.14.dist-info/METADATA,sha256=4ptdgEVUdwxzDN2LHBDb23wmVZNuXJ8BpxrI8FWT-OI,28054
-hivetrace-1.3.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-hivetrace-1.3.14.dist-info/top_level.txt,sha256=F6mZCzZ5CSftMc-M0NeOYWbwyTzjybR72P4qSBMyZZM,10
-hivetrace-1.3.14.dist-info/RECORD,,
+hivetrace-1.3.16.dist-info/licenses/LICENSE,sha256=8d3g3prbWPDLQ5AV0dtyWfYTj5QPl8MJ_wlr2l8pjEU,11333
+hivetrace-1.3.16.dist-info/METADATA,sha256=CqBk5XpY5jcW9PhX8S-Snv0RAZjwlSrlwGRVlwYRlPg,28054
+hivetrace-1.3.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+hivetrace-1.3.16.dist-info/top_level.txt,sha256=F6mZCzZ5CSftMc-M0NeOYWbwyTzjybR72P4qSBMyZZM,10
+hivetrace-1.3.16.dist-info/RECORD,,