agenta 0.17.5__tar.gz → 0.17.6a1__tar.gz

{agenta-0.17.5 → agenta-0.17.6a1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.17.5
+Version: 0.17.6a1
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering

{agenta-0.17.5 → agenta-0.17.6a1}/agenta/sdk/decorators/llm_entrypoint.py

@@ -83,9 +83,11 @@ class entrypoint(BaseDecorator):
                 {"config": config_params, "environment": "playground"}
             )
 
+            # Exceptions are all handled inside self.execute_function()
             llm_result = await self.execute_function(
                 func, *args, params=func_params, config_params=config_params
             )
+
             return llm_result
 
         @functools.wraps(func)

{agenta-0.17.5 → agenta-0.17.6a1}/agenta/sdk/decorators/tracing.py

@@ -58,19 +58,24 @@ class instrument(BaseDecorator):
             try:
                 result = await func(*args, **kwargs)
                 self.tracing.update_span_status(span=span, value="OK")
-            except Exception as e:
-                result = str(e)
-                self.tracing.set_span_attribute(
-                    {"traceback_exception": traceback.format_exc()}
-                )
-                self.tracing.update_span_status(span=span, value="ERROR")
-            finally:
                 self.tracing.end_span(
                     outputs=(
                         {"message": result} if not isinstance(result, dict) else result
                     )
                 )
-            return result
+                return result
+
+            except Exception as e:
+                result = {
+                    "message": str(e),
+                    "stacktrace": traceback.format_exc(),
+                }
+                self.tracing.set_span_attribute(
+                    {"traceback_exception": traceback.format_exc()}
+                )
+                self.tracing.update_span_status(span=span, value="ERROR")
+                self.tracing.end_span(outputs=result)
+                raise e
 
         @wraps(func)
         def sync_wrapper(*args, **kwargs):
@@ -89,17 +94,23 @@ class instrument(BaseDecorator):
             try:
                 result = func(*args, **kwargs)
                 self.tracing.update_span_status(span=span, value="OK")
-            except Exception as e:
-                result = str(e)
-                self.tracing.set_span_attribute(
-                    {"traceback_exception": traceback.format_exc()}
-                )
-                self.tracing.update_span_status(span=span, value="ERROR")
-            finally:
                 self.tracing.end_span(
                     outputs=(
                         {"message": result} if not isinstance(result, dict) else result
                     )
                 )
+                return result
+
+            except Exception as e:
+                result = {
+                    "message": str(e),
+                    "stacktrace": traceback.format_exc(),
+                }
+                self.tracing.set_span_attribute(
+                    {"traceback_exception": traceback.format_exc()}
+                )
+                self.tracing.update_span_status(span=span, value="ERROR")
+                self.tracing.end_span(outputs=result)
+                raise e
 
         return async_wrapper if is_coroutine_function else sync_wrapper

{agenta-0.17.5 → agenta-0.17.6a1}/agenta/sdk/tracing/callbacks.py

@@ -55,7 +55,9 @@ def litellm_handler():
                     "message": kwargs.get(
                         "complete_streaming_response"
                     ),  # the complete streamed response (only set if `completion(..stream=True)`)
-                    "usage": response_obj.usage.dict(),  # litellm calculates usage
+                    "usage": response_obj.usage.dict()
+                    if hasattr(response_obj, "usage")
+                    else None,  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -69,7 +71,9 @@ def litellm_handler():
             self._trace.end_span(
                 outputs={
                     "message": response_obj.choices[0].message.content,
-                    "usage": response_obj.usage.dict(),  # litellm calculates usage
+                    "usage": response_obj.usage.dict()
+                    if hasattr(response_obj, "usage")
+                    else None,  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -93,7 +97,9 @@ def litellm_handler():
             self._trace.end_span(
                 outputs={
                     "message": kwargs["exception"],  # the Exception raised
-                    "usage": response_obj.usage.dict(),  # litellm calculates usage
+                    "usage": response_obj.usage.dict()
+                    if hasattr(response_obj, "usage")
+                    else None,  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -109,7 +115,9 @@ def litellm_handler():
                     "message": kwargs.get(
                         "complete_streaming_response"
                     ),  # the complete streamed response (only set if `completion(..stream=True)`)
-                    "usage": response_obj.usage.dict(),  # litellm calculates usage
+                    "usage": response_obj.usage.dict()
+                    if hasattr(response_obj, "usage")
+                    else None,  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -123,7 +131,9 @@ def litellm_handler():
             self._trace.end_span(
                 outputs={
                     "message": response_obj.choices[0].message.content,
-                    "usage": response_obj.usage.dict(),  # litellm calculates usage
+                    "usage": response_obj.usage.dict()
+                    if hasattr(response_obj, "usage")
+                    else None,  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -147,7 +157,9 @@ def litellm_handler():
             self._trace.end_span(
                 outputs={
                     "message": kwargs["exception"],  # the Exception raised
-                    "usage": response_obj.usage.dict(),  # litellm calculates usage
+                    "usage": response_obj.usage.dict()
+                    if hasattr(response_obj, "usage")
+                    else None,  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost

{agenta-0.17.5 → agenta-0.17.6a1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.17.5"
+version = "0.17.6a1"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <mahmoud@agenta.ai>"]
@@ -34,7 +34,7 @@ cachetools = "^5.3.3"
 
 [tool.poetry.dev-dependencies]
 pytest = "^8.2"
-setuptools = "^70.1.0"
+setuptools = "^70.2.0"
 
 [build-system]
 requires = ["poetry-core"]