agenta 0.19.1a0__tar.gz → 0.19.2__tar.gz

{agenta-0.19.1a0 → agenta-0.19.2}/PKG-INFO

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

{agenta-0.19.1a0 → agenta-0.19.2}/agenta/sdk/agenta_init.py

@@ -26,16 +26,6 @@ class AgentaSingleton:
             cls._instance = super(AgentaSingleton, cls).__new__(cls)
         return cls._instance
 
-    @property
-    def client(self):
-        """API Backend client.
-
-        Returns:
-            AgentaAPI: instance of agenta api backend
-        """
-
-        return AgentaApi(base_url=self.host + "/api", api_key=self.api_key)
-
     def init(
         self,
         app_id: Optional[str] = None,
@@ -95,7 +85,7 @@ class AgentaSingleton:
 
 
 class Config:
-    def __init__(self, base_id: str, host: str, api_key: str = ""):
+    def __init__(self, base_id: str, host: str, api_key: Optional[str] = ""):
         self.base_id = base_id
         self.host = host
 
@@ -103,7 +93,9 @@ class Config:
             self.persist = False
         else:
             self.persist = True
-            self.client = AgentaApi(base_url=self.host + "/api", api_key=api_key)
+            self.client = AgentaApi(
+                base_url=self.host + "/api", api_key=api_key if api_key else ""
+            )
 
     def register_default(self, overwrite=False, **kwargs):
         """alias for default"""

{agenta-0.19.1a0 → agenta-0.19.2}/agenta/sdk/tracing/llm_tracing.py

@@ -178,7 +178,7 @@ class Tracing(metaclass=SingletonMeta):
         span.status = value
 
     def _update_span_cost(self, span: CreateSpan, cost: Optional[float]):
-        if span is not None and cost is not None and isinstance(cost, float):
+        if cost is not None and isinstance(cost, float):
             if span.cost is None:
                 span.cost = cost
             else:
@@ -187,7 +187,7 @@ class Tracing(metaclass=SingletonMeta):
     def _update_span_tokens(self, span: CreateSpan, tokens: Optional[dict]):
         if isinstance(tokens, LlmTokens):
             tokens = tokens.dict()
-        if span is not None and tokens is not None and isinstance(tokens, dict):
+        if tokens is not None and isinstance(tokens, dict):
             if span.tokens is None:
                 span.tokens = LlmTokens(**tokens)
             else:
@@ -218,7 +218,6 @@ class Tracing(metaclass=SingletonMeta):
 
         self.active_span.end_time = datetime.now(timezone.utc)
         self.active_span.outputs = [outputs.get("message", "")]
-
         if self.active_span.spankind in [
             "LLM",
             "RETRIEVER",
@@ -230,15 +229,45 @@ class Tracing(metaclass=SingletonMeta):
         self.pending_spans.append(self.active_span)
 
         active_span_parent_id = self.active_span.parent_span_id
+        if (
+            self.active_span.status == SpanStatusCode.ERROR.value
+            and active_span_parent_id is not None
+        ):
+            self.record_exception_and_end_trace(span_parent_id=active_span_parent_id)
 
         if active_span_parent_id is None:
             self.end_trace(parent_span=self.active_span)
+
         else:
             parent_span = self.span_dict[active_span_parent_id]
             self._update_span_cost(parent_span, self.active_span.cost)
             self._update_span_tokens(parent_span, self.active_span.tokens)
             self.active_span = parent_span
 
+    def record_exception_and_end_trace(self, span_parent_id: str):
+        """
+        Record an exception and end the trace.
+
+        Args:
+            span_parent_id (str): The ID of the parent span.
+
+        Returns:
+            None
+        """
+
+        parent_span = self.span_dict.get(span_parent_id)
+        if parent_span is not None:
+            # Update parent span of active span
+            parent_span.outputs = self.active_span.outputs  # type: ignore
+            parent_span.status = "ERROR"
+            parent_span.end_time = datetime.now(timezone.utc)
+
+            # Push parent span to list of recorded spans and end trace
+            self.pending_spans.append(parent_span)
+            self.end_trace(parent_span=parent_span)
+
+        # TODO: improve exception logic here.
+
     def end_trace(self, parent_span: CreateSpan):
         """
         Ends the active trace and sends the recorded spans for processing.
@@ -256,7 +285,7 @@ class Tracing(metaclass=SingletonMeta):
         if self.api_key == "":
             return
 
-        if self.active_trace_id is None:
+        if not self.active_trace_id:
             raise RuntimeError("No active trace to end.")
 
         self.llm_logger.info("Preparing to send recorded spans for processing.")

{agenta-0.19.1a0 → agenta-0.19.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.19.1a0"
+version = "0.19.2"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <mahmoud@agenta.ai>"]