agenta 0.19.9__py3-none-any.whl → 0.20.0a1__py3-none-any.whl

agenta/sdk/tracing/llm_tracing.py

@@ -1,11 +1,16 @@
 import os
+import copy
+import json
 from uuid import uuid4
 
+import traceback
 from threading import Lock
 from datetime import datetime, timezone
 from typing import Optional, Dict, Any, List
 
-from agenta.sdk.tracing.tracing_context import tracing_context
+from contextlib import contextmanager
+
+from agenta.sdk.tracing.tracing_context import tracing_context, TracingContext
 from agenta.sdk.tracing.logger import llm_logger as logging
 from agenta.sdk.tracing.tasks_manager import TaskQueue
 from agenta.client.backend.client import AsyncAgentaApi
@@ -20,7 +25,7 @@ from bson.objectid import ObjectId
 
 VARIANT_TRACKING_FEATURE_FLAG = False
 
-from agenta.sdk.utils.debug import debug, DEBUG, SHIFT
+from agenta.sdk.utils.debug import debug
 
 
 logging.setLevel("DEBUG")
@@ -91,6 +96,46 @@ class Tracing(metaclass=SingletonMeta):
             base_url=self.host, api_key=self.api_key, timeout=120  # type: ignore
         ).observability
 
+    ### --- Context Manager --- ###
+
+    @contextmanager
+    def Context(self, **kwargs):
+        # This will evolve as be work towards OTel compliance
+
+        token = None
+
+        try:
+            if tracing_context.get() is None:
+                token = tracing_context.set(TracingContext())
+
+            self.open_span(**kwargs)
+
+            yield
+
+            self.set_status(status="OK")
+
+        except Exception as e:
+            logging.error(e)
+
+            result = {
+                "message": str(e),
+                "stacktrace": traceback.format_exc(),
+            }
+
+            self.set_status(status="ERROR")
+            self.set_attributes({"traceback_exception": traceback.format_exc()})
+            self.store_outputs(result)
+
+            raise
+
+        finally:
+            self.close_span()
+
+            if token is not None:
+                self.flush_spans()
+
+                tracing_context.reset(token)
+
     ### --- API --- ###
 
     @debug()
@@ -150,6 +195,14 @@ class Tracing(metaclass=SingletonMeta):
 
         tracing.trace_tags.extend(tags)
 
+    @debug()
+    def is_trace_ready(self):
+        tracing = tracing_context.get()
+
+        are_spans_ready = [span.end_time is not None for span in tracing.spans.values()]
+
+        return all(are_spans_ready)
+
     @debug()
     def close_trace(self) -> None:
         """
@@ -177,11 +230,7 @@ class Tracing(metaclass=SingletonMeta):
         if not self.api_key:
             logging.error("No API key")
         else:
-            self._process_closed_spans()
-
-        self._clear_closed_spans()
-        self._clear_tracked_spans()
-        self._clear_active_span()
+            self._process_spans()
 
         self._clear_trace_tags()
 
@@ -193,6 +242,7 @@ class Tracing(metaclass=SingletonMeta):
         name: str,
         spankind: str,
         input: Dict[str, Any],
+        active: bool = True,
         config: Optional[Dict[str, Any]] = None,
         **kwargs,
     ) -> CreateSpan:
@@ -213,6 +263,7 @@ class Tracing(metaclass=SingletonMeta):
             attributes={},
             status=SpanStatusCode.UNSET.value,
             start_time=datetime.now(timezone.utc),
+            internals=None,
             outputs=None,
             tags=None,
             user=None,
@@ -224,12 +275,18 @@ class Tracing(metaclass=SingletonMeta):
         )
 
         if tracing.trace_id is None:
-            self.start_trace(span, config)
+            self.open_trace(span, config)
         else:
             span.parent_span_id = tracing.active_span.id  # type: ignore
 
-        tracing.tracked_spans[span.id] = span
-        tracing.active_span = span
+        tracing.spans[span.id] = span
+
+        if active:
+            tracing.active_span = span
+        else:
+            # DETACHED SPAN
+            pass
+
         ### --- TO BE CLEANED --- <<<
 
         logging.info(f"Opened  span  {span_id} {spankind.upper()}")
@@ -238,8 +295,7 @@ class Tracing(metaclass=SingletonMeta):
 
     @debug(req=True)
     def set_attributes(
-        self,
-        attributes: Dict[str, Any] = {},
+        self, attributes: Dict[str, Any] = {}, span_id: Optional[str] = None
     ) -> None:
         """
         Set attributes for the active span.
