agenta 0.19.1__py3-none-any.whl → 0.19.1a0__py3-none-any.whl

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/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.dist-info → agenta-0.19.1a0.dist-info}/METADATA

@@ -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.dist-info → agenta-0.19.1a0.dist-info}/RECORD

@@ -127,7 +127,7 @@ agenta/docker/docker-assets/lambda_function.py,sha256=h4UZSSfqwpfsCgERv6frqwm_4J
 agenta/docker/docker-assets/main.py,sha256=7MI-21n81U7N7A0GxebNi0cmGWtJKcR2sPB6FcH2QfA,251
 agenta/docker/docker_utils.py,sha256=5uHMCzXkCvIsDdEiwbnnn97KkzsFbBvyMwogCsv_Z5U,3509
 agenta/sdk/__init__.py,sha256=cF0de6DiH-NZWEm0XvPN8_TeC1whPBnDf1WYYE1qK2g,762
-agenta/sdk/agenta_init.py,sha256=zhMYCH6QMDzQADrOmFBaqZRvAYrA6CC35XsePY7IMxg,9789
+agenta/sdk/agenta_init.py,sha256=aKCJRoZ6nsY3dHWwvgxdxyJhYL3n5xEcD6ZJiqshchQ,9959
 agenta/sdk/client.py,sha256=trKyBOYFZRk0v5Eptxvh87yPf50Y9CqY6Qgv4Fy-VH4,2142
 agenta/sdk/context.py,sha256=q-PxL05-I84puunUAs9LGsffEXcYhDxhQxjuOz2vK90,901
 agenta/sdk/decorators/base.py,sha256=9aNdX5h8a2mFweuhdO-BQPwXGKY9ONPIdLRhSGAGMfY,217
@@ -137,7 +137,7 @@ agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/tracing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/tracing/callbacks.py,sha256=8XYcGC4EZQ0lBn2IWmtPH_yC5Hu6Yu-NkdZRuY7CoTw,6816
 agenta/sdk/tracing/context_manager.py,sha256=HskDaiORoOhjeN375gm05wYnieQzh5UnoIsnSAHkAyc,252
-agenta/sdk/tracing/llm_tracing.py,sha256=fHtc1tKgdR2CDnFMrfDIxSvyjwFW3Qk2RE8P-y_tCM0,11361
+agenta/sdk/tracing/llm_tracing.py,sha256=3zbyL0Spv6UFvze6deAb0VYcHWFllC5itwkt_wTkMBo,10389
 agenta/sdk/tracing/logger.py,sha256=GfH7V-jBHcn7h5dbdrnkDMe_ml3wkXFBeoQiqR4KVRc,474
 agenta/sdk/tracing/tasks_manager.py,sha256=ROrWIaqS2J2HHiJtRWiHKlLY8CCsqToP5VeXu7mamck,3748
 agenta/sdk/types.py,sha256=KMnQUOdjaHSWctDLIiMHnk0o3c-C47Vm4Mn2kIZ88YI,5740
@@ -159,7 +159,7 @@ agenta/templates/simple_prompt/app.py,sha256=kODgF6lhzsaJPdgL5b21bUki6jkvqjWZzWR
 agenta/templates/simple_prompt/env.example,sha256=g9AE5bYcGPpxawXMJ96gh8oenEPCHTabsiOnfQo3c5k,70
 agenta/templates/simple_prompt/requirements.txt,sha256=ywRglRy7pPkw8bljmMEJJ4aOOQKrt9FGKULZ-DGkoBU,23
 agenta/templates/simple_prompt/template.toml,sha256=DQBtRrF4GU8LBEXOZ-GGuINXMQDKGTEG5y37tnvIUIE,60
-agenta-0.19.1.dist-info/METADATA,sha256=5pVwD7MuunohvTraDMTtW3zlo7_y2UKnRJMHm1Wmrow,26458
-agenta-0.19.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-agenta-0.19.1.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.19.1.dist-info/RECORD,,
+agenta-0.19.1a0.dist-info/METADATA,sha256=C6yd9GdZsaTuuJ-gxd0tD3N5QONjKuPJiNcTeslgApg,26460
+agenta-0.19.1a0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+agenta-0.19.1a0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.19.1a0.dist-info/RECORD,,