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

agenta/cli/helper.py

@@ -15,6 +15,9 @@ import toml
 from agenta.client.backend.client import AgentaApi
 
 BACKEND_URL_SUFFIX = os.environ.get("BACKEND_URL_SUFFIX", "api")
+POSTHOG_KEY = os.environ.get(
+    "POSTHOG_KEY", "phc_hmVSxIjTW1REBHXgj2aw4HW9X6CXb6FzerBgP9XenC7"
+)
 
 
 def get_global_config(var_name: str) -> Optional[Any]:
@@ -111,7 +114,8 @@ def init_telemetry_config() -> None:
     ):
         set_global_config("telemetry_tracking_enabled", True)
         set_global_config(
-            "telemetry_api_key", "phc_hmVSxIjTW1REBHXgj2aw4HW9X6CXb6FzerBgP9XenC7"
+            "telemetry_api_key",
+            POSTHOG_KEY,
         )
 
 

agenta/sdk/agenta_init.py

@@ -31,7 +31,6 @@ class AgentaSingleton:
         self,
         *,
         host: Optional[str] = None,
-        project_id: Optional[str] = None,
         api_key: Optional[str] = None,
         config_fname: Optional[str] = None,
         #
@@ -84,24 +83,17 @@ class AgentaSingleton:
                 "3. As an environment variable 'AGENTA_APP_ID'."
             )
 
-        self.project_id = (
-            project_id
-            or os.environ.get("AGENTA_PROJECT_ID")
-            or config.get("project_id")
-        )
-
         self.api_key = (
             api_key or os.environ.get("AGENTA_API_KEY") or config.get("api_key")
         )
 
         self.tracing = Tracing(
-            url=f"{self.host}/api/observability/v1/traces",  # type: ignore
+            url=f"{self.host}/api/observability/v1/otlp/traces",  # type: ignore
         )
 
         self.tracing.configure(
-            project_id=self.project_id,
             api_key=self.api_key,
-            # DEPRECATED
+            # DEPRECATING
             app_id=self.app_id,
         )
 