@@ -248,41 +304,31 @@ class Tracing(metaclass=SingletonMeta):
             attributes (Dict[str, Any], optional): A dictionary of attributes to set. Defaults to {}.
         """
 
-        tracing = tracing_context.get()
-
-        if tracing.active_span is None:
-            logging.error(f"Cannot set attributes ({set(attributes)}), no active span")
-            return
+        span = self._get_target_span(span_id)
 
         logging.info(
-            f"Setting span  {tracing.active_span.id} {tracing.active_span.spankind.upper()} attributes={attributes}"
+            f"Setting span  {span.id} {span.spankind.upper()} attributes={attributes}"
         )
 
         for key, value in attributes.items():
-            tracing.active_span.attributes[key] = value  # type: ignore
+            span.attributes[key] = value  # type: ignore
 
     @debug()
-    def set_status(self, status: str) -> None:
+    def set_status(self, status: str, span_id: Optional[str] = None) -> None:
         """
         Set status for the active span.
 
         Args:
             status: Enum ( UNSET, OK, ERROR )
         """
-        tracing = tracing_context.get()
+        span = self._get_target_span(span_id)
 
-        if tracing.active_span is None:
-            logging.error(f"Cannot set status ({status}), no active span")
-            return
-
-        logging.info(
-            f"Setting span  {tracing.active_span.id} {tracing.active_span.spankind.upper()} status={status}"
-        )
+        logging.info(f"Setting span  {span.id} {span.spankind.upper()} status={status}")
 
-        tracing.active_span.status = status
+        span.status = status
 
     @debug()
-    def close_span(self, outputs: Dict[str, Any]) -> None:
+    def close_span(self, span_id: Optional[str] = None) -> None:
         """
         Ends the active span, if it is a parent span, ends the trace too.
 
