agenta 0.27.0a2__py3-none-any.whl → 0.27.0a6__py3-none-any.whl

agenta/sdk/tracing/inline.py

@@ -11,82 +11,11 @@ from enum import Enum
 from collections import OrderedDict
 
 
-NOF_CHARS = 8
-
-
-def _p_id(id):
-    return repr(str(id)[:NOF_CHARS])
-
-
-def _p_osa(o):
-    elements = []
-
-    for i in OrderedDict(sorted(o.items())).items():
-        if not i[0].startswith("_"):
-            if i[1].__class__.__module__ != "builtins":
-                if repr(i[1]).startswith("<"):
-                    elements.append(f"{i[0]}: {i[1].name}")
-                elif repr(i[1]).startswith("UUID("):
-                    elements.append(f"{i[0]}: {_p_id(i[1])}")
-                else:
-                    elements.append(f"{i[0]}: {i[1].__str__()}")
-            else:
-                if isinstance(i[1], list):
-                    elements.append(
-                        f"{i[0]}: [" + ", ".join([el.__str__() for el in i[1]]) + "]"
-                    )
-                elif isinstance(i[1], dict):
-                    elements.append(f"{i[0]}: {{{_p_osa(i[1])}}}")
-                else:
-                    if i[1] is not None:
-                        if i[0] == "slug":
-                            elements.append(f"{i[0]}: {repr(i[1][:8])}")
-                        else:
-                            elements.append(f"{i[0]}: {repr(i[1])}")
-
-    return ", ".join(elements)
-
-
-def _p_ora(o, open="{", close="}", sep=": ", foo=repr):
-    if o.__class__.__module__ != "builtins":
-        if o.__class__.__name__ == "UUID":
-            return repr(o)
-        if isinstance(o, Enum):
-            return o
-        if isinstance(o, datetime):
-            return o.isoformat()
-        return f"{o.__class__.__name__}({_p_ora(o.__dict__, open='', close='', sep='=', foo=lambda x : x)})"
-    elif isinstance(o, list):
-        return f"[{', '.join([repr(el) for el in o])}]"
-    elif isinstance(o, dict):
-        o = OrderedDict(sorted(o.items()))
-        return f"{open}{', '.join([f'{foo(elk)}{sep}{_p_ora(elv)}' for elk, elv in o.items()])}{close}"
-    else:
-        if o is not None:
-            return repr(o)
-
-
-def _str(o):
-    return f"{{{_p_osa(o.__dict__)}}}"
-
-
-def _repr(o):
-    return _p_ora(o)
-
-
-class DisplayBase(BaseModel):
-    def __str__(self):
-        return _str(self)
-
-    def __repr__(self):
-        return _repr(self)
-
-
-class ProjectScopeDTO(DisplayBase):
+class ProjectScopeDTO(BaseModel):
     project_id: UUID
 
 
-class LifecycleDTO(DisplayBase):
+class LifecycleDTO(BaseModel):
     created_at: datetime
     updated_at: Optional[datetime] = None
 
@@ -102,14 +31,14 @@ class LifecycleDTO(DisplayBase):
 ### services.observability.dtos ###
 ### --------------------------- ###
 
-from typing import List, Dict, Any, Union, Optional, Sequence
+from typing import List, Dict, Any, Union, Optional
 
 from enum import Enum
 from datetime import datetime
 from uuid import UUID
 
 
-class TimeDTO(DisplayBase):
+class TimeDTO(BaseModel):
     start: datetime
     end: datetime
     span: int
@@ -121,7 +50,7 @@ class StatusCode(Enum):
     ERROR = "ERROR"
 
 
-class StatusDTO(DisplayBase):
+class StatusDTO(BaseModel):
     code: StatusCode
     message: Optional[str] = None
     stacktrace: Optional[str] = None
@@ -155,16 +84,16 @@ class NodeType(Enum):
     # --- VARIANTS --- #
 
 
-class RootDTO(DisplayBase):
+class RootDTO(BaseModel):
     id: UUID
 
 
-class TreeDTO(DisplayBase):
+class TreeDTO(BaseModel):
     id: UUID
     type: Optional[TreeType] = None
 
 
-class NodeDTO(DisplayBase):
+class NodeDTO(BaseModel):
     id: UUID
     type: Optional[NodeType] = None
     name: str
@@ -177,13 +106,13 @@ Tags = Dict[str, str]
 Refs = Dict[str, str]
 
 
