agenta 0.26.0a0__py3-none-any.whl → 0.27.0a0__py3-none-any.whl

agenta/sdk/tracing/processors.py

@@ -0,0 +1,65 @@
+from typing import Optional, Any, Dict
+
+from opentelemetry.context import Context
+from opentelemetry.sdk.trace import Span
+from opentelemetry.sdk.trace.export import (
+    SpanExporter,
+    ReadableSpan,
+    BatchSpanProcessor,
+    _DEFAULT_EXPORT_TIMEOUT_MILLIS,
+    _DEFAULT_MAX_QUEUE_SIZE,
+)
+
+# LOAD CONTEXT, HERE
+
+
+class TraceProcessor(BatchSpanProcessor):
+    def __init__(
+        self,
+        span_exporter: SpanExporter,
+        references: Dict[str, str] = None,
+        max_queue_size: int = None,
+        schedule_delay_millis: float = None,
+        max_export_batch_size: int = None,
+        export_timeout_millis: float = None,
+    ):
+        super().__init__(
+            span_exporter,
+            _DEFAULT_MAX_QUEUE_SIZE,
+            60 * 60 * 1000,  # 1 hour
+            _DEFAULT_MAX_QUEUE_SIZE,
+            _DEFAULT_EXPORT_TIMEOUT_MILLIS,
+        )
+
+        self._registry = dict()
+        self._exporter = span_exporter
+        self.references = references or dict()
+
+    def on_start(self, span: Span, parent_context: Optional[Context] = None) -> None:
+        # ADD LINKS FROM CONTEXT, HERE
+
+        for key in self.references.keys():
+            span.set_attribute(f"ag.refs.{key}", self.references[key])
+
+        if span.context.trace_id not in self._registry:
+            self._registry[span.context.trace_id] = dict()
+
+        self._registry[span.context.trace_id][span.context.span_id] = True
+
+    def on_end(self, span: ReadableSpan):
+        super().on_end(span)
+
+        del self._registry[span.context.trace_id][span.context.span_id]
+
+        if self.is_ready(span.get_span_context().trace_id):
+            self.force_flush()
+
+    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]:
+        trace = self._exporter.fetch(trace_id)  # type: ignore
+
+        return trace

agenta/sdk/tracing/spans.py

@@ -0,0 +1,124 @@
+from typing import Optional, Union, Any, Dict
+
+from opentelemetry.trace import SpanContext
+from opentelemetry.trace.status import Status, StatusCode
+from opentelemetry.sdk.trace import Span
+
+from agenta.sdk.tracing.attributes import serialize
+
+
+class CustomSpan(Span):  # INHERITANCE FOR TYPING ONLY
+    def __init__(self, span: Span) -> None:
+        super().__init__(  # INHERITANCE FOR TYPING ONLY
+            name=span.name,
+            context=span.context,
+            parent=span.parent,
+            sampler=span._sampler,
+            trace_config=span._trace_config,
+            resource=span.resource,
+            attributes=span.attributes,
+            events=span.events,
+            links=span.links,
+            kind=span.kind,
+            span_processor=span._span_processor,
+            instrumentation_info=span.instrumentation_info,
+            record_exception=span._record_exception,
+            set_status_on_exception=span._set_status_on_exception,
+            limits=span._limits,
+            instrumentation_scope=span.instrumentation_scope,
+        )
+
+        self._span = span
+
+    ## --- PROXY METHODS --- ##
+
+    def get_span_context(self):
+        return self._span.get_span_context()
+
+    def is_recording(self) -> bool:
+        return self._span.is_recording()
+
+    def update_name(self, name: str) -> None:
+        self._span.update_name(name)
+
+    def set_status(
+        self,
+        status: Union[Status, StatusCode],
+        description: Optional[str] = None,
+    ) -> None:
+        self._span.set_status(status=status, description=description)
+
+    def end(self) -> None:
+        self._span.end()
+
+    ## --- CUSTOM METHODS W/ ATTRIBUTES SERALIZATION --- ##
+
+    def set_attributes(
+        self,
+        attributes: Dict[str, Any],
+        namespace: Optional[str] = None,
+        max_depth: Optional[int] = None,
+    ) -> None:
+        self._span.set_attributes(
+            attributes=serialize(
+                namespace=namespace,
+                attributes=attributes,
+                max_depth=max_depth,
+            )
+        )
+
+    def set_attribute(
+        self,
+        key: str,
+        value: Any,
+        namespace: Optional[str] = None,
+    ) -> None:
+        self.set_attributes({key: value}, namespace)
+
+    def add_event(
+        self,
+        name: str,
+        attributes: Optional[Dict[str, Any]] = None,
+        timestamp: Optional[int] = None,
+        namespace: Optional[str] = None,
+    ) -> None:
+        self._span.add_event(
+            name=name,
+            attributes=serialize(
+                namespace=namespace,
+                attributes=attributes,
+            ),
+            timestamp=timestamp,
+        )
+
+    def add_link(
+        self,
+        context: SpanContext,
+        attributes: Optional[Dict[str, Any]] = None,
+        namespace: Optional[str] = None,
+    ) -> None:
+        self._span.add_link(
+            context=context,
+            attributes=serialize(
+                namespace=namespace,
+                attributes=attributes,
+            ),
+        )
+
+    def record_exception(
+        self,
+        exception: BaseException,
+        attributes: Optional[Dict[str, Any]] = None,
+        timestamp: Optional[int] = None,
+        escaped: bool = False,
+        namespace: Optional[str] = None,
+    ) -> None:
+        self._span.record_exception(
+            exception=exception,
+            attributes=serialize(
+                namespace=namespace,
+                attributes=attributes,
+            ),
+            timestamp=timestamp,
+            escaped=escaped,
+        )

