agenta 0.24.2__tar.gz → 0.24.2a1__tar.gz

{agenta-0.24.2 → agenta-0.24.2a1}/PKG-INFO

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

{agenta-0.24.2 → agenta-0.24.2a1}/agenta/client/backend/types/create_span.py

@@ -34,11 +34,11 @@ class CreateSpan(UniversalBaseModel):
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=False
+            extra="allow", frozen=True
         )  # type: ignore # Pydantic v2
     else:
 
         class Config:
-            frozen = False
+            frozen = True
             smart_union = True
             extra = pydantic.Extra.allow

{agenta-0.24.2 → agenta-0.24.2a1}/agenta/sdk/tracing/llm_tracing.py

@@ -199,9 +199,7 @@ class Tracing(metaclass=SingletonMeta):
     def is_trace_ready(self):
         tracing = tracing_context.get()
 
-        are_spans_ready = [
-            (span.end_time == span.start_time) for span in tracing.spans.values()
-        ]
+        are_spans_ready = [span.end_time is not None for span in tracing.spans.values()]
 
         return all(are_spans_ready)
 
@@ -255,8 +253,6 @@ class Tracing(metaclass=SingletonMeta):
         logging.info(f"Opening span  {span_id} {spankind.upper()}")
 
         ### --- TO BE CLEANED --- >>>
-        now = datetime.now(timezone.utc)
-
         span = CreateSpan(
             id=span_id,
             inputs=input,
@@ -265,13 +261,13 @@ class Tracing(metaclass=SingletonMeta):
             config=config,
             spankind=spankind.upper(),
             attributes={},
-            status="UNSET",
-            start_time=now,
+            status=SpanStatusCode.UNSET.value,
+            start_time=datetime.now(timezone.utc),
             internals=None,
             outputs=None,
             tags=None,
             user=None,
-            end_time=now,
+            end_time=None,
             tokens=None,
             cost=None,
             token_consumption=None,
@@ -461,12 +457,18 @@ class Tracing(metaclass=SingletonMeta):
             trace["trace_id"] = tracing.trace_id
 
             for span in tracing.spans.values():
+                if span.end_time is None:
+                    span.end_time = span.start_time
+
                 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 - span.start_time).total_seconds()
+                    trace["latency"] = (
+                        (span.end_time if span.end_time else span.start_time)
+                        - span.start_time
+                    ).total_seconds()
 
             spans = (
                 []

{agenta-0.24.2 → agenta-0.24.2a1}/pyproject.toml

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