agenta 0.50.0__py3-none-any.whl → 0.50.2__py3-none-any.whl

agenta/sdk/decorators/routing.py

@@ -352,7 +352,7 @@ class entrypoint:
                     return await self.handle_success(result, inline)
 
                 except Exception as error:  # pylint: disable=broad-except
-                    self.handle_failure(error)
+                    await self.handle_failure(error, inline)
 
     async def handle_success(
         self,
@@ -360,8 +360,9 @@ class entrypoint:
         inline: bool,
     ):
         data = None
-        tree = None
         content_type = "text/plain"
+
+        tree = None
         tree_id = None
         trace_id = None
         span_id = None
@@ -369,6 +370,7 @@ class entrypoint:
         with suppress():
             if isinstance(result, (dict, list)):
                 content_type = "application/json"
+
             data = self.patch_result(result)
 
             (
@@ -381,8 +383,8 @@ class entrypoint:
         try:
             return BaseResponse(
                 data=data,
-                tree=tree,
                 content_type=content_type,
+                tree=tree,
                 tree_id=tree_id,
                 trace_id=trace_id,
                 span_id=span_id,
@@ -402,9 +404,10 @@ class entrypoint:
                     content_type=content_type,
                 )
 
-    def handle_failure(
+    async def handle_failure(
         self,
         error: Exception,
+        inline: bool,
     ):
         display_exception("Application Exception")
 
@@ -416,9 +419,29 @@ class entrypoint:
 
         stacktrace = format_exception(error, value=error, tb=error.__traceback__)  # type: ignore
 
+        tree = None
+        tree_id = None
+        trace_id = None
+        span_id = None
+
+        with suppress():
+            (
+                tree,
+                tree_id,
+                trace_id,
+                span_id,
+            ) = await self.fetch_inline_trace(inline)
+
         raise HTTPException(
             status_code=status_code,
-            detail={"message": str(error), "stacktrace": stacktrace},
+            detail=dict(
+                message=str(error),
+                stacktrace=stacktrace,
+                tree=tree,
+                tree_id=tree_id,
+                trace_id=trace_id,
+                span_id=span_id,
+            ),
         )
 
     def patch_result(

agenta/sdk/tracing/inline.py

@@ -1009,18 +1009,21 @@ def parse_inline_trace(
     ##############################################
 
     spans = [
-        loads(
-            span_dto.model_dump_json(
-                exclude_none=True,
-                exclude_defaults=True,
-            )
+        span_dto.model_dump(
+            mode="json",
+            exclude_none=True,
+            exclude_defaults=True,
         )
         for span_dto in agenta_span_dtos
     ]
     inline_trace = AgentaNodesResponse(
         version="1.0.0",
         nodes=[AgentaNodeDto(**span) for span in spans],
-    ).model_dump(exclude_none=True, exclude_unset=True)
+    ).model_dump(
+        mode="json",
+        exclude_none=True,
+        exclude_unset=True,
+    )
     return inline_trace
 
 

{agenta-0.50.0.dist-info → agenta-0.50.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agenta
-Version: 0.50.0
+Version: 0.50.2
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Keywords: LLMOps,LLM,evaluation,prompt engineering
 Author: Mahmoud Mabrouk

{agenta-0.50.0.dist-info → agenta-0.50.2.dist-info}/RECORD

@@ -313,7 +313,7 @@ agenta/sdk/context/exporting.py,sha256=16X8fgMhl58gehSlqANX97FiKxx4TkGiG4d2B0-7Z
 agenta/sdk/context/routing.py,sha256=FEsjw8EttI1SMyUo96ptcUsvHJnhoKwdr1szlkxxJNU,598
 agenta/sdk/context/tracing.py,sha256=xjErrXP1Nq1AfL-Cif1l-lNEfs12eQ3v_VCRgoKe7nY,743
 agenta/sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/decorators/routing.py,sha256=L3o0Q9flBfb2UIfCIx0OWzkP0W-45_GQ3-Ri8phwNlU,25678
+agenta/sdk/decorators/routing.py,sha256=7oWQTOuxV9uAbyBXvMThA3P_UnBdHx-T6mgZ50mFqVY,26178
 agenta/sdk/decorators/tracing.py,sha256=6xwN9AomDND_5wL21NXFoZsdwTBuITw2I39GNqNGD-4,10044
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
 agenta/sdk/litellm/litellm.py,sha256=dCjw0H_jD3L3UQ3l9SbLm5dfeIGnel6dPtQYJ78beYM,10202
@@ -340,7 +340,7 @@ agenta/sdk/tracing/__init__.py,sha256=rQNe5-zT5Kt7_CDhq-lnUIi1EYTBVzVf_MbfcIxVD9
 agenta/sdk/tracing/attributes.py,sha256=DwjjOk3mGOvz0jYu8EYr3hhItvawK5GX80_MfciqPrc,5559
 agenta/sdk/tracing/conventions.py,sha256=JBtznBXZ3aRkGKkLl7cPwdMNh3w1G-H2Ta2YrAxbr38,950
 agenta/sdk/tracing/exporters.py,sha256=hSQCyQScaIihBy3oBjhY2BZZxyP_zTm3xHwB-iQHnsI,3448
-agenta/sdk/tracing/inline.py,sha256=ShPAAjk_26I4hgrmC6y0n-I9gxB3Q4VeEYhhsLHInPU,31454
+agenta/sdk/tracing/inline.py,sha256=rvulLQqtCWrpJmzlg0fo31_YLhq8KBTTOPPPyQt5WCE,31477
 agenta/sdk/tracing/processors.py,sha256=wZWF8vTEouhbGxbfPGXVQJI53CcXFByqxHiCSIzJV-4,5407
 agenta/sdk/tracing/propagation.py,sha256=EeOqDMqnh_MoEhGd1do_vy_tQBYUcoC8kpLqVoZeqg0,2561
 agenta/sdk/tracing/spans.py,sha256=nqUOjjirBxB8Eacv8Qj4Ra_6rknGi3lbJdNyKmk5ODQ,3707
@@ -357,6 +357,6 @@ agenta/sdk/utils/logging.py,sha256=j4NzpFk2ilOM10sRBOxCjmansDHSx6HwMV8IAEreRb8,8
 agenta/sdk/utils/preinit.py,sha256=1TAAHhYyYnLLwvzwnf33Qwkou7tI3iITlVt-1kcyGaM,1186
 agenta/sdk/utils/singleton.py,sha256=17Ph7LGnnV8HkPjImruKita2ni03Ari5jr0jqm__4sc,312
 agenta/sdk/utils/timing.py,sha256=nZR-kudVUtKFlHuBhztgSGxj7FVnCB4Uv6sfg-1dkrQ,1556
-agenta-0.50.0.dist-info/METADATA,sha256=DnlZWnLgz_NRS6WF-vy2uSViLYgVUnBh_EQQOwliJvE,31512
-agenta-0.50.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-agenta-0.50.0.dist-info/RECORD,,
+agenta-0.50.2.dist-info/METADATA,sha256=Svx_2JUtKDhNYQP9AU3Da7zTCrbRn1fH1vWtUNYgdEY,31512
+agenta-0.50.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+agenta-0.50.2.dist-info/RECORD,,