-class LinkDTO(DisplayBase):
+class LinkDTO(BaseModel):
     type: str
     id: UUID
     tree_id: Optional[UUID] = None
 
 
-class ParentDTO(DisplayBase):
+class ParentDTO(BaseModel):
     id: UUID
 
 
@@ -205,25 +134,25 @@ class OTelStatusCode(Enum):
     STATUS_CODE_UNSET = "STATUS_CODE_UNSET"
 
 
-class OTelContextDTO(DisplayBase):
+class OTelContextDTO(BaseModel):
     trace_id: str
     span_id: str
 
 
-class OTelEventDTO(DisplayBase):
+class OTelEventDTO(BaseModel):
     name: str
     timestamp: datetime
 
     attributes: Optional[Attributes] = None
 
 
-class OTelLinkDTO(DisplayBase):
+class OTelLinkDTO(BaseModel):
     context: OTelContextDTO
 
     attributes: Optional[Attributes] = None
 
 
-class OTelExtraDTO(DisplayBase):
+class OTelExtraDTO(BaseModel):
     kind: Optional[str] = None
 
     attributes: Optional[Attributes] = None
@@ -231,7 +160,7 @@ class OTelExtraDTO(DisplayBase):
     links: Optional[List[OTelLinkDTO]] = None
 
 
-class SpanDTO(DisplayBase):
+class SpanDTO(BaseModel):
     scope: ProjectScopeDTO
 
     lifecycle: Optional[LifecycleDTO] = None
@@ -258,7 +187,7 @@ class SpanDTO(DisplayBase):
     nodes: Optional[Dict[str, Union["SpanDTO", List["SpanDTO"]]]] = None
 
 
-class OTelSpanDTO(DisplayBase):
+class OTelSpanDTO(BaseModel):
     context: OTelContextDTO
 
     name: str

agenta/sdk/tracing/processors.py

@@ -1,5 +1,4 @@
 from typing import Optional, Dict
-from traceback import format_exc
 
 from opentelemetry.context import Context
 from opentelemetry.sdk.trace import Span
@@ -7,7 +6,6 @@ from opentelemetry.sdk.trace.export import (
     SpanExporter,
     ReadableSpan,
     BatchSpanProcessor,
-    _DEFAULT_EXPORT_TIMEOUT_MILLIS,
     _DEFAULT_MAX_QUEUE_SIZE,
 )
 
@@ -38,7 +36,11 @@ class TraceProcessor(BatchSpanProcessor):
         self._exporter = span_exporter
         self.references = references or dict()
 
-    def on_start(self, span: Span, parent_context: Optional[Context] = None) -> None:
+    def on_start(
+        self,
+        span: Span,
+        parent_context: Optional[Context] = None,
+    ) -> None:
         # ADD LINKS FROM CONTEXT, HERE
 
         for key in self.references.keys():
@@ -49,7 +51,10 @@ class TraceProcessor(BatchSpanProcessor):
 
         self._registry[span.context.trace_id][span.context.span_id] = True
 
-    def on_end(self, span: ReadableSpan):
+    def on_end(
+        self,
+        span: ReadableSpan,
+    ):
         super().on_end(span)
 
         del self._registry[span.context.trace_id][span.context.span_id]
@@ -57,7 +62,10 @@ class TraceProcessor(BatchSpanProcessor):
         if self.is_ready(span.get_span_context().trace_id):
             self.force_flush()
 
-    def force_flush(self, timeout_millis: int = None) -> bool:
+    def force_flush(
+        self,
+        timeout_millis: int = None,
+    ) -> bool:
         ret = super().force_flush(timeout_millis)
 
         if not ret:
@@ -65,12 +73,18 @@ class TraceProcessor(BatchSpanProcessor):
             log.error("Agenta SDK - skipping export due to timeout.")
             log.error("--------------------------------------------")
 
-    def is_ready(self, trace_id: Optional[int] = None) -> bool:
+    def is_ready(
+        self,
+        trace_id: Optional[int] = None,
+    ) -> bool:
         is_ready = not len(self._registry.get(trace_id, {}))
 
         return is_ready
 
-    def fetch(self, trace_id: Optional[int] = None) -> Dict[str, ReadableSpan]:
+    def fetch(
+        self,
+        trace_id: Optional[int] = None,
+    ) -> Dict[str, ReadableSpan]:
         trace = self._exporter.fetch(trace_id)  # type: ignore
 
         return trace