@@ -299,45 +345,122 @@ class Tracing(metaclass=SingletonMeta):
         Returns:
             None
         """
+        span = self._get_target_span(span_id)
 
-        tracing = tracing_context.get()
-
-        if tracing.active_span is None:
-            logging.error("Cannot close span, no active span")
+        spankind = span.spankind
 
-        span_id = tracing.active_span.id
-        spankind = tracing.active_span.spankind
-
-        logging.info(f"Closing span  {span_id} {spankind}")
+        logging.info(f"Closing span  {span.id} {spankind}")
 
         ### --- TO BE CLEANED --- >>>
-        tracing.active_span.end_time = datetime.now(timezone.utc)
-
-        tracing.active_span.outputs = [outputs.get("message", "")]
+        span.end_time = datetime.now(timezone.utc)
 
-        if tracing.active_span.spankind.upper() in [
-            "LLM",
-            "RETRIEVER",
-        ]:  # TODO: Remove this whole part. Setting the cost should be done through set_span_attribute
-            self._update_span_cost(tracing.active_span, outputs.get("cost", None))
-            self._update_span_tokens(tracing.active_span, outputs.get("usage", None))
+        # TODO: Remove this whole part. Setting the cost should be done through set_span_attribute
+        if isinstance(span.outputs, dict):
+            self._update_span_cost(span, span.outputs.get("cost", None))
+            self._update_span_tokens(span, span.outputs.get("usage", None))
 
-        tracing.closed_spans.append(tracing.active_span)
+        span_parent_id = span.parent_span_id
 
-        active_span_parent_id = tracing.active_span.parent_span_id
+        if span_parent_id is not None:
+            tracing = tracing_context.get()
 
-        if active_span_parent_id is None:
-            self.end_trace(parent_span=tracing.active_span)
+            parent_span = tracing.spans[span_parent_id]
+            self._update_span_cost(parent_span, span.cost)
+            self._update_span_tokens(parent_span, span.tokens)
 
-        else:
-            parent_span = tracing.tracked_spans[active_span_parent_id]
-            self._update_span_cost(parent_span, tracing.active_span.cost)
-            self._update_span_tokens(parent_span, tracing.active_span.tokens)
-            tracing.active_span = parent_span
+            if span_id is None:
+                tracing.active_span = parent_span
         ### --- TO BE CLEANED --- <<<
 
         logging.info(f"Closed  span  {span_id} {spankind}")
 
+    @debug()
+    def store_internals(
+        self, internals: Dict[str, Any] = {}, span_id: Optional[str] = None
+    ) -> None:
+        """
+        Set internals for the active span.
+
+        Args:
+            internals (Dict[str, Any], optional): A dictionary of local variables to set. Defaults to {}.
+        """
+        span = self._get_target_span(span_id)
+
+        logging.info(
+            f"Setting span  {span.id} {span.spankind.upper()} internals={internals}"
+        )
+
+        if span.internals is None:
+            span.internals = dict()
+
+        for key, value in internals.items():
+            span.internals[key] = value  # type: ignore
+
+    @debug()
+    def store_outputs(
+        self, outputs: Dict[str, Any] = {}, span_id: Optional[str] = None
+    ) -> None:
+        """
+        Set outputs for the active span.
+
+        Args:
+            outputs (Dict[str, Any], optional): A dictionary of output variables to set. Defaults to {}.
+        """
+        span = self._get_target_span(span_id)
+
+        logging.info(
+            f"Setting span  {span.id} {span.spankind.upper()} outputs={outputs}"
+        )
+
+        span.outputs = outputs
+
+    @debug()
+    def dump_trace(self):
+        """
+        Collects and organizes tracing information into a dictionary.
+        This function retrieves the current tracing context and extracts relevant data such as `trace_id`, `cost`, `tokens`, and `latency` for the whole trace.
+        It also dumps detailed span information using the `dump_spans` method and includes it in the trace dictionary.
+        If an error occurs during the process, it logs the error message and stack trace.
+
+        Returns:
+            dict: A dictionary containing the trace information.
+        """
+        try:
+            trace = dict()
+
+            tracing = tracing_context.get()
+
+            trace["trace_id"] = tracing.trace_id
+
+            for span in tracing.spans.values():
+                if span.parent_span_id is None:
+                    trace["cost"] = span.cost
+                    trace["usage"] = (
+                        None if span.tokens is None else json.loads(span.tokens.json())
+                    )
+                    trace["latency"] = (
+                        (span.end_time if span.end_time else span.start_time)
+                        - span.start_time
+                    ).total_seconds()
+
+            spans = (
+                []
+                if len(tracing.spans) == 0
+                else [json.loads(span.json()) for span in tracing.spans.values()]
+            )
+
+            if spans is not None:
+                trace["spans"] = spans
+
+        except Exception as e:
+            logging.error(e)
+            logging.error(traceback.format_exc())
+
+        return trace
+
+    def flush_spans(self) -> None:
+        self.close_trace()
+
     ### --- Legacy API --- ###
 
     def start_trace(
@@ -348,7 +471,7 @@ class Tracing(metaclass=SingletonMeta):
     ) -> None:  # Legacy
         self.open_trace(span, config, **kwargs)
 
-    def end_trace(self, parent_span: CreateSpan) -> None:  # Legacy
+    def end_trace(self, _: CreateSpan) -> None:  # Legacy
         self.close_trace()
 
     def start_span(
@@ -371,7 +494,8 @@ class Tracing(metaclass=SingletonMeta):
         self.set_attributes(attributes)
 
     def end_span(self, outputs: Dict[str, Any]) -> None:  # Legacy
-        self.close_span(outputs)
+        self.store_outputs(outputs)
+        self.close_span()
 
     ### --- Helper Functions --- ###
 
@@ -400,46 +524,44 @@ class Tracing(metaclass=SingletonMeta):
         # return uuid4().hex[:16]
         return str(ObjectId())
 
-    def _process_closed_spans(self) -> None:
+    def _get_target_span(self, span_id) -> CreateSpan:
         tracing = tracing_context.get()
 
-        logging.info(f"Sending spans {tracing.trace_id} #={len(tracing.closed_spans)} ")
+        span = None
 
-        # async def mock_create_traces(trace, spans):
-        #    print("trace-id", trace)
-        #    print("spans", spans)
+        if span_id is None:
+            if tracing.active_span is None:
+                logging.error(f"Cannot set attributes, no active span")
+                return
 
-        self.tasks_manager.add_task(
-            tracing.trace_id,
-            "trace",
-            # mock_create_traces(
-            self.client.create_traces(
-                trace=tracing.trace_id,
-                spans=tracing.closed_spans,  # type: ignore
-            ),
-            self.client,
-        )
+            span = tracing.active_span
+        else:
+            if span_id not in tracing.spans.keys():
+                logging.error(f"Cannot set attributes, span ({span_id}) not found")
+                return
 
-        logging.info(f"Sent    spans {tracing.trace_id} #={len(tracing.closed_spans)}")
+            span = tracing.spans[span_id]
 
-    def _clear_closed_spans(self) -> None:
-        tracing = tracing_context.get()
-
-        tracing.closed_spans.clear()
+        return span
 
-    def _clear_tracked_spans(self) -> None:
+    def _process_spans(self) -> None:
         tracing = tracing_context.get()
 
-        tracing.tracked_spans.clear()
-
-    def _clear_active_span(self) -> None:
-        tracing = tracing_context.get()
+        spans = list(tracing.spans.values())
 
-        span_id = tracing.active_span.id
+        logging.info(f"Sending trace {tracing.trace_id} spans={len(spans)} ")
 
-        tracing.active_span = None
+        self.tasks_manager.add_task(
+            tracing.trace_id,
+            "send-trace",
+            # mock_create_traces(
+            self.client.create_traces(
+                trace=tracing.trace_id, spans=spans  # type: ignore
+            ),
+            self.client,
+        )
 
-        logging.debug(f"Cleared active span {span_id}")
+        logging.info(f"Sent    trace {tracing.trace_id}")
 
     def _update_span_cost(self, span: CreateSpan, cost: Optional[float]) -> None:
         if span is not None and cost is not None and isinstance(cost, float):
@@ -455,6 +577,16 @@ class Tracing(metaclass=SingletonMeta):
             if span.tokens is None:
                 span.tokens = LlmTokens(**tokens)
             else:
-                span.tokens.prompt_tokens += tokens["prompt_tokens"]
-                span.tokens.completion_tokens += tokens["completion_tokens"]
-                span.tokens.total_tokens += tokens["total_tokens"]
+                span.tokens.prompt_tokens += (
+                    tokens["prompt_tokens"]
+                    if tokens["prompt_tokens"] is not None
+                    else 0
+                )
+                span.tokens.completion_tokens += (
+                    tokens["completion_tokens"]
+                    if tokens["completion_tokens"] is not None
+                    else 0
+                )
+                span.tokens.total_tokens += (
+                    tokens["total_tokens"] if tokens["total_tokens"] is not None else 0
+                )

agenta/sdk/tracing/tasks_manager.py

@@ -105,8 +105,9 @@ class TaskQueue(object):
                 future = self._thread_pool.submit(asyncio.run, task.run())
                 future.result()
             except Exception as exc:
-                self._logger.error(f"Error running task: {str(exc)}")
-                self._logger.error(f"Recording {task.coroutine_type} status to ERROR.")
+                self._logger.error(
+                    f"Task '{task.coroutine_type}' failed with error: {str(exc)}"
+                )
                 break
             finally:
                 self.tasks.task_done()

agenta/sdk/tracing/tracing_context.py

@@ -1,6 +1,6 @@
-import contextvars
+from contextvars import ContextVar
 
-from typing import Optional, Dict, Any, List
+from typing import Optional, Dict, List
 
 from agenta.client.backend.types.create_span import CreateSpan
 
@@ -15,14 +15,13 @@ class TracingContext:
 
         ### --- SPANS --- ###
         self.active_span: Optional[CreateSpan] = None
-        self.tracked_spans: Dict[str, CreateSpan] = {}
-        self.closed_spans: List[CreateSpan] = []
+        self.spans: Dict[str, CreateSpan] = {}
 
     def __repr__(self) -> str:
-        return f"TracingContext(trace_id=[{self.trace_id}], active_span=[{self.active_span.id if self.active_span else None}{' ' + self.active_span.spankind if self.active_span else ''}])"
+        return f"TracingContext(trace='{self.trace_id}', spans={[f'{span.id} {span.spankind}' for span in self.spans.values()]})"
 
     def __str__(self) -> str:
         return self.__repr__()
 
 
-tracing_context = contextvars.ContextVar(CURRENT_TRACING_CONTEXT_KEY, default=None)
+tracing_context = ContextVar(CURRENT_TRACING_CONTEXT_KEY, default=None)

agenta/sdk/types.py

@@ -1,5 +1,5 @@
 import json
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, Any
 
 from pydantic import ConfigDict, BaseModel, HttpUrl
 
@@ -16,11 +16,10 @@ class LLMTokenUsage(BaseModel):
     total_tokens: int
 
 
-class FuncResponse(BaseModel):
-    message: str
-    usage: Optional[LLMTokenUsage]
-    cost: Optional[float]
-    latency: float
+class BaseResponse(BaseModel):
+    version: Optional[str] = "2.0"
+    data: Optional[Dict[str, Any]]
+    trace: Optional[Dict[str, Any]]
 
 
 class DictInput(dict):

{agenta-0.19.9.dist-info → agenta-0.20.0a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.19.9
+Version: 0.20.0a1
 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.9.dist-info → agenta-0.20.0a1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agenta/__init__.py,sha256=6ZfEnXNHoJVXO9HT_U-WuitFvE6Oq_UTrjAZGIRzaDY,887
+agenta/__init__.py,sha256=nL_Fogmx_64xS--WjhE4gE_0h5LIJDbDGKpjZoNJ9oI,894
 agenta/cli/evaluation_commands.py,sha256=fs6492tprPId9p8eGO02Xy-NCBm2RZNJLZWcUxugwd8,474
 agenta/cli/helper.py,sha256=vRxHyeNaltzNIGrfU2vO0H28_rXDzx9QqIZ_S-W6zL4,6212
 agenta/cli/main.py,sha256=Wz0ODhoeKK3Qg_CFUhu6D909szk05tc8ZVBB6H1-w7k,9763
@@ -54,7 +54,7 @@ agenta/client/backend/types/base_output.py,sha256=ynXhDBQKrkR6Lnkx-yv6Q8xW4wXmzX
 agenta/client/backend/types/body_import_testset.py,sha256=7dVF3mv3VO0Co8F0qxLAgu4jabqDPjebK4mYvcd_TuA,1061
 agenta/client/backend/types/config_db.py,sha256=P0cSYvVOn0ZxpYMIdvhWpQVjRuBS5APe6qlc69AXaF4,1028
 agenta/client/backend/types/create_app_output.py,sha256=pgnTnfZx35Q-8wZ1yTZBQ0ydYacGzFC9kyLug_UvymM,986
-agenta/client/backend/types/create_span.py,sha256=Ldb2zclVtVyBRKxM2Ap3YjE-FP3jbaOya96_ZsYw1cg,1794
+agenta/client/backend/types/create_span.py,sha256=CfTLCHRnDyicJS3-cioDM5ZWZQq8SEi6z3BEDfnQ9Hs,1913
 agenta/client/backend/types/create_trace_response.py,sha256=FO-Ii9JEn2AQ1nmZYmjnKRbACsNxRvY_-xn7Ys7Yo8A,1012
 agenta/client/backend/types/docker_env_vars.py,sha256=altCvA1k-zdAkKNYLwaCnmV48HZg9cwe2cHu_BGImac,986
 agenta/client/backend/types/environment_output.py,sha256=dl0GKodeqB7kWK5mH6Y4iBppkpwRzSTmtkXH1II4L6w,1257
@@ -85,7 +85,7 @@ agenta/client/backend/types/image.py,sha256=p7Vmp7HlMV3YyXe8SFdXYJjCbPNIypW6NfVG
 agenta/client/backend/types/invite_request.py,sha256=1nJTUHspzw2WYpUSd4UUtRnjDHM-dqDBvYewgw-hCQE,993
 agenta/client/backend/types/list_api_keys_response.py,sha256=ZNh7jKwHEMKNp8OV5WJ5XxtKn39DwqK1f8vlFKl54x4,1097
 agenta/client/backend/types/llm_run_rate_limit.py,sha256=mfT4lTczPxrJvd8ZCOAjPvw58QoM151p_uZT0PWNOJ4,1045
-agenta/client/backend/types/llm_tokens.py,sha256=J236Fgmz5TeFO0MQA1ZA1QozvR6d3kt8aEUyWkq3jLI,1070
+agenta/client/backend/types/llm_tokens.py,sha256=nDaiJRZs1jvsXVDDoc-CXGNLUP8hRhioY37P1CGEav0,1082
 agenta/client/backend/types/new_human_evaluation.py,sha256=lIgMjVccSp22RRfMGGLH4-yKjMtJeQvjhlwX9EtAxmY,1150
 agenta/client/backend/types/new_testset.py,sha256=9NOC1-f_UZASy4ptzidLNcRU6Odq609ayvSQxEva-40,1009
 agenta/client/backend/types/organization.py,sha256=vJf6Gbz8WCnqabPQmt_t_gfrWPpuvTXgTxKCJKJsrmc,1218
@@ -126,22 +126,22 @@ agenta/docker/docker-assets/entrypoint.sh,sha256=29XK8VQjQsx4hN2j-4JDy-6kQb5y4LC
 agenta/docker/docker-assets/lambda_function.py,sha256=h4UZSSfqwpfsCgERv6frqwm_4JrYu9rLz3I-LxCfeEg,83
 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/__init__.py,sha256=ewYNjm6AHlqkIrPfX2D_pXZMwShOdhEUcWXb7xGA2bk,769
 agenta/sdk/agenta_init.py,sha256=8MfDuypxohd0qRTdtGjX7L17KW-1UGmzNVdiqF15_ak,9790
 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
-agenta/sdk/decorators/llm_entrypoint.py,sha256=umcniOOQfKVdPgHVb_jRhoGKei14Yc3PIZmEC8CU2Wg,22996
-agenta/sdk/decorators/tracing.py,sha256=c9LwQJkhJcyO7Uq-sNpDSwfwOUTAmlqNuJjz4bSx-k0,6172
+agenta/sdk/decorators/llm_entrypoint.py,sha256=ukdw3jyKarkXHKNaQyQmzVnJsYCntzmgoAFccvCZQnA,27343
+agenta/sdk/decorators/tracing.py,sha256=BzXa_2b2VvbO5ZNTh126OkL3LFq8dMK-QFug8BHUBKA,2687
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/tracing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/tracing/callbacks.py,sha256=0rqkW-PGZSaNSMp_t4bGI7R9HQRgTyIy0gxmpGVJWpE,6915
+agenta/sdk/tracing/callbacks.py,sha256=xWP3RRtmj3ogXeqYI6zdj6gF5QGu70UlQ17WNwdXIx8,8063
 agenta/sdk/tracing/context_manager.py,sha256=HskDaiORoOhjeN375gm05wYnieQzh5UnoIsnSAHkAyc,252
-agenta/sdk/tracing/llm_tracing.py,sha256=OL9OZ9sKv58hAsSxf33E-VYFx2ZeJpr9jqe3Hvz1CX8,13706
+agenta/sdk/tracing/llm_tracing.py,sha256=A7_-Xx3Rioh_nfT7OjZvDPbKUvxEqruKhYRfdGwyo4w,17312
 agenta/sdk/tracing/logger.py,sha256=GfH7V-jBHcn7h5dbdrnkDMe_ml3wkXFBeoQiqR4KVRc,474
-agenta/sdk/tracing/tasks_manager.py,sha256=ROrWIaqS2J2HHiJtRWiHKlLY8CCsqToP5VeXu7mamck,3748
-agenta/sdk/tracing/tracing_context.py,sha256=EOi1zfqpb2cBjhBtHIphUkVHi4jiWes-RRNdbgk1kMc,906
-agenta/sdk/types.py,sha256=KMnQUOdjaHSWctDLIiMHnk0o3c-C47Vm4Mn2kIZ88YI,5740
+agenta/sdk/tracing/tasks_manager.py,sha256=FBSFOWIKBycyA4ShB2ZVMzrzYQ8pWGWWBClFX8nlZFA,3726
+agenta/sdk/tracing/tracing_context.py,sha256=nt3ewa-TK9BRJviGIZYazsAQUiG4daWxjtsbjeaDprs,789
+agenta/sdk/types.py,sha256=1rVy8ob-rTOrIFcSSseXrt0JQtqqhlJfVgVxCB2ErCk,5754
 agenta/sdk/utils/debug.py,sha256=QyuPsSoN0425UD13x_msPxSF_VT6YwHiQunZUibI-jg,2149
 agenta/sdk/utils/globals.py,sha256=JmhJcCOSbwvjQ6GDyUc2_SYR27DZk7YcrRH80ktHHOM,435
 agenta/sdk/utils/helper/openai_cost.py,sha256=1VkgvucDnNZm1pTfcVLz9icWunntp1d7zwMmnviy3Uw,5877
@@ -161,7 +161,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.9.dist-info/METADATA,sha256=y9sN35sLh1Mj04gwuGUvZMYC_2KSo935v2NgP3R-Isw,26460
-agenta-0.19.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-agenta-0.19.9.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.19.9.dist-info/RECORD,,
+agenta-0.20.0a1.dist-info/METADATA,sha256=s33kh03kzj-9YuQZipN4uhW57Tyhoj9O5g-g70w_aCY,26462
+agenta-0.20.0a1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+agenta-0.20.0a1.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.20.0a1.dist-info/RECORD,,