agenta/sdk/tracing/tracing.py

@@ -0,0 +1,174 @@
+from typing import Optional, Any, Dict
+
+from httpx import get as check
+
+from opentelemetry.trace import (
+    get_current_span,
+    set_tracer_provider,
+    get_tracer_provider,
+    Status,
+    StatusCode,
+)
+from opentelemetry.sdk.trace import Span, Tracer, TracerProvider
+from opentelemetry.sdk.resources import Resource
+
+from agenta.sdk.utils.singleton import Singleton
+from agenta.sdk.utils.exceptions import suppress  # USE IT !
+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.spans import CustomSpan
+from agenta.sdk.tracing.context import tracing_context
+from agenta.sdk.tracing.inline import parse_inline_trace
+
+
+class Tracing(metaclass=Singleton):
+    VERSION = "0.1.0"
+
+    Status = Status
+    StatusCode = StatusCode
+
+    def __init__(
+        self,
+        url: str,
+    ) -> None:
+        # ENDPOINT (OTLP)
+        self.otlp_url = url
+        # AUTHENTICATION (OTLP)
+        self.project_id: Optional[str] = None
+        # AUTHORIZATION (OTLP)
+        self.api_key: Optional[str] = None
+        # HEADERS (OTLP)
+        self.headers: Dict[str, str] = dict()
+        # REFERENCES
+        self.references: Dict[str, str] = dict()
+
+        # TRACER PROVIDER
+        self.tracer_provider: Optional[TracerProvider] = None
+        # TRACER
+        self.tracer: Optional[Tracer] = None
+        # INLINE SPANS for INLINE TRACES (INLINE PROCESSOR)
+        self.inline_spans: Dict[str, Any] = dict()
+
+    # PUBLIC
+
+    def configure(
+        self,
+        project_id: Optional[str] = None,
+        api_key: Optional[str] = None,
+        #
+        app_id: Optional[str] = None,
+    ):
+        # AUTHENTICATION (OTLP)
+        self.project_id = project_id  # "f7943e42-ec69-498e-bf58-8db034b9286e"
+        self.app_id = app_id
+        # AUTHORIZATION (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}
+
+        # 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),
+            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.")
+
+            _otlp = TraceProcessor(
+                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."
+            )
+        # 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,
+    ):
+        _span = get_current_span()
+
+        if _span.is_recording():
+            return CustomSpan(_span)
+
+        return _span
+
+    def store_internals(
+        self,
+        attributes: Dict[str, Any],
+        span: Optional[Span] = None,
+    ):
+        if span is None:
+            span = self.get_current_span()
+
+        span.set_attributes(attributes={"internals": attributes}, namespace="data")
+
+    def is_inline_trace_ready(
+        self,
+        trace_id: int,
+    ) -> bool:
+        is_ready = self.inline.is_ready(trace_id)
+
+        return is_ready
+
+    def get_inline_trace(
+        self,
+        trace_id: int,
+    ) -> Dict[str, Any]:
+        if trace_id is None:
+            return {}
+
+        is_ready = self.inline.is_ready(trace_id)
+
+        if is_ready is False:
+            return {}
+
+        otel_spans = self.inline.fetch(trace_id)
+
+        if not otel_spans:
+            return {}
+
+        inline_trace = parse_inline_trace(self.project_id or self.app_id, otel_spans)
+
+        return inline_trace
+
+
+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")
+
+    return tracing.tracer

