agenta 0.19.1__tar.gz → 0.19.1a0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.19.1
+Version: 0.19.1a0
 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.1 → agenta-0.19.1a0}/agenta/sdk/agenta_init.py

@@ -26,6 +26,16 @@ 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,
@@ -85,7 +95,7 @@ class AgentaSingleton:
 
 
 class Config:
-    def __init__(self, base_id: str, host: str, api_key: Optional[str] = ""):
+    def __init__(self, base_id: str, host: str, api_key: str = ""):
         self.base_id = base_id
         self.host = host
 
@@ -93,9 +103,7 @@ class Config:
             self.persist = False
         else:
             self.persist = True
-            self.client = AgentaApi(
-                base_url=self.host + "/api", api_key=api_key if api_key else ""
-            )
+            self.client = AgentaApi(base_url=self.host + "/api", api_key=api_key)
 
     def register_default(self, overwrite=False, **kwargs):
         """alias for default"""

{agenta-0.19.1 → agenta-0.19.1a0}/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 cost is not None and isinstance(cost, float):
+        if span is not None and 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 tokens is not None and isinstance(tokens, dict):
+        if span is not None and tokens is not None and isinstance(tokens, dict):
             if span.tokens is None:
                 span.tokens = LlmTokens(**tokens)
             else:
@@ -218,6 +218,7 @@ 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",
@@ -229,45 +230,15 @@ 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.
@@ -285,7 +256,7 @@ class Tracing(metaclass=SingletonMeta):
         if self.api_key == "":
             return
 
-        if not self.active_trace_id:
+        if self.active_trace_id is None:
             raise RuntimeError("No active trace to end.")
 
         self.llm_logger.info("Preparing to send recorded spans for processing.")

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

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