agenta/sdk/tracing/spans.py

@@ -8,7 +8,10 @@ from agenta.sdk.tracing.attributes import serialize
 
 
 class CustomSpan(Span):  # INHERITANCE FOR TYPING ONLY
-    def __init__(self, span: Span) -> None:
+    def __init__(
+        self,
+        span: Span,
+    ) -> None:
         super().__init__(  # INHERITANCE FOR TYPING ONLY
             name=span.name,
             context=span.context,
@@ -38,7 +41,10 @@ class CustomSpan(Span):  # INHERITANCE FOR TYPING ONLY
     def is_recording(self) -> bool:
         return self._span.is_recording()
 
-    def update_name(self, name: str) -> None:
+    def update_name(
+        self,
+        name: str,
+    ) -> None:
         self._span.update_name(name)
 
     def set_status(
@@ -46,7 +52,10 @@ class CustomSpan(Span):  # INHERITANCE FOR TYPING ONLY
         status: Union[Status, StatusCode],
         description: Optional[str] = None,
     ) -> None:
-        self._span.set_status(status=status, description=description)
+        self._span.set_status(
+            status=status,
+            description=description,
+        )
 
     def end(self) -> None:
         self._span.end()
@@ -73,7 +82,10 @@ class CustomSpan(Span):  # INHERITANCE FOR TYPING ONLY
         value: Any,
         namespace: Optional[str] = None,
     ) -> None:
-        self.set_attributes({key: value}, namespace)
+        self.set_attributes(
+            attributes={key: value},
+            namespace=namespace,
+        )
 
     def add_event(
         self,

agenta/sdk/tracing/tracing.py

@@ -1,5 +1,8 @@
 from typing import Optional, Any, Dict
 from enum import Enum
+from uuid import UUID
+
+# from traceback import format_exc
 
 from httpx import get as check
 
@@ -16,12 +19,10 @@ from opentelemetry.sdk.resources import Resource
 from agenta.sdk.utils.singleton import Singleton
 from agenta.sdk.utils.exceptions import suppress
 from agenta.sdk.utils.logging import log
-
 from agenta.sdk.tracing.processors import TraceProcessor
-from agenta.sdk.tracing.exporters import ConsoleExporter, InlineExporter, OTLPExporter
+from agenta.sdk.tracing.exporters import InlineExporter, OTLPExporter
 from agenta.sdk.tracing.spans import CustomSpan
 from agenta.sdk.tracing.inline import parse_inline_trace
-
 from agenta.sdk.tracing.conventions import Reference, is_valid_attribute_key
 
 
@@ -59,7 +60,7 @@ class Tracing(metaclass=Singleton):
         self,
         project_id: Optional[str] = None,
         api_key: Optional[str] = None,
-        #
+        # DEPRECATING
         app_id: Optional[str] = None,
     ):
         # AUTHENTICATION (OTLP)
@@ -75,53 +76,59 @@ class Tracing(metaclass=Singleton):
             self.headers.update(**{"AG-APP-ID": app_id})
         if api_key:
             self.headers.update(**{"Authorization": self.api_key})
+
         # REFERENCES
-        self.references = {"application.id": app_id}
+        self.references["application.id"] = app_id
 
         # TRACER PROVIDER
         self.tracer_provider = TracerProvider(
             resource=Resource(attributes={"service.name": "agenta-sdk"})
         )
-        # TRACE PROCESSORS -- CONSOLE
-        # _console = TraceProcessor(
-        #    ConsoleExporter(),
-        #    references=self.references,
-        # )
-        # self.tracer_provider.add_span_processor(_console)
         # TRACE PROCESSORS -- INLINE
         self.inline = TraceProcessor(
-            InlineExporter(registry=self.inline_spans),
+            InlineExporter(
+                registry=self.inline_spans,
+            ),
             references=self.references,
         )
         self.tracer_provider.add_span_processor(self.inline)
         # TRACE PROCESSORS -- OTLP
         try:
-            log.info(f"Connecting to the remote trace receiver at {self.otlp_url}...")
-
-            check(self.otlp_url, headers=self.headers, timeout=1)
-
-            log.info(f"Connection established.")
+            log.info("--------------------------------------------")
+            log.info(f"Agenta SDK - connecting to otlp receiver at: {self.otlp_url}")
+            log.info("--------------------------------------------")
+
+            check(
+                self.otlp_url,
+                headers=self.headers,
+                timeout=1,
+            )
 
             _otlp = TraceProcessor(
-                OTLPExporter(endpoint=self.otlp_url, headers=self.headers),
+                OTLPExporter(
+                    endpoint=self.otlp_url,
+                    headers=self.headers,
+                ),
                 references=self.references,
             )
 
             self.tracer_provider.add_span_processor(_otlp)
-        except Exception as e:
-            log.error(e)
-            log.warning(f"Connection failed.")
-            log.warning(
-                f"Warning: Your traces will not be exported since {self.otlp_url} is unreachable."
-            )
+
+            log.info(f"Success: traces will be exported.")
+            log.info("--------------------------------------------")
+
+        except:
+            # log.warning(format_exc().strip("\n"))
+            # log.warning("--------------------------------------------")
+            log.warning(f"Failure: traces will not be exported.")
+            log.warning("--------------------------------------------")
+
         # GLOBAL TRACER PROVIDER -- INSTRUMENTATION LIBRARIES
         set_tracer_provider(self.tracer_provider)
         # TRACER
         self.tracer: Tracer = self.tracer_provider.get_tracer("agenta.tracer")
 
-    def get_current_span(
-        self,
-    ):
+    def get_current_span(self):
         _span = None
 
         with suppress():
@@ -141,7 +148,10 @@ class Tracing(metaclass=Singleton):
             if span is None:
                 span = self.get_current_span()
 