agenta/sdk/types.py

@@ -186,15 +186,3 @@ class FileInputURL(HttpUrl):
     @classmethod
     def __schema_type_properties__(cls) -> dict:
         return {"x-parameter": "file_url", "type": "string"}
-
-
-class Context(BaseModel):
-    model_config = ConfigDict(extra="allow")
-
-    def to_json(self):
-        return self.model_dump()
-
-    @classmethod
-    def from_json(cls, json_str: str):
-        data = json.loads(json_str)
-        return cls(**data)

agenta/sdk/utils/{helper/openai_cost.py → costs.py}

@@ -49,6 +49,9 @@ MODEL_COST_PER_1K_TOKENS = {
     "gpt-35-turbo-16k-completion": 0.004,
     "gpt-35-turbo-16k-0613-completion": 0.004,
     # Others
+    "text-embedding-ada-002": 0.1,
+    "text-ada-002": 0.1,
+    "adav2": 0.1,
     "text-ada-001": 0.0004,
     "ada": 0.0004,
     "text-babbage-001": 0.0005,

agenta/sdk/utils/debug.py

@@ -1,7 +1,7 @@
 from inspect import iscoroutinefunction
 from functools import wraps
 
-from agenta.sdk.tracing.logger import llm_logger as logging
+from agenta.sdk.utils.logging import log
 
 DEBUG = False
 SHIFT = 7
@@ -14,7 +14,7 @@ def debug(shift=1, req=False, res=False, chars=[">", "<"]):
         @wraps(f)
         async def async_log_wrapper(*args, **kwargs):
             if DEBUG:
-                logging.debug(
+                log.debug(
                     " ".join(
                         [
                             chars[0] * shift + " " * (SHIFT - shift),
@@ -26,7 +26,7 @@ def debug(shift=1, req=False, res=False, chars=[">", "<"]):
                 )
             result = await f(*args, **kwargs)
             if DEBUG:
-                logging.debug(
+                log.debug(
                     " ".join(
                         [
                             chars[1] * shift + " " * (SHIFT - shift),
@@ -40,7 +40,7 @@ def debug(shift=1, req=False, res=False, chars=[">", "<"]):
         @wraps(f)
         def log_wrapper(*args, **kwargs):
             if DEBUG:
-                logging.debug(
+                log.debug(
                     " ".join(
                         [
                             chars[0] * shift + " " * (SHIFT - shift),
@@ -52,7 +52,7 @@ def debug(shift=1, req=False, res=False, chars=[">", "<"]):
                 )
             result = f(*args, **kwargs)
             if DEBUG:
-                logging.debug(
+                log.debug(
                     " ".join(
                         [
                             chars[1] * shift + " " * (SHIFT - shift),

agenta/sdk/utils/exceptions.py

@@ -0,0 +1,19 @@
+from contextlib import AbstractContextManager
+from traceback import format_exc
+from agenta.sdk.utils.logging import log
+
+
+class suppress(AbstractContextManager):
+    def __init__(self):
+        pass
+
+    def __enter__(self):
+        pass
+
+    def __exit__(self, exc_type, exc_value, exc_tb):
+        if exc_type is None:
+            return
+        else:
+            print("Agenta SDK - Tracing Exception")
+            log.error(f"{exc_type.__name__}: {exc_value}\n{format_exc()}")
+            return

agenta/sdk/utils/globals.py

@@ -1,15 +1,13 @@
 import agenta
 
 
-def set_global(setup=None, config=None, tracing=None):
-    """Allows usage of agenta.config and agenta.setup in the user's code.
+def set_global(config=None, tracing=None):
+    """Allows usage of agenta.config and agenta.tracing in the user's code.
 
     Args:
-        setup: _description_. Defaults to None.
         config: _description_. Defaults to None.
+        tracing: _description_. Defaults to None.
     """
-    if setup is not None:
-        agenta.setup = setup
     if config is not None:
         agenta.config = config
     if tracing is not None:

agenta/sdk/{tracing/logger.py → utils/logging.py}

@@ -1,12 +1,11 @@
 import logging
 
 
-class LLMLogger:
-    def __init__(self, name="LLMLogger", level=logging.DEBUG):
+class Logger:
+    def __init__(self, name="agenta.logger", level=logging.DEBUG):
         self.logger = logging.getLogger(name)
         self.logger.setLevel(level)
 
-        # Add a stream logger to view the logs in the console
         console_handler = logging.StreamHandler()
         self.logger.addHandler(console_handler)
 
@@ -15,5 +14,4 @@ class LLMLogger:
         return self.logger
 
 
-# Initialize llm logger
-llm_logger = LLMLogger().log
+log = Logger().log

agenta/sdk/utils/singleton.py

@@ -0,0 +1,13 @@
+from threading import Lock
+
+
+class Singleton(type):
+    _instances = {}
+
+    _lock: Lock = Lock()
+
+    def __call__(cls, *args, **kwargs):
+        with cls._lock:
+            if cls not in cls._instances:
+                cls._instances[cls] = super().__call__(*args, **kwargs)
+        return cls._instances[cls]

{agenta-0.26.0a0.dist-info → agenta-0.27.0a0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.26.0a0
+Version: 0.27.0a0
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering
@@ -23,6 +23,10 @@ Requires-Dist: fastapi (>=0.100.0)
 Requires-Dist: httpx (>=0.24,<0.28)
 Requires-Dist: importlib-metadata (>=8.0.0,<9.0)
 Requires-Dist: ipdb (>=0.13)
+Requires-Dist: litellm (>=1.48.0,<2.0.0)
+Requires-Dist: opentelemetry-api (>=1.27.0,<2.0.0)
+Requires-Dist: opentelemetry-exporter-otlp (>=1.27.0,<2.0.0)
+Requires-Dist: opentelemetry-sdk (>=1.27.0,<2.0.0)
 Requires-Dist: posthog (>=3.1.0,<4.0.0)
 Requires-Dist: pydantic (>=2)
 Requires-Dist: pymongo (>=4.6.3,<5.0.0)

{agenta-0.26.0a0.dist-info → agenta-0.27.0a0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-agenta/__init__.py,sha256=MCeFBzYtrVNdhElvl1fBjypsqhgEaAMKy6ZR-B3YmJk,993
+agenta/__init__.py,sha256=cd3c7bpyfZC8ho2Dc299bg8l_BDRQ3TFbd9Tnn2UQTY,923
 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
@@ -14,7 +14,7 @@ agenta/client/backend/apps/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2t
 agenta/client/backend/apps/client.py,sha256=9rWax7LlEIpXdqq4McPAkvwHweSTomnyrFxFo8ehYy0,54136
 agenta/client/backend/bases/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
 agenta/client/backend/bases/client.py,sha256=BZsz5eXaa2voZdJXqgd5J5hPUuYvWwIcPCWyl49w-oY,6028
-agenta/client/backend/client.py,sha256=NqKTA7oKQFI4eU0TNfB-SEDILMgVEJSZFodGsl3hWZI,102435
+agenta/client/backend/client.py,sha256=k262wVUWPYWbZhLiZec7sTLfh8CD7tfCYdFKyLqUMi0,102659
 agenta/client/backend/configs/__init__.py,sha256=9mUnTDeA1TxYvkj1l01A1prqsJV0ERRY2tzkY1fA4MQ,64
 agenta/client/backend/configs/client.py,sha256=qtrRdRuNIZBjh9Ykj7eNKhocp7dFdFQoDSbUduVjKJ0,19089
 agenta/client/backend/containers/__init__.py,sha256=Haw2PwiPhNvM26PLQN57jY0bN-QqPoDG4VA-P_uGL3A,153
@@ -26,7 +26,7 @@ agenta/client/backend/core/api_error.py,sha256=TtMtCdxXjd7Tasc9c8ooFg124nPrb2MXG
 agenta/client/backend/core/client_wrapper.py,sha256=mU22mZBjz2QXalKJLDGo1qNeAt1awLxRMO8fzbPbnr8,1829
 agenta/client/backend/core/datetime_utils.py,sha256=BHjt_H3WVslcuPsr6qjJoVif_SsdLvFN0c43ABE5UiQ,1069
 agenta/client/backend/core/file.py,sha256=vliNmlB7PbDfi4EKiVPNq5QaGXJ4zlDBGupv7Qciy7g,1520
-agenta/client/backend/core/http_client.py,sha256=LHzx7H12q9WOUquq8QZYSUPCt4TRdwxcRbwQm2JboeE,20278
+agenta/client/backend/core/http_client.py,sha256=1ORQxbKS0c6nMbeiANucMrlwUImmLXlCfjBbR_W-LSY,20440
 agenta/client/backend/core/jsonable_encoder.py,sha256=SHXw4G4n-f0IPgNkxj_-Fip3kN8NUAI-YrKxdZw8kl0,3662
 agenta/client/backend/core/pydantic_utilities.py,sha256=sLzQZQLS_y66SZSBMtJE7YrQK8INda-ta1tKwAHabGk,7540
 agenta/client/backend/core/query_encoder.py,sha256=8qYl5VPl1jU4cDF0X7oSU_DXjlVWY5ayigFBpNTMGOA,2150
@@ -132,28 +132,38 @@ 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=kO1q2_IR0fEAo4M-2Pt_v-zC7GxxnkLogjKFhU869Ps,3555
-agenta/sdk/__init__.py,sha256=XyxMUm8-fCH6G6KAFU0iAfiKTGEzNzBtbAfCp9i1Y7o,831
-agenta/sdk/agenta_init.py,sha256=8MfDuypxohd0qRTdtGjX7L17KW-1UGmzNVdiqF15_ak,9790
+agenta/sdk/__init__.py,sha256=k6lzJF6gzaFdzq_FJ3chBu6ug8pNCT3_-bfeB1ZqGkg,1165
+agenta/sdk/agenta_init.py,sha256=Wg9LOD8pQiz4F6-QepnqWmwdBKe_FJK-AHs6_HvO1x4,10603
 agenta/sdk/assets.py,sha256=Zv4i8MVUSB3jMODQon1mzJtYxuntmrCNjLGk8f-2fls,2856
 agenta/sdk/client.py,sha256=trKyBOYFZRk0v5Eptxvh87yPf50Y9CqY6Qgv4Fy-VH4,2142
-agenta/sdk/config_manager.py,sha256=n3UTpIhiB1J6FIQ5IvlucfGVpxoIIemnmdI68p34Yvw,8003
-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=jXXTQS-sCwmkKvF0bNYTG9pkK13B2odnHsrYepTU_x8,35572
-agenta/sdk/decorators/tracing.py,sha256=e0olx2EEdjXY0NqpIoDJSVxCnUmv0woewTUuCJXy2tM,4166
-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/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
-agenta/sdk/types.py,sha256=1AXUKTURFFgPH5iWeTYcNcg6dxVo1RrD0WuVe9e-NgU,5919
+agenta/sdk/config_manager.py,sha256=HFOJpJKBkhlA0C-KPuxb4-bHNZeZqdpmx_beoX4lQw8,7997
+agenta/sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+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=4vJsFwyXUx1cuzILVOzELDfv17QnYTXpIE2NwFnC__I,35427
+agenta/sdk/decorators/tracing.py,sha256=hGh8xnoiQBC-ycWOTHm-a6bX60c1aEkN3NGV9JZ0F2w,11202
+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=0Ld0A4eL6EAWwbhBY0Qp7MLWXXbBsd8EDtbBuwIy1PI,4775
+agenta/sdk/tracing/context.py,sha256=KDQYjRcmbpb6e0xEuX-bAQGhEjwx4H2nwybRPOKCDjo,578
+agenta/sdk/tracing/conventions.py,sha256=YFkKPls77sl5YOXrRjfn4erEjczMXThc7b-rRaVVTuo,722
+agenta/sdk/tracing/exporters.py,sha256=e-SfpN8dKQ0vuBZQTZ6qgIMBhyriOrVeLYJkv-us0Zs,1328
+agenta/sdk/tracing/inline.py,sha256=n9_XM6VJzg9v_z7xjFHyK6aKABn_YpEINXyvXSyOgcw,35821
+agenta/sdk/tracing/processors.py,sha256=PRQpj8aOmOjvf71M48AtCurIxVFg2ZZWRtJLtlB4xyE,1974
+agenta/sdk/tracing/spans.py,sha256=AnQuuyuCGST6n_He4SF5WdrtMts9JlE8v6rVRdm2rRg,3570
+agenta/sdk/tracing/tracing.py,sha256=5r0whzLzF2bwYKZkUN3QDjq8AketEb-9sM4nqnO9zzM,5168
+agenta/sdk/types.py,sha256=B7C7Jpeq0CpmLwmN1dmpggJsZX8WpxyyM42PkdSsx3s,5669
 agenta/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-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
+agenta/sdk/utils/costs.py,sha256=i8C7ud__pThLS55XkN4YW8czXtGeXr2mx7jjcOFeiXg,5955
+agenta/sdk/utils/debug.py,sha256=DxiCAeXxxrcEZT2CjlNA6BMvujGP4nzQ-rfb-_mLMck,2114
+agenta/sdk/utils/exceptions.py,sha256=kzU5bNBMZvVv0xnwvc5osVMsb_Ujm0qqeem9wpNo2z0,500
+agenta/sdk/utils/globals.py,sha256=2HhyzWn55BbYNCZ3rT8dAxk1GGXuGQPbtq_THjaHbBw,372
+agenta/sdk/utils/logging.py,sha256=FKzAO5eHjR_qvpEnc4tKXUA6WftmwdwYwuJK6CbSc84,379
 agenta/sdk/utils/preinit.py,sha256=YlJL7RLfel0R7DFp-jK7OV-z4ZIQJM0oupYlk7g8b5o,1278
+agenta/sdk/utils/singleton.py,sha256=17Ph7LGnnV8HkPjImruKita2ni03Ari5jr0jqm__4sc,312
 agenta/templates/compose_email/README.md,sha256=ss7vZPpI1Hg0VmYtFliwq_r5LnqbCy_S5OQDXg8UoIA,308
 agenta/templates/compose_email/app.py,sha256=SG8QnBkC49MtvWa2z5_YQ2d8wxZxhO_hWs15j3CsWvc,2384
 agenta/templates/compose_email/env.example,sha256=g9AE5bYcGPpxawXMJ96gh8oenEPCHTabsiOnfQo3c5k,70
@@ -169,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.26.0a0.dist-info/METADATA,sha256=X9sEaKeywN8OI24YukCQYnvywurjaK_1Rz6AZdWOuDc,31534
-agenta-0.26.0a0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-agenta-0.26.0a0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.26.0a0.dist-info/RECORD,,
+agenta-0.27.0a0.dist-info/METADATA,sha256=O3yhji_Fs-vomvmT-XV3jDff3pQYFgpkFt1zGT0X-Y4,31738
+agenta-0.27.0a0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+agenta-0.27.0a0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.27.0a0.dist-info/RECORD,,

agenta/sdk/context.py

@@ -1,41 +0,0 @@
-import json
-import sqlite3
-from typing import List
-from .types import Context
-
-
-def setup_db():
-    conn = sqlite3.connect("context.db")
-    c = conn.cursor()
-    c.execute(
-        """
-    CREATE TABLE IF NOT EXISTS contexts
-    (id INTEGER PRIMARY KEY AUTOINCREMENT, context TEXT)
-    """
-    )
-    conn.commit()
-    conn.close()
-
-
-def get_contexts() -> List[Context]:
-    contexts = []
-    conn = sqlite3.connect("context.db")
-    c = conn.cursor()
-    for row in c.execute("SELECT * FROM contexts"):
-        context_data = json.loads(row[1])
-        contexts.append(Context.parse_obj(context_data))
-    conn.close()
-    return contexts
-
-
-def save_context(result: Context):
-    conn = sqlite3.connect("context.db")
-    c = conn.cursor()
-    c.execute(
-        """
-    INSERT INTO contexts (context) VALUES (?)
-    """,
-        (json.dumps(result.dict()),),
-    )
-    conn.commit()
-    conn.close()