agenta 0.24.2__py3-none-any.whl → 0.24.2a1__py3-none-any.whl

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

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

@@ -56,7 +56,7 @@ agenta/client/backend/types/body_import_testset.py,sha256=J-AH8SkA6mWgNZ5vIz12xp
 agenta/client/backend/types/config_db.py,sha256=nCM_3WKTCP3zko26wys5wTSn2cun9OYclwMp_vY5qjE,666
 agenta/client/backend/types/correct_answer.py,sha256=khEHspf9nszpzhswc7rSruCrcmvVdHFm7ti8JRdpXyI,592
 agenta/client/backend/types/create_app_output.py,sha256=0_avs1u3pIU14P5ut1WOUrravDwpg2GkSj8kYsAMK9Y,600
-agenta/client/backend/types/create_span.py,sha256=ka5f5ia8N_VjlSqdkapbDGaYobdqfduKCLNP6clbPig,1567
+agenta/client/backend/types/create_span.py,sha256=IKpkEmZiK_D6xot-pQYGKbbD5iOGRvuew9ikY-7EQpo,1565
 agenta/client/backend/types/create_trace_response.py,sha256=Pr1NM9nLHiPVzCP8m7VQ1NIb92fOhJTmMlEUK6FrykI,643
 agenta/client/backend/types/docker_env_vars.py,sha256=td2vhkHyYCwIC_TjlUu7hDuIXsfSJH2L7J_wGaNvw_c,600
 agenta/client/backend/types/environment_output.py,sha256=BuQNgdo07gAlHgC8RSV3Aq-IRpSwi3jai-jbyrTi8OQ,913
@@ -144,7 +144,7 @@ agenta/sdk/decorators/tracing.py,sha256=e0olx2EEdjXY0NqpIoDJSVxCnUmv0woewTUuCJXy
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/tracing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/tracing/callbacks.py,sha256=ug9xeXLgJEpx4l3x1BrJD2bFrH83I3kajlGsU8WtG4U,7699
-agenta/sdk/tracing/llm_tracing.py,sha256=5EeIuRu2wgGknlsZD-AkuJH_gOBrmsrCrWK0-h0Uov0,17941
+agenta/sdk/tracing/llm_tracing.py,sha256=R_EBG9-anIGlxPNZcFHATh3O7SEmIUjf0ZGtiK6eFO4,18116
 agenta/sdk/tracing/logger.py,sha256=GfH7V-jBHcn7h5dbdrnkDMe_ml3wkXFBeoQiqR4KVRc,474
 agenta/sdk/tracing/tasks_manager.py,sha256=FBSFOWIKBycyA4ShB2ZVMzrzYQ8pWGWWBClFX8nlZFA,3726
 agenta/sdk/tracing/tracing_context.py,sha256=nt3ewa-TK9BRJviGIZYazsAQUiG4daWxjtsbjeaDprs,789
@@ -169,7 +169,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.24.2.dist-info/METADATA,sha256=7YOxwqvE0HYtGOaVWpu3wnt6ZL4qsP0fvvAw9sNm1NM,26499
-agenta-0.24.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-agenta-0.24.2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.24.2.dist-info/RECORD,,
+agenta-0.24.2a1.dist-info/METADATA,sha256=FXUIOTkixRaNN7vzg4Am9xWdtchfOnFGAzyICo5Thc4,26501
+agenta-0.24.2a1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+agenta-0.24.2a1.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.24.2a1.dist-info/RECORD,,