@@ -252,7 +244,6 @@ class Config:
 
 def init(
     host: Optional[str] = None,
-    project_id: Optional[str] = None,
     api_key: Optional[str] = None,
     config_fname: Optional[str] = None,
     # DEPRECATED
@@ -284,7 +275,6 @@ def init(
 
     singleton.init(
         host=host,
-        project_id=project_id,
         api_key=api_key,
         config_fname=config_fname,
         # DEPRECATED

agenta/sdk/tracing/inline.py

@@ -161,7 +161,7 @@ class OTelExtraDTO(BaseModel):
 
 
 class SpanDTO(BaseModel):
-    scope: ProjectScopeDTO
+    scope: Optional[ProjectScopeDTO] = None
 
     lifecycle: Optional[LifecycleDTO] = None
 
@@ -755,11 +755,8 @@ def _parse_from_attributes(
 
 
 def parse_from_otel_span_dto(
-    project_id: str,
     otel_span_dto: OTelSpanDTO,
 ) -> SpanDTO:
-    scope = ProjectScopeDTO(project_id=UUID(project_id))
-
     lifecyle = LifecycleDTO(
         created_at=datetime.now(),
     )
@@ -831,7 +828,6 @@ def parse_from_otel_span_dto(
     )
 
     span_dto = SpanDTO(
-        scope=scope,
         lifecycle=lifecyle,
         root=root,
         tree=tree,
@@ -902,7 +898,6 @@ from opentelemetry.sdk.trace import ReadableSpan
 
 
 def parse_inline_trace(
-    project_id: str,
     spans: Dict[str, ReadableSpan],
 ):
     otel_span_dtos = _parse_readable_spans(spans)
@@ -911,8 +906,7 @@ def parse_inline_trace(
     ### apis.fastapi.observability.api.otlp_collect_traces() ###
     ### ---------------------------------------------------- ###
     span_dtos = [
-        parse_from_otel_span_dto(project_id, otel_span_dto)
-        for otel_span_dto in otel_span_dtos
+        parse_from_otel_span_dto(otel_span_dto) for otel_span_dto in otel_span_dtos
     ]
     ### ---------------------------------------------------- ###
     ### apis.fastapi.observability.api.otlp_collect_traces() ###

agenta/sdk/tracing/tracing.py

@@ -2,8 +2,6 @@ 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
 
 from opentelemetry.trace import (
@@ -38,9 +36,7 @@ class Tracing(metaclass=Singleton):
     ) -> None:
         # ENDPOINT (OTLP)
         self.otlp_url = url
-        # AUTHENTICATION (OTLP)
-        self.project_id: Optional[str] = None
-        # AUTHORIZATION (OTLP)
+        # AITH (OTLP)
         self.api_key: Optional[str] = None
         # HEADERS (OTLP)
         self.headers: Dict[str, str] = dict()
@@ -49,6 +45,8 @@ class Tracing(metaclass=Singleton):
 
         # TRACER PROVIDER
         self.tracer_provider: Optional[TracerProvider] = None
+        # TRACE PROCESSORS -- INLINE
+        self.inline: Optional[TraceProcessor] = None
         # TRACER
         self.tracer: Optional[Tracer] = None
         # INLINE SPANS for INLINE TRACES (INLINE PROCESSOR)
@@ -58,25 +56,18 @@ class Tracing(metaclass=Singleton):
 
     def configure(
         self,
-        project_id: Optional[str] = None,
         api_key: Optional[str] = None,
         # DEPRECATING
         app_id: Optional[str] = None,
     ):
-        # AUTHENTICATION (OTLP)
-        self.project_id = project_id  # "f7943e42-ec69-498e-bf58-8db034b9286e"
-        self.app_id = app_id
-        # AUTHORIZATION (OTLP)
+        # AUTH (OTLP)
         self.api_key = api_key
         # HEADERS (OTLP)
         self.headers = {}
-        if project_id:
-            self.headers.update(**{"AG-PROJECT-ID": project_id})
         if app_id:
             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
 
@@ -95,7 +86,10 @@ class Tracing(metaclass=Singleton):
         # TRACE PROCESSORS -- OTLP
         try:
             log.info("--------------------------------------------")
-            log.info(f"Agenta SDK - connecting to otlp receiver at: {self.otlp_url}")
+            log.info(
+                "Agenta SDK - connecting to otlp receiver at: %s",
+                self.otlp_url,
+            )
             log.info("--------------------------------------------")
 
             check(
@@ -114,13 +108,11 @@ class Tracing(metaclass=Singleton):
 
             self.tracer_provider.add_span_processor(_otlp)
 
-            log.info(f"Success: traces will be exported.")
+            log.info("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.")
+        except:  # pylint: disable=bare-except
+            log.warning("Failure: traces will not be exported.")
             log.warning("--------------------------------------------")
 
         # GLOBAL TRACER PROVIDER -- INSTRUMENTATION LIBRARIES
@@ -163,12 +155,12 @@ class Tracing(metaclass=Singleton):
                 span = self.get_current_span()
 
             for key in refs.keys():
-                if key in Reference:
+                if key in [_.value for _ in Reference.__members__.values()]:
                     # TYPE AND FORMAT CHECKING
                     if key.endswith(".id"):
                         try:
                             refs[key] = str(UUID(refs[key]))
-                        except:
+                        except:  # pylint: disable=bare-except
                             refs[key] = None
 
                     refs[key] = str(refs[key])
@@ -243,14 +235,14 @@ class Tracing(metaclass=Singleton):
                 otel_spans = self.inline.fetch(trace_id)
 
                 if otel_spans:
-                    _inline_trace = parse_inline_trace(
-                        self.project_id or self.app_id, otel_spans
-                    )
+                    _inline_trace = parse_inline_trace(otel_spans)
 
         return _inline_trace
 
 
-def get_tracer(tracing: Tracing) -> Tracer:
+def get_tracer(
+    tracing: Tracing,
+) -> Tracer:
     if tracing is None or tracing.tracer is None or tracing.tracer_provider is None:
         return get_tracer_provider().get_tracer("default.tracer")
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.27.0a6
+Version: 0.27.0a8
 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.0a6.dist-info → agenta-0.27.0a8.dist-info}/RECORD

@@ -1,6 +1,6 @@
 agenta/__init__.py,sha256=6NgtqH14bL5M25Kqy2F8KueChz1snciKiIxho__i_UI,970
 agenta/cli/evaluation_commands.py,sha256=fs6492tprPId9p8eGO02Xy-NCBm2RZNJLZWcUxugwd8,474
-agenta/cli/helper.py,sha256=vRxHyeNaltzNIGrfU2vO0H28_rXDzx9QqIZ_S-W6zL4,6212
+agenta/cli/helper.py,sha256=P97HbNb_qzOyl5CM_MjAqWEBCdgebU6M81G_4UCmF1A,6288
 agenta/cli/main.py,sha256=Wz0ODhoeKK3Qg_CFUhu6D909szk05tc8ZVBB6H1-w7k,9763
 agenta/cli/telemetry.py,sha256=GaFFRsE_NtrcSSJ10r2jhgFs5Sk8gf2C09Ox3gOr3eU,1317
 agenta/cli/variant_commands.py,sha256=HfKRZsajKOXwZD2OyzjSfNtSx1yI01wI1cfqpvoHETI,17400
@@ -133,7 +133,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=kO1q2_IR0fEAo4M-2Pt_v-zC7GxxnkLogjKFhU869Ps,3555
 agenta/sdk/__init__.py,sha256=IopYEonqvpdGXsAYh2wQ4Ai_2YzsY7aF88wTzrclV4M,1208
-agenta/sdk/agenta_init.py,sha256=hBE78BIu6QFfvaxWY2x7Y7EXAFKx6TSPEtEj2S6qWNU,10606
+agenta/sdk/agenta_init.py,sha256=vQcaWvvrMR6xMNJvX6V-9eNIvx3_JsjPA6xcxiahX30,10308
 agenta/sdk/assets.py,sha256=Zv4i8MVUSB3jMODQon1mzJtYxuntmrCNjLGk8f-2fls,2856
 agenta/sdk/client.py,sha256=trKyBOYFZRk0v5Eptxvh87yPf50Y9CqY6Qgv4Fy-VH4,2142
 agenta/sdk/config_manager.py,sha256=HFOJpJKBkhlA0C-KPuxb4-bHNZeZqdpmx_beoX4lQw8,7997
@@ -151,10 +151,10 @@ agenta/sdk/tracing/attributes.py,sha256=zh8JQZSeYCLBeIRSopKJx6QQ-WEgw08Cr64DS_WO
 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/inline.py,sha256=0ZLy89k5wkmTUaz69vqc3vP51ZdqmLGMj3Qz2o3hieE,32856
 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/tracing/tracing.py,sha256=S-7rt1LMHCF3YzV1XzAnhdkeocCffj9CCwlHScU7mqU,7350
 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.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,,
+agenta-0.27.0a8.dist-info/METADATA,sha256=HDnjYrKr75LSj7gY05n9EzkuNi_y7wqCR8I7Tscm44o,31738
+agenta-0.27.0a8.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+agenta-0.27.0a8.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.27.0a8.dist-info/RECORD,,