-            span.set_attributes(attributes={"internals": attributes}, namespace="data")
+            span.set_attributes(
+                attributes={"internals": attributes},
+                namespace="data",
+            )
 
     def store_refs(
         self,
@@ -154,12 +164,22 @@ class Tracing(metaclass=Singleton):
 
             for key in refs.keys():
                 if key in Reference:
+                    # TYPE AND FORMAT CHECKING
+                    if key.endswith(".id"):
+                        try:
+                            refs[key] = str(UUID(refs[key]))
+                        except:
+                            refs[key] = None
+
+                    refs[key] = str(refs[key])
+
                     # ADD REFERENCE TO THIS SPAN
                     span.set_attribute(
                         key.value if isinstance(key, Enum) else key,
                         refs[key],
                         namespace="refs",
                     )
+
                     # AND TO ALL SPANS CREATED AFTER THIS ONE
                     self.references[key] = refs[key]
                     # TODO: THIS SHOULD BE REPLACED BY A TRACE CONTEXT !!!
@@ -175,7 +195,11 @@ class Tracing(metaclass=Singleton):
 
             for key in meta.keys():
                 if is_valid_attribute_key(key):
-                    span.set_attribute(key, meta[key], namespace="meta")
+                    span.set_attribute(
+                        key,
+                        meta[key],
+                        namespace="meta",
+                    )
 
     def store_metrics(
         self,
@@ -188,7 +212,11 @@ class Tracing(metaclass=Singleton):
 
             for key in metrics.keys():
                 if is_valid_attribute_key(key):
-                    span.set_attribute(key, metrics[key], namespace="metrics")
+                    span.set_attribute(
+                        key,
+                        metrics[key],
+                        namespace="metrics",
+                    )
 
     def is_inline_trace_ready(
         self,

{agenta-0.27.0a2.dist-info → agenta-0.27.0a6.dist-info}/METADATA

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

{agenta-0.27.0a2.dist-info → agenta-0.27.0a6.dist-info}/RECORD

@@ -141,20 +141,20 @@ agenta/sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 agenta/sdk/context/routing.py,sha256=gOoOM88hSjIjzQ3ni68TjUoq7WtwSH3kB07YjCYvW2c,631
 agenta/sdk/context/tracing.py,sha256=UmmW15UFFsvxS0myS6aD9wBk5iNepNlQi4tEQ_ejfYM,96
 agenta/sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/decorators/routing.py,sha256=L3_bYJ254x6uB4d8_TK47jKIsHrz4kibf3Dlty9UHQM,36026
-agenta/sdk/decorators/tracing.py,sha256=-A1GjhPs9OicDtXidtxAEGnUJRyuzZR1oHA95rsNrzw,11457
+agenta/sdk/decorators/routing.py,sha256=ERtZDJoCoR_s73gOt0h_Z_Kr8FCskMkrl_ILD85VV04,35873
+agenta/sdk/decorators/tracing.py,sha256=iXcQKoQ3VcV-FBOSUJ6UXChGaYkOQf20VMrfkfme0LI,7564
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
 agenta/sdk/litellm/litellm.py,sha256=j4WyRTQdxLkvelaPm5BfUbaUxkbQXDhyr3yp9yXo1RQ,8289
 agenta/sdk/router.py,sha256=mOguvtOwl2wmyAgOuWTsf98pQwpNiUILKIo67W_hR3A,119
 agenta/sdk/tracing/__init__.py,sha256=rQNe5-zT5Kt7_CDhq-lnUIi1EYTBVzVf_MbfcIxVD98,41
-agenta/sdk/tracing/attributes.py,sha256=eYVzsAt_2ekDCO_RQ7VrZHuySftFbcRPKaf9lEvXx0E,3815
-agenta/sdk/tracing/context.py,sha256=KDQYjRcmbpb6e0xEuX-bAQGhEjwx4H2nwybRPOKCDjo,578
-agenta/sdk/tracing/conventions.py,sha256=W7-FVjLjnBGiY9cvxFzW4tlsdFwW9l5LLGm8AgMynmM,938
-agenta/sdk/tracing/exporters.py,sha256=hTGbdkuH1rIqT7aRsSIYolQPCq2MxmEfT9IGBn20mRU,1370
-agenta/sdk/tracing/inline.py,sha256=OtYkGMO2XKdLH7t2vqVqteLs8zQCGyM93sCIU-xYD8Y,35163
-agenta/sdk/tracing/processors.py,sha256=PpCJcyhZdr_hNjA_QlM6jtTZj6neUMdA_C5rnx97BfM,2390
-agenta/sdk/tracing/spans.py,sha256=AnQuuyuCGST6n_He4SF5WdrtMts9JlE8v6rVRdm2rRg,3570
-agenta/sdk/tracing/tracing.py,sha256=XIBYRAsQ6bb0hkcybKJNP0uVaPl3dE4NcTYzwlwAOOA,6942
+agenta/sdk/tracing/attributes.py,sha256=zh8JQZSeYCLBeIRSopKJx6QQ-WEgw08Cr64DS_WOcT8,3833
+agenta/sdk/tracing/context.py,sha256=PSJdhcaOXSMAuGUBySpLKPKyx8duF3TJzhUEk2ufqPc,777
+agenta/sdk/tracing/conventions.py,sha256=JBtznBXZ3aRkGKkLl7cPwdMNh3w1G-H2Ta2YrAxbr38,950
+agenta/sdk/tracing/exporters.py,sha256=LzNGjOARwRrJ5gTFjdG9ziZlqnA0WeMjpkUfpsl8BSs,1383
+agenta/sdk/tracing/inline.py,sha256=imtlQ0rqfX1rD-SYo1YVAZCctRjxZskdfeA1Pm__wcw,32980
+agenta/sdk/tracing/processors.py,sha256=uqH8AKdRJkv-MY5FcTD-Qrrr0ARMlAAsImPPKolteQ0,2444
+agenta/sdk/tracing/spans.py,sha256=nqUOjjirBxB8Eacv8Qj4Ra_6rknGi3lbJdNyKmk5ODQ,3707
+agenta/sdk/tracing/tracing.py,sha256=g8LKWOLdeEM6HifrmZAdVYi31-sNFAamioaVwUQ-qrs,7712
 agenta/sdk/types.py,sha256=B7C7Jpeq0CpmLwmN1dmpggJsZX8WpxyyM42PkdSsx3s,5669
 agenta/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/utils/costs.py,sha256=i8C7ud__pThLS55XkN4YW8czXtGeXr2mx7jjcOFeiXg,5955
@@ -179,7 +179,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.27.0a2.dist-info/METADATA,sha256=ApO4L8qDn3H-cS4dfvbP2g5yyieP1Rc-rkUfWsdfRn0,31738
-agenta-0.27.0a2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-agenta-0.27.0a2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.27.0a2.dist-info/RECORD,,
+agenta-0.27.0a6.dist-info/METADATA,sha256=P8n7lccwkGHLB_YYeKmr_mpCr41oqW4eC_QRJC8PrXs,31738
+agenta-0.27.0a6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+agenta-0.27.0a6.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.27.0a6.dist-info/RECORD,,